Run report generation through module snapshots

internal/app/app_test.go

@@ -166,12 +166,18 @@ func TestGenerateReportWritesReportAndPreflight(t *testing.T) {
 		runBody: "# Daily Report\n\nRain this morning.\n",
 	}
 	outputPath := filepath.Join(t.TempDir(), "daily.md")
+	filesystemStore, err := state.NewFilesystemStore(cfg.Workspace)
+	if err != nil {
+		t.Fatalf("NewFilesystemStore() error = %v", err)
+	}
+	store := &recordingStore{Store: filesystemStore}
 
 	result, err := GenerateReport(context.Background(), ReportRequest{
 		Config:     cfg,
 		Resolved:   resolved,
 		OutputPath: outputPath,
 		Renderer:   renderer,
+		Store:      store,
 	})
 	if err != nil {
 		t.Fatalf("GenerateReport() error = %v", err)
@@ -195,6 +201,9 @@ func TestGenerateReportWritesReportAndPreflight(t *testing.T) {
 	if renderer.runRequest.OutputPath != result.ReportPath {
 		t.Fatalf("run OutputPath = %q, want managed report path %q", renderer.runRequest.OutputPath, result.ReportPath)
 	}
+	if got, want := strings.Join(store.calls, ","), "module_snapshot,briefing,data_package,preflight,metadata,prepare_report,metadata"; !strings.HasPrefix(got, want) {
+		t.Fatalf("store calls = %v, want prefix %s", store.calls, want)
+	}
 	assertPathsExist(t, result.BriefingPath, result.ModuleSnapshotPath, result.DataPackagePath, result.PreflightPath, result.ReportPath, result.MetadataPath, outputPath)
 	snapshotData, err := os.ReadFile(result.ModuleSnapshotPath)
 	if err != nil {
@@ -1536,6 +1545,41 @@ type recordingRenderer struct {
 	runBody       string
 }
 
+type recordingStore struct {
+	state.Store
+	calls []string
+}
+
+func (s *recordingStore) SaveBriefing(ctx context.Context, resolved report.Resolved, pkg briefing.Package) (string, error) {
+	s.calls = append(s.calls, "briefing")
+	return s.Store.SaveBriefing(ctx, resolved, pkg)
+}
+
+func (s *recordingStore) SaveModuleSnapshot(ctx context.Context, resolved report.Resolved, snapshot module.Snapshot) (string, error) {
+	s.calls = append(s.calls, "module_snapshot")
+	return s.Store.SaveModuleSnapshot(ctx, resolved, snapshot)
+}
+
+func (s *recordingStore) SaveDataPackage(ctx context.Context, resolved report.Resolved, pkg promptinput.Package) (string, error) {
+	s.calls = append(s.calls, "data_package")
+	return s.Store.SaveDataPackage(ctx, resolved, pkg)
+}
+
+func (s *recordingStore) SavePreflight(ctx context.Context, resolved report.Resolved, artifact state.PreflightArtifact) (string, error) {
+	s.calls = append(s.calls, "preflight")
+	return s.Store.SavePreflight(ctx, resolved, artifact)
+}
+
+func (s *recordingStore) PrepareRenderedReport(ctx context.Context, resolved report.Resolved) (string, error) {
+	s.calls = append(s.calls, "prepare_report")
+	return s.Store.PrepareRenderedReport(ctx, resolved)
+}
+
+func (s *recordingStore) SaveMetadata(ctx context.Context, metadata state.Metadata) (string, error) {
+	s.calls = append(s.calls, "metadata")
+	return s.Store.SaveMetadata(ctx, metadata)
+}
+
 func successfulRenderer(body string) *recordingRenderer {
 	return &recordingRenderer{
 		renderResult: &scriptorium.RenderResult{ExitCode: 0},