fix reboot metabox scenario

canonical · Oct 18, 2023 · b4cd5ad · b4cd5ad
1 parent 782265d
commit b4cd5ad
Showing 1 changed file with 21 additions and 19 deletions.
diff --git a/metabox/metabox/scenarios/restart/launcher.py b/metabox/metabox/scenarios/restart/launcher.py
@@ -21,26 +21,28 @@
 
 from metabox.core.actions import AssertPrinted
 from metabox.core.scenario import Scenario
+from metabox.core.utils import tag
 
-class Reboot(Scenario):
 
-    modes = ['remote']
-    launcher = textwrap.dedent("""
-    [launcher]
-    launcher_version = 1
-    stock_reports = text
-    [test plan]
-    unit = com.canonical.certification::power-automated
-    forced = yes
-    [test selection]
-    forced = yes
-    exclude = .*cold.*
-    [ui]
-    type = silent
-    """)
+@tag("reboot")
+class Reboot(Scenario):
+    modes = ["remote"]
+    launcher = textwrap.dedent(
+        """
+        [launcher]
+        launcher_version = 1
+        stock_reports = text
+        [test plan]
+        unit = com.canonical.certification::power-automated
+        forced = yes
+        [test selection]
+        forced = yes
+        exclude = .*cold.*
+        [ui]
+        type = silent
+        """
+    )
     steps = [
-        AssertPrinted('Connection lost!'),
-        AssertPrinted('Reconnecting'),
-        AssertPrinted("Reconnected\s+\(took"),
-        AssertPrinted('job passed   : Warm reboot'),
+        AssertPrinted("Connection lost!"),
+        AssertPrinted("job passed   : Warm reboot"),
     ]