Fixed integration tests

fzhinkin · Mar 23, 2024 · f62a152 · f62a152
1 parent c79e77b
commit f62a152
Showing 1 changed file with 18 additions and 3 deletions.
diff --git a/integration/build.gradle.kts b/integration/build.gradle.kts
@@ -1,6 +1,7 @@
 import org.gradle.process.internal.ExecException
 import java.io.ByteArrayOutputStream
 import java.lang.IllegalStateException
+import java.nio.charset.StandardCharsets
 
 plugins {
     id("java")
@@ -26,6 +27,19 @@ val hasXCtrace: Boolean by lazy {
     }
 }
 
+val hasPMC: Boolean by lazy {
+    try {
+        val out = ByteArrayOutputStream()
+        val res = exec {
+            commandLine("sysctl", "-n",  "hw.cpufamily")
+            standardOutput = out
+        }
+        res.exitValue == 0 && out.toByteArray().decodeToString().lineSequence().first() != "0"
+    } catch (e: ExecException) {
+        false
+    }
+}
+
 tasks {
     create("runAsmProfWithDefaultArgs", JavaExec::class.java) {
         onlyIf { hasXCtrace && !project.hasProperty("noPMU") }
@@ -89,17 +103,18 @@ tasks {
             }
             val kind = if (hasXCtrace) "supported" else "unsupported"
             checkHasXCTraceAsmProfiler(filter(out.lineSequence()), kind)
-            checkHasXCTraceNormProfiler(filter(out.lineSequence()), kind)
+            val normKind = if (hasPMC) "supported" else "unsupported"
+            checkHasXCTraceNormProfiler(filter(out.lineSequence()), normKind)
         }
     }
     create("runNormProfiler", JavaExec::class.java) {
-        onlyIf { hasXCtrace }
+        onlyIf { hasPMC }
         dependsOn.add("jmhJar")
         classpath += jmhJar.get().outputs.files
         mainClass = "io.github.fzhinkin.XCTraceNormProfilerTest"
     }
     create("listPmuEvents", JavaExec::class.java) {
-        onlyIf { hasXCtrace }
+        onlyIf { hasPMC }
         dependsOn.add("jmhJar")
         classpath += jmhJar.get().outputs.files
         mainClass = "io.github.fzhinkin.XCTraceNormProfilerListEventsTest"