From 75e612b6e8f5564929753c9d46aeff340fcb643a Mon Sep 17 00:00:00 2001
From: Doug Davis <dug@microsoft.com>
Date: Tue, 11 Apr 2023 11:34:54 +0000
Subject: [PATCH] Trying to fix a race warning

Signed-off-by: Doug Davis <dug@microsoft.com>
---
 protocol/kafka_sarama/v2/sender_test.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/protocol/kafka_sarama/v2/sender_test.go b/protocol/kafka_sarama/v2/sender_test.go
index 4aeec39cc..cdb1e02a8 100644
--- a/protocol/kafka_sarama/v2/sender_test.go
+++ b/protocol/kafka_sarama/v2/sender_test.go
@@ -74,6 +74,8 @@ func (s *syncProducerMock) AbortTxn() error {
 }
 
 func (s *syncProducerMock) TxnStatus() sarama.ProducerTxnStatusFlag {
+	s.lock.Lock()
+	defer s.lock.Unlock()
 	return s.status
 }