mirror of
https://github.com/kevinveenbirkenbach/computer-playbook.git
synced 2025-06-25 19:55:31 +02:00
100 lines
3.0 KiB
Python
100 lines
3.0 KiB
Python
import os
|
|
import sys
|
|
import unittest
|
|
|
|
# Add the filter_plugins directory to the import path
|
|
dir_path = os.path.abspath(
|
|
os.path.join(os.path.dirname(__file__), '../../filter_plugins')
|
|
)
|
|
sys.path.insert(0, dir_path)
|
|
|
|
from ansible.errors import AnsibleFilterError
|
|
from domain_redirect_mappings import FilterModule
|
|
|
|
class TestDomainMappings(unittest.TestCase):
|
|
def setUp(self):
|
|
self.filter = FilterModule()
|
|
self.primary = 'example.com'
|
|
|
|
def test_empty_apps(self):
|
|
apps = {}
|
|
result = self.filter.domain_mappings(apps, self.primary)
|
|
self.assertEqual(result, [])
|
|
|
|
def test_app_without_domains(self):
|
|
apps = {'app1': {}}
|
|
# no domains key → no mappings
|
|
result = self.filter.domain_mappings(apps, self.primary)
|
|
self.assertEqual(result, [])
|
|
|
|
def test_empty_domains_cfg(self):
|
|
apps = {'app1': {'domains': {}}}
|
|
default = 'app1.example.com'
|
|
expected = []
|
|
result = self.filter.domain_mappings(apps, self.primary)
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_explicit_aliases(self):
|
|
apps = {
|
|
'app1': {
|
|
'domains': {'aliases': ['alias.com']}
|
|
}
|
|
}
|
|
default = 'app1.example.com'
|
|
expected = [
|
|
{'source': 'alias.com', 'target': default},
|
|
]
|
|
result = self.filter.domain_mappings(apps, self.primary)
|
|
# order not important
|
|
self.assertCountEqual(result, expected)
|
|
|
|
def test_canonical_not_default(self):
|
|
apps = {
|
|
'app1': {
|
|
'domains': {'canonical': ['foo.com']}
|
|
}
|
|
}
|
|
expected = [
|
|
{'source': 'app1.example.com', 'target': 'foo.com'}
|
|
]
|
|
result = self.filter.domain_mappings(apps, self.primary)
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_canonical_dict(self):
|
|
apps = {
|
|
'app1': {
|
|
'domains': {
|
|
'canonical': {'one': 'one.com', 'two': 'two.com'}
|
|
}
|
|
}
|
|
}
|
|
# first canonical key 'one' → one.com
|
|
expected = [
|
|
{'source': 'app1.example.com', 'target': 'one.com'}
|
|
]
|
|
result = self.filter.domain_mappings(apps, self.primary)
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_multiple_apps(self):
|
|
apps = {
|
|
'app1': {'domains': {'aliases': ['a1.com']}},
|
|
'app2': {'domains': {'canonical': ['c2.com']}},
|
|
}
|
|
expected = [
|
|
{'source': 'a1.com', 'target': 'app1.example.com'},
|
|
{'source': 'app2.example.com', 'target': 'c2.com'},
|
|
]
|
|
result = self.filter.domain_mappings(apps, self.primary)
|
|
self.assertCountEqual(result, expected)
|
|
|
|
def test_invalid_aliases_type(self):
|
|
apps = {
|
|
'app1': {'domains': {'aliases': 123}}
|
|
}
|
|
with self.assertRaises(AnsibleFilterError):
|
|
self.filter.domain_mappings(apps, self.primary)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|