Janady13/missionmatch / internal/integration/graph_lifecycle_integration_test.go

1//go:build integration 2 3package integration 4 5import ( 6 "context" 7 "encoding/json" 8 "testing" 9 "time" 10 11 "github.com/google/uuid" 12 "github.com/jackc/pgx/v5/pgtype" 13 "github.com/shopspring/decimal" 14 15 "github.com/missionworkforce/missionmatch/internal/modules/ai" 16 "github.com/missionworkforce/missionmatch/internal/platform/database/dbgen" 17) 18 19// seedLifecyclePhase inserts one opportunity_records phase (forecast, active_solicitation, or contract_award) 20// through the migrator (the global catalog the app role may only read), carrying the FULL (agency, naics, 21// psc, pop_state, set_aside) segment tuple plus the shared dedupe_key that links the phases, so the G9 22// succession has a real lifecycle to walk. The contract_award phase carries the prior estimated_value and 23// incumbent UEI; the forecast phase carries the estimated_award_date. It returns the generated uuid so the 24// anchor solicitation can be walked. 25func seedLifecyclePhase(t *testing.T, srcID, dedupe, kind, agency, naics, psc, popState, setAside string, 26 estVal *decimal.Decimal, incumbentUEI, incumbentName string, awardDate, estAwardDate, popEnd pgtype.Date) string { 27 t.Helper() 28 if err := dbgen.New(migrator).UpsertOpportunityRecord(context.Background(), dbgen.UpsertOpportunityRecordParams{ 29 SourceName: "SAM.gov", SourceRecordID: srcID, DedupeKey: dedupe, RecordKind: kind, 30 Title: "Mission support " + kind, Agency: agency, Naics: naics, Psc: psc, PopState: popState, 31 SetAside: setAside, EstimatedValue: estVal, ValueSource: "operator", 32 IncumbentUei: incumbentUEI, IncumbentName: incumbentName, 33 AwardDate: awardDate, EstimatedAwardDate: estAwardDate, PopEndDate: popEnd, 34 IsPlanning: kind == "forecast", FetchedAt: time.Now().UTC(), 35 }); err != nil { 36 t.Fatalf("seed lifecycle phase %s (%s): %v", srcID, kind, err) 37 } 38 var id string 39 if err := migrator.QueryRow(context.Background(), 40 "SELECT id::text FROM opportunity_records WHERE source_name='SAM.gov' AND source_record_id=$1", srcID).Scan(&id); err != nil { 41 t.Fatalf("resolve lifecycle phase id %s: %v", srcID, err) 42 } 43 return id 44} 45 46// TestG9ProcurementSuccession is the live proof of the recompete lifecycle walk against real Postgres. It 47// seeds one lifecycle in the SAME (agency, naics, psc, pop_state, set_aside) segment sharing one dedupe_key, a 48// forecast, an active_solicitation anchor, and a predecessor contract_award, then proves the succession from 49// the solicitation returns is_recompete true (a predecessor was found), the prior award amount as an exact 50// decimal, the prior prime UEI explicitly labeled prime-of-record, and the announcing forecast, with the cited 51// record ids. A second solicitation with no predecessor returns is_new true. 52func TestG9ProcurementSuccession(t *testing.T) { 53 org := seedOrg(t) 54 run := uuid.NewString()[:8] 55 agency := "DEPT OF THE AIR FORCE " + run 56 naics, psc, popState, setAside := "541512", "R425", "HI", "SBA" 57 dedupe := "LIFECYCLE-DEDUPE-" + run 58 59 priorAmt := decimal.RequireFromString("7500000.00") 60 awardDate := pgtype.Date{Time: time.Now().UTC().AddDate(-3, 0, 0), Valid: true} 61 estAwardDate := pgtype.Date{Time: time.Now().UTC().AddDate(0, 4, 0), Valid: true} 62 63 // The predecessor contract award (the prior prime of record), the announcing forecast, and the anchor 64 // solicitation, all sharing the dedupe_key and the full segment tuple. 65 predID := seedLifecyclePhase(t, "G9-AWD-"+run, dedupe, "contract_award", agency, naics, psc, popState, setAside, 66 &priorAmt, "UEIPRIORPRIME", "Prior Prime LLC", awardDate, pgtype.Date{}, pgtype.Date{}) 67 fcastID := seedLifecyclePhase(t, "G9-FCAST-"+run, dedupe, "forecast", agency, naics, psc, popState, setAside, 68 nil, "", "", pgtype.Date{}, estAwardDate, pgtype.Date{}) 69 solID := seedLifecyclePhase(t, "G9-SOL-"+run, dedupe, "active_solicitation", agency, naics, psc, popState, setAside, 70 nil, "", "", pgtype.Date{}, pgtype.Date{}, pgtype.Date{}) 71 72 rows, err := ai.ProcurementSuccession(ctxOrg(org), appDB, solID) 73 if err != nil { 74 t.Fatalf("procurement succession: %v", err) 75 } 76 if len(rows) != 1 { 77 t.Fatalf("the succession must return exactly the one anchored solicitation row, got %d", len(rows)) 78 } 79 r := rows[0] 80 if r.SolicitationID != solID { 81 t.Errorf("the anchor must be the named solicitation, got %q", r.SolicitationID) 82 } 83 if !r.IsRecompete || r.IsNew { 84 t.Errorf("a solicitation with a predecessor award must read is_recompete true and is_new false, got recompete %v new %v", r.IsRecompete, r.IsNew) 85 } 86 if r.PredecessorAwardID != predID { 87 t.Errorf("the cited predecessor must be the matched award, got %q want %q", r.PredecessorAwardID, predID) 88 } 89 if r.IncumbentUEI != "UEIPRIORPRIME" { 90 t.Errorf("the prior prime UEI must be the predecessor incumbent, got %q", r.IncumbentUEI) 91 } 92 if r.IncumbentRole != "prime-of-record" { 93 t.Errorf("the incumbent edge must be labeled prime-of-record, got %q", r.IncumbentRole) 94 } 95 if r.PriorAwardAmount == nil || !r.PriorAwardAmount.Equal(priorAmt) { 96 t.Errorf("the prior award amount must be the decimal 7500000.00, got %v", r.PriorAwardAmount) 97 } 98 if r.ForecastID != fcastID { 99 t.Errorf("the announcing forecast must be cited, got %q want %q", r.ForecastID, fcastID) 100 } 101 102 // A second solicitation in a fresh segment with no predecessor reads as a new start. 103 freshDedupe := "LIFECYCLE-NEW-" + run 104 newSolID := seedLifecyclePhase(t, "G9-NEW-"+run, freshDedupe, "active_solicitation", 105 "DEPT OF THE NAVY "+run, "541330", "R408", "VA", "FULL", nil, "", "", pgtype.Date{}, pgtype.Date{}, pgtype.Date{}) 106 newRows, err := ai.ProcurementSuccession(ctxOrg(org), appDB, newSolID) 107 if err != nil { 108 t.Fatalf("procurement succession (new start): %v", err) 109 } 110 if len(newRows) != 1 { 111 t.Fatalf("the new start succession must return one row, got %d", len(newRows)) 112 } 113 if newRows[0].IsRecompete || !newRows[0].IsNew { 114 t.Errorf("a solicitation with no predecessor must read is_new true and is_recompete false, got recompete %v new %v", 115 newRows[0].IsRecompete, newRows[0].IsNew) 116 } 117 if newRows[0].PriorAwardAmount != nil { 118 t.Errorf("a new start must carry no prior award amount, got %v", newRows[0].PriorAwardAmount) 119 } 120 121 // Byte stable on a re run (the determinism mandate). 122 again, err := ai.ProcurementSuccession(ctxOrg(org), appDB, solID) 123 if err != nil { 124 t.Fatalf("procurement succession re run: %v", err) 125 } 126 first, _ := json.Marshal(rows) 127 second, _ := json.Marshal(again) 128 if string(first) != string(second) { 129 t.Errorf("the succession must be byte stable on a re run.\nfirst: %s\nsecond: %s", first, second) 130 } 131} 132 133// TestG9RecompeteClock proves the recompete clock returns the opportunities and awards whose period of 134// performance end falls inside the forward window for an agency, ordered by that end date with a record id 135// tiebreaker, each row carrying its source so an award is never conflated with an opportunity, the incumbent 136// prime labeled prime-of-record, the dollars exact decimals, and the result byte stable. The award arm keys 137// ONLY on awarding_agency, the only agency key usaspending_awards carries. 138func TestG9RecompeteClock(t *testing.T) { 139 org := seedOrg(t) 140 run := uuid.NewString()[:8] 141 agency := "DEPT OF DEFENSE " + run 142 143 insideWindow := time.Now().UTC().AddDate(0, 6, 0) // inside the 24 month window 144 outsideWindow := time.Now().UTC().AddDate(5, 0, 0) // past the window 145 146 // An award ending inside the window and one far outside it (must not appear), reusing the G1 award seeder. 147 seedMarketAward(t, "G9-RC-IN-"+run, "PIID-IN", agency, "541512", "UEIAWARDIN999", "Award In Window LLC", "3000000.00", insideWindow, false) 148 seedMarketAward(t, "G9-RC-OUT-"+run, "PIID-OUT", agency, "541512", "UEIAWARDOUT99", "Award Out Window LLC", "3000000.00", outsideWindow, false) 149 150 // An opportunity ending inside the window. 151 estVal := decimal.RequireFromString("4250000.00") 152 _ = seedLifecyclePhase(t, "G9-RC-OPP-"+run, "G9-RC-OPP-"+run, "active_solicitation", 153 agency, "541512", "R425", "HI", "SBA", &estVal, "UEIOPPINCUMB9", "Opp Incumbent LLC", 154 pgtype.Date{}, pgtype.Date{}, pgtype.Date{Time: insideWindow, Valid: true}) 155 156 rows, err := ai.RecompeteClock(ctxOrg(org), appDB, agency, ai.DefaultRecompeteWindowMonths) 157 if err != nil { 158 t.Fatalf("recompete clock: %v", err) 159 } 160 if len(rows) != 2 { 161 t.Fatalf("the clock must return exactly the in window award and opportunity, got %d", len(rows)) 162 } 163 164 var sawAward, sawOpp bool 165 prev := time.Time{} 166 for _, r := range rows { 167 end, perr := time.Parse("2006-01-02", r.PopEndDate) 168 if perr != nil { 169 t.Fatalf("the pop end date must be a stable ISO date, got %q", r.PopEndDate) 170 } 171 if end.Before(prev) { 172 t.Errorf("the clock must be ordered by pop end date ascending; %s precedes %s", end, prev) 173 } 174 prev = end 175 if r.IncumbentRole != "prime-of-record" { 176 t.Errorf("every clock row must label the incumbent prime-of-record, got %q", r.IncumbentRole) 177 } 178 switch r.SourceTable { 179 case "usaspending_awards": 180 sawAward = true 181 if r.RecordID != "G9-RC-IN-"+run { 182 t.Errorf("only the in window award may appear, got %q", r.RecordID) 183 } 184 case "opportunity_records": 185 sawOpp = true 186 if r.Dollars == nil || !r.Dollars.Equal(decimal.RequireFromString("4250000.00")) { 187 t.Errorf("the opportunity dollars must be the decimal 4250000.00, got %v", r.Dollars) 188 } 189 default: 190 t.Errorf("a clock row must carry a known source table, got %q", r.SourceTable) 191 } 192 } 193 if !sawAward || !sawOpp { 194 t.Errorf("the clock must include both the in window award and opportunity (award %v, opp %v)", sawAward, sawOpp) 195 } 196 197 again, err := ai.RecompeteClock(ctxOrg(org), appDB, agency, ai.DefaultRecompeteWindowMonths) 198 if err != nil { 199 t.Fatalf("recompete clock re run: %v", err) 200 } 201 first, _ := json.Marshal(rows) 202 second, _ := json.Marshal(again) 203 if string(first) != string(second) { 204 t.Errorf("the recompete clock must be byte stable on a re run.\nfirst: %s\nsecond: %s", first, second) 205 } 206} 207 208// TestG9ForecastRunway proves the forecast runway returns the forecasts whose estimated award date falls 209// inside the forward window, joined to the federal mission area and carrying the place of performance state, 210// ordered by estimated award date with an id tiebreaker, the dollars exact decimals, and byte stable. A 211// forecast landing outside the window does not appear. 212func TestG9ForecastRunway(t *testing.T) { 213 org := seedOrg(t) 214 run := uuid.NewString()[:8] 215 agency := "DEPT OF THE ARMY " + run 216 217 inside := pgtype.Date{Time: time.Now().UTC().AddDate(0, 5, 0), Valid: true} // inside a 12 month window 218 outside := pgtype.Date{Time: time.Now().UTC().AddDate(3, 0, 0), Valid: true} // 3 years out 219 220 estVal := decimal.RequireFromString("9100000.00") 221 insideID := seedLifecyclePhase(t, "G9-FR-IN-"+run, "G9-FR-IN-"+run, "forecast", 222 agency, "541512", "R425", "NM", "8A", &estVal, "", "", pgtype.Date{}, inside, pgtype.Date{}) 223 _ = seedLifecyclePhase(t, "G9-FR-OUT-"+run, "G9-FR-OUT-"+run, "forecast", 224 agency, "541512", "R425", "NM", "8A", nil, "", "", pgtype.Date{}, outside, pgtype.Date{}) 225 226 // Tag the in window forecast SIGINT so the runway carries the readable mission area. 227 if err := dbgen.New(migrator).UpsertFederalMissionTag(context.Background(), dbgen.UpsertFederalMissionTagParams{ 228 RecordTable: "opportunity_records", RecordID: insideID, MissionCode: "SIGINT", 229 Mechanism: "keyword", Evidence: "signals intelligence", Confidence: decimal.RequireFromString("1.00"), 230 }); err != nil { 231 t.Fatalf("tag forecast: %v", err) 232 } 233 234 rows, err := ai.ForecastRunway(ctxOrg(org), appDB, 12) 235 if err != nil { 236 t.Fatalf("forecast runway: %v", err) 237 } 238 239 var sawInside, sawOutside bool 240 prev := time.Time{} 241 for _, r := range rows { 242 d, perr := time.Parse("2006-01-02", r.EstimatedAwardDate) 243 if perr != nil { 244 t.Fatalf("the estimated award date must be a stable ISO date, got %q", r.EstimatedAwardDate) 245 } 246 if d.Before(prev) { 247 t.Errorf("the runway must be ordered by estimated award date ascending; %s precedes %s", d, prev) 248 } 249 prev = d 250 if r.ForecastID == insideID { 251 sawInside = true 252 if r.PopState != "NM" { 253 t.Errorf("the forecast must carry its place of performance state, got %q", r.PopState) 254 } 255 if r.MissionCode != "SIGINT" || r.MissionArea == "" { 256 t.Errorf("the in window forecast must carry the SIGINT mission area, got code %q area %q", r.MissionCode, r.MissionArea) 257 } 258 if r.Dollars == nil || !r.Dollars.Equal(decimal.RequireFromString("9100000.00")) { 259 t.Errorf("the forecast dollars must be the decimal 9100000.00, got %v", r.Dollars) 260 } 261 } 262 if r.ForecastID == "G9-FR-OUT-"+run { 263 sawOutside = true 264 } 265 } 266 if !sawInside { 267 t.Error("the runway must include the in window forecast") 268 } 269 if sawOutside { 270 t.Error("the runway must NOT include the forecast landing outside the window") 271 } 272 273 again, err := ai.ForecastRunway(ctxOrg(org), appDB, 12) 274 if err != nil { 275 t.Fatalf("forecast runway re run: %v", err) 276 } 277 first, _ := json.Marshal(rows) 278 second, _ := json.Marshal(again) 279 if string(first) != string(second) { 280 t.Errorf("the forecast runway must be byte stable on a re run.\nfirst: %s\nsecond: %s", first, second) 281 } 282} 283