Added workaround and debug details for setFrequency issue

2 files changed, 53 insertions, 18 deletions

diff --git a/lab_control/test/jds6600_test.py b/lab_control/test/jds6600_test.py
index 817ccf4..59e2d33 100644
--- a/lab_control/test/jds6600_test.py
+++ b/lab_control/test/jds6600_test.py
@@ -17,7 +17,7 @@ def uut(mockDevice):
 
 def checkNumericalParameter(testValues, writeValue, valueInMock):
     for ch in JDS6600.AVAILABLE_CHANNELS:
-        assert valueInMock(ch) == None
+        assert valueInMock(ch) is None
 
         for value in testValues:
             writeValue(ch, value)
@@ -75,3 +75,22 @@ def test_invalidChannel(uut):
         with pytest.raises(AssertionError):
             uut.setOff(ch)
 
+def test_setFrequencySingleFailure(uut, mockDevice):
+    testFrequency = 1000.0
+    testChannel = 1
+    assert mockDevice.getFrequency(testChannel) is None
+
+    mockDevice.injectFailures(1)
+    uut.setFrequency(testChannel, testFrequency)
+
+    assert mockDevice.getFrequency(testChannel) == testFrequency
+
+def test_setFrequencyMultipleFailures(uut, mockDevice):
+    testFrequency = 1000.0
+    testChannel = 1
+    assert mockDevice.getFrequency(testChannel) is None
+
+    mockDevice.injectFailures(2)
+    uut.setFrequency(testChannel, testFrequency)
+
+    assert mockDevice.getFrequency(testChannel) == 0.0
diff --git a/lab_control/test/mock_jds6600_device.py b/lab_control/test/mock_jds6600_device.py
index 979ec69..c027573 100644
--- a/lab_control/test/mock_jds6600_device.py
+++ b/lab_control/test/mock_jds6600_device.py
@@ -18,6 +18,8 @@ class MockJDS6600Device():
 
         self._portName = os.ttyname(self._slave)
         self._channels = [MockJDS6600Device.ChannelState() for i in [1, 2]]
+
+        self._injectedFailureCounter = 0
         
         self._mainThread = threading.Thread(target=self._mainLoop)
         self._mainThread.start()
@@ -44,37 +46,48 @@ class MockJDS6600Device():
         function = int(m.group("function"))
         args = m.group("args").split(",")
 
-        if opcode == "w":
-            # channel on/off
-            if function == 20:
+        # channel on/off
+        if function == 20:
+            if opcode == "w":
                 self._channels[0].on = args[0] == "1"
                 self._channels[1].on = args[1] == "1"
+                return ":ok\r\n"
+            elif opcode == "r":
+                return f":r20={int(self._channels[0].on)},{int(self._channels[1].on)}.\r\n"
 
-            # channel frequency
-            elif function == 23 or function == 24:
+        # channel frequency
+        elif function == 23 or function == 24:
+            ch = function - 23
+            if opcode == "w":
                 # Actual device takes a second argument for scaling, here we ignore it and always use 0 (Hz)
-                ch = function - 23
                 frequency = float(args[0]) / 100.0
-                self._channels[ch].frequency = frequency
 
-            # channel amplitude
-            elif function == 25 or function == 26:
+                if self._injectedFailureCounter > 0:
+                    self._channels[ch].frequency = 0.0
+                    self._injectedFailureCounter -= 1
+                else:
+                    self._channels[ch].frequency = frequency
+                return ":ok\r\n"
+            elif opcode == "r":
+                frequency = self._channels[ch].frequency
+                return f":r{function}={int(frequency)},0.\r\n"
+
+        # channel amplitude
+        elif function == 25 or function == 26:
+            if opcode == "w":
                 ch = function - 25
                 amplitude = float(args[0]) / 1000.0
                 self._channels[ch].amplitude = amplitude
+                return ":ok\r\n"
 
-            # channel function shape
-            elif function == 21 or function == 22:
+        # channel function shape
+        elif function == 21 or function == 22:
+            if opcode == "w":
                 ch = function - 21
                 shape = int(args[0])
                 self._channels[ch].function = shape
+                return ":ok\r\n"
 
-            return ":ok\r\n"
-
-        elif opcode == "r":
-            if function == 20:
-                return f":r20={int(self._channels[0].on)},{int(self._channels[1].on)}.\r\n"
-        
         # Unknown request format, no response
         return None
 
@@ -109,3 +122,6 @@ class MockJDS6600Device():
     def getFunction(self, ch: int) -> int:
         return self._channels[ch - 1].function
 
+    def injectFailures(self, count: int) -> None:
+        self._injectedFailureCounter += count
+