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