installers/charm/osm-nbi/tests/unit/test_charm.py

   1 #!/usr/bin/env python3
   2 # Copyright 2022 Canonical Ltd.
   3 #
   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
   7 #
   8 #         http://www.apache.org/licenses/LICENSE-2.0
   9 #
  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
  14 # under the License.
  15 #
  16 # For those usages not covered by the Apache License, Version 2.0 please
  17 # contact: legal@canonical.com
  18 #
  19 # To get in touch with the maintainers, please contact:
  20 # osm-charmers@lists.launchpad.net
  21 #
  22 # Learn more about testing at: https://juju.is/docs/sdk/testing
  23
  24 import pytest
  25 from ops.model import ActiveStatus, BlockedStatus
  26 from ops.testing import Harness
  27 from pytest_mock import MockerFixture
  28
  29 from charm import CharmError, OsmNbiCharm, check_service_active
  30
  31 container_name = "nbi"
  32 service_name = "nbi"
  33
  34
  35 @pytest.fixture
  36 def harness(mocker: MockerFixture):
  37     mocker.patch("charm.KubernetesServicePatch", lambda x, y: None)
  38     harness = Harness(OsmNbiCharm)
  39     harness.begin()
  40     yield harness
  41     harness.cleanup()
  42
  43
  44 def test_missing_relations(harness: Harness):
  45     harness.charm.on.config_changed.emit()
  46     assert type(harness.charm.unit.status) == BlockedStatus
  47     assert all(
  48         relation in harness.charm.unit.status.message
  49         for relation in ["mongodb", "kafka", "prometheus", "keystone"]
  50     )
  51
  52
  53 def test_ready(harness: Harness):
  54     _add_relations(harness)
  55     assert harness.charm.unit.status == ActiveStatus()
  56
  57
  58 def test_container_stops_after_relation_broken(harness: Harness):
  59     harness.charm.on[container_name].pebble_ready.emit(container_name)
  60     container = harness.charm.unit.get_container(container_name)
  61     relation_ids = _add_relations(harness)
  62     check_service_active(container, service_name)
  63     harness.remove_relation(relation_ids[0])
  64     with pytest.raises(CharmError):
  65         check_service_active(container, service_name)
  66
  67
  68 def _add_relations(harness: Harness):
  69     relation_ids = []
  70     # Add mongo relation
  71     relation_id = harness.add_relation("mongodb", "mongodb")
  72     harness.add_relation_unit(relation_id, "mongodb/0")
  73     harness.update_relation_data(
  74         relation_id, "mongodb/0", {"connection_string": "mongodb://:1234"}
  75     )
  76     relation_ids.append(relation_id)
  77     # Add kafka relation
  78     relation_id = harness.add_relation("kafka", "kafka")
  79     harness.add_relation_unit(relation_id, "kafka/0")
  80     harness.update_relation_data(relation_id, "kafka", {"host": "kafka", "port": 9092})
  81     relation_ids.append(relation_id)
  82     # Add prometheus relation
  83     relation_id = harness.add_relation("prometheus", "prometheus")
  84     harness.add_relation_unit(relation_id, "prometheus/0")
  85     harness.update_relation_data(
  86         relation_id, "prometheus", {"hostname": "prometheus", "port": 9090}
  87     )
  88     relation_ids.append(relation_id)
  89     # Add keystone relation
  90     relation_id = harness.add_relation("keystone", "keystone")
  91     harness.add_relation_unit(relation_id, "keystone/0")
  92     harness.update_relation_data(
  93         relation_id,
  94         "keystone",
  95         {
  96             "host": "host",
  97             "port": "port",
  98             "user_domain_name": "user_domain_name",
  99             "project_domain_name": "project_domain_name",
 100             "username": "username",
 101             "password": "password",
 102             "service": "service",
 103             "keystone_db_password": "keystone_db_password",
 104             "region_id": "region_id",
 105             "admin_username": "admin_username",
 106             "admin_password": "admin_password",
 107             "admin_project_name": "admin_project_name",
 108         },
 109     )
 110     relation_ids.append(relation_id)
 111     return relation_ids