def setUp(self):
     self._protocol = Protocol('Test')
     self._protocol.add(UInt(1, 'id', 1))
     self._protocol.add(UInt(2, 'length', None))
     self._protocol.add(PDU('length-2'))
     self._msg = MessageTemplate('FooRequest', self._protocol, {'id': '0xaa'})
     self._msg.add(UInt(1, 'field_1', None))
     self._msg.add(UInt(1, 'field_2', None))
     byte_stream = MockStream(to_bin('0xff0004cafe aa0004dead dd0004beef'))
     self._msg_stream = MessageStream(byte_stream, self._protocol)
 def setUp(self):
     self._protocol = Protocol('Test')
     self._protocol.add(UInt(1, 'id', 1))
     self._protocol.add(UInt(2, 'length', None))
     self._protocol.add(PDU('length-2'))
     self._msg = MessageTemplate('FooRequest', self._protocol, {'id': '0xaa'})
     self._msg.add(UInt(1, 'field_1', None))
     self._msg.add(UInt(1, 'field_2', None))
     byte_stream = MockStream(to_bin('0xff0004cafe aa0004dead dd0004beef'))
     self._msg_stream = MessageStream(byte_stream, self._protocol)
 def setUp(self):
     self._protocol = Protocol("Test")
     self._protocol.add(UInt(1, "id", 1))
     self._protocol.add(UInt(2, "length", None))
     self._protocol.add(PDU("length-2"))
     self._msg = MessageTemplate("FooRequest", self._protocol, {"id": "0xaa"})
     self._msg.add(UInt(1, "field_1", None))
     self._msg.add(UInt(1, "field_2", None))
     byte_stream = MockStream(to_bin("0xff0004cafe aa0004dead dd0004beef"))
     self._msg_stream = MessageStream(byte_stream, self._protocol, threading.RLock())
class TestMessageStream(TestCase):

    def setUp(self):
        self._protocol = Protocol('Test')
        self._protocol.add(UInt(1, 'id', 1))
        self._protocol.add(UInt(2, 'length', None))
        self._protocol.add(PDU('length-2'))
        self._msg = MessageTemplate('FooRequest', self._protocol, {'id': '0xaa'})
        self._msg.add(UInt(1, 'field_1', None))
        self._msg.add(UInt(1, 'field_2', None))
        byte_stream = MockStream(to_bin('0xff0004cafe aa0004dead dd0004beef'))
        self._msg_stream = MessageStream(byte_stream, self._protocol)

    def test_get_message(self):
        msg = self._msg_stream.get(self._msg, header_filter='id')
        self.assertEquals(msg.field_1.hex, '0xde')

    def test_get_message_from_cache(self):
        _ = self._msg_stream.get(self._msg, header_filter='id')
        self._msg.header_parameters = {'id': '0xdd'}
        msg = self._msg_stream.get(self._msg, header_filter='id')
        self.assertEquals(msg.field_1.hex, '0xbe')

    def test_empty_message_stream(self):
        _ = self._msg_stream.get(self._msg, header_filter='id')
        self._msg_stream.empty()
        self.assertRaises(socket.timeout, self._msg_stream.get, self._msg, timeout=0.1)

    def test_filter_by_unset_field_fails(self):
        self._msg.header_parameters = {}
        self.assertRaises(AssertionError, self._msg_stream.get, self._msg, header_filter='id')

    def test_timeout_goes_to_stream(self):
        self._msg.header_parameters = {'id': '0x00'}
        self.assertRaises(socket.timeout, self._msg_stream.get, self._msg, timeout=0.1, header_filter='id')
class TestProtocolMessageReceiving(TestCase):
    def setUp(self):
        self._protocol = Protocol("Test")
        self._protocol.add(UInt(1, "id", 1))
        self._protocol.add(UInt(2, "length", None))
        self._protocol.add(PDU("length-2"))

    def test_read_header_and_pdu(self):
        stream = MockStream(to_bin("0xff0004cafe"))
        header, data = self._protocol.read(stream)
        self.assertEquals(header.id.hex, "0xff")
        self.assertEquals(data, "\xca\xfe")
Example #6
0
class TestProtocolMessageReceiving(TestCase):
    def setUp(self):
        self._protocol = Protocol('Test')
        self._protocol.add(UInt(1, 'id', 1))
        self._protocol.add(UInt(2, 'length', None))
        self._protocol.add(PDU('length-2'))

    def test_read_header_and_pdu(self):
        stream = MockStream(to_bin('0xff0004cafe'))
        header, data = self._protocol.read(stream)
        self.assertEquals(header.id.hex, '0xff')
        self.assertEquals(data, '\xca\xfe')
class TestProtocolMessageReceiving(TestCase):

    def setUp(self):
        self._protocol = Protocol('Test')
        self._protocol.add(UInt(1, 'id', 1))
        self._protocol.add(UInt(2, 'length', None))
        self._protocol.add(PDU('length-2'))

    def test_read_header_and_pdu(self):
        stream = MockStream(to_bin('0xff0004cafe'))
        header, data = self._protocol.read(stream)
        self.assertEquals(header.id.hex, '0xff')
        self.assertEquals(data, '\xca\xfe')
