fix: Use tables with primary key in delete-stale tests (#1195)

Tested by:
* cloudquery/cloudquery#13568
* cloudquery/cloudquery#13600
BEGIN_COMMIT_OVERRIDE
fix: Use tables with primary key in `delete-stale` tests (#1195)
fix: Return a pointer to the actual table column instead of a copy in `schema.Table.Column` (#1195)
END_COMMIT_OVERRIDE

Author: candiduslynx

plugin/testing_write_delete.go

@@ -19,7 +19,7 @@ func (s *WriterTestSuite) testDeleteStaleBasic(ctx context.Context) {
 	table := &schema.Table{
 		Name: tableName,
 		Columns: schema.ColumnList{
-			schema.Column{Name: "id", Type: arrow.PrimitiveTypes.Int64},
+			schema.Column{Name: "id", Type: arrow.PrimitiveTypes.Int64, PrimaryKey: true, NotNull: true},
 			schema.CqSourceNameColumn,
 			schema.CqSyncTimeColumn,
 		},
@@ -85,6 +85,7 @@ func (s *WriterTestSuite) testDeleteStaleAll(ctx context.Context) {
 	syncTime := time.Now().UTC().Truncate(time.Microsecond)
 	table := schema.TestTable(tableName, s.genDatOptions)
 	table.Columns = append(schema.ColumnList{schema.CqSourceNameColumn, schema.CqSyncTimeColumn}, table.Columns...)
+	table.Columns[table.Columns.Index("id")].PrimaryKey = true
 	require.NoErrorf(s.t, s.plugin.writeOne(ctx, &message.WriteMigrateTable{Table: table}), "failed to create table")
 
 	tg := schema.NewTestDataGenerator()