From 4bfbef73bc868c96c21849ab852d134d75878151 Mon Sep 17 00:00:00 2001
From: Sudheer Shanka <sudheersai@google.com>
Date: Fri, 20 May 2022 15:15:35 -0700
Subject: [PATCH] Keep the screen on while the test activity is on top.

Bug: 231440256
Test: atest com.android.cts.net.HostsideConnOnActivityStartTest
Ignore-AOSP-First: Will cherrypick afterwards
Change-Id: Ibb0cc4c8a7cc2338ca938c775706f6b779fa96d3
---
 .../app2/src/com/android/cts/net/hostside/app2/MyActivity.java | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/cts/hostside/app2/src/com/android/cts/net/hostside/app2/MyActivity.java b/tests/cts/hostside/app2/src/com/android/cts/net/hostside/app2/MyActivity.java
index bffa7b0165..aa58ff9c29 100644
--- a/tests/cts/hostside/app2/src/com/android/cts/net/hostside/app2/MyActivity.java
+++ b/tests/cts/hostside/app2/src/com/android/cts/net/hostside/app2/MyActivity.java
@@ -27,6 +27,7 @@ import android.content.IntentFilter;
 import android.os.Bundle;
 import android.os.RemoteCallback;
 import android.util.Log;
+import android.view.WindowManager;
 
 import androidx.annotation.GuardedBy;
 
@@ -42,6 +43,8 @@ public class MyActivity extends Activity {
     protected void onCreate(Bundle savedInstanceState) {
         super.onCreate(savedInstanceState);
         Log.d(TAG, "MyActivity.onCreate()");
+
+        getWindow().addFlags(WindowManager.LayoutParams.FLAG_KEEP_SCREEN_ON);
     }
 
     @Override