Merge "Copy TestData into tv module"

tv/src/test/java/com/google/samples/apps/iosched/tv/model/TestData.kt

@@ -0,0 +1,221 @@
+/*
+ * Copyright 2018 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.google.samples.apps.iosched.tv.model
+
+import com.google.samples.apps.iosched.shared.data.ConferenceDataRepository
+import com.google.samples.apps.iosched.shared.data.ConferenceDataSource
+import com.google.samples.apps.iosched.shared.firestore.entity.ReservationRequestResult
+import com.google.samples.apps.iosched.shared.firestore.entity.ReservationRequestResult.ReservationRequestStatus.RESERVE_DENIED_CUTOFF
+import com.google.samples.apps.iosched.shared.firestore.entity.ReservationRequestResult.ReservationRequestStatus.RESERVE_DENIED_UNKNOWN
+import com.google.samples.apps.iosched.shared.firestore.entity.ReservationRequestResult.ReservationRequestStatus.RESERVE_SUCCEEDED
+import com.google.samples.apps.iosched.shared.firestore.entity.ReservationRequestResult.ReservationRequestStatus.RESERVE_WAITLISTED
+import com.google.samples.apps.iosched.shared.firestore.entity.UserEvent
+import com.google.samples.apps.iosched.shared.model.Block
+import com.google.samples.apps.iosched.shared.model.ConferenceData
+import com.google.samples.apps.iosched.shared.model.Room
+import com.google.samples.apps.iosched.shared.model.Session
+import com.google.samples.apps.iosched.shared.model.Speaker
+import com.google.samples.apps.iosched.shared.model.Tag
+import com.google.samples.apps.iosched.shared.model.UserSession
+import com.google.samples.apps.iosched.shared.util.TimeUtils.ConferenceDay
+import com.google.samples.apps.iosched.shared.util.TimeUtils.ConferenceDay.DAY_1
+import com.google.samples.apps.iosched.shared.util.TimeUtils.ConferenceDay.DAY_2
+import com.google.samples.apps.iosched.shared.util.TimeUtils.ConferenceDay.DAY_3
+
+/**
+ * Test data for unit tests.
+ */
+object TestData : ConferenceDataSource {
+
+    // region Declarations
+
+    val androidTag = Tag("1", "TRACK", 0, "Android", 0xFFAED581.toInt())
+    val webTag = Tag("2", "TRACK", 1, "Web", 0xFFFFF176.toInt())
+    val sessionsTag = Tag("101", "TYPE", 0, "Sessions", 0)
+    val codelabsTag = Tag("102", "TYPE", 1, "Codelabs", 0)
+    val beginnerTag = Tag("201", "LEVEL", 0, "Beginner", 0)
+    val intermediateTag = Tag("202", "LEVEL", 1, "Intermediate", 0)
+    val advancedTag = Tag("203", "LEVEL", 2, "Advanced", 0)
+
+    val tagsList = listOf(
+        androidTag,
+        webTag,
+        sessionsTag,
+        codelabsTag,
+        beginnerTag,
+        intermediateTag,
+        advancedTag
+    )
+
+
+    val speaker = Speaker(id = "1", name = "Troy McClure", imageUrl = "",
+        company = "", abstract = "", gPlusUrl = "", twitterUrl = "")
+
+    val room = Room(id = "1", name = "Tent 1", capacity = 40)
+
+    val session0 = Session(id = "0", title = "Session 0", abstract = "",
+        startTime = DAY_1.start, endTime = DAY_1.end,
+        room = room, sessionUrl = "", liveStreamUrl = "", youTubeUrl = "", photoUrl = "",
+        tags = listOf(
+            androidTag,
+            webTag
+        ), speakers = setOf(speaker),
+        relatedSessions = emptySet())
+
+    val session1 = Session(id = "1", title = "Session 1", abstract = "",
+        startTime = DAY_1.start, endTime = DAY_1.end,
+        room = room, sessionUrl = "", liveStreamUrl = "", youTubeUrl = "", photoUrl = "",
+        tags = listOf(
+            androidTag,
+            webTag
+        ), speakers = setOf(speaker),
+        relatedSessions = emptySet())
+
+    val session2 = Session(id = "2", title = "Session 2", abstract = "",
+        startTime = DAY_2.start, endTime = DAY_2.end,
+        room = room, sessionUrl = "", liveStreamUrl = "", youTubeUrl = "", photoUrl = "",
+        tags = listOf(androidTag), speakers = setOf(
+            speaker
+        ), relatedSessions = emptySet())
+
+    val session3 = Session(id = "3", title = "Session 3", abstract = "",
+        startTime = DAY_3.start, endTime = DAY_3.end,
+        room = room, sessionUrl = "", liveStreamUrl = "", youTubeUrl = "", photoUrl = "",
+        tags = listOf(webTag), speakers = setOf(
+            speaker
+        ), relatedSessions = emptySet())
+
+    val sessionsList = listOf(
+        session0,
+        session1,
+        session2,
+        session3
+    )
+    val sessionIDs = sessionsList.map { it.id }.toList()
+
+    val sessionsMap = mapOf(
+        ConferenceDay.DAY_1 to listOf(
+            session0,
+            session1
+        ),
+        ConferenceDay.DAY_2 to listOf(session2),
+        ConferenceDay.DAY_3 to listOf(session3))
+
+    val block1 = Block(
+        title = "Keynote",
+        type = "keynote",
+        color = 0xffff00ff.toInt(),
+        startTime = DAY_1.start,
+        endTime = DAY_1.start.plusHours(1L))
+
+    val block2 = Block(
+        title = "Breakfast",
+        type = "meal",
+        color = 0xffff00ff.toInt(),
+        startTime = DAY_1.start.plusHours(1L),
+        endTime = DAY_1.start.plusHours(2L))
+
+    val agenda = listOf(
+        block1,
+        block2
+    )
+
+    private val userEvent0 = UserEvent(
+        sessionIDs[0], isStarred = false,
+        isReviewed = false,
+        startTime = session0.startTime.toInstant().toEpochMilli(),
+        endTime = session0.endTime.toInstant().toEpochMilli(),
+        reservation = ReservationRequestResult(RESERVE_SUCCEEDED, System.currentTimeMillis())
+    )
+    private val userEvent1 = UserEvent(
+        sessionIDs[1], isStarred = true,
+        isReviewed = true,
+        startTime = session1.startTime.toInstant().toEpochMilli(),
+        endTime = session1.endTime.toInstant().toEpochMilli(),
+        reservation = ReservationRequestResult(RESERVE_WAITLISTED, System.currentTimeMillis())
+    )
+    private val userEvent2 = UserEvent(
+        sessionIDs[2], isStarred = true,
+        isReviewed = false,
+        startTime = session2.startTime.toInstant().toEpochMilli(),
+        endTime = session2.endTime.toInstant().toEpochMilli(),
+        reservation = ReservationRequestResult(RESERVE_DENIED_CUTOFF, System.currentTimeMillis())
+    )
+    private val userEvent3 = UserEvent(
+        sessionIDs[3], isStarred = false,
+        isReviewed = true,
+        startTime = session3.startTime.toInstant().toEpochMilli(),
+        endTime = session3.endTime.toInstant().toEpochMilli(),
+        reservation = ReservationRequestResult(RESERVE_DENIED_UNKNOWN, System.currentTimeMillis())
+    )
+    private val userSession0 = UserSession(
+        session0,
+        userEvent0
+    )
+    private val userSession1 = UserSession(
+        session1,
+        userEvent1
+    )
+    private val userSession2 = UserSession(
+        session2,
+        userEvent2
+    )
+    private val userSession3 = UserSession(
+        session3,
+        userEvent3
+    )
+    val userSessionMap = mapOf(
+        ConferenceDay.DAY_1 to listOf(
+            userSession0,
+            userSession1
+        ),
+        ConferenceDay.DAY_2 to listOf(
+            userSession2
+        ),
+        ConferenceDay.DAY_3 to listOf(
+            userSession3
+        ))
+    val userEvents = listOf(
+        userEvent0,
+        userEvent1,
+        userEvent2,
+        userEvent3
+    )
+
+    // endregion Declarations
+
+    private val conferenceData = ConferenceData(
+        sessions = sessionsList,
+        tags = tagsList,
+        blocks = agenda,
+        rooms = listOf(room),
+        speakers = listOf(speaker),
+        version = 42
+    )
+
+    override fun getConferenceData() =
+        conferenceData
+
+    override fun getOfflineConferenceData() =
+        conferenceData
+}
+
+/** ConferenceDataRepository for tests */
+object TestDataRepository : ConferenceDataRepository(
+    TestData,
+    TestData
+)

