From b94172c8bc44202dcaa73a41adbb9f3c563e4d1d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Piotr=20Wilczy=C5=84ski?= <wilczynskip@google.com>
Date: Tue, 24 Jan 2023 16:16:39 +0000
Subject: [PATCH] Add a test for `shouldDeviceBePutToSleep` when performing the
 first state transition

Bug: 265088477
Test: atest LogicalDisplayMapperTest
Change-Id: I5bb4df5f2e2197be087ebab373a1a476af15a7e0
---
 .../android/server/display/LogicalDisplayMapperTest.java    | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/services/tests/servicestests/src/com/android/server/display/LogicalDisplayMapperTest.java b/services/tests/servicestests/src/com/android/server/display/LogicalDisplayMapperTest.java
index 650eef0338e6a..a7da2417b0bcc 100644
--- a/services/tests/servicestests/src/com/android/server/display/LogicalDisplayMapperTest.java
+++ b/services/tests/servicestests/src/com/android/server/display/LogicalDisplayMapperTest.java
@@ -16,6 +16,7 @@
 
 package com.android.server.display;
 
+import static android.hardware.devicestate.DeviceStateManager.INVALID_DEVICE_STATE;
 import static android.view.Display.DEFAULT_DISPLAY;
 import static android.view.Display.DEFAULT_DISPLAY_GROUP;
 import static android.view.Display.TYPE_EXTERNAL;
@@ -441,6 +442,11 @@ public class LogicalDisplayMapperTest {
                 /* isOverrideActive= */false,
                 /* isInteractive= */true,
                 /* isBootCompleted= */true));
+        assertFalse(mLogicalDisplayMapper.shouldDeviceBePutToSleep(DEVICE_STATE_CLOSED,
+                INVALID_DEVICE_STATE,
+                /* isOverrideActive= */false,
+                /* isInteractive= */true,
+                /* isBootCompleted= */true));
     }
 
     @Test