def setUp(self): """ Sets up and connects the _client. """ TrhphClientTest._end_client_if_any() super(TrhphClientTest, self).setUp() host = self.device_address port = self.device_port self._samples_recd = 0 outfile = file('trhph_output.txt', 'a') prefix_state = True _client = TrhphClient(host, port, outfile, prefix_state) # set the class and instance variables to refer to this object: TrhphClientTest._trhph_client = self._client = _client # prepare client including going to the main menu _client.set_data_listener(self._data_listener) _client.set_generic_timeout(self._timeout) log.info("connecting") _client.connect()
def configure(self, *args, **kwargs): """ Configure the driver for communications with the device via port agent / logger (valid but unconnected connection object). @param config comms config dict. @raises InstrumentStateException if command not allowed in current state @throws InstrumentParameterException if missing comms or invalid config dict. """ if log.isEnabledFor(logging.DEBUG): log.debug("args=%s kwargs=%s" % (str(args), str(kwargs))) self._assert_state(TrhphDriverState.UNCONFIGURED) config = kwargs.get('config', None) if config is None: # raise InstrumentParameterException(msg="'config' parameter required") config = args[0] # Verify dict and construct connection client. try: addr = config['addr'] port = config['port'] if isinstance(addr, str) and \ isinstance(port, int) and len(addr) > 0: # self._connection = LoggerClient(addr, port) def _data_listener(sample): log.info("_data_listener: sample = %s" % str(sample)) host = addr outfile = file('trhph_output.txt', 'a') log.info("setting TrhphClient to connect to %s:%s" % (host, port)) self.trhph_client = TrhphClient(host, port, outfile, True) self.trhph_client.set_data_listener(_data_listener) else: raise InstrumentParameterException( msg='Invalid comms config dict') except (TypeError, KeyError): raise InstrumentParameterException( msg='Invalid comms config dict.') self._state = TrhphDriverState.DISCONNECTED