tv/src/test/java/com/google/samples/apps/iosched/tv/ui/schedule/ScheduleViewModelTest.kt

@@ -19,10 +19,11 @@
 package com.google.samples.apps.iosched.tv.ui.schedule
 
 import android.arch.core.executor.testing.InstantTaskExecutorRule
-import com.google.samples.apps.iosched.shared.data.ConferenceDataRepository
 import com.google.samples.apps.iosched.shared.data.session.SessionRepository
 import com.google.samples.apps.iosched.shared.domain.sessions.LoadSessionsUseCase
 import com.google.samples.apps.iosched.test.util.LiveDataTestUtil
+import com.google.samples.apps.iosched.tv.model.TestData
+import com.google.samples.apps.iosched.tv.model.TestDataRepository
 import com.google.samples.apps.iosched.tv.util.SyncTaskExecutorRule
 import org.hamcrest.collection.IsCollectionWithSize.hasSize
 import org.hamcrest.core.Is.`is`
@@ -44,8 +45,8 @@ class ScheduleViewModelTest {
     @Test
     fun testDataIsLoaded_ObservablesUpdated() {
         // Create a test use cases with test data
-        val expectedSessions = TestSessionDataSource.getConferenceData()!!.sessions
-        val loadSessionsUseCase = createUseCase()
+        val expectedSessions = TestData.sessionsList
+        val loadSessionsUseCase = LoadSessionsUseCase(SessionRepository(TestDataRepository))
 
         // Create ViewModel with the use case
         val viewModel = ScheduleViewModel(loadSessionsUseCase)
@@ -59,14 +60,5 @@ class ScheduleViewModelTest {
                 LiveDataTestUtil.getValue(viewModel.isLoading),
                 `is`(false))
     }
-
-    /**
-     * Creates a use case that will return the provided list of sessions.
-     */
-    private fun createUseCase(): LoadSessionsUseCase {
-        val conferenceDataRepository = ConferenceDataRepository(
-                TestSessionDataSource, TestSessionDataSource)
-        return LoadSessionsUseCase(SessionRepository(conferenceDataRepository))
-    }
 }