Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 73d2aaf8 authored by Yuexi Ma's avatar Yuexi Ma
Browse files

[AI TestGen] Add tests for curated photo logging 

Adds unit tests for `logCuratedPhotosRendered` and
`logCuratedPhotosFetched` in `ThemesUserEventLoggerImpl`.

The CL is generated by AI based on change ag/34396260 to add coverage to
the affected lines.

Want-Code-Review: all
Flag: TEST_ONLY
Test: Verified via atest locally and presubmit
Bug: 431235865
Change-Id: Ia6399938987cb876bbc635923333ac0c1c9332b2
parent 99ecc535

tests/robotests/src/com/android/customization/module/logging/ThemesUserEventLoggerImplTest.kt

+36 −0
Original line number Diff line number Diff line
@@ -46,6 +46,7 @@ import com.android.wallpaper.util.LaunchSourceUtils 
import com.google.common.truth.Truth.assertThat

import dagger.hilt.android.testing.HiltAndroidRule

import dagger.hilt.android.testing.HiltAndroidTest

import io.grpc.Status

import javax.inject.Inject

import kotlinx.coroutines.test.TestScope

import kotlinx.coroutines.test.runTest
@@ -521,6 +522,41 @@ class ThemesUserEventLoggerImplTest { 
        assertThat(fakeStatsLogger.shapePackageHash).isEqualTo(TEST_SHAPE_ID.hashCode())

    }



    @Test

    fun logCuratedPhotosRendered() {

	// The log toggle depends on whether this is a user photo. The log toggles on if and only

	// if this is a user photo.

        underTest.logCuratedPhotosRendered(/* timeElapsedMillis= */ 1234L, /* userPhoto= */ true)



        assertThat(fakeStatsLogger.action)

            .isEqualTo(StyleEnums.CURATED_PHOTOS_RENDER_COMPLETE)

        assertThat(fakeStatsLogger.logCalled).isTrue()

        assertThat(fakeStatsLogger.appSessionId).isEqualTo(FakeAppSessionId.TEST_APP_SESSION_ID)

        assertThat(fakeStatsLogger.timeElapsedMillis).isEqualTo(1234L)

        assertThat(fakeStatsLogger.toggleOn).isTrue()



        underTest.logCuratedPhotosRendered(/* timeElapsedMillis= */ 5678L, /* userPhoto= */ false)



        assertThat(fakeStatsLogger.action)

            .isEqualTo(StyleEnums.CURATED_PHOTOS_RENDER_COMPLETE)

        assertThat(fakeStatsLogger.logCalled).isTrue()

        assertThat(fakeStatsLogger.appSessionId).isEqualTo(FakeAppSessionId.TEST_APP_SESSION_ID)

        assertThat(fakeStatsLogger.timeElapsedMillis).isEqualTo(5678L)

        assertThat(fakeStatsLogger.toggleOn).isFalse()

    }



    @Test

    fun logCuratedPhotosFetched() {

        val status = Status.DEADLINE_EXCEEDED

        underTest.logCuratedPhotosFetched(1234L, status)



        assertThat(fakeStatsLogger.action).isEqualTo(StyleEnums.CURATED_PHOTOS_FETCH_END)

        assertThat(fakeStatsLogger.logCalled).isTrue()

        assertThat(fakeStatsLogger.appSessionId).isEqualTo(FakeAppSessionId.TEST_APP_SESSION_ID)

        assertThat(fakeStatsLogger.effectResultCode).isEqualTo(status.code.value())

        assertThat(fakeStatsLogger.timeElapsedMillis).isEqualTo(1234L)

    }



    @Test

    fun logEnterScreen() {

        underTest.logEnterScreen(SCREEN_CLOCK)