add simple test

OpenTMI · Aug 1, 2024 · a7b977f · a7b977f
1 parent 31677bf
commit a7b977f
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/test/test_StfClient.py b/test/test_StfClient.py
@@ -107,6 +107,14 @@ class MockResp:
         with self.assertRaises(DeviceNotFound):
             self.client.find_and_allocate({})
 
+    def test_fail_fast(self):
+
+        class MockResp:
+            devices = [{'serial': 123, 'present': True, 'ready': True, 'using': False, 'owner': None, 'status': 1}]
+        self.DevicesApi.return_value.get_devices = MagicMock(return_value=MockResp())
+        with self.assertRaises(DeviceNotFound):
+            self.client.find_and_allocate({})
+
     def test_list_devices(self):
         available = {'serial': 123, 'present': True, 'ready': True, 'using': False, 'owner': None, 'status': 3}
         self.client.get_devices = MagicMock(return_value=[available])