Function TestTableMetadata_DropColumn

backend/store/model/database_test.go:299–335

(t *testing.T)

}

func TestTableMetadata_DropColumn(t *testing.T) {
	metadata := &storepb.DatabaseSchemaMetadata{
		Name: "testdb",
		Schemas: []*storepb.SchemaMetadata{
			{
				Name: "public",
				Tables: []*storepb.TableMetadata{
					{
						Name: "users",
						Columns: []*storepb.ColumnMetadata{
							{Name: "id", Type: "int"},
							{Name: "email", Type: "varchar"},
							{Name: "name", Type: "varchar"},
						},
					},
				},
			},
		},
	}

	schema := NewDatabaseMetadata(metadata, nil, nil, storepb.Engine_POSTGRES, true)
	schemaMeta := schema.GetSchemaMetadata("public")
	tableMeta := schemaMeta.GetTable("users")

	// Drop column
	err := tableMeta.DropColumn("email")

	require.Nil(t, err)

	// Verify column is gone
	retrieved := tableMeta.GetColumn("email")
	require.Nil(t, retrieved)

	// Verify other columns still exist
	require.NotNil(t, tableMeta.GetColumn("id"))
	require.NotNil(t, tableMeta.GetColumn("name"))
}

func TestTableMetadata_DropColumn_NotExists(t *testing.T) {
	metadata := &storepb.DatabaseSchemaMetadata{