CharmHub and new kafka and zookeeper charms

[osm/devops.git] / installers / charm / nbi / tests / test_charm.py

diff --git a/installers/charm/nbi/tests/test_charm.py b/installers/charm/nbi/tests/test_charm.py
index 2b4ea0f..92c2980 100644 (file)
--- a/installers/charm/nbi/tests/test_charm.py
+++ b/installers/charm/nbi/tests/test_charm.py
@@ -43,11 +43,12 @@ class TestCharm(unittest.TestCase):
             "enable_test": False,
             "auth_backend": "internal",
             "database_commonkey": "key",
+            "mongodb_uri": "",
             "log_level": "INFO",
             "max_file_size": 0,
             "ingress_whitelist_source_range": "",
             "tls_secret_name": "",
-            "site_url": "https://nbi.192.168.100.100.xip.io",
+            "site_url": "https://nbi.192.168.100.100.nip.io",
             "cluster_issuer": "vault-issuer",
         }
         self.harness.update_config(self.config)
@@ -78,6 +79,16 @@ class TestCharm(unittest.TestCase):
         # Assertions
         self.assertIsInstance(self.harness.charm.unit.status, ActiveStatus)
 
+    def test_with_relations_internal_and_mongodb_config(
+        self,
+    ) -> NoReturn:
+        "Test with relations and mongodb config (internal)"
+        self.initialize_kafka_relation()
+        self.initialize_mongo_config()
+        self.initialize_prometheus_relation()
+        # Verifying status
+        self.assertNotIsInstance(self.harness.charm.unit.status, BlockedStatus)
+
     def test_with_relations_internal(
         self,
     ) -> NoReturn:
@@ -88,6 +99,18 @@ class TestCharm(unittest.TestCase):
         # Verifying status
         self.assertNotIsInstance(self.harness.charm.unit.status, BlockedStatus)
 
+    def test_with_relations_and_mongodb_config_with_keystone_missing(
+        self,
+    ) -> NoReturn:
+        "Test with relations and mongodb config (keystone)"
+        self.harness.update_config({"auth_backend": "keystone"})
+        self.initialize_kafka_relation()
+        self.initialize_mongo_config()
+        self.initialize_prometheus_relation()
+        # Verifying status
+        self.assertIsInstance(self.harness.charm.unit.status, BlockedStatus)
+        self.assertTrue("keystone" in self.harness.charm.unit.status.message)
+
     def test_with_relations_keystone_missing(
         self,
     ) -> NoReturn:
@@ -100,6 +123,18 @@ class TestCharm(unittest.TestCase):
         self.assertIsInstance(self.harness.charm.unit.status, BlockedStatus)
         self.assertTrue("keystone" in self.harness.charm.unit.status.message)
 
+    def test_with_relations_and_mongodb_config_with_keystone(
+        self,
+    ) -> NoReturn:
+        "Test with relations (keystone)"
+        self.harness.update_config({"auth_backend": "keystone"})
+        self.initialize_kafka_relation()
+        self.initialize_mongo_config()
+        self.initialize_prometheus_relation()
+        self.initialize_keystone_relation()
+        # Verifying status
+        self.assertNotIsInstance(self.harness.charm.unit.status, BlockedStatus)
+
     def test_with_relations_keystone(
         self,
     ) -> NoReturn:
@@ -112,13 +147,24 @@ class TestCharm(unittest.TestCase):
         # Verifying status
         self.assertNotIsInstance(self.harness.charm.unit.status, BlockedStatus)
 
+    def test_mongodb_exception_relation_and_config(
+        self,
+    ) -> NoReturn:
+        self.initialize_mongo_config()
+        self.initialize_mongo_relation()
+        # Verifying status
+        self.assertIsInstance(self.harness.charm.unit.status, BlockedStatus)
+
     def initialize_kafka_relation(self):
         kafka_relation_id = self.harness.add_relation("kafka", "kafka")
         self.harness.add_relation_unit(kafka_relation_id, "kafka/0")
         self.harness.update_relation_data(
-            kafka_relation_id, "kafka/0", {"host": "kafka", "port": 9092}
+            kafka_relation_id, "kafka", {"host": "kafka", "port": 9092}
         )
 
+    def initialize_mongo_config(self):
+        self.harness.update_config({"mongodb_uri": "mongodb://mongo:27017"})
+
     def initialize_mongo_relation(self):
         mongodb_relation_id = self.harness.add_relation("mongodb", "mongodb")
         self.harness.add_relation_unit(mongodb_relation_id, "mongodb/0")