diff --git a/tests/postgres/generator_test.go b/tests/postgres/generator_test.go index a1ea307..fe1407f 100644 --- a/tests/postgres/generator_test.go +++ b/tests/postgres/generator_test.go @@ -222,10 +222,18 @@ func TestGenerator_TableMetadata(t *testing.T) { // Spot check the actor table and assert that the emitted // properties are as expected. var got metadata.Table + var specialFeatures metadata.Column for _, table := range schema.TablesMetaData { if table.Name == "actor" { got = table } + if table.Name == "film" { + for _, column := range table.Columns { + if column.Name == "special_features" { + specialFeatures = column + } + } + } } want := metadata.Table{ @@ -238,6 +246,7 @@ func TestGenerator_TableMetadata(t *testing.T) { }, } require.Equal(t, want, got) + require.Equal(t, metadata.ArrayType, specialFeatures.DataType.Kind) } func TestGeneratorSpecialCharacters(t *testing.T) {