Example #8
0
class TestMessageStream(TestCase):
    def setUp(self):
        self._protocol = Protocol('Test')
        self._protocol.add(UInt(1, 'id', 1))
        self._protocol.add(UInt(2, 'length', None))
        self._protocol.add(PDU('length-2'))
        self._msg = MessageTemplate('FooRequest', self._protocol,
                                    {'id': '0xaa'})
        self._msg.add(UInt(1, 'field_1', None))
        self._msg.add(UInt(1, 'field_2', None))
        byte_stream = MockStream(to_bin('0xff0004cafe aa0004dead dd0004beef'))
        self._msg_stream = MessageStream(byte_stream, self._protocol,
                                         threading.RLock())

    def test_get_message(self):
        msg = self._msg_stream.get(self._msg, header_filter='id')
        self.assertEquals(msg.field_1.hex, '0xde')

    def test_get_message_from_cache(self):
        _ = self._msg_stream.get(self._msg, header_filter='id')
        self._msg.header_parameters = {'id': '0xdd'}
        msg = self._msg_stream.get(self._msg, header_filter='id')
        self.assertEquals(msg.field_1.hex, '0xbe')

    def test_empty_message_stream(self):
        _ = self._msg_stream.get(self._msg, header_filter='id')
        self._msg_stream.empty()
        self.assertRaises(socket.timeout,
                          self._msg_stream.get,
                          self._msg,
                          timeout=0.1)

    def test_filter_by_unset_field_fails(self):
        self._msg.header_parameters = {}
        self.assertRaises(AssertionError,
                          self._msg_stream.get,
                          self._msg,
                          header_filter='id')

    def test_timeout_goes_to_stream(self):
        self._msg.header_parameters = {'id': '0x00'}
        self.assertRaises(socket.timeout,
                          self._msg_stream.get,
                          self._msg,
                          timeout=0.1,
                          header_filter='id')

    def test_get_messages_count_from_cache_two_messages(self):
        _ = self._msg_stream.get(self._msg, header_filter='id')
        self._msg.header_parameters = {'id': '0xdd'}
        count = self._msg_stream.get_messages_count_in_cache()
        self.assertEquals(count, 2)

    def test_get_messages_count_from_cache_three_messages(self):
        count = self._msg_stream.get_messages_count_in_cache()
        self.assertEquals(count, 3)
class TestMessageStream(TestCase):
    def setUp(self):
        self._protocol = Protocol("Test")
        self._protocol.add(UInt(1, "id", 1))
        self._protocol.add(UInt(2, "length", None))
        self._protocol.add(PDU("length-2"))
        self._msg = MessageTemplate("FooRequest", self._protocol, {"id": "0xaa"})
        self._msg.add(UInt(1, "field_1", None))
        self._msg.add(UInt(1, "field_2", None))
        byte_stream = MockStream(to_bin("0xff0004cafe aa0004dead dd0004beef"))
        self._msg_stream = MessageStream(byte_stream, self._protocol, threading.RLock())

    def test_get_message(self):
        msg = self._msg_stream.get(self._msg, header_filter="id")
        self.assertEquals(msg.field_1.hex, "0xde")

    def test_get_message_from_cache(self):
        _ = self._msg_stream.get(self._msg, header_filter="id")
        self._msg.header_parameters = {"id": "0xdd"}
        msg = self._msg_stream.get(self._msg, header_filter="id")
        self.assertEquals(msg.field_1.hex, "0xbe")

    def test_empty_message_stream(self):
        _ = self._msg_stream.get(self._msg, header_filter="id")
        self._msg_stream.empty()
        self.assertRaises(socket.timeout, self._msg_stream.get, self._msg, timeout=0.1)

    def test_filter_by_unset_field_fails(self):
        self._msg.header_parameters = {}
        self.assertRaises(AssertionError, self._msg_stream.get, self._msg, header_filter="id")

    def test_timeout_goes_to_stream(self):
        self._msg.header_parameters = {"id": "0x00"}
        self.assertRaises(socket.timeout, self._msg_stream.get, self._msg, timeout=0.1, header_filter="id")

    def test_get_messages_count_from_cache_two_messages(self):
        _ = self._msg_stream.get(self._msg, header_filter="id")
        self._msg.header_parameters = {"id": "0xdd"}
        count = self._msg_stream.get_messages_count_in_cache()
        self.assertEquals(count, 2)

    def test_get_messages_count_from_cache_three_messages(self):
        count = self._msg_stream.get_messages_count_in_cache()
        self.assertEquals(count, 3)
 def setUp(self):
     self._protocol = Protocol('Test')
     self._protocol.add(UInt(1, 'id', 1))
     self._protocol.add(UInt(2, 'length', None))
     self._protocol.add(PDU('length-2'))
 def setUp(self):
     self._protocol = Protocol("Test")
     self._protocol.add(UInt(1, "id", 1))
     self._protocol.add(UInt(2, "length", None))
     self._protocol.add(PDU("length-2"))
 def setUp(self):
     self._protocol = Protocol('Test')
     self._protocol.add(UInt(1, 'id', 1))
     self._protocol.add(UInt(2, 'length', None))
     self._protocol.add(PDU('length-2'))