updated unit tests to include notifications in common package

kubernetes-sigs · Oct 9, 2024 · 5a08a3c · 5a08a3c
1 parent a3b7fba
commit 5a08a3c
Show file tree

Hide file tree

Showing 2 changed files with 277 additions and 134 deletions.
diff --git a/pkg/i2gw/notifications/notifications.go b/pkg/i2gw/notifications/notifications.go
@@ -27,7 +27,7 @@ import (
 )
 
 func init() {
-	NotificationAggr = NotificationAggregator{Notifications: map[string][]Notification{}}
+	NotificationAggr = BuildNotificationAggregator()
 }
 
 const (
@@ -55,6 +55,11 @@ var NotificationAggr NotificationAggregator
 // package without the common package having knowledge about which provider is making a call it
 type NotificationCallback func(mType MessageType, message string, CallingObjects ...client.Object)
 
+// BuildNotificationAggregator returns an instance of initialized NotificationAggregator
+func BuildNotificationAggregator() NotificationAggregator {
+	return NotificationAggregator{Notifications: map[string][]Notification{}}
+}
+
 // DispatchNotification is used to send a notification to the NotificationAggregator
 func (na *NotificationAggregator) DispatchNotification(notification Notification, ProviderName string) {
 	na.mutex.Lock()