Skip to content

Commit 6f4b271

Browse files
Add missing monkeypatch method for unit tests
1 parent ee4b098 commit 6f4b271

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

tests/unit/conftest.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -77,8 +77,10 @@ def start(self, services: list[str] = None, *_, **__):
7777
for service in services:
7878
assert service in ("mysqlrouter-service", "mysqlrouter-exporter")
7979

80-
self.services["mysqlrouter-service"]["active"] = "mysqlrouter-service" in services
81-
self.services["mysqlrouter-exporter"]["active"] = "mysqlrouter-exporter" in services
80+
if "mysqlrouter-service" in services:
81+
self.services["mysqlrouter-service"]["active"] = True
82+
if "mysqlrouter-exporter" in services:
83+
self.services["mysqlrouter-exporter"]["active"] = True
8284

8385
def stop(self, services: list[str] = None, *_, **__):
8486
for service in services:
@@ -89,6 +91,12 @@ def stop(self, services: list[str] = None, *_, **__):
8991
if "mysqlrouter-exporter" in services:
9092
self.services["mysqlrouter-exporter"]["active"] = False
9193

94+
def restart(self, services: list[str] = []):
95+
if "mysqlrouter-service" in services:
96+
self.services["mysqlrouter-service"]["active"] = True
97+
if "mysqlrouter-exporter" in services:
98+
self.services["mysqlrouter-exporter"]["active"] = True
99+
92100
monkeypatch.setattr(snap, "_snap", Snap())
93101

94102
monkeypatch.setattr(

0 commit comments

Comments
 (0)