113 lines
4.0 KiB
Go
113 lines
4.0 KiB
Go
package postgres
|
|
|
|
import (
|
|
"database/sql"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestMapObservationParentRowNullables(t *testing.T) {
|
|
observedAt := time.Date(2026, 3, 19, 23, 45, 0, 0, time.FixedZone("CST", -6*3600))
|
|
isDay := true
|
|
temp := 18.25
|
|
|
|
obs := mapObservationParentRow(observationParentRow{
|
|
StationID: sql.NullString{String: "KSTL", Valid: true},
|
|
StationName: sql.NullString{String: "St. Louis", Valid: true},
|
|
ObservedAt: observedAt,
|
|
ConditionCode: 2,
|
|
IsDay: sql.NullBool{Bool: isDay, Valid: true},
|
|
TemperatureC: sql.NullFloat64{Float64: temp, Valid: true},
|
|
TextDescription: sql.NullString{String: "Partly Cloudy", Valid: true},
|
|
})
|
|
|
|
if obs.StationID != "KSTL" {
|
|
t.Fatalf("expected station id KSTL, got %q", obs.StationID)
|
|
}
|
|
if obs.IsDay == nil || !*obs.IsDay {
|
|
t.Fatalf("expected isDay pointer true, got %v", obs.IsDay)
|
|
}
|
|
if obs.TemperatureC == nil || *obs.TemperatureC != temp {
|
|
t.Fatalf("expected temperature %v, got %v", temp, obs.TemperatureC)
|
|
}
|
|
if obs.DewpointC != nil {
|
|
t.Fatalf("expected nil dewpoint, got %v", *obs.DewpointC)
|
|
}
|
|
if got := obs.Timestamp.Location().String(); got != "UTC" {
|
|
t.Fatalf("expected UTC timestamp, got %s", got)
|
|
}
|
|
}
|
|
|
|
func TestMapObservationPresentWeatherRow(t *testing.T) {
|
|
row := observationPresentWeatherRow{
|
|
WeatherIndex: 1,
|
|
RawText: sql.NullString{String: `{"code":61,"text":"rain"}`, Valid: true},
|
|
}
|
|
|
|
pw, err := mapObservationPresentWeatherRow(row)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if pw.Raw == nil {
|
|
t.Fatalf("expected raw map to be populated")
|
|
}
|
|
if got, ok := pw.Raw["text"].(string); !ok || got != "rain" {
|
|
t.Fatalf("expected raw text rain, got %#v", pw.Raw["text"])
|
|
}
|
|
}
|
|
|
|
func TestMapForecastPeriodRowNullables(t *testing.T) {
|
|
start := time.Date(2026, 3, 20, 0, 0, 0, 0, time.UTC)
|
|
end := start.Add(1 * time.Hour)
|
|
period := mapForecastPeriodRow(forecastPeriodRow{
|
|
StartTime: start,
|
|
EndTime: end,
|
|
ConditionCode: 80,
|
|
Name: sql.NullString{String: "Midnight", Valid: true},
|
|
TemperatureC: sql.NullFloat64{Float64: 12.5, Valid: true},
|
|
TemperatureCMin: sql.NullFloat64{Valid: false},
|
|
})
|
|
|
|
if period.Name != "Midnight" {
|
|
t.Fatalf("expected period name Midnight, got %q", period.Name)
|
|
}
|
|
if period.TemperatureC == nil || *period.TemperatureC != 12.5 {
|
|
t.Fatalf("expected temperature pointer 12.5, got %v", period.TemperatureC)
|
|
}
|
|
if period.TemperatureCMin != nil {
|
|
t.Fatalf("expected nil TemperatureCMin, got %v", *period.TemperatureCMin)
|
|
}
|
|
if !period.StartTime.Equal(start) || !period.EndTime.Equal(end) {
|
|
t.Fatalf("unexpected time range: %s - %s", period.StartTime, period.EndTime)
|
|
}
|
|
}
|
|
|
|
func TestAttachAlertReferencesPreservesOrder(t *testing.T) {
|
|
sent1 := time.Date(2026, 3, 20, 1, 0, 0, 0, time.UTC)
|
|
sent2 := sent1.Add(10 * time.Minute)
|
|
|
|
alerts := []indexedAlert{
|
|
{Index: 4, Alert: mapAlertRow(alertRow{AlertIndex: 4, AlertID: "a-4"}).Alert},
|
|
{Index: 9, Alert: mapAlertRow(alertRow{AlertIndex: 9, AlertID: "a-9"}).Alert},
|
|
}
|
|
references := []indexedAlertReference{
|
|
mapAlertReferenceRow(alertReferenceRow{AlertIndex: 4, Identifier: sql.NullString{String: "r1", Valid: true}, Sent: sql.NullTime{Time: sent1, Valid: true}}),
|
|
mapAlertReferenceRow(alertReferenceRow{AlertIndex: 4, Identifier: sql.NullString{String: "r2", Valid: true}, Sent: sql.NullTime{Time: sent2, Valid: true}}),
|
|
mapAlertReferenceRow(alertReferenceRow{AlertIndex: 9, Identifier: sql.NullString{String: "r9", Valid: true}}),
|
|
}
|
|
|
|
out := attachAlertReferences(alerts, references)
|
|
if len(out) != 2 {
|
|
t.Fatalf("expected 2 alerts, got %d", len(out))
|
|
}
|
|
if len(out[0].References) != 2 {
|
|
t.Fatalf("expected first alert to have 2 references, got %d", len(out[0].References))
|
|
}
|
|
if out[0].References[0].Identifier != "r1" || out[0].References[1].Identifier != "r2" {
|
|
t.Fatalf("unexpected first alert reference order: %+v", out[0].References)
|
|
}
|
|
if len(out[1].References) != 1 || out[1].References[0].Identifier != "r9" {
|
|
t.Fatalf("unexpected second alert references: %+v", out[1].References)
|
|
}
|
|
}
|