Beispiel #1
0
    def setUp(self):
        self.mock = DotDict()
        self.mock.port_agent = Mock(name='port_agent_client')
        self.mock.callback = Mock(name='callback')

        def mock_set(values, startup=True):
            assert isinstance(values, dict)
            return_string = ""
            for key in values.keys():
                return_string += "%s=%s" % (key, values[key]
                                            )  # inefficient, I know
            return return_string

        self.driver = SingleConnectionInstrumentDriver(self.mock.callback)

        # set some values
        self.driver._protocol = InstrumentProtocol(self.mock.callback)
        self.driver.set_resource = mock_set

        self.driver._protocol._param_dict.add(
            "foo",
            r'foo=(.*)',
            lambda match: int(match.group(1)),
            lambda x: str(x),
            direct_access=True,
            startup_param=True,
            default_value=10)
        self.driver._protocol._param_dict.set_default("foo")
        self.driver._protocol._param_dict.add(
            "bar",
            r'bar=(.*)',
            lambda match: int(match.group(1)),
            lambda x: str(x),
            direct_access=False,
            startup_param=True,
            default_value=15)
        self.driver._protocol._param_dict.set_default("bar")
        self.driver._protocol._param_dict.add(
            "baz",
            r'baz=(.*)',
            lambda match: int(match.group(1)),
            lambda x: str(x),
            direct_access=True,
            default_value=30)
        self.driver._protocol._param_dict.set_default("baz")
        self.driver._protocol._param_dict.add(
            "bat",
            r'bat=(.*)',
            lambda match: int(match.group(1)),
            lambda x: str(x),
            startup_param=False,
            default_value=40)
        self.driver._protocol._param_dict.set_default("bat")

        self.driver._protocol._cmd_dict.add("cmd1")
        self.driver._protocol._cmd_dict.add("cmd2")

        self.driver._protocol._driver_dict.add(
            DriverDictKey.VENDOR_SW_COMPATIBLE, True)
    def setUp(self):
        """
        """
        self.callback_result = None
        self._trigger_count = 0
        self._events = []

        self.protocol = InstrumentProtocol(self.event_callback)