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 NbiCharm
29 from ops
.model
import ActiveStatus
, BlockedStatus
30 from ops
.testing
import Harness
33 class TestCharm(unittest
.TestCase
):
34 """Prometheus Charm unit tests."""
36 def setUp(self
) -> NoReturn
:
38 self
.image_info
= sys
.modules
["oci_image"].OCIImageResource().fetch()
39 self
.harness
= Harness(NbiCharm
)
40 self
.harness
.set_leader(is_leader
=True)
44 "auth_backend": "internal",
45 "database_commonkey": "key",
49 "ingress_whitelist_source_range": "",
50 "tls_secret_name": "",
51 "site_url": "https://nbi.192.168.100.100.nip.io",
52 "cluster_issuer": "vault-issuer",
54 self
.harness
.update_config(self
.config
)
56 def test_config_changed_no_relations(
59 """Test ingress resources without HTTP."""
61 self
.harness
.charm
.on
.config_changed
.emit()
64 self
.assertIsInstance(self
.harness
.charm
.unit
.status
, BlockedStatus
)
67 relation
in self
.harness
.charm
.unit
.status
.message
68 for relation
in ["mongodb", "kafka", "prometheus"]
72 def test_config_changed_non_leader(
75 """Test ingress resources without HTTP."""
76 self
.harness
.set_leader(is_leader
=False)
77 self
.harness
.charm
.on
.config_changed
.emit()
80 self
.assertIsInstance(self
.harness
.charm
.unit
.status
, ActiveStatus
)
82 def test_with_relations_internal_and_mongodb_config(
85 "Test with relations and mongodb config (internal)"
86 self
.initialize_kafka_relation()
87 self
.initialize_mongo_config()
88 self
.initialize_prometheus_relation()
90 self
.assertNotIsInstance(self
.harness
.charm
.unit
.status
, BlockedStatus
)
92 def test_with_relations_internal(
95 "Test with relations (internal)"
96 self
.initialize_kafka_relation()
97 self
.initialize_mongo_relation()
98 self
.initialize_prometheus_relation()
100 self
.assertNotIsInstance(self
.harness
.charm
.unit
.status
, BlockedStatus
)
102 def test_with_relations_and_mongodb_config_with_keystone_missing(
105 "Test with relations and mongodb config (keystone)"
106 self
.harness
.update_config({"auth_backend": "keystone"})
107 self
.initialize_kafka_relation()
108 self
.initialize_mongo_config()
109 self
.initialize_prometheus_relation()
111 self
.assertIsInstance(self
.harness
.charm
.unit
.status
, BlockedStatus
)
112 self
.assertTrue("keystone" in self
.harness
.charm
.unit
.status
.message
)
114 def test_with_relations_keystone_missing(
117 "Test with relations (keystone)"
118 self
.harness
.update_config({"auth_backend": "keystone"})
119 self
.initialize_kafka_relation()
120 self
.initialize_mongo_relation()
121 self
.initialize_prometheus_relation()
123 self
.assertIsInstance(self
.harness
.charm
.unit
.status
, BlockedStatus
)
124 self
.assertTrue("keystone" in self
.harness
.charm
.unit
.status
.message
)
126 def test_with_relations_and_mongodb_config_with_keystone(
129 "Test with relations (keystone)"
130 self
.harness
.update_config({"auth_backend": "keystone"})
131 self
.initialize_kafka_relation()
132 self
.initialize_mongo_config()
133 self
.initialize_prometheus_relation()
134 self
.initialize_keystone_relation()
136 self
.assertNotIsInstance(self
.harness
.charm
.unit
.status
, BlockedStatus
)
138 def test_with_relations_keystone(
141 "Test with relations (keystone)"
142 self
.harness
.update_config({"auth_backend": "keystone"})
143 self
.initialize_kafka_relation()
144 self
.initialize_mongo_relation()
145 self
.initialize_prometheus_relation()
146 self
.initialize_keystone_relation()
148 self
.assertNotIsInstance(self
.harness
.charm
.unit
.status
, BlockedStatus
)
150 def test_mongodb_exception_relation_and_config(
153 self
.initialize_mongo_config()
154 self
.initialize_mongo_relation()
156 self
.assertIsInstance(self
.harness
.charm
.unit
.status
, BlockedStatus
)
158 def initialize_kafka_relation(self
):
159 kafka_relation_id
= self
.harness
.add_relation("kafka", "kafka")
160 self
.harness
.add_relation_unit(kafka_relation_id
, "kafka/0")
161 self
.harness
.update_relation_data(
162 kafka_relation_id
, "kafka", {"host": "kafka", "port": 9092}
165 def initialize_mongo_config(self
):
166 self
.harness
.update_config({"mongodb_uri": "mongodb://mongo:27017"})
168 def initialize_mongo_relation(self
):
169 mongodb_relation_id
= self
.harness
.add_relation("mongodb", "mongodb")
170 self
.harness
.add_relation_unit(mongodb_relation_id
, "mongodb/0")
171 self
.harness
.update_relation_data(
174 {"connection_string": "mongodb://mongo:27017"},
177 def initialize_keystone_relation(self
):
178 keystone_relation_id
= self
.harness
.add_relation("keystone", "keystone")
179 self
.harness
.add_relation_unit(keystone_relation_id
, "keystone/0")
180 self
.harness
.update_relation_data(
181 keystone_relation_id
,
186 "user_domain_name": "ud",
187 "project_domain_name": "pd",
191 "keystone_db_password": "something",
192 "region_id": "something",
193 "admin_username": "something",
194 "admin_password": "something",
195 "admin_project_name": "something",
199 def initialize_prometheus_relation(self
):
200 prometheus_relation_id
= self
.harness
.add_relation("prometheus", "prometheus")
201 self
.harness
.add_relation_unit(prometheus_relation_id
, "prometheus/0")
202 self
.harness
.update_relation_data(
203 prometheus_relation_id
,
205 {"hostname": "prometheus", "port": 9090},
209 if __name__
== "__main__":
213 # class TestCharm(unittest.TestCase):
214 # """Prometheus Charm unit tests."""
216 # def setUp(self) -> NoReturn:
218 # self.image_info = sys.modules["oci_image"].OCIImageResource().fetch()
219 # self.harness = Harness(NbiCharm)
220 # self.harness.set_leader(is_leader=True)
221 # self.harness.begin()
223 # "enable_ng_ro": True,
224 # "database_commonkey": "commonkey",
225 # "log_level": "INFO",
226 # "vim_database": "db_name",
227 # "ro_database": "ro_db_name",
228 # "openmano_tenant": "mano",
231 # def test_config_changed_no_relations(
234 # """Test ingress resources without HTTP."""
236 # self.harness.charm.on.config_changed.emit()
239 # self.assertIsInstance(self.harness.charm.unit.status, BlockedStatus)
242 # relation in self.harness.charm.unit.status.message
243 # for relation in ["mongodb", "kafka"]
248 # self.harness.update_config({"enable_ng_ro": False})
249 # self.assertIsInstance(self.harness.charm.unit.status, BlockedStatus)
252 # relation in self.harness.charm.unit.status.message
253 # for relation in ["mysql"]
257 # def test_config_changed_non_leader(
260 # """Test ingress resources without HTTP."""
261 # self.harness.set_leader(is_leader=False)
262 # self.harness.charm.on.config_changed.emit()
265 # self.assertIsInstance(self.harness.charm.unit.status, ActiveStatus)
267 # def test_with_relations_ng(
270 # "Test with relations (ng-ro)"
272 # # Initializing the kafka relation
273 # kafka_relation_id = self.harness.add_relation("kafka", "kafka")
274 # self.harness.add_relation_unit(kafka_relation_id, "kafka/0")
275 # self.harness.update_relation_data(
276 # kafka_relation_id, "kafka/0", {"host": "kafka", "port": 9092}
279 # # Initializing the mongo relation
280 # mongodb_relation_id = self.harness.add_relation("mongodb", "mongodb")
281 # self.harness.add_relation_unit(mongodb_relation_id, "mongodb/0")
282 # self.harness.update_relation_data(
283 # mongodb_relation_id,
285 # {"connection_string": "mongodb://mongo:27017"},
288 # self.harness.charm.on.config_changed.emit()
291 # self.assertNotIsInstance(self.harness.charm.unit.status, BlockedStatus)
294 # if __name__ == "__main__":