From c2188c044bec1389bc6f8ba075ed7851fc33a2a0 Mon Sep 17 00:00:00 2001
From: Dragomir Penev <dragomir.penev@canonical.com>
Date: Tue, 19 Nov 2024 16:29:52 +0200
Subject: [PATCH] Integration test

---
 .../new_relations/test_new_relations.py       | 38 ++++++++++++++++++-
 1 file changed, 37 insertions(+), 1 deletion(-)

diff --git a/tests/integration/new_relations/test_new_relations.py b/tests/integration/new_relations/test_new_relations.py
index 24307164eb..6070fc469e 100644
--- a/tests/integration/new_relations/test_new_relations.py
+++ b/tests/integration/new_relations/test_new_relations.py
@@ -13,7 +13,16 @@
 from pytest_operator.plugin import OpsTest
 
 from .. import markers
-from ..helpers import CHARM_BASE, assert_sync_standbys, get_leader_unit, scale_application
+from ..helpers import (
+    CHARM_BASE,
+    assert_sync_standbys,
+    get_leader_unit,
+    get_machine_from_unit,
+    get_primary,
+    scale_application,
+    start_machine,
+    stop_machine,
+)
 from ..juju_ import juju_major_version
 from .helpers import (
     build_connection_string,
@@ -184,6 +193,33 @@ async def test_database_relation_with_charm_libraries(ops_test: OpsTest):
             cursor.execute("DROP TABLE test;")
 
 
+@pytest.mark.group("new_relations_tests")
+@pytest.mark.abort_on_fail
+async def test_filter_out_degraded_replicas(ops_test: OpsTest):
+    primary = await get_primary(ops_test, f"{DATABASE_APP_NAME}/0")
+    replica = next(
+        unit.name
+        for unit in ops_test.model.applications[DATABASE_APP_NAME].units
+        if unit.name != primary
+    )
+    machine = await get_machine_from_unit(ops_test, replica)
+    await stop_machine(ops_test, machine)
+
+    # Topology observer runs every half a minute
+    await asyncio.sleep(60)
+
+    data = await get_application_relation_data(
+        ops_test,
+        APPLICATION_APP_NAME,
+        FIRST_DATABASE_RELATION_NAME,
+        "read-only-endpoints",
+    )
+    assert len(data.split(",")) == 1
+
+    await start_machine(ops_test, machine)
+    await ops_test.model.wait_for_idle(apps=DATABASE_APP_NAME, status="active", timeout=200)
+
+
 @pytest.mark.group("new_relations_tests")
 async def test_user_with_extra_roles(ops_test: OpsTest):
     """Test superuser actions and the request for more permissions."""