128 lines
3.7 KiB
Go
128 lines
3.7 KiB
Go
package sources
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"gitea.maximumdirect.net/ejr/feedkit/config"
|
|
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/spc"
|
|
)
|
|
|
|
func TestRegisterBuiltinsRegistersNWSHourlyForecastDriver(t *testing.T) {
|
|
reg := fksource.NewRegistry()
|
|
RegisterBuiltins(reg)
|
|
|
|
in, err := reg.BuildInput(sourceConfigForDriver(nws.DriverForecastHourly))
|
|
if err != nil {
|
|
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) {
|
|
reg := fksource.NewRegistry()
|
|
RegisterBuiltins(reg)
|
|
|
|
in, err := reg.BuildInput(sourceConfigForDriver(nws.DriverForecastNarrative))
|
|
if err != nil {
|
|
t.Fatalf("BuildInput(nws_forecast_narrative) error = %v", err)
|
|
}
|
|
if _, ok := in.(fksource.PollSource); !ok {
|
|
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)
|
|
}
|
|
}
|
|
|
|
func TestRegisterBuiltinsDoesNotRegisterLegacyNWSForecastDriver(t *testing.T) {
|
|
reg := fksource.NewRegistry()
|
|
RegisterBuiltins(reg)
|
|
|
|
_, err := reg.BuildInput(sourceConfigForDriver("nws_forecast"))
|
|
if err == nil {
|
|
t.Fatalf("BuildInput(nws_forecast) expected unknown driver error")
|
|
}
|
|
if !strings.Contains(err.Error(), `unknown source driver: "nws_forecast"`) {
|
|
t.Fatalf("error = %q, want unknown source driver for nws_forecast", err)
|
|
}
|
|
}
|
|
|
|
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 {
|
|
url := "https://example.invalid"
|
|
if driver == openweather.DriverObservation {
|
|
url = "https://example.invalid?units=metric"
|
|
}
|
|
params := map[string]any{
|
|
"url": url,
|
|
"user_agent": "test-agent",
|
|
}
|
|
if driver == spc.DriverConvectiveOutlook {
|
|
params["latitude"] = 38.6239
|
|
params["longitude"] = -90.3571
|
|
}
|
|
return config.SourceConfig{
|
|
Name: "test-source",
|
|
Driver: driver,
|
|
Mode: config.SourceModePoll,
|
|
Params: params,
|
|
}
|
|
}
|