Lint

armadaproject · Nov 20, 2023 · a774dc0 · a774dc0
1 parent adb9eb4
commit a774dc0
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/internal/scheduler/simulator/simulator.go b/internal/scheduler/simulator/simulator.go
@@ -682,7 +682,9 @@ func (s *Simulator) handleJobRunLeased(txn *jobdb.Txn, e *armadaevents.JobRunLea
 	)
 
 	updatedJob := job.WithUpdatedRun(job.LatestRun().WithRunning(true))
-	txn.Upsert([]*jobdb.Job{updatedJob})
+	if err := txn.Upsert([]*jobdb.Job{updatedJob}); err != nil {
+		return nil, false, err
+	}
 	return updatedJob, true, nil
 }
 
@@ -817,7 +819,9 @@ func (s *Simulator) handleJobRunPreempted(txn *jobdb.Txn, e *armadaevents.JobRun
 	)
 	s.jobTemplateByJobId[retryJobId.String()] = jobTemplate
 	updatedJob := job.WithUpdatedRun(job.LatestRun().WithReturned(true))
-	txn.Upsert([]*jobdb.Job{updatedJob})
+	if err := txn.Upsert([]*jobdb.Job{updatedJob}); err != nil {
+		return nil, false, err
+	}
 	return updatedJob, true, nil
 }