computer-playbook/tests/unit/test_domain_mappings.py

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_multiple_aliases(self):
        apps = {
            'app1': {'domains': {'aliases': ['a1.com','a2.com']}}
        }
        expected = [
            {'source': 'a1.com', 'target': 'app1.example.com'},
            {'source': 'a2.com', 'target': 'app1.example.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()