diff --git a/tests/rulecheck/test_domainrulecheck.py b/tests/rulecheck/test_domainrulecheck.py index 702862f78b..ebcf5ca44e 100644 --- a/tests/rulecheck/test_domainrulecheck.py +++ b/tests/rulecheck/test_domainrulecheck.py @@ -21,22 +21,24 @@ from tests import unittest class DomainRuleCheckerTestCase(unittest.TestCase): - def test_allowed(self): config = { "default": False, "domain_mapping": { "source_one": ["target_one", "target_two"], - "source_two": ["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")) + 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 = { @@ -44,50 +46,56 @@ class DomainRuleCheckerTestCase(unittest.TestCase): "domain_mapping": { "source_one": ["target_one", "target_two"], "source_two": ["target_two"], - "source_four": [] - } + "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")) + 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"] - } + "source_two": ["target_two"], + }, } check = DomainRuleChecker(config) - self.assertTrue(check.user_may_invite("test:source_three", - "test:target_one", "room")) + 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"] - } + "source_two": ["target_two"], + }, } check = DomainRuleChecker(config) - self.assertFalse(check.user_may_invite("test:source_three", - "test:target_one", "room")) + 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"] - } + "source_two": ["target_two"], + }, } self.assertEquals(config, DomainRuleChecker.parse_config(config)) @@ -95,7 +103,7 @@ class DomainRuleCheckerTestCase(unittest.TestCase): config = { "domain_mapping": { "source_one": ["target_one", "target_two"], - "source_two": ["target_two"] + "source_two": ["target_two"], } } self.assertRaises(ConfigError, DomainRuleChecker.parse_config, config)