Eric Rakestraw
356c3be648
All checks were successful
ci/woodpecker/push/build-image Pipeline was successful
61 lines
1.7 KiB
Go
61 lines
1.7 KiB
Go
package sources
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"gitea.maximumdirect.net/ejr/feedkit/config"
|
|
fksource "gitea.maximumdirect.net/ejr/feedkit/sources"
|
|
)
|
|
|
|
func TestRegisterBuiltinsRegistersNWSHourlyForecastDriver(t *testing.T) {
|
|
reg := fksource.NewRegistry()
|
|
RegisterBuiltins(reg)
|
|
|
|
in, err := reg.BuildInput(sourceConfigForDriver("nws_forecast_hourly"))
|
|
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_forecast_narrative"))
|
|
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 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 sourceConfigForDriver(driver string) config.SourceConfig {
|
|
return config.SourceConfig{
|
|
Name: "test-source",
|
|
Driver: driver,
|
|
Mode: config.SourceModePoll,
|
|
Params: map[string]any{
|
|
"url": "https://example.invalid",
|
|
"user_agent": "test-agent",
|
|
},
|
|
}
|
|
}
|