Function TestTableMetadata_CreateColumn

backend/store/model/database_test.go:227–264

(t *testing.T)

}

func TestTableMetadata_CreateColumn(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"},
						},
					},
				},
			},
		},
	}

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

	// Create a new column
	columnProto := &storepb.ColumnMetadata{
		Name:     "email",
		Type:     "varchar",
		Nullable: true,
	}
	err := tableMeta.CreateColumn(columnProto, nil /* columnCatalog */)

	require.Nil(t, err)

	// Verify column is now accessible
	retrieved := tableMeta.GetColumn("email")
	require.NotNil(t, retrieved)
	require.Equal(t, "email", retrieved.GetProto().Name)
	require.Equal(t, "varchar", retrieved.GetProto().Type)
}

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