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)