Eric Rakestraw
fd820fd964
All checks were successful
ci/woodpecker/push/build-image Pipeline was successful
80 lines
2.2 KiB
Go
80 lines
2.2 KiB
Go
package postgres
|
|
|
|
import "testing"
|
|
|
|
func TestWeatherPostgresSchemaShape(t *testing.T) {
|
|
s := PostgresSchema()
|
|
if s.MapEvent == nil {
|
|
t.Fatalf("PostgresSchema().MapEvent is nil")
|
|
}
|
|
|
|
wantTables := map[string]bool{
|
|
tableObservations: true,
|
|
tableObservationPresentWeather: true,
|
|
tableForecasts: true,
|
|
tableForecastPeriods: true,
|
|
tableForecastDiscussions: true,
|
|
tableForecastDiscussionKeyMessages: true,
|
|
tableWeatherStoryRuns: true,
|
|
tableWeatherStories: true,
|
|
tableAlertRuns: true,
|
|
tableAlerts: true,
|
|
tableAlertReferences: true,
|
|
}
|
|
|
|
if len(s.Tables) != len(wantTables) {
|
|
t.Fatalf("PostgresSchema().Tables len = %d, want %d", len(s.Tables), len(wantTables))
|
|
}
|
|
|
|
seenIndexes := map[string]bool{}
|
|
for _, tbl := range s.Tables {
|
|
if !wantTables[tbl.Name] {
|
|
t.Fatalf("unexpected table %q in schema", tbl.Name)
|
|
}
|
|
if tbl.PruneColumn == "" {
|
|
t.Fatalf("table %q missing prune column", tbl.Name)
|
|
}
|
|
for _, idx := range tbl.Indexes {
|
|
if seenIndexes[idx.Name] {
|
|
t.Fatalf("duplicate index name %q", idx.Name)
|
|
}
|
|
seenIndexes[idx.Name] = true
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestWeatherPostgresSchemaIncludesWeatherStoryColumns(t *testing.T) {
|
|
runColumns := columnsForTable(t, tableWeatherStoryRuns)
|
|
if !runColumns["as_of"] {
|
|
t.Fatalf("%s missing as_of column", tableWeatherStoryRuns)
|
|
}
|
|
if !runColumns["story_count"] {
|
|
t.Fatalf("%s missing story_count column", tableWeatherStoryRuns)
|
|
}
|
|
|
|
storyColumns := columnsForTable(t, tableWeatherStories)
|
|
for _, col := range []string{"start_time", "end_time", "updated_at", "title", "description", "alt_text", "priority", "story_order", "download_url"} {
|
|
if !storyColumns[col] {
|
|
t.Fatalf("%s missing %s column", tableWeatherStories, col)
|
|
}
|
|
}
|
|
}
|
|
|
|
func columnsForTable(t *testing.T, table string) map[string]bool {
|
|
t.Helper()
|
|
|
|
schema := PostgresSchema()
|
|
for _, tbl := range schema.Tables {
|
|
if tbl.Name != table {
|
|
continue
|
|
}
|
|
cols := make(map[string]bool, len(tbl.Columns))
|
|
for _, col := range tbl.Columns {
|
|
cols[col.Name] = true
|
|
}
|
|
return cols
|
|
}
|
|
t.Fatalf("missing table %q", table)
|
|
return nil
|
|
}
|