From b0e11d0e0c07b364fa4d393b519abc23eb536f29 Mon Sep 17 00:00:00 2001 From: Adam Shannon Date: Thu, 9 Mar 2023 08:56:26 -0600 Subject: [PATCH] pipeline: test file receiver shouldAutocommit() --- internal/pipeline/file_receiver_test.go | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/internal/pipeline/file_receiver_test.go b/internal/pipeline/file_receiver_test.go index d8bd589..f48efa6 100644 --- a/internal/pipeline/file_receiver_test.go +++ b/internal/pipeline/file_receiver_test.go @@ -56,6 +56,24 @@ func testFileReceiver(t *testing.T) *FileReceiver { return fileRec } +func TestFileReceiver__shouldAutocommit(t *testing.T) { + fr := testFileReceiver(t) + + // Ensure the setup is as we expect + require.Nil(t, fr.cfg.Inbound.Kafka) + require.False(t, fr.shouldAutocommit()) + + // Set a config with AutoCommit disabled + fr.cfg.Inbound.Kafka = &service.KafkaConfig{ + AutoCommit: false, + } + require.False(t, fr.shouldAutocommit()) + + // Set .AutoCommit to true + fr.cfg.Inbound.Kafka.AutoCommit = true + require.True(t, fr.shouldAutocommit()) +} + func TestFileReceiver__contains(t *testing.T) { err := errors.New("pubsub (code=Unknown): write tcp 10.100.53.92:45360->12.132.211.32:2222: write: broken pipe")