Add gRPC conversion support for BillingCard events (gravitational#7303)

wiredcraft-ops · Jun 15, 2021 · 42beccb · 42beccb
1 parent a441275
commit 42beccb
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/api/types/events/oneof.go b/api/types/events/oneof.go
@@ -213,6 +213,14 @@ func ToOneOf(in AuditEvent) (*OneOf, error) {
 		out.Event = &OneOf_MFADeviceDelete{
 			MFADeviceDelete: e,
 		}
+	case *BillingCardCreate:
+		out.Event = &OneOf_BillingCardCreate{
+			BillingCardCreate: e,
+		}
+	case *BillingCardDelete:
+		out.Event = &OneOf_BillingCardDelete{
+			BillingCardDelete: e,
+		}
 	default:
 		return nil, trace.BadParameter("event type %T is not supported", in)
 	}
@@ -311,6 +319,10 @@ func FromOneOf(in OneOf) (AuditEvent, error) {
 		return e, nil
 	} else if e := in.GetMFADeviceDelete(); e != nil {
 		return e, nil
+	} else if e := in.GetBillingCardCreate(); e != nil {
+		return e, nil
+	} else if e := in.GetBillingCardDelete(); e != nil {
+		return e, nil
 	} else {
 		if in.Event == nil {
 			return nil, trace.BadParameter("failed to parse event, session record is corrupted")