diff --git a/udp/flow_test.go b/udp/flow_test.go index 0c61220..ebcc94b 100644 --- a/udp/flow_test.go +++ b/udp/flow_test.go @@ -107,7 +107,9 @@ func TestFlow_Produce(t *testing.T) { flowA.isAlive = true go func() { - _, err := flowA.readPacket(context.Background(), testConn.SideB()) + p, err := flowA.readPacket(context.Background(), testConn.SideB()) + assert.Nil(t, err) + err = flowA.queueDatagram(context.Background(), p) assert.Nil(t, err) }() p, err := flowA.Produce(context.Background()) @@ -143,7 +145,9 @@ func TestFlow_Produce(t *testing.T) { flowA.isAlive = true go func() { - _, err := flowA.readPacket(context.Background(), testConn.SideB()) + p, err := flowA.readPacket(context.Background(), testConn.SideB()) + assert.Nil(t, err) + err = flowA.queueDatagram(context.Background(), p) assert.Nil(t, err) }() p, err := flowA.Produce(context.Background())