fix tests

Author: lukaszcl

tools/flakeguard/reports/data.go

@@ -179,6 +179,15 @@ func aggregate(reportChan <-chan *TestReport) (*TestReport, error) {
 	return fullReport, nil
 }
 
+func aggregateFromReports(reports ...*TestReport) (*TestReport, error) {
+	reportChan := make(chan *TestReport, len(reports))
+	for _, report := range reports {
+		reportChan <- report
+	}
+	close(reportChan)
+	return aggregate(reportChan)
+}
+
 func mergeTestResults(a, b TestResult) TestResult {
 	a.Runs += b.Runs
 	a.Durations = append(a.Durations, b.Durations...)

tools/flakeguard/reports/data_test.go

@@ -287,7 +287,7 @@ func TestAggregate(t *testing.T) {
 		},
 	}
 
-	aggregatedReport, err := aggregate(report1, report2)
+	aggregatedReport, err := aggregateFromReports(report1, report2)
 	if err != nil {
 		t.Fatalf("Error aggregating reports: %v", err)
 	}
@@ -371,7 +371,7 @@ func TestAggregateOutputs(t *testing.T) {
 		},
 	}
 
-	aggregatedReport, err := aggregate(report1, report2)
+	aggregatedReport, err := aggregateFromReports(report1, report2)
 	if err != nil {
 		t.Fatalf("Error aggregating reports: %v", err)
 	}
@@ -432,7 +432,7 @@ func TestAggregateIdenticalOutputs(t *testing.T) {
 		},
 	}
 
-	aggregatedReport, err := aggregate(report1, report2)
+	aggregatedReport, err := aggregateFromReports(report1, report2)
 	if err != nil {
 		t.Fatalf("Error aggregating reports: %v", err)
 	}
@@ -569,7 +569,7 @@ func TestAggregate_AllSkippedTests(t *testing.T) {
 		},
 	}
 
-	aggregatedReport, err := aggregate(report1, report2)
+	aggregatedReport, err := aggregateFromReports(report1, report2)
 	if err != nil {
 		t.Fatalf("Error aggregating reports: %v", err)
 	}