update unit tests

Author: carlcsaposs-canonical

tests/unit/conftest.py

@@ -25,6 +25,20 @@ def disable_tenacity_retry(monkeypatch):
         monkeypatch.setattr(f"tenacity.{retry_class}.__call__", lambda *args, **kwargs: False)
 
 
+class _MockRefresh:
+    in_progress = False
+    next_unit_allowed_to_refresh = True
+    workload_allowed_to_start = True
+    app_status_higher_priority = None
+    unit_status_higher_priority = None
+
+    def __init__(self, _, /):
+        pass
+
+    def unit_status_lower_priority(self, *, workload_is_running=True):
+        return None
+
+
 @pytest.fixture(autouse=True)
 def patch(monkeypatch):
     monkeypatch.setattr(
@@ -37,9 +51,14 @@ def patch(monkeypatch):
         "mysql_shell.Shell.get_mysql_router_user_for_unit", lambda *args, **kwargs: None
     )
     monkeypatch.setattr("mysql_shell.Shell.is_router_in_cluster_set", lambda *args, **kwargs: True)
-    monkeypatch.setattr("upgrade.Upgrade.in_progress", False)
-    monkeypatch.setattr("upgrade.Upgrade.versions_set", True)
-    monkeypatch.setattr("upgrade.Upgrade.is_compatible", True)
+    monkeypatch.setattr("charm_refresh.Kubernetes", _MockRefresh)
+    monkeypatch.setattr(
+        "charm_refresh.CharmSpecificCommon.__post_init__", lambda *args, **kwargs: None
+    )
+    monkeypatch.setattr(
+        "relations.database_requires.RelationEndpoint.does_relation_exist",
+        lambda *args, **kwargs: True,
+    )
 
 
 @pytest.fixture(autouse=True)

tests/unit/scenario_/database_relations/test_database_relations.py

@@ -39,7 +39,7 @@ def output_states(*, relations: list[scenario.Relation]) -> typing.Iterable[scen
         relations=[
             *relations,
             scenario.PeerRelation(endpoint="mysql-router-peers"),
-            scenario.PeerRelation(endpoint="upgrade-version-a"),
+            scenario.PeerRelation(endpoint="refresh-v-three"),
         ],
         containers=[container],
         leader=True,

tests/unit/scenario_/database_relations/test_database_relations_breaking.py

@@ -21,7 +21,7 @@ def output_state(
         relations=[
             *relations,
             scenario.PeerRelation(endpoint="mysql-router-peers"),
-            scenario.PeerRelation(endpoint="upgrade-version-a"),
+            scenario.PeerRelation(endpoint="refresh-v-three"),
         ],
         containers=[container],
         secrets=secrets,

tests/unit/scenario_/test_start.py

@@ -20,7 +20,7 @@ def test_start_sets_status_if_no_relations(leader, can_connect, unit_status):
         leader=leader,
         relations=[
             scenario.PeerRelation(endpoint="mysql-router-peers"),
-            scenario.PeerRelation(endpoint="upgrade-version-a"),
+            scenario.PeerRelation(endpoint="refresh-v-three"),
         ],
     )
     output_state = context.run("start", input_state)