Function TestBatchMessageHandler_Handle

pkg/dequeuer/batch_handler_test.go:30–59

(t *testing.T)

)

func TestBatchMessageHandler_Handle(t *testing.T) {
	t.Parallel()
	awsClient := testAWSClient(t)

	var callCount int
	server := httptest.NewServer(
		http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			callCount++
			w.WriteHeader(http.StatusOK)
		}),
	)

	logger := newLogger(t)
	defer func() { _ = logger.Sync() }()

	batchHandler := NewBatchMessageHandler(BatchMessageHandlerConfig{
		APIName:   "test",
		JobID:     "12345",
		Region:    _localStackDefaultRegion,
		TargetURL: server.URL,
	}, awsClient, &statsd.NoOpClient{}, logger)

	err := batchHandler.Handle(&sqs.Message{
		Body:      aws.String(""),
		MessageId: aws.String("1"),
	})

	require.Equal(t, callCount, 1)
	require.NoError(t, err)
}

func TestBatchMessageHandler_Handle_OnJobComplete(t *testing.T) {
	t.Parallel()