1 files changed, 38 insertions, 0 deletions
diff --git a/packages/SystemUI/tests/src/com/android/systemui/statusbar/notification/collection/inflation/NotifUiAdjustmentProviderTest.kt b/packages/SystemUI/tests/src/com/android/systemui/statusbar/notification/collection/inflation/NotifUiAdjustmentProviderTest.kt
new file mode 100644
index 000000000000..dd15caebf9e6
--- /dev/null
+++ b/packages/SystemUI/tests/src/com/android/systemui/statusbar/notification/collection/inflation/NotifUiAdjustmentProviderTest.kt
@@ -0,0 +1,38 @@
+package com.android.systemui.statusbar.notification.collection.inflation
+
+import android.testing.AndroidTestingRunner
+import android.testing.TestableLooper.RunWithLooper
+import androidx.test.filters.SmallTest
+import com.android.systemui.SysuiTestCase
+import com.android.systemui.statusbar.NotificationLockscreenUserManager
+import com.android.systemui.statusbar.notification.SectionClassifier
+import com.android.systemui.util.mockito.mock
+import com.android.systemui.util.mockito.withArgCaptor
+import org.junit.Test
+import org.junit.runner.RunWith
+import org.mockito.Mockito.verify
+
+@SmallTest
+@RunWith(AndroidTestingRunner::class)
+@RunWithLooper
+class NotifUiAdjustmentProviderTest : SysuiTestCase() {
+    private val lockscreenUserManager: NotificationLockscreenUserManager = mock()
+    private val sectionClassifier: SectionClassifier = mock()
+
+    private val adjustmentProvider = NotifUiAdjustmentProvider(
+        lockscreenUserManager,
+        sectionClassifier,
+    )
+
+    @Test
+    fun notifLockscreenStateChangeWillNotifDirty() {
+        val dirtyListener = mock<Runnable>()
+        adjustmentProvider.addDirtyListener(dirtyListener)
+        val notifLocksreenStateChangeListener =
+            withArgCaptor<NotificationLockscreenUserManager.NotificationStateChangedListener> {
+                verify(lockscreenUserManager).addNotificationStateChangedListener(capture())
+            }
+        notifLocksreenStateChangeListener.onNotificationStateChanged()
+        verify(dirtyListener).run();
+    }
+}