Fixed cancellation bug.
diff --git a/utils/utils.go b/utils/utils.go index 270b539..c082630 100644 --- a/utils/utils.go +++ b/utils/utils.go
@@ -107,7 +107,12 @@ if diff := cmp.Diff(got.GetPaymentInformation(), want.GetPaymentInformation(), cmp.Comparer(proto.Equal)); diff != "" { return fmt.Errorf("payment information differs (-got +want)\n%s", diff) } - if diff := cmp.Diff(got.GetStatus(), mpb.BookingStatus_CONFIRMED); diff != "" { + // BookingStatus_CONFIRMED is the default case unless want overrides it. + wantStatus := mpb.BookingStatus_CONFIRMED + if want.GetStatus() != mpb.BookingStatus_BOOKING_STATUS_UNSPECIFIED { + wantStatus = want.GetStatus() + } + if diff := cmp.Diff(got.GetStatus(), wantStatus); diff != "" { return fmt.Errorf("status differs (-got +want)\n%s", diff) } return nil