examples: decorate strategy's transition() with @raise_if_broken

In order to get the word out about the @raise_if_broken decorator,
decorate the strategies in labgrid with it.

Signed-off-by: Bastian Krause <[email protected]>

Author: Bastian-Krause

examples/qemu-networking/qemunetworkstrategy.py

@@ -17,7 +17,7 @@
 import attr
 
 from labgrid import target_factory, step
-from labgrid.strategy import Strategy, StrategyError
+from labgrid.strategy import Strategy, StrategyError, raise_if_broken
 from labgrid.util import get_free_port
 
 
@@ -77,6 +77,7 @@ def update_network_service(self):
                 networkservice.address = new_address
                 networkservice.port = self.__remote_port
 
+    @raise_if_broken
     @step(args=["state"])
     def transition(self, state, *, step):
         if not isinstance(state, Status):

examples/strategy/bareboxrebootstrategy.py

@@ -5,7 +5,7 @@
 from labgrid import target_factory, step
 from labgrid.driver import BareboxDriver, ShellDriver
 from labgrid.protocol import PowerProtocol
-from labgrid.strategy import Strategy
+from labgrid.strategy import Strategy, raise_if_broken
 
 
 @attr.s(eq=False)
@@ -58,6 +58,7 @@ class BareboxRebootStrategy(Strategy):
     def __attrs_post_init__(self):
         super().__attrs_post_init__()
 
+    @raise_if_broken
     @step(args=["new_status"])
     def transition(self, new_status, *, step):
         if not isinstance(new_status, Status):

examples/strategy/quartusstrategy.py

@@ -5,7 +5,7 @@
 from labgrid import target_factory, step
 from labgrid.driver import QuartusHPSDriver, SerialDriver
 from labgrid.protocol import PowerProtocol
-from labgrid.strategy import Strategy
+from labgrid.strategy import Strategy, raise_if_broken
 
 
 @attr.s(eq=False)
@@ -37,6 +37,7 @@ class QuartusHPSStrategy(Strategy):
     def __attrs_post_init__(self):
         super().__attrs_post_init__()
 
+    @raise_if_broken
     @step(args=["status"])
     def transition(self, status, *, step):
         if not isinstance(status, Status):

examples/usbpower/examplestrategy.py

@@ -5,7 +5,7 @@
 from labgrid.driver import BareboxDriver, ShellDriver, USBSDMuxDriver
 from labgrid import step, target_factory
 from labgrid.protocol import PowerProtocol
-from labgrid.strategy import Strategy
+from labgrid.strategy import Strategy, raise_if_broken
 
 
 @attr.s(eq=False)
@@ -36,6 +36,7 @@ class ExampleStrategy(Strategy):
     def __attrs_post_init__(self):
         super().__attrs_post_init__()
 
+    @raise_if_broken
     @step(args=["status"])
     def transition(self, status, *, step):
         if not isinstance(status, Status):