From fbdccf4dd7ca1747629c3e49412a1f2a7123cf6a Mon Sep 17 00:00:00 2001
From: Klaus Wakonig <klaus.wakonig@psi.ch>
Date: Thu, 8 Sep 2022 21:54:34 +0200
Subject: [PATCH] replaced staging in tests by direct controller.on calls

---
 tests/test_galil.py   | 2 +-
 tests/test_smaract.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_galil.py b/tests/test_galil.py
index f4814a2..ba98480 100644
--- a/tests/test_galil.py
+++ b/tests/test_galil.py
@@ -21,7 +21,7 @@ def test_axis_get(pos, msg, sign):
         socket_cls=SocketMock,
     )
     leyey.controller.sock.flush_buffer()
-    leyey.stage()
+    leyey.controller.on()
     leyey.controller.sock.buffer_recv = msg
     val = leyey.read()
     assert val["leyey"]["value"] == pos
diff --git a/tests/test_smaract.py b/tests/test_smaract.py
index af3f624..6ad5588 100644
--- a/tests/test_smaract.py
+++ b/tests/test_smaract.py
@@ -203,7 +203,7 @@ def test_move_axis(pos, get_msg, return_msg):
         sign=1,
         socket_cls=SocketMock,
     )
-    lsmarA.stage()
+    lsmarA.controller.on()
     controller = lsmarA.controller
     controller.sock.flush_buffer()
     controller.sock.buffer_recv = return_msg