From 070e4a8b7decec69bc6801e3efae7f24cf4988d7 Mon Sep 17 00:00:00 2001
From: Ilya Matveev <ilya.matveev@jetbrains.com>
Date: Fri, 2 Aug 2019 19:37:29 +0700
Subject: [PATCH] Tests: Take two-stage mode into account in coverage tests

---
 backend.native/tests/build.gradle | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/backend.native/tests/build.gradle b/backend.native/tests/build.gradle
index 3a2fbf8c305..738b8c1c2da 100644
--- a/backend.native/tests/build.gradle
+++ b/backend.native/tests/build.gradle
@@ -89,7 +89,8 @@ project.convention.plugins.executor = ExecutorServiceKt.create(project)
 ext.konanNoRun = true
 
 // Enable two-stage compilation if the test_two_stage property is set.
-if (project.hasProperty("test_two_stage")) {
+ext.twoStageEnabled = project.hasProperty("test_two_stage")
+if (twoStageEnabled) {
     tasks.withType(KonanCompileNativeBinary.class) {
         enableTwoStageCompilation = true
     }
@@ -3677,11 +3678,21 @@ standaloneTest("coverage_basic_program") {
     description = "Test that `-Xcoverage` generates correct __llvm_coverage information"
 
     def dir = buildDir.absolutePath
-    
-    flags = ["-Xcoverage-file=$dir/program.profraw", "-Xcoverage", "-entry", "coverage.basic.program.main"]
+    def coverageFile = "$dir/program.profraw"
+
+    flags = ["-Xcoverage-file=$coverageFile", "-Xcoverage", "-entry", "coverage.basic.program.main"]
     source = "$projectDir/coverage/basic/program/main.kt"
 
+    doFirst { file(coverageFile).delete() }
+
     doLast {
+        if (twoStageEnabled) {
+            // If two-stage compilation is enabled, no sources are passed to the compiler when a final binary is generated.
+            // So according to -Xcoverage flag semantics it must provide no coverage info.
+            assert !file(coverageFile).exists()
+            return
+        }
+
         execLlvm("llvm-profdata") {
             args "merge"
             args "$dir/program.profraw"