diff options
| author | Nathan West <nathan.west@okstate.edu> | 2015-10-05 18:17:09 -0400 |
|---|---|---|
| committer | Nathan West <nathan.west@okstate.edu> | 2015-10-07 20:44:02 -0400 |
| commit | 928fd29a5e19570c7ed55d9a83a5eb487e75fe56 (patch) | |
| tree | 79a9c56942efad2c00341e849ab82f3b4fc50ff0 /gr-blocks/python | |
| parent | bde02e814e7f68e6ebf945d7f294014605fd3239 (diff) | |
blocks: add socket_pdu test using a flowgraph
Diffstat (limited to 'gr-blocks/python')
| -rwxr-xr-x | gr-blocks/python/blocks/qa_socket_pdu.py | 62 |
1 files changed, 61 insertions, 1 deletions
diff --git a/gr-blocks/python/blocks/qa_socket_pdu.py b/gr-blocks/python/blocks/qa_socket_pdu.py index 4dba2bfd45..db9f53c71e 100755 --- a/gr-blocks/python/blocks/qa_socket_pdu.py +++ b/gr-blocks/python/blocks/qa_socket_pdu.py @@ -22,7 +22,8 @@ from gnuradio import gr, gr_unittest, blocks import random - +import pmt +import time class qa_socket_pdu (gr_unittest.TestCase): @@ -40,6 +41,65 @@ class qa_socket_pdu (gr_unittest.TestCase): self.pdu_send = None self.pdu_recv = None + def test_002 (self): + # Send a PDU through a pair of UDP sockets + port = str(random.Random().randint(0, 30000) + 10000) + srcdata = (0x64, 0x6f, 0x67, 0x65) + data = pmt.init_u8vector(srcdata.__len__(), srcdata) + pdu_msg = pmt.cons(pmt.PMT_NIL, data) + + self.pdu_source = blocks.message_strobe(pdu_msg, 500) + self.pdu_recv = blocks.socket_pdu("UDP_SERVER", "localhost", port) + self.pdu_send = blocks.socket_pdu("UDP_CLIENT", "localhost", port) + + self.dbg = blocks.message_debug() + + self.tb.msg_connect(self.pdu_source, "strobe", self.pdu_send, "pdus") + self.tb.msg_connect(self.pdu_recv, "pdus", self.dbg, "store") + + self.tb.start () + time.sleep(1) + self.tb.stop() + self.tb.wait() + self.pdu_send = None + self.pdu_recv = None + + received = self.dbg.get_message(0) + received_data = pmt.cdr(received) + msg_data = [] + for i in xrange(4): + msg_data.append(pmt.u8vector_ref(received_data, i)) + self.assertEqual(srcdata, tuple(msg_data)) + + def test_003 (self): + # Test that block stops when interacting with streaming interface + port = str(random.Random().randint(0, 30000) + 10000) + srcdata = (0x73, 0x75, 0x63, 0x68, 0x74, 0x65, 0x73, 0x74, 0x76, 0x65, 0x72, 0x79, 0x70, 0x61, 0x73, 0x73) + tag_dict = {"offset": 0} + tag_dict["key"] = pmt.intern("len") + tag_dict["value"] = pmt.from_long(8) + tag1 = gr.python_to_tag(tag_dict) + tag_dict["offset"] = 8 + tag2 = gr.python_to_tag(tag_dict) + tags = [tag1, tag2] + + src = blocks.vector_source_b(srcdata, False, 1, tags) + ts_to_pdu = blocks.tagged_stream_to_pdu(blocks.byte_t, "len") + pdu_send = blocks.socket_pdu("UDP_CLIENT", "localhost", "4141") + #pdu_recv = blocks.socket_pdu("UDP_SERVER", "localhost", port) + pdu_to_ts = blocks.pdu_to_tagged_stream(blocks.byte_t, "len") + head = blocks.head(gr.sizeof_char, 10) + sink = blocks.vector_sink_b(1) + + self.tb.connect(src, ts_to_pdu) + self.tb.msg_connect(ts_to_pdu, "pdus", pdu_send, "pdus") + # a UDP socket connects pdu_send to pdu_recv + # TODO: test that the recv socket can be destroyed from downstream + # that signals DONE. Also that we get the PDUs we sent + #self.tb.msg_connect(pdu_recv, "pdus", pdu_to_ts, "pdus") + #self.tb.connect(pdu_to_ts, head, sink) + self.tb.run() + if __name__ == '__main__': gr_unittest.run(qa_socket_pdu, "qa_socket_pdu.xml") |