diff options
author | Nick Chalko <nchalko@google.com> | 2015-08-03 15:39:56 -0700 |
---|---|---|
committer | Nick Chalko <nchalko@google.com> | 2015-08-03 15:53:37 -0700 |
commit | 816a4be1a0f34f6a48877c8afd3dbbca19eac435 (patch) | |
tree | 4f18dda269764494942f5313acc93db4a35d47db /tests/unit/src/com/android/tv/recommendation/ChannelRecordTest.java | |
parent | 6edd2b09e5d16a29c703a5fcbd2e88c5cf5e55b7 (diff) | |
download | TV-816a4be1a0f34f6a48877c8afd3dbbca19eac435.tar.gz |
Migrate Live Channels App Src to AOSP branch
Bug: 21625152
Change-Id: I07e2830b27440556dc757e6340b4f77d1c0cbc66
Diffstat (limited to 'tests/unit/src/com/android/tv/recommendation/ChannelRecordTest.java')
-rw-r--r-- | tests/unit/src/com/android/tv/recommendation/ChannelRecordTest.java | 118 |
1 files changed, 118 insertions, 0 deletions
diff --git a/tests/unit/src/com/android/tv/recommendation/ChannelRecordTest.java b/tests/unit/src/com/android/tv/recommendation/ChannelRecordTest.java new file mode 100644 index 00000000..9b0e2805 --- /dev/null +++ b/tests/unit/src/com/android/tv/recommendation/ChannelRecordTest.java @@ -0,0 +1,118 @@ +/* + * Copyright (C) 2015 The Android Open Source Project + * + * 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 + * + * http://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.android.tv.recommendation; + +import android.test.AndroidTestCase; + +import java.util.Random; +import java.util.concurrent.TimeUnit; + +/** + * Unit tests for {@link ChannelRecord}. + */ +public class ChannelRecordTest extends AndroidTestCase { + private static final int CHANNEL_RECORD_MAX_HISTORY_SIZE = ChannelRecord.MAX_HISTORY_SIZE; + + private Random mRandom; + private ChannelRecord mChannelRecord; + private long mLatestWatchEndTimeMs; + + public void setUp() throws Exception { + super.setUp(); + mLatestWatchEndTimeMs = System.currentTimeMillis() - TimeUnit.DAYS.toMillis(1); + mChannelRecord = new ChannelRecord(getContext(), null, false); + mRandom = RecommendationUtils.createTestRandom(); + } + + public void testGetLastWatchEndTime_noHistory() { + assertEquals(0, mChannelRecord.getLastWatchEndTimeMs()); + } + + public void testGetLastWatchEndTime_oneHistory() { + addWatchLog(); + + assertEquals(mLatestWatchEndTimeMs, mChannelRecord.getLastWatchEndTimeMs()); + } + + public void testGetLastWatchEndTime_maxHistories() { + for (int i = 0; i < CHANNEL_RECORD_MAX_HISTORY_SIZE; ++i) { + addWatchLog(); + } + + assertEquals(mLatestWatchEndTimeMs, mChannelRecord.getLastWatchEndTimeMs()); + } + + public void testGetLastWatchEndTime_moreThanMaxHistories() { + for (int i = 0; i < CHANNEL_RECORD_MAX_HISTORY_SIZE + 1; ++i) { + addWatchLog(); + } + + assertEquals(mLatestWatchEndTimeMs, mChannelRecord.getLastWatchEndTimeMs()); + } + + public void testGetTotalWatchDuration_noHistory() { + assertEquals(0, mChannelRecord.getTotalWatchDurationMs()); + } + + public void testGetTotalWatchDuration_oneHistory() { + long durationMs = addWatchLog(); + + assertEquals(durationMs, mChannelRecord.getTotalWatchDurationMs()); + } + + public void testGetTotalWatchDuration_maxHistories() { + long totalWatchTimeMs = 0; + for (int i = 0; i < CHANNEL_RECORD_MAX_HISTORY_SIZE; ++i) { + long durationMs = addWatchLog(); + totalWatchTimeMs += durationMs; + } + + assertEquals(totalWatchTimeMs, mChannelRecord.getTotalWatchDurationMs()); + } + + public void testGetTotalWatchDuration_moreThanMaxHistories() { + long totalWatchTimeMs = 0; + long firstDurationMs = 0; + for (int i = 0; i < CHANNEL_RECORD_MAX_HISTORY_SIZE + 1; ++i) { + long durationMs = addWatchLog(); + totalWatchTimeMs += durationMs; + if (i == 0) { + firstDurationMs = durationMs; + } + } + + // Only latest CHANNEL_RECORD_MAX_HISTORY_SIZE logs are remained. + assertEquals(totalWatchTimeMs - firstDurationMs, mChannelRecord.getTotalWatchDurationMs()); + } + + /** + * Add new log history to channelRecord which its duration is lower than 1 minute. + * + * @return New watch log's duration time in milliseconds. + */ + private long addWatchLog() { + // Time hopping with random seconds. + mLatestWatchEndTimeMs += TimeUnit.SECONDS.toMillis(mRandom.nextInt(60) + 1); + + long durationMs = TimeUnit.SECONDS.toMillis(mRandom.nextInt(60) + 1); + mChannelRecord.logWatchHistory(new WatchedProgram(null, + mLatestWatchEndTimeMs, mLatestWatchEndTimeMs + durationMs)); + mLatestWatchEndTimeMs += durationMs; + + return durationMs; + } +} |