1- from catkit2 .catkit_bindings import LocalMemory , MessageBroker , MessageSubscriptionMode
1+ from catkit2 .catkit_bindings import LocalMemory , LocalMessageBroker , MessageSubscriptionMode
22import numpy as np
33import pytest
44
@@ -7,7 +7,7 @@ def broker():
77 header = LocalMemory .create (1024 * 1024 * 512 )
88 block = LocalMemory .create (1024 * 1024 * 1024 )
99
10- broker = MessageBroker .create (header , [block ])
10+ broker = LocalMessageBroker .create (header , [block ])
1111 yield broker
1212
1313def test_message_subscription (broker ):
@@ -22,9 +22,6 @@ def test_message_subscription(broker):
2222 message .payload = arr
2323 broker .publish_message (message )
2424
25- assert subscription_newest .next_message_id == 2
26- assert subscription_sequential .next_message_id == 0
27-
2825 message_1 = subscription_newest .get_next_message (0.01 )
2926 message_2 = subscription_sequential .get_next_message (0.01 )
3027 message_3 = subscription_sequential .get_next_message (0.01 )
@@ -33,6 +30,9 @@ def test_message_subscription(broker):
3330 assert message_2 is not None
3431 assert message_3 is not None
3532
33+ assert message_1 .frame_id == 2
34+ assert message_2 .frame_id == 0
35+
3636 assert message_1 .payload [0 ] == 12
3737 assert message_2 .payload [0 ] == 10
3838 assert message_3 .payload [0 ] == 11
@@ -45,11 +45,13 @@ def test_message_subscription(broker):
4545
4646 assert subscription_sequential .try_get_next_message () is None
4747
48- subscription_newest2 = broker .subscribe (topic , starting_frame_id = 1 , mode = MessageSubscriptionMode .NewestOnly )
49- assert subscription_newest2 .next_message_id == 2
48+ subscription_newest2 = broker .subscribe (topic , preferred_next_frame_id = 1 , mode = MessageSubscriptionMode .NewestOnly )
49+ m = subscription_newest2 .get_next_message (0.01 )
50+ assert m .frame_id == 2
5051
51- subscription_sequential2 = broker .subscribe (topic , starting_frame_id = 1 , mode = MessageSubscriptionMode .Sequential )
52- assert subscription_sequential2 .next_message_id == 1
52+ subscription_sequential2 = broker .subscribe (topic , preferred_next_frame_id = 1 , mode = MessageSubscriptionMode .Sequential )
53+ m = subscription_sequential2 .get_next_message (0.01 )
54+ assert m .frame_id == 1
5355
5456dtypes = ['int8' , 'uint8' , 'int16' , 'uint16' , 'int32' , 'uint32' , 'int64' , 'uint64' , 'float32' , 'float64' , 'complex64' , 'complex128' ]
5557shapes = [[10 ], [10 , 10 ], [10 , 10 , 10 ], [10 , 10 , 10 , 10 ]]
@@ -65,7 +67,7 @@ def test_message_dtype_and_shape(broker, shape, dtype):
6567 message .payload = arr
6668 broker .publish_message (message )
6769
68- retrieved_message = broker .get_newest_message (topic )
70+ retrieved_message = broker .get_current_message (topic )
6971
7072 assert np .all (arr == retrieved_message .payload )
7173 assert retrieved_message .payload .dtype == dtype
@@ -98,7 +100,7 @@ def test_message_broker_publish(broker):
98100
99101 broker .publish_data (topic , data )
100102
101- retrieved_message = broker .get_newest_message (topic )
103+ retrieved_message = broker .get_current_message (topic )
102104
103105 assert (retrieved_message .payload == arr ).all ()
104106
@@ -107,14 +109,14 @@ def test_message_broker_trace_id(broker):
107109 data = b'hello world'
108110
109111 broker .publish_data (topic , data )
110- message_1 = broker .get_newest_message (topic )
112+ message_1 = broker .get_current_message (topic )
111113
112114 broker .publish_data (topic , data * 2 )
113- message_2 = broker .get_newest_message (topic )
115+ message_2 = broker .get_current_message (topic )
114116
115117 assert str (message_2 .trace_id ) != str (message_1 .trace_id )
116118
117119 broker .publish_data (topic , data * 3 , trace_id = message_1 .trace_id )
118- message_3 = broker .get_newest_message (topic )
120+ message_3 = broker .get_current_message (topic )
119121
120122 assert str (message_3 .trace_id ) == str (message_1 .trace_id )
0 commit comments