From d37252c9b6044ce540db54e8b480a85dfc1ec25d Mon Sep 17 00:00:00 2001
From: Eddy Pedroni <eddy@0xf7.com>
Date: Sun, 5 Jun 2022 16:49:33 +0200
Subject: Moved mocks to separate module

---
 lab_control/test/mock_jds6600_device.py | 85 ---------------------------------
 1 file changed, 85 deletions(-)
 delete mode 100644 lab_control/test/mock_jds6600_device.py

(limited to 'lab_control/test/mock_jds6600_device.py')

diff --git a/lab_control/test/mock_jds6600_device.py b/lab_control/test/mock_jds6600_device.py
deleted file mode 100644
index 8b7b440..0000000
--- a/lab_control/test/mock_jds6600_device.py
+++ /dev/null
@@ -1,85 +0,0 @@
-import re
-
-class MockJDS6600Device:
-    class ChannelState:
-        def __init__(self):
-            self.on = False
-            self.frequency = None
-            self.amplitude = None
-            self.function = None
-
-    def __init__(self):
-        self._channels = [MockJDS6600Device.ChannelState() for i in [1, 2]]
-        self._injectedFailureCounter = 0
-
-    def _handleRequest(self, request: str) -> str:
-        pattern = r":(?P<opcode>[wrab])(?P<function>\d+)=(?P<args>.*)\."
-        m = re.search(pattern, request)
-        
-        if not m:
-            return None
-
-        opcode = m.group("opcode")
-        function = int(m.group("function"))
-        args = m.group("args").split(",")
-
-        # 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:
-            ch = function - 23
-            if opcode == "w":
-                # Actual device takes a second argument for scaling, here we ignore it and always use 0 (Hz)
-                frequency = float(args[0]) / 100.0
-
-                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:
-            if opcode == "w":
-                ch = function - 21
-                shape = int(args[0])
-                self._channels[ch].function = shape
-                return ":ok\r\n"
-
-        # Unknown request format, no response
-        return None
-
-    def isOn(self, ch: int) -> bool:
-        return self._channels[ch - 1].on
-
-    def getFrequency(self, ch: int) -> float:
-        return self._channels[ch - 1].frequency
-
-    def getAmplitude(self, ch: int) -> float:
-        return self._channels[ch - 1].amplitude
-
-    def getFunction(self, ch: int) -> int:
-        return self._channels[ch - 1].function
-
-    def injectFailures(self, count: int) -> None:
-        self._injectedFailureCounter += count
-
-- 
cgit v1.2.3