fix one test

OpenTMI · Aug 1, 2024 · 31677bf · 31677bf
1 parent c5a5217
commit 31677bf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_StfClient.py b/test/test_StfClient.py
@@ -205,7 +205,7 @@ def test_allocation_context_first_success(self):
     @patch('time.sleep', side_effect=MagicMock())
     def test_allocation_context_wait_success(self, mock_sleep):
         dev1 = {'serial': '123', 'present': True, 'ready': True, 'using': False, 'owner': None, 'status': 3}
-        self.client.get_devices = MagicMock(side_effect=[[], [dev1]])
+        self.client.get_devices = MagicMock(side_effect=[[dev1], [], [dev1]])
         url = '123'
 
         with self.client.allocation_context({"serial": '123'}) as device: