2 # Copyright 2021 Canonical Ltd.
4 # Licensed under the Apache License, Version 2.0 (the "License"); you may
5 # not use this file except in compliance with the License. You may obtain
6 # a copy of the License at
8 # http://www.apache.org/licenses/LICENSE-2.0
10 # Unless required by applicable law or agreed to in writing, software
11 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
12 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
13 # License for the specific language governing permissions and limitations
16 # For those usages not covered by the Apache License, Version 2.0 please
17 # contact: legal@canonical.com
19 # To get in touch with the maintainers, please contact:
20 # osm-charmers@lists.launchpad.net
24 from typing
import NoReturn
28 from charm
import PlaCharm
29 from ops
.model
import ActiveStatus
, BlockedStatus
30 from ops
.testing
import Harness
33 class TestCharm(unittest
.TestCase
):
34 """Pla Charm unit tests."""
36 def setUp(self
) -> NoReturn
:
38 self
.image_info
= sys
.modules
["oci_image"].OCIImageResource().fetch()
39 self
.harness
= Harness(PlaCharm
)
40 self
.harness
.set_leader(is_leader
=True)
45 self
.harness
.update_config(self
.config
)
47 def test_config_changed_no_relations(
50 """Test ingress resources without HTTP."""
52 self
.harness
.charm
.on
.config_changed
.emit()
55 self
.assertIsInstance(self
.harness
.charm
.unit
.status
, BlockedStatus
)
58 relation
in self
.harness
.charm
.unit
.status
.message
59 for relation
in ["mongodb", "kafka"]
63 def test_config_changed_non_leader(
66 """Test ingress resources without HTTP."""
67 self
.harness
.set_leader(is_leader
=False)
68 self
.harness
.charm
.on
.config_changed
.emit()
71 self
.assertIsInstance(self
.harness
.charm
.unit
.status
, ActiveStatus
)
73 def test_with_relations(
76 "Test with relations (internal)"
77 self
.initialize_kafka_relation()
78 self
.initialize_mongo_relation()
80 self
.assertNotIsInstance(self
.harness
.charm
.unit
.status
, BlockedStatus
)
82 def initialize_kafka_relation(self
):
83 kafka_relation_id
= self
.harness
.add_relation("kafka", "kafka")
84 self
.harness
.add_relation_unit(kafka_relation_id
, "kafka/0")
85 self
.harness
.update_relation_data(
86 kafka_relation_id
, "kafka/0", {"host": "kafka", "port": 9092}
89 def initialize_mongo_relation(self
):
90 mongodb_relation_id
= self
.harness
.add_relation("mongodb", "mongodb")
91 self
.harness
.add_relation_unit(mongodb_relation_id
, "mongodb/0")
92 self
.harness
.update_relation_data(
95 {"connection_string": "mongodb://mongo:27017"},
99 if __name__
== "__main__":