CharmHub and new kafka and zookeeper charms

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

diff --git a/installers/charm/pla/tests/test_charm.py b/installers/charm/pla/tests/test_charm.py
index b1a7820..d577e9f 100644 (file)
--- a/installers/charm/pla/tests/test_charm.py
+++ b/installers/charm/pla/tests/test_charm.py
@@ -23,10 +23,11 @@
 import sys
 from typing import NoReturn
 import unittest
-from ops.model import ActiveStatus, BlockedStatus
-from ops.testing import Harness
+
 
 from charm import PlaCharm
+from ops.model import ActiveStatus, BlockedStatus
+from ops.testing import Harness
 
 
 class TestCharm(unittest.TestCase):
@@ -40,6 +41,7 @@ class TestCharm(unittest.TestCase):
         self.harness.begin()
         self.config = {
             "log_level": "INFO",
+            "mongodb_uri": "",
         }
         self.harness.update_config(self.config)
 
@@ -69,6 +71,15 @@ class TestCharm(unittest.TestCase):
         # Assertions
         self.assertIsInstance(self.harness.charm.unit.status, ActiveStatus)
 
+    def test_with_relations_and_mongodb_config(
+        self,
+    ) -> NoReturn:
+        "Test with relations and mongodb config (internal)"
+        self.initialize_kafka_relation()
+        self.initialize_mongo_config()
+        # Verifying status
+        self.assertNotIsInstance(self.harness.charm.unit.status, BlockedStatus)
+
     def test_with_relations(
         self,
     ) -> NoReturn:
@@ -78,13 +89,25 @@ class TestCharm(unittest.TestCase):
         # Verifying status
         self.assertNotIsInstance(self.harness.charm.unit.status, BlockedStatus)
 
+    def test_exception_mongodb_relation_and_config(
+        self,
+    ) -> NoReturn:
+        "Test with relation and config for Mongodb. Test must fail"
+        self.initialize_mongo_relation()
+        self.initialize_mongo_config()
+        # 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")