#!/usr/bin/env python3
# Copyright 2022 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
# a copy of the License at
#
#         http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
#
# For those usages not covered by the Apache License, Version 2.0 please
# contact: legal@canonical.com
#
# To get in touch with the maintainers, please contact:
# osm-charmers@lists.launchpad.net
#
# Learn more about testing at: https://juju.is/docs/sdk/testing

import pytest
from ops.model import ActiveStatus, BlockedStatus
from ops.testing import Harness
from pytest_mock import MockerFixture

from charm import CharmError, OsmNbiCharm, check_service_active

container_name = "nbi"
service_name = "nbi"


@pytest.fixture
def harness(mocker: MockerFixture):
    mocker.patch("charm.KubernetesServicePatch", lambda x, y: None)
    harness = Harness(OsmNbiCharm)
    harness.begin()
    yield harness
    harness.cleanup()


def test_missing_relations(harness: Harness):
    harness.charm.on.config_changed.emit()
    assert type(harness.charm.unit.status) == BlockedStatus
    assert all(
        relation in harness.charm.unit.status.message
        for relation in ["mongodb", "kafka", "prometheus", "keystone"]
    )


def test_ready(harness: Harness):
    _add_relations(harness)
    assert harness.charm.unit.status == ActiveStatus()


def test_container_stops_after_relation_broken(harness: Harness):
    harness.charm.on[container_name].pebble_ready.emit(container_name)
    container = harness.charm.unit.get_container(container_name)
    relation_ids = _add_relations(harness)
    check_service_active(container, service_name)
    harness.remove_relation(relation_ids[0])
    with pytest.raises(CharmError):
        check_service_active(container, service_name)


def test_nbi_relation_joined(harness: Harness):
    harness.set_leader(True)
    _add_relations(harness)
    relation_id = harness.add_relation("nbi", "ng-ui")
    harness.add_relation_unit(relation_id, "ng-ui/0")
    relation_data = harness.get_relation_data(relation_id, harness.charm.app.name)
    assert harness.charm.unit.status == ActiveStatus()
    assert relation_data == {"host": harness.charm.app.name, "port": "9999"}


def _add_relations(harness: Harness):
    relation_ids = []
    # Add mongo relation
    relation_id = harness.add_relation("mongodb", "mongodb")
    harness.add_relation_unit(relation_id, "mongodb/0")
    harness.update_relation_data(
        relation_id, "mongodb/0", {"connection_string": "mongodb://:1234"}
    )
    relation_ids.append(relation_id)
    # Add kafka relation
    relation_id = harness.add_relation("kafka", "kafka")
    harness.add_relation_unit(relation_id, "kafka/0")
    harness.update_relation_data(relation_id, "kafka", {"host": "kafka", "port": 9092})
    relation_ids.append(relation_id)
    # Add prometheus relation
    relation_id = harness.add_relation("prometheus", "prometheus")
    harness.add_relation_unit(relation_id, "prometheus/0")
    harness.update_relation_data(
        relation_id, "prometheus", {"hostname": "prometheus", "port": 9090}
    )
    relation_ids.append(relation_id)
    # Add keystone relation
    relation_id = harness.add_relation("keystone", "keystone")
    harness.add_relation_unit(relation_id, "keystone/0")
    harness.update_relation_data(
        relation_id,
        "keystone",
        {
            "host": "host",
            "port": "port",
            "user_domain_name": "user_domain_name",
            "project_domain_name": "project_domain_name",
            "username": "username",
            "password": "password",
            "service": "service",
            "keystone_db_password": "keystone_db_password",
            "region_id": "region_id",
            "admin_username": "admin_username",
            "admin_password": "admin_password",
            "admin_project_name": "admin_project_name",
        },
    )
    relation_ids.append(relation_id)
    return relation_ids
