Strip .prod.global.gc.ca from hostnames

2026-02-06 16:59:56 -05:00
parent dcddd88cbc
commit a13fc5b282
3 changed files with 23 additions and 1 deletions
--- a/test_inventory_keys.py
+++ b/test_inventory_keys.py
@@ -55,5 +55,12 @@ class TestInventoryKeys(unittest.TestCase):
        self.assertIn("good_name", hosts)
        self.assertNotIn("bad_name", hosts)

+    def test_suffix_stripping(self):
+        from wif2ansible.parsers import clean_hostname
+        self.assertEqual(clean_hostname("server.prod.global.gc.ca"), "server")
+        self.assertEqual(clean_hostname("server.PROD.GLOBAL.GC.CA"), "server")
+        self.assertEqual(clean_hostname("nosuffix"), "nosuffix")
+        self.assertEqual(clean_hostname("other.suffix.com"), "other.suffix.com")
+
 if __name__ == '__main__':
    unittest.main()