123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101 |
- # -*- coding: utf-8 -*-
- # Copyright 2018 New Vector Ltd
- #
- # Licensed under the Apache License, Version 2.0 (the "License");
- # you may not use this file except in compliance with the License.
- # You may obtain a copy of the License at
- #
- # http://www.apache.org/licenses/LICENSE-2.0
- #
- # Unless required by applicable law or agreed to in writing, software
- # distributed under the License is distributed on an "AS IS" BASIS,
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- # See the License for the specific language governing permissions and
- # limitations under the License.
- from tests import unittest
- from synapse.config._base import ConfigError
- from synapse.rulecheck.domain_rule_checker import DomainRuleChecker
- class DomainRuleCheckerTestCase(unittest.TestCase):
- def test_allowed(self):
- config = {
- "default": False,
- "domain_mapping": {
- "source_one": ["target_one", "target_two"],
- "source_two": ["target_two"]
- }
- }
- check = DomainRuleChecker(config)
- self.assertTrue(check.user_may_invite("test:source_one",
- "test:target_one", "room"))
- self.assertTrue(check.user_may_invite("test:source_one",
- "test:target_two", "room"))
- self.assertTrue(check.user_may_invite("test:source_two",
- "test:target_two", "room"))
- def test_disallowed(self):
- config = {
- "default": True,
- "domain_mapping": {
- "source_one": ["target_one", "target_two"],
- "source_two": ["target_two"],
- "source_four": []
- }
- }
- check = DomainRuleChecker(config)
- self.assertFalse(check.user_may_invite("test:source_one",
- "test:target_three", "room"))
- self.assertFalse(check.user_may_invite("test:source_two",
- "test:target_three", "room"))
- self.assertFalse(check.user_may_invite("test:source_two",
- "test:target_one", "room"))
- self.assertFalse(check.user_may_invite("test:source_four",
- "test:target_one", "room"))
- def test_default_allow(self):
- config = {
- "default": True,
- "domain_mapping": {
- "source_one": ["target_one", "target_two"],
- "source_two": ["target_two"]
- }
- }
- check = DomainRuleChecker(config)
- self.assertTrue(check.user_may_invite("test:source_three",
- "test:target_one", "room"))
- def test_default_deny(self):
- config = {
- "default": False,
- "domain_mapping": {
- "source_one": ["target_one", "target_two"],
- "source_two": ["target_two"]
- }
- }
- check = DomainRuleChecker(config)
- self.assertFalse(check.user_may_invite("test:source_three",
- "test:target_one", "room"))
- def test_config_parse(self):
- config = {
- "default": False,
- "domain_mapping": {
- "source_one": ["target_one", "target_two"],
- "source_two": ["target_two"]
- }
- }
- self.assertEquals(config, DomainRuleChecker.parse_config(config))
- def test_config_parse_failure(self):
- config = {
- "domain_mapping": {
- "source_one": ["target_one", "target_two"],
- "source_two": ["target_two"]
- }
- }
- self.assertRaises(ConfigError, DomainRuleChecker.parse_config, config)
|