78 lines
1.6 KiB
Python
78 lines
1.6 KiB
Python
import unittest
|
|
|
|
from . import config
|
|
|
|
TEST_CONFIG = """
|
|
{
|
|
"mails_path": "/var/tmp/mails",
|
|
"matches": [
|
|
{
|
|
"name": "mydomain",
|
|
"addr_rexs": [
|
|
".*@mydomain.com",
|
|
".*@m.mydomain.com"
|
|
]
|
|
},
|
|
{
|
|
"name": "personal",
|
|
"addrs": [
|
|
"first.last@mydomain.com",
|
|
"secret.name@mydomain.com"
|
|
]
|
|
}
|
|
],
|
|
"boxes": [
|
|
{
|
|
"name": "spam",
|
|
"rules": [
|
|
{
|
|
"match_name": "mydomain",
|
|
"negate": true,
|
|
"stop_check": true
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "important",
|
|
"rules": [
|
|
{
|
|
"match_name": "personal"
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "all",
|
|
"rules": [
|
|
{
|
|
"match_name": "default_match_all"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
"""
|
|
|
|
|
|
class TestConfig(unittest.TestCase):
|
|
|
|
def test_config(self) -> None:
|
|
cfg = config.Config(TEST_CONFIG)
|
|
self.assertEqual(cfg.mails_path, "/var/tmp/mails")
|
|
|
|
def test_parse_rules(self) -> None:
|
|
cfg = config.Config(TEST_CONFIG)
|
|
op = config.parse_checkers(cfg)
|
|
self.assertEqual(len(op), 3)
|
|
|
|
def test_get_mboxes(self) -> None:
|
|
cfg = config.Config(TEST_CONFIG)
|
|
rules = config.parse_checkers(cfg)
|
|
self.assertEqual(config.get_mboxes("foo@bar.com", rules), ['spam'])
|
|
self.assertEqual(config.get_mboxes("foo@mydomain.com", rules), ['all'])
|
|
self.assertEqual(config.get_mboxes("first.last@mydomain.com", rules),
|
|
['important', 'all'])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|