@@ -34,6 +34,7 @@ func TestSuiteConsumeOperator[InMsg any, Msg any](
34
34
var queueName string
35
35
var serializer MessageSerializer [InMsg , Msg ]
36
36
var callbackCount atomic.Int64
37
+ var callbackValue Msg
37
38
var callback Callback [Msg ]
38
39
39
40
queueName = "testQueueName"
@@ -71,6 +72,13 @@ func TestSuiteConsumeOperator[InMsg any, Msg any](
71
72
})
72
73
73
74
assert .NotNil (t , operator .Serializer (), serializer )
75
+ value , err := serializer .Serialize (rawMessageProvider ())
76
+ if err != nil {
77
+ return
78
+ }
79
+
80
+ assert .NotNil (t , value )
81
+ assert .Nil (t , err )
74
82
})
75
83
76
84
t .Run ("Callback" , func (t * testing.T ) {
@@ -94,16 +102,21 @@ func TestSuiteConsumeOperator[InMsg any, Msg any](
94
102
callbackCount = atomic.Int64 {}
95
103
callback = func (msg Msg ) {
96
104
callbackCount .Add (1 )
105
+ callbackValue = msg
97
106
}
98
107
operator = operatorProvider (queueName , serializer , callback )
99
108
})
100
109
101
110
msg := rawMessageProvider ()
102
111
assert .NotNil (t , msg )
103
112
113
+ expectedSerializedValue , err := serializer .Serialize (msg )
114
+ assert .Nil (t , err )
115
+
104
116
operator .Consume (msg )
105
117
assert .Equal (t , serializer .(* mockSerializer [InMsg , Msg ]).Count (), int64 (1 ))
106
118
assert .Equal (t , callbackCount .Load (), int64 (1 ))
119
+ assert .Equal (t , callbackValue , expectedSerializedValue )
107
120
})
108
121
109
122
t .Run ("StartConsume" , func (t * testing.T ) {
0 commit comments