Clean up.

artie-labs · Jul 30, 2024 · 866ce73 · 866ce73
1 parent 3999984
commit 866ce73
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/models/event/event_test.go b/models/event/event_test.go
@@ -58,6 +58,12 @@ func (e *EventsTestSuite) TestEvent_IsValid() {
 }
 
 func (e *EventsTestSuite) TestEvent_TableName() {
+	{
+		// Don't pass in tableName.
+		evt, err := ToMemoryEvent(e.fakeEvent, idMap, &kafkalib.TopicConfig{}, config.Replication)
+		assert.NoError(e.T(), err)
+		assert.Equal(e.T(), e.fakeEvent.GetTableName(), evt.Table)
+	}
 	{
 		// Now pass it in, it should override.
 		evt, err := ToMemoryEvent(e.fakeEvent, idMap, &kafkalib.TopicConfig{TableName: "orders"}, config.Replication)

diff --git a/models/event/events_suite_test.go b/models/event/events_suite_test.go
@@ -34,7 +34,7 @@ func (e *EventsTestSuite) SetupTest() {
 	fakeEvent := &mocks.FakeEvent{}
 	fakeEvent.GetDataReturns(map[string]any{constants.DeleteColumnMarker: false, constants.OnlySetDeleteColumnMarker: false}, nil)
 	fakeEvent.GetColumnsReturns(&columns.Columns{}, nil)
-
+	fakeEvent.GetTableNameReturns("foo")
 	e.fakeEvent = fakeEvent
 }