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
25 from typing
import NoReturn
28 from charm
import MonCharm
29 from ops
.model
import ActiveStatus
, BlockedStatus
30 from ops
.testing
import Harness
33 def encode(content
: str):
34 return base64
.b64encode(content
.encode("ascii")).decode("utf-8")
37 certificate_pem
= encode(
39 -----BEGIN CERTIFICATE-----
40 MIIDazCCAlOgAwIBAgIUf1b0s3UKtrxHXH2rge7UaQyfJAMwDQYJKoZIhvcNAQEL
41 BQAwRTELMAkGA1UEBhMCQVUxEzARBgNVBAgMClNvbWUtU3RhdGUxITAfBgNVBAoM
42 GEludGVybmV0IFdpZGdpdHMgUHR5IEx0ZDAeFw0yMTAzMjIxNzEyMjdaFw0zMTAz
43 MjAxNzEyMjdaMEUxCzAJBgNVBAYTAkFVMRMwEQYDVQQIDApTb21lLVN0YXRlMSEw
44 HwYDVQQKDBhJbnRlcm5ldCBXaWRnaXRzIFB0eSBMdGQwggEiMA0GCSqGSIb3DQEB
45 AQUAA4IBDwAwggEKAoIBAQCgCfCBgYAN6ON0yHDXuW407rFtJVRf0u46Jrp0Dk7J
46 kkSZ1e7Kq14r7yFHazEBWv78oOdwBocvWrd8leLuf3bYGcHR65hRy6A/fbYm5Aje
47 cKpwlFwaqfR4BLelwJl79jZ2rJX738cCBVrIk1nAVdOxGrXV4MTWUaKR2c+uKKvc
48 OKRT+5VqCeP4N5FWeATZ/KqGu8uV9E9WhFgwIZyStemLyLaDbn5PmAQ6S9oeR5jJ
49 o2gEEp/lDKvsqOWs76KFumSKa9hQs5Dw2lj0mb1UoyYK1gYc4ubzVChJadv44AU8
50 MYtIjlFn1X1P+RjaKZNUIAGXkoLwYn6SizF6y6LiuFS9AgMBAAGjUzBRMB0GA1Ud
51 DgQWBBRl+/23CB+FXczeAZRQyYcfOdy9YDAfBgNVHSMEGDAWgBRl+/23CB+FXcze
52 AZRQyYcfOdy9YDAPBgNVHRMBAf8EBTADAQH/MA0GCSqGSIb3DQEBCwUAA4IBAQAd
53 dkeDym6lRN8kWFtfu3IyiLF8G8sn91qNbH3Yr4TuTBhgcjYyW6PgisSbrNgA9ysE
54 GoaF7ohb8GeVfCsQdK23+NpAlj/+DZ3OnGcxwXj1RUAz4yr9kanV1yuEtr1q2xJI
55 UaECWr8HZlwGBAKNTGx2EXT2/2aFzgULpDcxzTKD+MRpKpMUrWhf9ULvVrclvHWe
56 POLYhobUFuBHuo6rt5Rcq16j67zCX9EVTlAE3o2OECIWByK22sXdeOidYMpTkl4q
57 8FrOqjNsx5d+SBPJBv/pqtBm4bA47Vx1P8tbWOQ4bXS0UmXgwpeBOU/O/ot30+KS
59 -----END CERTIFICATE-----
64 class TestCharm(unittest
.TestCase
):
65 """Prometheus Charm unit tests."""
67 def setUp(self
) -> NoReturn
:
69 self
.image_info
= sys
.modules
["oci_image"].OCIImageResource().fetch()
70 self
.harness
= Harness(MonCharm
)
71 self
.harness
.set_leader(is_leader
=True)
74 "vca_host": "192.168.0.13",
76 "vca_secret": "admin",
77 "vca_cacert": "cacert",
78 "database_commonkey": "commonkey",
81 "openstack_default_granularity": 10,
82 "global_request_timeout": 10,
83 "collector_interval": 30,
84 "evaluator_interval": 30,
85 "keystone_enabled": True,
86 "certificates": f
"cert1:{certificate_pem}",
88 self
.harness
.update_config(self
.config
)
90 def test_config_changed_no_relations(
93 """Test ingress resources without HTTP."""
95 self
.harness
.charm
.on
.config_changed
.emit()
98 self
.assertIsInstance(self
.harness
.charm
.unit
.status
, BlockedStatus
)
101 relation
in self
.harness
.charm
.unit
.status
.message
102 for relation
in ["mongodb", "kafka", "prometheus", "keystone"]
106 def test_config_changed_non_leader(
109 """Test ingress resources without HTTP."""
110 self
.harness
.set_leader(is_leader
=False)
111 self
.harness
.charm
.on
.config_changed
.emit()
114 self
.assertIsInstance(self
.harness
.charm
.unit
.status
, ActiveStatus
)
116 def test_with_relations_and_mongodb_config(
119 "Test with relations (internal)"
120 self
.initialize_kafka_relation()
121 self
.initialize_mongo_config()
122 self
.initialize_prometheus_relation()
123 self
.initialize_keystone_relation()
125 self
.assertNotIsInstance(self
.harness
.charm
.unit
.status
, BlockedStatus
)
127 def test_with_relations(
130 "Test with relations (internal)"
131 self
.initialize_kafka_relation()
132 self
.initialize_mongo_relation()
133 self
.initialize_prometheus_relation()
134 self
.initialize_keystone_relation()
136 self
.assertNotIsInstance(self
.harness
.charm
.unit
.status
, BlockedStatus
)
138 def test_exception_mongodb_relation_and_config(
141 "Test with relations and config for mongodb. Must fail"
142 self
.initialize_mongo_relation()
143 self
.initialize_mongo_config()
145 self
.assertIsInstance(self
.harness
.charm
.unit
.status
, BlockedStatus
)
147 def initialize_kafka_relation(self
):
148 kafka_relation_id
= self
.harness
.add_relation("kafka", "kafka")
149 self
.harness
.add_relation_unit(kafka_relation_id
, "kafka/0")
150 self
.harness
.update_relation_data(
151 kafka_relation_id
, "kafka", {"host": "kafka", "port": 9092}
154 def initialize_mongo_config(self
):
155 self
.harness
.update_config({"mongodb_uri": "mongodb://mongo:27017"})
157 def initialize_mongo_relation(self
):
158 mongodb_relation_id
= self
.harness
.add_relation("mongodb", "mongodb")
159 self
.harness
.add_relation_unit(mongodb_relation_id
, "mongodb/0")
160 self
.harness
.update_relation_data(
163 {"connection_string": "mongodb://mongo:27017"},
166 def initialize_prometheus_relation(self
):
167 prometheus_relation_id
= self
.harness
.add_relation("prometheus", "prometheus")
168 self
.harness
.add_relation_unit(prometheus_relation_id
, "prometheus/0")
169 self
.harness
.update_relation_data(
170 prometheus_relation_id
,
172 {"hostname": "prometheus", "port": 9090},
175 def initialize_keystone_relation(self
):
176 keystone_relation_id
= self
.harness
.add_relation("keystone", "keystone")
177 self
.harness
.add_relation_unit(keystone_relation_id
, "keystone/0")
178 self
.harness
.update_relation_data(
179 keystone_relation_id
,
184 "user_domain_name": "ud",
185 "project_domain_name": "pd",
189 "keystone_db_password": "something",
190 "region_id": "something",
191 "admin_username": "something",
192 "admin_password": "something",
193 "admin_project_name": "something",
198 if __name__
== "__main__":
202 # class TestCharm(unittest.TestCase):
203 # """MON Charm unit tests."""
205 # def setUp(self) -> NoReturn:
207 # self.harness = Harness(MonCharm)
208 # self.harness.set_leader(is_leader=True)
209 # self.harness.begin()
211 # def test_on_start_without_relations(self) -> NoReturn:
212 # """Test installation without any relation."""
213 # self.harness.charm.on.start.emit()
216 # self.assertIsInstance(self.harness.charm.unit.status, BlockedStatus)
218 # # Verifying status message
219 # self.assertGreater(len(self.harness.charm.unit.status.message), 0)
221 # self.harness.charm.unit.status.message.startswith("Waiting for ")
223 # self.assertIn("kafka", self.harness.charm.unit.status.message)
224 # self.assertIn("mongodb", self.harness.charm.unit.status.message)
225 # self.assertIn("prometheus", self.harness.charm.unit.status.message)
226 # self.assertTrue(self.harness.charm.unit.status.message.endswith(" relations"))
228 # def test_on_start_with_relations(self) -> NoReturn:
229 # """Test deployment without keystone."""
230 # expected_result = {
235 # "imageDetails": self.harness.charm.image.fetch(),
236 # "imagePullPolicy": "Always",
240 # "containerPort": 8000,
245 # "ALLOW_ANONYMOUS_LOGIN": "yes",
246 # "OSMMON_OPENSTACK_DEFAULT_GRANULARITY": 300,
247 # "OSMMON_GLOBAL_REQUEST_TIMEOUT": 10,
248 # "OSMMON_GLOBAL_LOGLEVEL": "INFO",
249 # "OSMMON_COLLECTOR_INTERVAL": 30,
250 # "OSMMON_EVALUATOR_INTERVAL": 30,
251 # "OSMMON_MESSAGE_DRIVER": "kafka",
252 # "OSMMON_MESSAGE_HOST": "kafka",
253 # "OSMMON_MESSAGE_PORT": 9092,
254 # "OSMMON_DATABASE_DRIVER": "mongo",
255 # "OSMMON_DATABASE_URI": "mongodb://mongo:27017",
256 # "OSMMON_DATABASE_COMMONKEY": "osm",
257 # "OSMMON_PROMETHEUS_URL": "http://prometheus:9090",
258 # "OSMMON_VCA_HOST": "admin",
259 # "OSMMON_VCA_USER": "admin",
260 # "OSMMON_VCA_SECRET": "secret",
261 # "OSMMON_VCA_CACERT": "",
265 # "kubernetesResources": {"ingressResources": []},
268 # self.harness.charm.on.start.emit()
270 # # Check if kafka datastore is initialized
271 # self.assertIsNone(self.harness.charm.state.message_host)
272 # self.assertIsNone(self.harness.charm.state.message_port)
274 # # Check if mongodb datastore is initialized
275 # self.assertIsNone(self.harness.charm.state.database_uri)
277 # # Check if prometheus datastore is initialized
278 # self.assertIsNone(self.harness.charm.state.prometheus_host)
279 # self.assertIsNone(self.harness.charm.state.prometheus_port)
281 # # Initializing the kafka relation
282 # kafka_relation_id = self.harness.add_relation("kafka", "kafka")
283 # self.harness.add_relation_unit(kafka_relation_id, "kafka/0")
284 # self.harness.update_relation_data(
285 # kafka_relation_id, "kafka/0", {"host": "kafka", "port": 9092}
288 # # Initializing the mongo relation
289 # mongodb_relation_id = self.harness.add_relation("mongodb", "mongodb")
290 # self.harness.add_relation_unit(mongodb_relation_id, "mongodb/0")
291 # self.harness.update_relation_data(
292 # mongodb_relation_id,
294 # {"connection_string": "mongodb://mongo:27017"},
297 # # Initializing the prometheus relation
298 # prometheus_relation_id = self.harness.add_relation("prometheus", "prometheus")
299 # self.harness.add_relation_unit(prometheus_relation_id, "prometheus/0")
300 # self.harness.update_relation_data(
301 # prometheus_relation_id,
303 # {"hostname": "prometheus", "port": 9090},
306 # # Checking if kafka data is stored
307 # self.assertEqual(self.harness.charm.state.message_host, "kafka")
308 # self.assertEqual(self.harness.charm.state.message_port, 9092)
310 # # Checking if mongodb data is stored
311 # self.assertEqual(self.harness.charm.state.database_uri, "mongodb://mongo:27017")
313 # # Checking if prometheus data is stored
314 # self.assertEqual(self.harness.charm.state.prometheus_host, "prometheus")
315 # self.assertEqual(self.harness.charm.state.prometheus_port, 9090)
318 # self.assertNotIsInstance(self.harness.charm.unit.status, BlockedStatus)
320 # pod_spec, _ = self.harness.get_pod_spec()
322 # self.assertDictEqual(expected_result, pod_spec)
324 # def test_on_kafka_unit_relation_changed(self) -> NoReturn:
325 # """Test to see if kafka relation is updated."""
326 # self.harness.charm.on.start.emit()
328 # self.assertIsNone(self.harness.charm.state.message_host)
329 # self.assertIsNone(self.harness.charm.state.message_port)
331 # relation_id = self.harness.add_relation("kafka", "kafka")
332 # self.harness.add_relation_unit(relation_id, "kafka/0")
333 # self.harness.update_relation_data(
334 # relation_id, "kafka/0", {"host": "kafka", "port": 9092}
337 # self.assertEqual(self.harness.charm.state.message_host, "kafka")
338 # self.assertEqual(self.harness.charm.state.message_port, 9092)
341 # self.assertIsInstance(self.harness.charm.unit.status, BlockedStatus)
343 # # Verifying status message
344 # self.assertGreater(len(self.harness.charm.unit.status.message), 0)
346 # self.harness.charm.unit.status.message.startswith("Waiting for ")
348 # self.assertNotIn("kafka", self.harness.charm.unit.status.message)
349 # self.assertIn("mongodb", self.harness.charm.unit.status.message)
350 # self.assertIn("prometheus", self.harness.charm.unit.status.message)
351 # self.assertTrue(self.harness.charm.unit.status.message.endswith(" relations"))
353 # def test_on_mongodb_unit_relation_changed(self) -> NoReturn:
354 # """Test to see if mongodb relation is updated."""
355 # self.harness.charm.on.start.emit()
357 # self.assertIsNone(self.harness.charm.state.database_uri)
359 # relation_id = self.harness.add_relation("mongodb", "mongodb")
360 # self.harness.add_relation_unit(relation_id, "mongodb/0")
361 # self.harness.update_relation_data(
362 # relation_id, "mongodb/0", {"connection_string": "mongodb://mongo:27017"}
365 # self.assertEqual(self.harness.charm.state.database_uri, "mongodb://mongo:27017")
368 # self.assertIsInstance(self.harness.charm.unit.status, BlockedStatus)
370 # # Verifying status message
371 # self.assertGreater(len(self.harness.charm.unit.status.message), 0)
373 # self.harness.charm.unit.status.message.startswith("Waiting for ")
375 # self.assertIn("kafka", self.harness.charm.unit.status.message)
376 # self.assertNotIn("mongodb", self.harness.charm.unit.status.message)
377 # self.assertIn("prometheus", self.harness.charm.unit.status.message)
378 # self.assertTrue(self.harness.charm.unit.status.message.endswith(" relations"))
380 # def test_on_prometheus_unit_relation_changed(self) -> NoReturn:
381 # """Test to see if prometheus relation is updated."""
382 # self.harness.charm.on.start.emit()
384 # self.assertIsNone(self.harness.charm.state.prometheus_host)
385 # self.assertIsNone(self.harness.charm.state.prometheus_port)
387 # relation_id = self.harness.add_relation("prometheus", "prometheus")
388 # self.harness.add_relation_unit(relation_id, "prometheus/0")
389 # self.harness.update_relation_data(
390 # relation_id, "prometheus", {"hostname": "prometheus", "port": 9090}
393 # self.assertEqual(self.harness.charm.state.prometheus_host, "prometheus")
394 # self.assertEqual(self.harness.charm.state.prometheus_port, 9090)
397 # self.assertIsInstance(self.harness.charm.unit.status, BlockedStatus)
399 # # Verifying status message
400 # self.assertGreater(len(self.harness.charm.unit.status.message), 0)
402 # self.harness.charm.unit.status.message.startswith("Waiting for ")
404 # self.assertIn("kafka", self.harness.charm.unit.status.message)
405 # self.assertIn("mongodb", self.harness.charm.unit.status.message)
406 # self.assertNotIn("prometheus", self.harness.charm.unit.status.message)
407 # self.assertTrue(self.harness.charm.unit.status.message.endswith(" relations"))
410 # if __name__ == "__main__":