ejr/weatherfeeder
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Table-drive source registry tests

Browse Source
This commit is contained in:
Eric Rakestraw
2026-06-11 02:10:13 +00:00
parent b7277e0c02
commit 33541a71fc
1 changed files with 14 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

88
internal/sources/builtins_test.go
View File

@@ -6,61 +6,29 @@ import (
"gitea.maximumdirect.net/ejr/feedkit/config" "gitea.maximumdirect.net/ejr/feedkit/config"
fksource "gitea.maximumdirect.net/ejr/feedkit/sources" fksource "gitea.maximumdirect.net/ejr/feedkit/sources"
"gitea.maximumdirect.net/ejr/weatherfeeder/internal/sources/nws"
"gitea.maximumdirect.net/ejr/weatherfeeder/internal/sources/openmeteo"
"gitea.maximumdirect.net/ejr/weatherfeeder/internal/sources/openweather" "gitea.maximumdirect.net/ejr/weatherfeeder/internal/sources/openweather"
"gitea.maximumdirect.net/ejr/weatherfeeder/internal/sources/spc" "gitea.maximumdirect.net/ejr/weatherfeeder/internal/sources/spc"
) )
func TestRegisterBuiltinsRegistersNWSHourlyForecastDriver(t *testing.T) { func TestRegisterBuiltinsRegistersCurrentPollDrivers(t *testing.T) {
reg := fksource.NewRegistry() reg := fksource.NewRegistry()
RegisterBuiltins(reg) RegisterBuiltins(reg)
in, err := reg.BuildInput(sourceConfigForDriver(nws.DriverForecastHourly)) if len(pollDriverRegistrations) == 0 {
if err != nil { t.Fatalf("pollDriverRegistrations is empty")
t.Fatalf("BuildInput(nws_forecast_hourly) error = %v", err)
} }
if _, ok := in.(fksource.PollSource); !ok {
t.Fatalf("BuildInput(nws_forecast_hourly) type = %T, want PollSource", in)
}
}
func TestRegisterBuiltinsRegistersNWSNarrativeForecastDriver(t *testing.T) { for _, tt := range pollDriverRegistrations {
reg := fksource.NewRegistry() tt := tt
RegisterBuiltins(reg) t.Run(tt.driver, func(t *testing.T) {
in, err := reg.BuildInput(sourceConfigForDriver(tt.driver))
in, err := reg.BuildInput(sourceConfigForDriver(nws.DriverForecastNarrative)) if err != nil {
if err != nil { t.Fatalf("BuildInput(%q) error = %v", tt.driver, err)
t.Fatalf("BuildInput(nws_forecast_narrative) error = %v", err) }
} if _, ok := in.(fksource.PollSource); !ok {
if _, ok := in.(fksource.PollSource); !ok { t.Fatalf("BuildInput(%q) type = %T, want PollSource", tt.driver, in)
t.Fatalf("BuildInput(nws_forecast_narrative) type = %T, want PollSource", in) }
} })
}
func TestRegisterBuiltinsRegistersNWSForecastDiscussionDriver(t *testing.T) {
reg := fksource.NewRegistry()
RegisterBuiltins(reg)
in, err := reg.BuildInput(sourceConfigForDriver(nws.DriverForecastDiscussion))
if err != nil {
t.Fatalf("BuildInput(nws_forecast_discussion) error = %v", err)
}
if _, ok := in.(fksource.PollSource); !ok {
t.Fatalf("BuildInput(nws_forecast_discussion) type = %T, want PollSource", in)
}
}
func TestRegisterBuiltinsRegistersNWSWeatherStoriesDriver(t *testing.T) {
reg := fksource.NewRegistry()
RegisterBuiltins(reg)
in, err := reg.BuildInput(sourceConfigForDriver(nws.DriverWeatherStories))
if err != nil {
t.Fatalf("BuildInput(nws_weatherstories) error = %v", err)
}
if _, ok := in.(fksource.PollSource); !ok {
t.Fatalf("BuildInput(nws_weatherstories) type = %T, want PollSource", in)
} }
} }
@@ -77,34 +45,6 @@ func TestRegisterBuiltinsDoesNotRegisterLegacyNWSForecastDriver(t *testing.T) {
} }
} }
func TestRegisterBuiltinsRegistersAllCurrentDrivers(t *testing.T) {
reg := fksource.NewRegistry()
RegisterBuiltins(reg)
drivers := []string{
nws.DriverObservation,
nws.DriverAlerts,
nws.DriverForecastHourly,
nws.DriverForecastNarrative,
nws.DriverForecastDiscussion,
nws.DriverWeatherStories,
openmeteo.DriverObservation,
openmeteo.DriverForecast,
openweather.DriverObservation,
spc.DriverConvectiveOutlook,
}
for _, driver := range drivers {
in, err := reg.BuildInput(sourceConfigForDriver(driver))
if err != nil {
t.Fatalf("BuildInput(%s) error = %v", driver, err)
}
if _, ok := in.(fksource.PollSource); !ok {
t.Fatalf("BuildInput(%s) type = %T, want PollSource", driver, in)
}
}
}
func sourceConfigForDriver(driver string) config.SourceConfig { func sourceConfigForDriver(driver string) config.SourceConfig {
url := "https://example.invalid" url := "https://example.invalid"
if driver == openweather.DriverObservation { if driver == openweather.DriverObservation {