add context with timeout to method_logger_test in binary log

Author: hasson82

internal/binarylog/method_logger_test.go

@@ -33,6 +33,8 @@ import (
 	"google.golang.org/protobuf/types/known/durationpb"
 )
 
+const defaultTestTimeout = 10 * time.Second
+
 func (s) TestLog(t *testing.T) {
 	idGen.reset()
 	ml := NewTruncatingMethodLogger(10, 10)
@@ -336,7 +338,9 @@ func (s) TestLog(t *testing.T) {
 	for i, tc := range testCases {
 		buf.Reset()
 		tc.want.SequenceIdWithinCall = uint64(i + 1)
-		ml.Log(context.Background(), tc.config)
+		ctx, cancel := context.WithTimeout(context.Background(), defaultTestTimeout)
+		defer cancel()
+		ml.Log(ctx, tc.config)
 		inSink := new(binlogpb.GrpcLogEntry)
 		if err := proto.Unmarshal(buf.Bytes()[4:], inSink); err != nil {
 			t.Errorf("failed to unmarshal bytes in sink to proto: %v", err)