PMM-13477 Fix tests.

Author: BupycHuk

exporter/diagnostic_data_collector_test.go

@@ -421,7 +421,7 @@ func cleanTestData(ctx context.Context, client *mongo.Client, count int) error {
 }
 
 func TestDisconnectedDiagnosticDataCollector(t *testing.T) {
-	ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
 	defer cancel()
 
 	client := tu.DefaultTestClient(ctx, t)
@@ -434,7 +434,7 @@ func TestDisconnectedDiagnosticDataCollector(t *testing.T) {
 	ti := labelsGetterMock{}
 
 	dbBuildInfo, err := retrieveMongoDBBuildInfo(ctx, client, logger.WithField("component", "test"))
-	require.NoError(t, err)
+	require.Error(t, err)
 
 	c := newDiagnosticDataCollector(ctx, client, logger, true, ti, dbBuildInfo)
 

exporter/v1_compatibility.go

@@ -852,6 +852,9 @@ func specialMetrics(ctx context.Context, client *mongo.Client, m bson.M, nodeTyp
 }
 
 func retrieveMongoDBBuildInfo(ctx context.Context, client *mongo.Client, l *logrus.Entry) (buildInfo, error) {
+	if client == nil {
+		return buildInfo{}, errors.New("cannot get mongo build info: client is nil")
+	}
 	buildInfoCmd := bson.D{bson.E{Key: "buildInfo", Value: 1}}
 	res := client.Database("admin").RunCommand(ctx, buildInfoCmd)