From 8a995beac65c16050d9d2ccd6ff9afaa24d40460 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Alexander=20K=C3=B6plinger?= <alex.koeplinger@outlook.com>
Date: Thu, 6 Apr 2023 11:21:45 +0200
Subject: [PATCH] Fix eventpipe/enabledisable.cs test on Android

It needs the same customization that we have in IpcTraceTest.cs
---
 .../Microsoft.Diagnostics.NETCore.Client/AssemblyInfo.cs      | 1 +
 src/tests/tracing/eventpipe/enabledisable/enabledisable.cs    | 4 ++++
 .../tracing/eventpipe/enabledisable/enabledisable.csproj      | 1 +
 3 files changed, 6 insertions(+)

diff --git a/src/tests/tracing/eventpipe/common/Microsoft.Diagnostics.NETCore.Client/AssemblyInfo.cs b/src/tests/tracing/eventpipe/common/Microsoft.Diagnostics.NETCore.Client/AssemblyInfo.cs
index a53e9716f896ba..a24514a125582e 100644
--- a/src/tests/tracing/eventpipe/common/Microsoft.Diagnostics.NETCore.Client/AssemblyInfo.cs
+++ b/src/tests/tracing/eventpipe/common/Microsoft.Diagnostics.NETCore.Client/AssemblyInfo.cs
@@ -5,4 +5,5 @@
 using System.Runtime.CompilerServices;
 
 [assembly: InternalsVisibleTo("common")]
+[assembly: InternalsVisibleTo("enabledisable")]
 #endif
\ No newline at end of file
diff --git a/src/tests/tracing/eventpipe/enabledisable/enabledisable.cs b/src/tests/tracing/eventpipe/enabledisable/enabledisable.cs
index 80b0b540125818..b90468e7918d62 100644
--- a/src/tests/tracing/eventpipe/enabledisable/enabledisable.cs
+++ b/src/tests/tracing/eventpipe/enabledisable/enabledisable.cs
@@ -72,6 +72,10 @@ public static int Main()
             };
 
             DiagnosticsClient client = new DiagnosticsClient(Process.GetCurrentProcess().Id);
+#if DIAGNOSTICS_RUNTIME
+            if (OperatingSystem.IsAndroid())
+                client = new DiagnosticsClient(new IpcEndpointConfig("127.0.0.1:9000", IpcEndpointConfig.TransportType.TcpSocket, IpcEndpointConfig.PortType.Listen));
+#endif
             using (EventPipeSession session1 = client.StartEventPipeSession(providers))
             {
                 EventPipeEventSource source1 = new EventPipeEventSource(session1.EventStream);
diff --git a/src/tests/tracing/eventpipe/enabledisable/enabledisable.csproj b/src/tests/tracing/eventpipe/enabledisable/enabledisable.csproj
index 88ecb83bf1abb2..7eecf6a83db18f 100644
--- a/src/tests/tracing/eventpipe/enabledisable/enabledisable.csproj
+++ b/src/tests/tracing/eventpipe/enabledisable/enabledisable.csproj
@@ -3,6 +3,7 @@
     <TargetFrameworkIdentifier>.NETCoreApp</TargetFrameworkIdentifier>
     <OutputType>exe</OutputType>
     <AllowUnsafeBlocks>true</AllowUnsafeBlocks>
+    <DefineConstants Condition="'$(GitHubRepositoryName)' == 'runtime'">$(DefineConstants);DIAGNOSTICS_RUNTIME</DefineConstants>
     <JitOptimizationSensitive>true</JitOptimizationSensitive>
     <UnloadabilityIncompatible>true</UnloadabilityIncompatible>
     <!-- GCStress timeout: https://github.com/dotnet/runtime/issues/51133 -->