@@ -101,8 +101,8 @@ impl MessageExt for Message {
101101#[ cfg( test) ]
102102mod tests {
103103 use super :: * ;
104- use crate :: mqtt_producer_record:: MessageRecord ;
105104
105+ use crate :: headers:: MqttVersion :: { MQTT_3 , MQTT_5 } ;
106106 use crate :: MessageBuilderExt ;
107107 use chrono:: Utc ;
108108 use cloudevents:: event:: Data ;
@@ -127,27 +127,23 @@ mod tests {
127127 . build ( )
128128 . unwrap ( ) ;
129129
130- let message_record = MessageRecord :: from_event (
131- EventBuilderV10 :: new ( )
132- . id ( "0001" )
133- . ty ( "example.test" )
134- . time ( time)
135- . source ( "http://localhost" )
136- . extension ( "someint" , "10" )
137- . data ( "application/json" , json ! ( { "hello" : "world" } ) )
138- . build ( )
139- . unwrap ( ) ,
140- headers:: MqttVersion :: V5 ,
141- )
142- . unwrap ( ) ;
130+ let event = EventBuilderV10 :: new ( )
131+ . id ( "0001" )
132+ . ty ( "example.test" )
133+ . time ( time)
134+ . source ( "http://localhost" )
135+ . extension ( "someint" , "10" )
136+ . data ( "application/json" , json ! ( { "hello" : "world" } ) )
137+ . build ( )
138+ . unwrap ( ) ;
143139
144140 let msg = MessageBuilder :: new ( )
145141 . topic ( "test" )
146- . message_record ( & message_record )
142+ . event ( event , MQTT_5 )
147143 . qos ( 1 )
148144 . finalize ( ) ;
149145
150- assert_eq ! ( msg. to_event( headers :: MqttVersion :: V5 ) . unwrap( ) , expected)
146+ assert_eq ! ( msg. to_event( ) . unwrap( ) , expected)
151147 }
152148
153149 #[ test]
@@ -172,18 +168,12 @@ mod tests {
172168 . build ( )
173169 . unwrap ( ) ;
174170
175- let serialized_event =
176- StructuredDeserializer :: deserialize_structured ( input, MessageRecord :: new ( ) ) . unwrap ( ) ;
177-
178171 let msg = MessageBuilder :: new ( )
179172 . topic ( "test" )
180- . message_record ( & serialized_event )
173+ . event ( input , MQTT_3 )
181174 . qos ( 1 )
182175 . finalize ( ) ;
183176
184- assert_eq ! (
185- msg. to_event( headers:: MqttVersion :: V3_1_1 ) . unwrap( ) ,
186- expected
187- )
177+ assert_eq ! ( msg. to_event( ) . unwrap( ) , expected)
188178 }
189179}
0 commit comments