You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
115 lines
4.1 KiB
115 lines
4.1 KiB
2 months ago
|
# License: Apache 2.0. See LICENSE file in root directory.
|
||
|
|
# Copyright(c) 2022 Intel Corporation. All Rights Reserved.
|
||
|
|
|
||
|
|
#test:donotrun:!dds
|
||
|
|
#test:retries:gha 2
|
||
|
|
|
||
|
|
import pyrealdds as dds
|
||
|
|
from rspy import log, test
|
||
|
|
|
||
|
|
dds.debug( log.is_debug_on(), 'C ' )
|
||
|
|
log.nested = 'C '
|
||
|
|
|
||
|
|
from time import sleep
|
||
|
|
import json
|
||
|
|
import flexible
|
||
|
|
|
||
|
|
participant = dds.participant()
|
||
|
|
participant.init( 123, "test-streaming" )
|
||
|
|
|
||
|
|
|
||
|
|
sample1 = { 'id' : 'test-message', 'n' : 1, 'content' : { 'int' : 1, 'float' : 2.0, 'array' : [ 1, 2.0, 'hello' ] } }
|
||
|
|
sample1_json = json.dumps( sample1 )
|
||
|
|
sample2 = { 'id' : 'test-message-2', 'n' : 2 }
|
||
|
|
sample2_json = json.dumps( sample2 )
|
||
|
|
sample3 = { 'id' : 'test-message', 'n' : 3, 'hmm' : [] }
|
||
|
|
sample3_json = json.dumps( sample3 )
|
||
|
|
|
||
|
|
|
||
|
|
import os.path
|
||
|
|
cwd = os.path.dirname(os.path.realpath(__file__))
|
||
|
|
remote_script = os.path.join( cwd, 'streaming-server.py' )
|
||
|
|
with test.remote( remote_script, nested_indent=" S" ) as remote:
|
||
|
|
remote.wait_until_ready()
|
||
|
|
#
|
||
|
|
#############################################################################################
|
||
|
|
#
|
||
|
|
test.start( "Test 1W:1R, 1 message..." )
|
||
|
|
try:
|
||
|
|
remote.run( 'topic = flexible.writer( participant, "1w1r" )' )
|
||
|
|
# NOTE: technically, we could combine everything into a single command:
|
||
|
|
# Topic("blah").write( ... )
|
||
|
|
# and it would do everything, including destroy the Topic instance with the writer etc.
|
||
|
|
# This doesn't work! If the writer is destroyed before we've had a chance to get the message,
|
||
|
|
# the message will get lost...
|
||
|
|
topic = flexible.reader( participant, "1w1r" )
|
||
|
|
# We may wait for the data before we even see the writer... so wait for the writer first!
|
||
|
|
topic.wait_for_writers()
|
||
|
|
# Same for the writer:
|
||
|
|
remote.run( 'topic.wait_for_readers()' )
|
||
|
|
remote.run( 'topic.write( """' + sample1_json + '""" )' )
|
||
|
|
msg = topic.read()
|
||
|
|
test.check( topic.empty() )
|
||
|
|
test.check_equal( msg.json_data(), sample1 )
|
||
|
|
except:
|
||
|
|
test.unexpected_exception()
|
||
|
|
remote.run( 'topic.stop()' )
|
||
|
|
topic.wait_for_writers( 0 )
|
||
|
|
topic = None
|
||
|
|
test.finish()
|
||
|
|
#
|
||
|
|
#############################################################################################
|
||
|
|
#
|
||
|
|
test.start( "Test 1W:1R, N messages..." )
|
||
|
|
try:
|
||
|
|
remote.run( 'topic = flexible.writer( participant, "1w1rNm" )' )
|
||
|
|
topic = flexible.reader( participant, "1w1rNm" )
|
||
|
|
topic.wait_for_writers()
|
||
|
|
remote.run( 'topic.wait_for_readers()' )
|
||
|
|
remote.run( 'topic.write( """' + sample1_json + '""" )' )
|
||
|
|
remote.run( 'topic.write( """' + sample2_json + '""" )' )
|
||
|
|
remote.run( 'topic.write( """' + sample3_json + '""" )' )
|
||
|
|
msg1 = topic.read()
|
||
|
|
msg2 = topic.read()
|
||
|
|
msg3 = topic.read()
|
||
|
|
test.check( topic.empty() )
|
||
|
|
test.check_equal( msg1.json_data(), sample1 )
|
||
|
|
test.check_equal( msg2.json_data(), sample2 )
|
||
|
|
test.check_equal( msg3.json_data(), sample3 )
|
||
|
|
except:
|
||
|
|
test.unexpected_exception()
|
||
|
|
remote.run( 'topic.stop()' )
|
||
|
|
topic.wait_for_writers( 0 )
|
||
|
|
topic = None
|
||
|
|
test.finish()
|
||
|
|
#
|
||
|
|
#############################################################################################
|
||
|
|
#
|
||
|
|
test.start( "Test 1W:2R, one message..." )
|
||
|
|
try:
|
||
|
|
remote.run( 'topic = flexible.writer( participant, "1w2r" )' )
|
||
|
|
r1 = flexible.reader( participant, "1w2r" )
|
||
|
|
r2 = flexible.reader( participant, r1.handle )
|
||
|
|
r1.wait_for_writers()
|
||
|
|
r2.wait_for_writers()
|
||
|
|
remote.run( 'topic.wait_for_readers( 2 )' )
|
||
|
|
remote.run( 'topic.write( """' + sample1_json + '""" )' )
|
||
|
|
msg1 = r1.read()
|
||
|
|
msg2 = r2.read()
|
||
|
|
test.check( r1.empty() )
|
||
|
|
test.check( r2.empty() )
|
||
|
|
test.check_equal( msg1.json_data(), sample1 )
|
||
|
|
test.check_equal( msg2.json_data(), sample1 )
|
||
|
|
except:
|
||
|
|
test.unexpected_exception()
|
||
|
|
remote.run( 'topic.stop()' )
|
||
|
|
r1.wait_for_writers( 0 )
|
||
|
|
r2.wait_for_writers( 0 )
|
||
|
|
r1 = r2 = None
|
||
|
|
test.finish()
|
||
|
|
#
|
||
|
|
#############################################################################################
|
||
|
|
|
||
|
|
|
||
|
|
test.print_results_and_exit()
|