@@ -24,6 +24,7 @@ import (
2424 "github.com/google/go-cmp/cmp"
2525
2626 "github.com/cloudevents/sdk-go/v2/binding"
27+ "github.com/cloudevents/sdk-go/v2/binding/test"
2728 "github.com/cloudevents/sdk-go/v2/client"
2829 "github.com/cloudevents/sdk-go/v2/event"
2930 "github.com/cloudevents/sdk-go/v2/protocol"
@@ -431,25 +432,22 @@ func TestClientStartReceiverWithAckMalformedEvent(t *testing.T) {
431432
432433 c , err := client .New (receiver , tc .opts ... )
433434 if err != nil {
434- t .Errorf ("failed to construct client: %v" , err )
435+ t .Fatalf ("failed to construct client: %v" , err )
435436 }
436437
437438 go c .StartReceiver (ctx , func (ctx context.Context , e event.Event ) protocol.Result {
438439 t .Error ("receiver callback called unexpectedly" )
439440 return nil
440441 })
441442
442- // wait for receive to occur
443- time .Sleep (time .Millisecond )
444-
445443 ctx , cancelTimeout := context .WithTimeout (ctx , time .Second )
446444 defer cancelTimeout ()
447445
448446 select {
449447 case <- receiver .finished :
450448 // continue to rest of the test
451449 case <- ctx .Done ():
452- t .Errorf ( "timeoued out waiting for receiver to complete" )
450+ t .Fatalf ( "timed out waiting for receiver to complete" )
453451 }
454452
455453 if tc .expectedAck {
@@ -555,18 +553,6 @@ func isImportantHeader(h string) bool {
555553 return true
556554}
557555
558- type mockMessage struct {}
559-
560- func (m * mockMessage ) ReadEncoding () binding.Encoding {
561- return binding .EncodingUnknown
562- }
563-
564- func (m * mockMessage ) ReadStructured (ctx context.Context , writer binding.StructuredWriter ) error {
565- return nil
566- }
567- func (m * mockMessage ) ReadBinary (ctx context.Context , writer binding.BinaryWriter ) error { return nil }
568- func (m * mockMessage ) Finish (err error ) error { return nil }
569-
570556type mockReceiver struct {
571557 mu sync.Mutex
572558 count int
@@ -584,7 +570,7 @@ func (m *mockReceiver) Receive(ctx context.Context) (binding.Message, error) {
584570
585571 m .count ++
586572
587- return binding .WithFinish (& mockMessage {} , func (err error ) {
573+ return binding .WithFinish (test . UnknownMessage , func (err error ) {
588574 m .result = err
589575 close (m .finished )
590576 }), nil
0 commit comments