# 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()