(fix) internal/civisibility/coverage: [APMLP-379] fix TestCollectCove…

…rageAfterTestExecution flaky test - v2 (#3170)

internal/civisibility/integrations/gotesting/coverage/test_coverage.go

@@ -215,11 +215,8 @@ func (t *testCoverage) CollectCoverageAfterTestExecution() {
 		return
 	}
 
-	t.postCoverageFilename = filepath.Join(temporaryDir, fmt.Sprintf("%d-%d-%d-post.out", t.moduleID, t.suiteID, t.testID))
-	_, err := tearDown(t.postCoverageFilename, "")
-	if err != nil {
-		log.Debug("civisibility.coverage: error getting coverage file: %v", err)
-		telemetry.CodeCoverageErrors()
+	if t.getCoverageData() != nil {
+		return
 	}
 
 	var pChannel = make(chan struct{})
@@ -232,6 +229,18 @@ func (t *testCoverage) CollectCoverageAfterTestExecution() {
 	}()
 }
 
+// getCoverageData gets the coverage data.
+func (t *testCoverage) getCoverageData() error {
+	t.postCoverageFilename = filepath.Join(temporaryDir, fmt.Sprintf("%d-%d-%d-post.out", t.moduleID, t.suiteID, t.testID))
+	_, err := tearDown(t.postCoverageFilename, "")
+	if err != nil {
+		log.Debug("civisibility.coverage: error getting coverage file: %v", err)
+		telemetry.CodeCoverageErrors()
+	}
+
+	return err
+}
+
 // processCoverageData processes the coverage data.
 func (t *testCoverage) processCoverageData() {
 	if t.preCoverageFilename == "" ||

internal/civisibility/integrations/gotesting/coverage/test_coverage_test.go

@@ -306,7 +306,10 @@ func TestCollectCoverageAfterTestExecution(t *testing.T) {
 	f.WriteString("mode: count\n")
 	f.Close()
 
-	tc.CollectCoverageAfterTestExecution()
+	err = tc.getCoverageData()
+	if err != nil {
+		t.Errorf("getCoverageData returned error: %v", err)
+	}
 
 	if tc.postCoverageFilename == "" {
 		t.Error("postCoverageFilename is empty after CollectCoverageAfterTestExecution")