[Sweep GHA Fix] The GitHub Actions run failed on d260290 (master) with the following error logs:

pumps/logzio.go

@@ -150,15 +150,21 @@
 			"raw_response":    decoded.RawResponse,
 			"ip_address":      decoded.IPAddress,
 		}
+			"ip_address":      decoded.IPAddress,
+		}
+	p.log.Info(p.GetName() + " Initialized")
 
 		event, err := json.Marshal(mapping)
 		if err != nil {
 			return fmt.Errorf("failed to marshal decoded data: %s", err)
 		}
 
-		p.sender.Send(event)
+		err = p.sender.Send(event)
+if err != nil {
+    return fmt.Errorf("failed to send event: %s", err)
+}
 	}
 	p.log.Info("Purged ", len(data), " records...")
 
 	return nil
 }

pumps/mgo_helper_test.go

@@ -37,7 +37,7 @@ func (c *Conn) ConnectDb() {
 	if c.Store == nil {
 		var err error
 		c.Store, err = persistent.NewPersistentStorage(&persistent.ClientOpts{
-			Type:             "mgo",
+			Type:             "mongo-go",
 			ConnectionString: dbAddr,
 		})
 		if err != nil {

pumps/mongo.go

@@ -353,11 +353,6 @@ func (m *MongoPump) ensureIndexes(collectionName string) error {
 }
 
 func (m *MongoPump) connect() {
-	if m.dbConf.MongoDriverType == "" {
-		// Default to mgo
-		m.dbConf.MongoDriverType = persistent.Mgo
-	}
-
 	store, err := persistent.NewPersistentStorage(&persistent.ClientOpts{
 		ConnectionString:         m.dbConf.MongoURL,
 		UseSSL:                   m.dbConf.MongoUseSSL,
@@ -367,7 +362,7 @@ func (m *MongoPump) connect() {
 		SSLPEMKeyfile:            m.dbConf.MongoSSLPEMKeyfile,
 		SessionConsistency:       m.dbConf.MongoSessionConsistency,
 		ConnectionTimeout:        m.timeout,
-		Type:                     m.dbConf.MongoDriverType,
+		Type:                     getMongoDriverType(m.dbConf.MongoDriverType),
 		DirectConnection:         m.dbConf.MongoDirectConnection,
 	})
 	if err != nil {
@@ -547,3 +542,12 @@ func (m *MongoPump) WriteUptimeData(data []interface{}) {
 		m.log.Error("Problem inserting to mongo collection: ", err)
 	}
 }
+
+func getMongoDriverType(driverType string) string {
+	if driverType == "" {
+		// Default to mongo-go
+		return persistent.OfficialMongo
+	}
+
+	return driverType
+}

pumps/mongo_aggregate.go

@@ -215,11 +215,6 @@ func (m *MongoAggregatePump) Init(config interface{}) error {
 func (m *MongoAggregatePump) connect() {
 	var err error
 
-	if m.dbConf.MongoDriverType == "" {
-		// Default to mgo
-		m.dbConf.MongoDriverType = persistent.Mgo
-	}
-
 	m.store, err = persistent.NewPersistentStorage(&persistent.ClientOpts{
 		ConnectionString:         m.dbConf.MongoURL,
 		UseSSL:                   m.dbConf.MongoUseSSL,
@@ -229,7 +224,7 @@ func (m *MongoAggregatePump) connect() {
 		SSLPEMKeyfile:            m.dbConf.MongoSSLPEMKeyfile,
 		SessionConsistency:       m.dbConf.MongoSessionConsistency,
 		ConnectionTimeout:        m.timeout,
-		Type:                     m.dbConf.MongoDriverType,
+		Type:                     getMongoDriverType(m.dbConf.MongoDriverType),
 		DirectConnection:         m.dbConf.MongoDirectConnection,
 	})
 	if err != nil {

pumps/mongo_selective.go

@@ -113,11 +113,6 @@ func (m *MongoSelectivePump) Init(config interface{}) error {
 func (m *MongoSelectivePump) connect() {
 	var err error
 
-	if m.dbConf.MongoDriverType == "" {
-		// Default to mgo
-		m.dbConf.MongoDriverType = persistent.Mgo
-	}
-
 	m.store, err = persistent.NewPersistentStorage(&persistent.ClientOpts{
 		ConnectionString:         m.dbConf.MongoURL,
 		UseSSL:                   m.dbConf.MongoUseSSL,
@@ -127,7 +122,7 @@ func (m *MongoSelectivePump) connect() {
 		SSLPEMKeyfile:            m.dbConf.MongoSSLPEMKeyfile,
 		SessionConsistency:       m.dbConf.MongoSessionConsistency,
 		ConnectionTimeout:        m.timeout,
-		Type:                     m.dbConf.MongoDriverType,
+		Type:                     getMongoDriverType(m.dbConf.MongoDriverType),
 		DirectConnection:         m.dbConf.MongoDirectConnection,
 	})
 	if err != nil {

pumps/mongo_test.go

@@ -657,3 +657,35 @@ func TestMongoPump_WriteData(t *testing.T) {
 		return records
 	}))
 }
+func TestGetMongoDriverType(t *testing.T) {
+	tests := []struct {
+		name       string
+		driverType string
+		want       string
+	}{
+		{
+			name:       "Empty driver type",
+			driverType: "",
+			want:       persistent.OfficialMongo,
+		},
+		{
+			name:       "mongo-go driver type",
+			driverType: persistent.OfficialMongo,
+			want:       persistent.OfficialMongo,
+		},
+		{
+			name:       "mgo driver type",
+			driverType: persistent.Mgo,
+			want:       persistent.Mgo,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got := getMongoDriverType(tt.driverType)
+			if got != tt.want {
+				t.Errorf("got %v, want %v", got, tt.want)
+			}
+		})
+	}
+}