X-Git-Url: https://osm.etsi.org/gitweb/?p=osm%2Fdevops.git;a=blobdiff_plain;f=installers%2Fcharm%2Fpla%2Ftests%2Ftest_charm.py;h=b1a782032d57d2d6c95faf8e46c9580cdea92061;hp=dbc7be3bcefeb50857c515929c0f3391d38e2ad5;hb=49379ced23b5e344a773ce77ac9cb59c1864e19b;hpb=e5fd208248a496cd798a66cb27216e83f2852a76 diff --git a/installers/charm/pla/tests/test_charm.py b/installers/charm/pla/tests/test_charm.py index dbc7be3b..b1a78203 100644 --- a/installers/charm/pla/tests/test_charm.py +++ b/installers/charm/pla/tests/test_charm.py @@ -1,5 +1,5 @@ #!/usr/bin/env python3 -# Copyright 2020 Canonical Ltd. +# Copyright 2021 Canonical Ltd. # # Licensed under the Apache License, Version 2.0 (the "License"); you may # not use this file except in compliance with the License. You may obtain @@ -20,27 +20,80 @@ # osm-charmers@lists.launchpad.net ## +import sys from typing import NoReturn import unittest - +from ops.model import ActiveStatus, BlockedStatus from ops.testing import Harness -from charm import PLACharm +from charm import PlaCharm class TestCharm(unittest.TestCase): - """PLA Charm unit tests.""" + """Pla Charm unit tests.""" def setUp(self) -> NoReturn: """Test setup""" - self.harness = Harness(PLACharm) + self.image_info = sys.modules["oci_image"].OCIImageResource().fetch() + self.harness = Harness(PlaCharm) self.harness.set_leader(is_leader=True) self.harness.begin() + self.config = { + "log_level": "INFO", + } + self.harness.update_config(self.config) + + def test_config_changed_no_relations( + self, + ) -> NoReturn: + """Test ingress resources without HTTP.""" + + self.harness.charm.on.config_changed.emit() + + # Assertions + self.assertIsInstance(self.harness.charm.unit.status, BlockedStatus) + self.assertTrue( + all( + relation in self.harness.charm.unit.status.message + for relation in ["mongodb", "kafka"] + ) + ) - def test_on_start_without_relations(self) -> NoReturn: - """Test installation without any relation.""" + def test_config_changed_non_leader( + self, + ) -> NoReturn: + """Test ingress resources without HTTP.""" + self.harness.set_leader(is_leader=False) self.harness.charm.on.config_changed.emit() + # Assertions + self.assertIsInstance(self.harness.charm.unit.status, ActiveStatus) + + def test_with_relations( + self, + ) -> NoReturn: + "Test with relations (internal)" + self.initialize_kafka_relation() + self.initialize_mongo_relation() + # Verifying status + self.assertNotIsInstance(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} + ) + + def initialize_mongo_relation(self): + mongodb_relation_id = self.harness.add_relation("mongodb", "mongodb") + self.harness.add_relation_unit(mongodb_relation_id, "mongodb/0") + self.harness.update_relation_data( + mongodb_relation_id, + "mongodb/0", + {"connection_string": "mongodb://mongo:27017"}, + ) + if __name__ == "__main__": unittest.main()