Clean up

artie-labs · Mar 29, 2024 · b0bd5a8 · b0bd5a8
1 parent f9d3ef5
commit b0bd5a8
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/lib/debezium/transformer/transformer_test.go b/lib/debezium/transformer/transformer_test.go
@@ -58,8 +58,7 @@ func (m mockAdatper) NewIterator() (RowsIterator, error) {
 	return m.iter, nil
 }
 
-type errorIterator struct {
-}
+type errorIterator struct{}
 
 func (m *errorIterator) HasNext() bool {
 	return true
@@ -197,7 +196,7 @@ func TestDebeziumTransformer_Next(t *testing.T) {
 		transformer, err := NewDebeziumTransformer(mockAdatper{
 			fieldConverters: fieldConverters,
 			partitionKeys:   []string{"foo"},
-			iter:            lib.NewBatchIterator([][]Row{{{"foo": "bar"}}}),
+			iter:            lib.NewSingleBatchIterator([]Row{{"foo": "bar"}}),
 		},
 		)
 		assert.NoError(t, err)