Function TestDelete

plugins/destination/sqlite/client/delete_test.go:21–74

(t *testing.T)

)

func TestDelete(t *testing.T) {
	testCases := []struct {
		name          string
		insertValues  []string
		deleteValues  []string
		deleteAll     bool
		expectedCount int
	}{
		{
			name:          "delete single record",
			insertValues:  []string{"foo", "bar"},
			deleteValues:  []string{"foo"},
			expectedCount: 1,
		},
		{
			name:          "delete both records",
			insertValues:  []string{"foo", "bar"},
			deleteValues:  []string{"foo", "bar"},
			expectedCount: 0,
		},
		{
			name:          "delete none",
			insertValues:  []string{"foo"},
			deleteValues:  []string{"bar"},
			expectedCount: 1,
		},
		{
			name:          "delete all records",
			insertValues:  []string{"foo", "bar"},
			deleteAll:     true,
			expectedCount: 0,
		},
	}
	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			r := require.New(t)
			ctx := context.Background()
			client := withPluginClient(ctx, r)

			table := createTestTable()
			r.NoError(client.MigrateTables(ctx, message.WriteMigrateTables{{Table: table}}))

			writeInserts := createInsertMessages(tc.insertValues, table)
			r.NoError(client.WriteTableBatch(ctx, "", writeInserts))

			writeDeletes := createDeleteMessages(tc.deleteAll, table, tc.deleteValues)
			r.NoError(client.DeleteRecord(ctx, writeDeletes))

			count, err := countAllRows(ctx, client, table)
			r.NoError(err)
			r.EqualValues(tc.expectedCount, count, "unexpected amount of items after delete with match")
		})
	}
}

func countAllRows(ctx context.Context, client *Client, table *schema.Table) (int64, error) {
	var err error