Run black on tests/rulecheck/test_domainrulecheck.py

This commit is contained in:
Erik Johnston 2019-03-18 14:23:46 +00:00
parent b85ff4b894
commit e64f7c0188

View File

@ -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)