Add near-term generate command
This commit is contained in:
@@ -1162,23 +1162,59 @@ func TestInspectMissingMetadata(t *testing.T) {
|
||||
func TestResolveGenerateMapsCommandToReportDefinition(t *testing.T) {
|
||||
cfg := config.Defaults()
|
||||
cfg.WeatherAPI.Timezone = "America/Chicago"
|
||||
now := mustParse("2026-05-29T18:00:00-05:00")
|
||||
now := mustParse("2026-05-29T08:00:00-05:00")
|
||||
|
||||
resolved, err := ResolveGenerate(GenerateRequest{
|
||||
Config: cfg,
|
||||
Report: ReportTomorrow,
|
||||
}, now)
|
||||
if err != nil {
|
||||
t.Fatalf("ResolveGenerate() error = %v", err)
|
||||
tests := []struct {
|
||||
name string
|
||||
kind ReportKind
|
||||
wantID report.ID
|
||||
wantPrompt string
|
||||
wantStart string
|
||||
wantEnd string
|
||||
requestDate time.Time
|
||||
}{
|
||||
{
|
||||
name: "tomorrow",
|
||||
kind: ReportTomorrow,
|
||||
wantID: report.DailyTomorrow,
|
||||
wantPrompt: "weather.daily_report",
|
||||
wantStart: "2026-05-30T00:00:00-05:00",
|
||||
wantEnd: "2026-05-31T00:00:00-05:00",
|
||||
requestDate: time.Time{},
|
||||
},
|
||||
{
|
||||
name: "near-term",
|
||||
kind: ReportNearTerm,
|
||||
wantID: report.NearTerm,
|
||||
wantPrompt: "weather.near_term_report",
|
||||
wantStart: "2026-05-29T08:00:00-05:00",
|
||||
wantEnd: "2026-05-29T14:00:00-05:00",
|
||||
},
|
||||
}
|
||||
if resolved.Definition.ID != report.DailyTomorrow {
|
||||
t.Fatalf("ID = %q, want daily_tomorrow", resolved.Definition.ID)
|
||||
}
|
||||
if resolved.Definition.PromptID != "weather.daily_report" {
|
||||
t.Fatalf("PromptID = %q, want weather.daily_report", resolved.Definition.PromptID)
|
||||
}
|
||||
if got := resolved.ValidPeriod.Start.Format("2006-01-02"); got != "2026-05-30" {
|
||||
t.Fatalf("valid start date = %s, want 2026-05-30", got)
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
resolved, err := ResolveGenerate(GenerateRequest{
|
||||
Config: cfg,
|
||||
Report: tt.kind,
|
||||
Date: tt.requestDate,
|
||||
}, now)
|
||||
if err != nil {
|
||||
t.Fatalf("ResolveGenerate() error = %v", err)
|
||||
}
|
||||
if resolved.Definition.ID != tt.wantID {
|
||||
t.Fatalf("ID = %q, want %q", resolved.Definition.ID, tt.wantID)
|
||||
}
|
||||
if resolved.Definition.PromptID != tt.wantPrompt {
|
||||
t.Fatalf("PromptID = %q, want %q", resolved.Definition.PromptID, tt.wantPrompt)
|
||||
}
|
||||
if got := resolved.ValidPeriod.Start.Format(time.RFC3339); got != tt.wantStart {
|
||||
t.Fatalf("valid start = %s, want %s", got, tt.wantStart)
|
||||
}
|
||||
if got := resolved.ValidPeriod.End.Format(time.RFC3339); got != tt.wantEnd {
|
||||
t.Fatalf("valid end = %s, want %s", got, tt.wantEnd)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user