def __init__(self, connection_string):
num_occurrences = connection_string.count(':')
if num_occurrences == 0:
self._host = connection_string
self._port = self.DEFAULT_PORT
elif num_occurrences == 1:
self._host, self._port = connection_string.split(':')
self._port = int(self._port)
else:
raise Exception("Invalid connection string {}".format(connection_string))
QueuedLineSender.__init__(self)
def setUp(self):
self.protocol = QueuedLineSender()
self.transport = proto_helpers.StringTransport()
self.protocol.makeConnection(self.transport)
self._reactor = hamjab.lib._reactor
class QueuedLineSenderTestCase(unittest.TestCase):
def setUp(self):
self.protocol = QueuedLineSender()
self.transport = proto_helpers.StringTransport()
self.protocol.makeConnection(self.transport)
self._reactor = hamjab.lib._reactor
def tearDown(self):
hamjab.lib._reactor = self._reactor
def test_commands_queueing(self):
d = self.protocol.sendLine('test1')
self.assertEqual('test1\r', self.transport.value())
self.transport.clear()
d.addCallback(self.assertEqual, 'answer')
d2 = self.protocol.sendLine('test2')
self.assertEqual('', self.transport.value())
d.addCallback(
lambda x: self.assertEqual('test2\r', self.transport.value()))
d2.addCallback(self.assertEqual, 'answer2')
self.protocol.dataReceived('answer\r')
self.protocol.dataReceived('answer2\r')
return d2
def test_simple_command(self):
d = self.protocol.sendLine('test')
self.assertEqual('test\r', self.transport.value())
self.transport.clear()
d.addCallback(self.assertEqual, 'answer')
self.protocol.dataReceived('answer\r')
return d
def test_simple_command_broken_answer(self):
d = self.protocol.sendLine('test')
self.assertEqual('test\r', self.transport.value())
self.transport.clear()
d.addCallback(self.assertEqual, 'answer')
self.protocol.dataReceived('an')
self.protocol.dataReceived('swer')
self.protocol.dataReceived('\r')
return d
def test_simple_command_timeout(self):
hamjab.lib._reactor = Clock()
d = self.protocol.sendLine('test')
self.assertEqual('test\r', self.transport.value())
self.transport.clear()
d.addCallback(self.assertEqual, 'TIMEOUT')
self.protocol.dataReceived('an')
self.protocol.dataReceived('swer')
hamjab.lib._reactor.advance(self.protocol.timeout)
return d
def test_unsolicited(self):
d = self.protocol.getUnsolicitedData()
d.addCallback(self.assertEqual, 'data')
self.protocol.dataReceived('data\r')
return d
def __init__(self, com_port, com_options={}):
self.com_port = com_port
self.com_options = com_options
QueuedLineSender.__init__(self)
def setUp(self):
self.protocol = QueuedLineSender()
self.transport = proto_helpers.StringTransport()
self.protocol.makeConnection(self.transport)
self._reactor = hamjab.lib._reactor
class QueuedLineSenderTestCase(unittest.TestCase):
def setUp(self):
self.protocol = QueuedLineSender()
self.transport = proto_helpers.StringTransport()
self.protocol.makeConnection(self.transport)
self._reactor = hamjab.lib._reactor
def tearDown(self):
hamjab.lib._reactor = self._reactor
def test_commands_queueing(self):
d = self.protocol.sendLine('test1')
self.assertEqual('test1\r', self.transport.value())
self.transport.clear()
d.addCallback(self.assertEqual, 'answer')
d2 = self.protocol.sendLine('test2')
self.assertEqual('', self.transport.value())
d.addCallback(lambda x: self.assertEqual('test2\r', self.transport.value()))
d2.addCallback(self.assertEqual, 'answer2')
self.protocol.dataReceived('answer\r')
self.protocol.dataReceived('answer2\r')
return d2
def test_simple_command(self):
d = self.protocol.sendLine('test')
self.assertEqual('test\r', self.transport.value())
self.transport.clear()
d.addCallback(self.assertEqual, 'answer')
self.protocol.dataReceived('answer\r')
return d
def test_simple_command_broken_answer(self):
d = self.protocol.sendLine('test')
self.assertEqual('test\r', self.transport.value())
self.transport.clear()
d.addCallback(self.assertEqual, 'answer')
self.protocol.dataReceived('an')
self.protocol.dataReceived('swer')
self.protocol.dataReceived('\r')
return d
def test_simple_command_timeout(self):
hamjab.lib._reactor = Clock()
d = self.protocol.sendLine('test')
self.assertEqual('test\r', self.transport.value())
self.transport.clear()
d.addCallback(self.assertEqual, 'TIMEOUT')
self.protocol.dataReceived('an')
self.protocol.dataReceived('swer')
hamjab.lib._reactor.advance(self.protocol.timeout)
return d
def test_unsolicited(self):
d = self.protocol.getUnsolicitedData()
d.addCallback(self.assertEqual, 'data')
self.protocol.dataReceived('data\r')
return d
