Put Prod into review automatically, too. (#183)

Fixes #182
mozilla · Oct 21, 2020 · d477ea1 · d477ea1
1 parent 67ca945
commit d477ea1
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/ccadb2OneCRL/main.go b/ccadb2OneCRL/main.go
@@ -315,6 +315,7 @@ func (u *Updater) Update() error {
 		Then(u.UpdateRecordsWithBugID()).
 		Then(u.PutStagingIntoReview()).
 		Then(u.PushToProduction()).
+		Then(u.PutProductionIntoReview()).
 		AutoRollbackOnError(true).
 		AutoClose(true).
 		Commit()
@@ -639,6 +640,14 @@ func (u *Updater) PushToProduction() transaction.Transactor {
 	})
 }
 
+func (u *Updater) PutProductionIntoReview() transaction.Transactor {
+	return transaction.NewTransaction().WithCommit(func() error {
+		return errors.WithStack(u.production.ToReview(ProductionCollection()))
+	}).WithRollback(func(_ error) error {
+		return errors.WithStack(u.production.ToRollBack(ProductionCollection()))
+	})
+}
+
 func (u *Updater) BlastEmails(intersection *onecrl.Set) {
 	bugIDs := make(map[int]bool, 0)
 	builder := strings.Builder{}