mirror of
https://mau.dev/maunium/synapse.git
synced 2024-10-01 01:36:05 -04:00
Run black on tests/rulecheck/test_domainrulecheck.py
This commit is contained in:
parent
b85ff4b894
commit
e64f7c0188
@ -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)
|
||||
|
Loading…
Reference in New Issue
Block a user