--- /dev/null
+#!/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 asyncio
+import json
+import logging
+from pathlib import Path
+
+import pytest
+import yaml
+from pytest_operator.plugin import OpsTest
+
+logger = logging.getLogger(__name__)
+
+METADATA = yaml.safe_load(Path("./metadata.yaml").read_text())
+MON_APP = METADATA["name"]
+GRAFANA_CHARM = "grafana-k8s"
+GRAFANA_APP = "grafana"
+DS_NAME = "osm_prometheus"
+DS_URL = "http://prometheus:9090"
+
+
+async def _run_action_create_datasource(ops_test: OpsTest, name: str, url: str):
+ action = (
+ await ops_test.model.applications[MON_APP]
+ .units[0]
+ .run_action(
+ action_name="create-datasource",
+ name=name,
+ url=url,
+ )
+ )
+ return await action.wait()
+
+
+async def _run_action_list_datasources(ops_test: OpsTest):
+ action = (
+ await ops_test.model.applications[MON_APP]
+ .units[0]
+ .run_action(action_name="list-datasources")
+ )
+ return await action.wait()
+
+
+async def _run_action_delete_datasource(ops_test: OpsTest, name: str):
+ action = (
+ await ops_test.model.applications[MON_APP]
+ .units[0]
+ .run_action(action_name="delete-datasource", name=name)
+ )
+ return await action.wait()
+
+
+@pytest.mark.abort_on_fail
+async def test_mon_and_grafana_are_idle(ops_test: OpsTest):
+ charm = await ops_test.build_charm(".")
+ resources = {"mon-image": METADATA["resources"]["mon-image"]["upstream-source"]}
+
+ await asyncio.gather(
+ ops_test.model.deploy(
+ charm, resources=resources, application_name=MON_APP, series="jammy"
+ ),
+ ops_test.model.deploy(GRAFANA_CHARM, application_name=GRAFANA_APP, channel="stable"),
+ )
+ async with ops_test.fast_forward():
+ await ops_test.model.wait_for_idle(apps=[GRAFANA_APP], status="active")
+ await ops_test.model.wait_for_idle(apps=[MON_APP], status="blocked")
+
+
+@pytest.mark.abort_on_fail
+async def test_mon_cannot_create_datasource_grafana_url_is_not_set(ops_test: OpsTest):
+ action = await _run_action_create_datasource(ops_test, DS_NAME, DS_URL)
+ assert action.status == "failed"
+ assert "Invalid URL" in action.message
+
+
+@pytest.mark.abort_on_fail
+async def test_mon_cannot_list_datasources_grafana_url_is_not_set(ops_test: OpsTest):
+ action = await _run_action_list_datasources(ops_test)
+ assert action.status == "failed"
+ assert "Invalid URL" in action.message
+
+
+@pytest.mark.abort_on_fail
+async def test_mon_cannot_delete_datasource_grafana_url_is_not_set(ops_test: OpsTest):
+ action = await _run_action_delete_datasource(ops_test, "prometheus")
+ assert action.status == "failed"
+ assert "Invalid URL" in action.message
+
+
+@pytest.mark.abort_on_fail
+async def test_mon_cannot_create_datasource_due_to_invalid_grafana_password(ops_test: OpsTest):
+ await ops_test.model.applications[MON_APP].set_config(
+ {"grafana-url": f"http://{GRAFANA_APP}:3000", "grafana-user": "admin"}
+ )
+ async with ops_test.fast_forward():
+ await ops_test.model.wait_for_idle(apps=[MON_APP], status="blocked")
+
+ action = await _run_action_create_datasource(ops_test, DS_NAME, DS_URL)
+ assert action.status == "failed"
+ assert action.message == "invalid username or password"
+
+
+@pytest.mark.abort_on_fail
+async def test_mon_cannot_create_datasource_invalid_url_fails(ops_test: OpsTest):
+ await ops_test.model.applications[MON_APP].set_config(
+ {"grafana-url": f"http://{GRAFANA_APP}:3000"}
+ )
+ async with ops_test.fast_forward():
+ await ops_test.model.wait_for_idle(apps=[MON_APP], status="blocked")
+
+ action = await _run_action_create_datasource(ops_test, DS_NAME, "prometheus:9090")
+ assert action.status == "failed"
+ assert action.message == "Invalid datasource url 'prometheus:9090'"
+
+
+async def _get_grafana_admin_password(ops_test: OpsTest):
+ action = (
+ await ops_test.model.applications[GRAFANA_APP].units[0].run_action("get-admin-password")
+ )
+ admin_password = (await action.wait()).results["admin-password"]
+ logger.info(f"Password obtained from {GRAFANA_APP} : {admin_password}")
+ assert admin_password != "Admin password has been changed by an administrator"
+ return admin_password
+
+
+@pytest.mark.abort_on_fail
+async def test_grafana_password_is_set_in_mon_config(ops_test: OpsTest):
+ admin_password = await _get_grafana_admin_password(ops_test)
+
+ default_password = None
+ grafana_password_in_mon = (
+ (await ops_test.model.applications[MON_APP].get_config())
+ .get("grafana-password")
+ .get("value")
+ )
+ assert grafana_password_in_mon == default_password
+
+ await ops_test.model.applications[MON_APP].set_config({"grafana-password": admin_password})
+
+ async with ops_test.fast_forward():
+ await ops_test.model.wait_for_idle(apps=[GRAFANA_APP], status="active")
+ await ops_test.model.wait_for_idle(apps=[MON_APP], status="blocked")
+
+ grafana_password_in_mon = (
+ (await ops_test.model.applications[MON_APP].get_config())
+ .get("grafana-password")
+ .get("value")
+ )
+ assert grafana_password_in_mon == admin_password
+
+
+@pytest.mark.abort_on_fail
+async def test_mon_create_datasource_successfully(ops_test: OpsTest):
+ action = await _run_action_create_datasource(ops_test, DS_NAME, DS_URL)
+ assert action.status == "completed"
+ assert action.results["datasource-name"] == DS_NAME
+
+
+@pytest.mark.abort_on_fail
+async def test_mon_list_datasources_returns_osm_prometheus(ops_test: OpsTest):
+ action = await _run_action_list_datasources(ops_test)
+ assert action.status == "completed"
+ datasources = json.loads(action.results.get("datasources"))
+ assert len(datasources) == 1
+ assert datasources[0].get("name") == DS_NAME
+ assert datasources[0].get("type") == "prometheus"
+
+
+@pytest.mark.abort_on_fail
+async def test_mon_create_datasource_that_already_exists_returns_error_message(ops_test: OpsTest):
+ action = await _run_action_create_datasource(ops_test, DS_NAME, DS_URL)
+ assert action.status == "failed"
+ assert action.message == "data source with the same name already exists"
+
+
+@pytest.mark.abort_on_fail
+async def test_mon_delete_existing_datasource_returns_success_message(ops_test: OpsTest):
+ action = await _run_action_delete_datasource(ops_test, DS_NAME)
+ assert action.status == "completed"
+
+
+@pytest.mark.abort_on_fail
+async def test_mon_list_datasources_is_empty(ops_test: OpsTest):
+ action = await _run_action_list_datasources(ops_test)
+ assert action.status == "completed"
+ datasources = json.loads(action.results.get("datasources"))
+ assert len(datasources) == 0
+
+
+@pytest.mark.abort_on_fail
+async def test_mon_delete_non_existing_datasource_returns_error_message(ops_test: OpsTest):
+ action = await _run_action_delete_datasource(ops_test, DS_NAME)
+ assert action.status == "failed"
+ assert action.message == "Data source not found"
+
+
+@pytest.mark.abort_on_fail
+async def test_mon_create_datasource_incorrect_grafana_host_fails(ops_test: OpsTest):
+ await ops_test.model.applications[MON_APP].set_config(
+ {"grafana-url": f"http://{GRAFANA_APP}-k8s:3000"}
+ )
+ async with ops_test.fast_forward():
+ await ops_test.model.wait_for_idle(apps=[MON_APP], status="blocked")
+ action = await _run_action_create_datasource(ops_test, DS_NAME, DS_URL)
+ assert action.status == "failed"
+ assert f"Failed to resolve '{GRAFANA_APP}-k8s'" in action.message
+
+
+@pytest.mark.abort_on_fail
+async def test_mon_list_datasources_incorrect_grafana_host_fails(ops_test: OpsTest):
+ action = await _run_action_list_datasources(ops_test)
+ assert action.status == "failed"
+ assert f"Failed to resolve '{GRAFANA_APP}-k8s'" in action.message
+
+
+@pytest.mark.abort_on_fail
+async def test_mon_delete_datasource_incorrect_grafana_host_fails(ops_test: OpsTest):
+ action = await _run_action_delete_datasource(ops_test, DS_NAME)
+ assert action.status == "failed"
+ assert f"Failed to resolve '{GRAFANA_APP}-k8s'" in action.message
+
+
+@pytest.mark.abort_on_fail
+async def test_mon_incorrect_grafana_port_returns_timeout_message(ops_test: OpsTest):
+ await ops_test.model.applications[MON_APP].set_config(
+ {"grafana-url": f"http://{GRAFANA_APP}:3001"}
+ )
+ async with ops_test.fast_forward():
+ await ops_test.model.wait_for_idle(apps=[MON_APP], status="blocked")
+ action = await _run_action_create_datasource(ops_test, DS_NAME, DS_URL)
+ assert action.status == "failed"
+ assert f"Connection to {GRAFANA_APP} timed out" in action.message
+
+
+@pytest.mark.abort_on_fail
+async def test_mon_list_datasources_incorrect_user_fails(ops_test: OpsTest):
+ await ops_test.model.applications[MON_APP].set_config(
+ {"grafana-url": f"http://{GRAFANA_APP}:3000", "grafana-user": "some_user"}
+ )
+ async with ops_test.fast_forward():
+ await ops_test.model.wait_for_idle(apps=[MON_APP], status="blocked")
+ action = await _run_action_list_datasources(ops_test)
+ assert action.status == "failed"
+ assert action.message == "invalid username or password"
+
+
+@pytest.mark.abort_on_fail
+async def test_mon_delete_datasource_incorrect_user_fails(ops_test: OpsTest):
+ action = await _run_action_delete_datasource(ops_test, DS_NAME)
+ assert action.status == "failed"
+ assert action.message == "invalid username or password"