| /* |
| * Copyright (C) 2017 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 android.support.media.tv; |
| |
| import static org.junit.Assert.assertEquals; |
| import static org.junit.Assert.assertNotNull; |
| import static org.junit.Assert.assertTrue; |
| |
| import android.content.ContentResolver; |
| import android.content.ContentUris; |
| import android.content.ContentValues; |
| import android.database.Cursor; |
| import android.database.MatrixCursor; |
| import android.media.tv.TvContentRating; |
| import android.net.Uri; |
| import android.os.Build; |
| import android.support.media.tv.TvContractCompat.Channels; |
| import android.support.media.tv.TvContractCompat.Programs; |
| import android.support.test.InstrumentationRegistry; |
| import android.support.test.filters.SdkSuppress; |
| import android.support.test.filters.SmallTest; |
| import android.support.test.runner.AndroidJUnit4; |
| |
| import org.junit.After; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| |
| import java.util.Arrays; |
| import java.util.Objects; |
| |
| /** |
| * Tests that programs can be created using the Builder pattern and correctly obtain |
| * values from them. |
| */ |
| @SmallTest |
| @RunWith(AndroidJUnit4.class) |
| @SdkSuppress(minSdkVersion = Build.VERSION_CODES.LOLLIPOP) |
| public class ProgramTest { |
| @After |
| public void tearDown() { |
| if (!Utils.hasTvInputFramework(InstrumentationRegistry.getContext())) { |
| return; |
| } |
| ContentResolver resolver = InstrumentationRegistry.getContext().getContentResolver(); |
| resolver.delete(Channels.CONTENT_URI, null, null); |
| } |
| |
| @Test |
| public void testEmptyProgram() { |
| Program emptyProgram = new Program.Builder() |
| .build(); |
| ContentValues contentValues = emptyProgram.toContentValues(); |
| compareProgram(emptyProgram, |
| Program.fromCursor(getProgramCursor(Program.PROJECTION, contentValues)), true); |
| } |
| |
| @Test |
| public void testSampleProgram() { |
| Program sampleProgram = new Program.Builder() |
| .setPackageName("My package") |
| .setTitle("Program Title") |
| .setDescription("This is a sample program") |
| .setEpisodeNumber(5) |
| .setSeasonNumber("The Final Season", 7) |
| .setThumbnailUri(Uri.parse("http://www.example.com/programs/poster.png")) |
| .setChannelId(3) |
| .setStartTimeUtcMillis(0) |
| .setEndTimeUtcMillis(1000) |
| .build(); |
| ContentValues contentValues = sampleProgram.toContentValues(); |
| compareProgram(sampleProgram, |
| Program.fromCursor(getProgramCursor(Program.PROJECTION, contentValues)), true); |
| |
| Program clonedSampleProgram = new Program.Builder(sampleProgram).build(); |
| compareProgram(sampleProgram, clonedSampleProgram, true); |
| } |
| |
| @Test |
| public void testFullyPopulatedProgram() { |
| Program fullyPopulatedProgram = createFullyPopulatedProgram(3); |
| |
| ContentValues contentValues = fullyPopulatedProgram.toContentValues(); |
| compareProgram(fullyPopulatedProgram, |
| Program.fromCursor(getProgramCursor(Program.PROJECTION, contentValues)), true); |
| |
| Program clonedFullyPopulatedProgram = new Program.Builder(fullyPopulatedProgram).build(); |
| compareProgram(fullyPopulatedProgram, clonedFullyPopulatedProgram, true); |
| } |
| |
| @Test |
| public void testChannelWithSystemContentProvider() { |
| if (!Utils.hasTvInputFramework(InstrumentationRegistry.getContext())) { |
| return; |
| } |
| Channel channel = new Channel.Builder() |
| .setInputId("TestInputService") |
| .setType(TvContractCompat.Channels.TYPE_OTHER) |
| .build(); |
| ContentResolver resolver = InstrumentationRegistry.getContext().getContentResolver(); |
| Uri channelUri = resolver.insert(Channels.CONTENT_URI, channel.toContentValues()); |
| assertNotNull(channelUri); |
| |
| Program fullyPopulatedProgram = |
| createFullyPopulatedProgram(ContentUris.parseId(channelUri)); |
| Uri programUri = resolver.insert(Programs.CONTENT_URI, |
| fullyPopulatedProgram.toContentValues()); |
| |
| Program programFromSystemDb = loadProgramFromContentProvider(resolver, programUri); |
| compareProgram(fullyPopulatedProgram, programFromSystemDb, false); |
| } |
| |
| @Test |
| public void testProgramUpdateWithContentProvider() { |
| if (!Utils.hasTvInputFramework(InstrumentationRegistry.getContext())) { |
| return; |
| } |
| Channel channel = new Channel.Builder() |
| .setInputId("TestInputService") |
| .setType(TvContractCompat.Channels.TYPE_OTHER) |
| .build(); |
| ContentResolver resolver = InstrumentationRegistry.getContext().getContentResolver(); |
| Uri channelUri = resolver.insert(Channels.CONTENT_URI, channel.toContentValues()); |
| assertNotNull(channelUri); |
| |
| Program fullyPopulatedProgram = |
| createFullyPopulatedProgram(ContentUris.parseId(channelUri)); |
| Uri programUri = resolver.insert(Programs.CONTENT_URI, |
| fullyPopulatedProgram.toContentValues()); |
| |
| Program programFromSystemDb = loadProgramFromContentProvider(resolver, programUri); |
| compareProgram(fullyPopulatedProgram, programFromSystemDb, false); |
| |
| // Update a field from a fully loaded program. |
| Program updatedProgram = new Program.Builder(programFromSystemDb) |
| .setDescription("description1").build(); |
| assertEquals(1, resolver.update( |
| programUri, updatedProgram.toContentValues(), null, null)); |
| programFromSystemDb = loadProgramFromContentProvider(resolver, programUri); |
| compareProgram(updatedProgram, programFromSystemDb, false); |
| |
| // Update a field with null from a fully loaded program. |
| updatedProgram = new Program.Builder(updatedProgram) |
| .setLongDescription(null).build(); |
| assertEquals(1, resolver.update( |
| programUri, updatedProgram.toContentValues(), null, null)); |
| programFromSystemDb = loadProgramFromContentProvider(resolver, programUri); |
| compareProgram(updatedProgram, programFromSystemDb, false); |
| |
| // Update a field without referencing fully loaded program. |
| ContentValues values = new Program.Builder().setDescription("description2").build() |
| .toContentValues(); |
| assertEquals(1, values.size()); |
| assertEquals(1, resolver.update(programUri, values, null, null)); |
| programFromSystemDb = loadProgramFromContentProvider(resolver, programUri); |
| Program expectedProgram = new Program.Builder(programFromSystemDb) |
| .setDescription("description2").build(); |
| compareProgram(expectedProgram, programFromSystemDb, false); |
| } |
| |
| @Test |
| public void testProgramEquals() { |
| assertEquals(createFullyPopulatedProgram(1), createFullyPopulatedProgram(1)); |
| } |
| |
| private static Program loadProgramFromContentProvider( |
| ContentResolver resolver, Uri programUri) { |
| try (Cursor cursor = resolver.query(programUri, null, null, null, null)) { |
| assertNotNull(cursor); |
| assertEquals(1, cursor.getCount()); |
| cursor.moveToNext(); |
| return Program.fromCursor(cursor); |
| } |
| } |
| |
| private static Program createFullyPopulatedProgram(long channelId) { |
| return new Program.Builder() |
| .setSearchable(false) |
| .setThumbnailUri(Uri.parse("http://example.com/thumbnail.png")) |
| .setAudioLanguages(new String [] {"eng", "kor"}) |
| .setCanonicalGenres(new String[] {TvContractCompat.Programs.Genres.MOVIES}) |
| .setContentRatings(new TvContentRating[] { |
| TvContentRating.createRating("com.android.tv", "US_TV", "US_TV_Y7")}) |
| .setDescription("This is a sample program") |
| .setEpisodeNumber("Pilot", 0) |
| .setEpisodeTitle("Hello World") |
| .setLongDescription("This is a longer description than the previous description") |
| .setPosterArtUri(Uri.parse("http://example.com/poster.png")) |
| .setSeasonNumber("The Final Season", 7) |
| .setSeasonTitle("The Final Season") |
| .setTitle("Google") |
| .setVideoHeight(1080) |
| .setVideoWidth(1920) |
| .setInternalProviderFlag1(0x4) |
| .setInternalProviderFlag2(0x3) |
| .setInternalProviderFlag3(0x2) |
| .setInternalProviderFlag4(0x1) |
| .setReviewRatingStyle(Programs.REVIEW_RATING_STYLE_PERCENTAGE) |
| .setReviewRating("83.9") |
| .setChannelId(channelId) |
| .setStartTimeUtcMillis(0) |
| .setEndTimeUtcMillis(1000) |
| .setBroadcastGenres(new String[] {"Music", "Family"}) |
| .setRecordingProhibited(false) |
| .build(); |
| } |
| |
| private static void compareProgram(Program programA, Program programB, |
| boolean includeIdAndProtectedFields) { |
| assertTrue(Arrays.equals(programA.getAudioLanguages(), programB.getAudioLanguages())); |
| assertTrue(Arrays.deepEquals(programA.getBroadcastGenres(), programB.getBroadcastGenres())); |
| assertTrue(Arrays.deepEquals(programA.getCanonicalGenres(), programB.getCanonicalGenres())); |
| assertEquals(programA.getChannelId(), programB.getChannelId()); |
| assertTrue(Arrays.deepEquals(programA.getContentRatings(), programB.getContentRatings())); |
| assertEquals(programA.getDescription(), programB.getDescription()); |
| assertEquals(programA.getEndTimeUtcMillis(), programB.getEndTimeUtcMillis()); |
| assertEquals(programA.getEpisodeNumber(), programB.getEpisodeNumber()); |
| assertEquals(programA.getEpisodeTitle(), programB.getEpisodeTitle()); |
| assertEquals(programA.getLongDescription(), programB.getLongDescription()); |
| assertEquals(programA.getPosterArtUri(), programB.getPosterArtUri()); |
| assertEquals(programA.getSeasonNumber(), programB.getSeasonNumber()); |
| assertEquals(programA.getStartTimeUtcMillis(), programB.getStartTimeUtcMillis()); |
| assertEquals(programA.getThumbnailUri(), programB.getThumbnailUri()); |
| assertEquals(programA.getTitle(), programB.getTitle()); |
| assertEquals(programA.getVideoHeight(), programB.getVideoHeight()); |
| assertEquals(programA.getVideoWidth(), programB.getVideoWidth()); |
| if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.M) { |
| assertEquals(programA.isSearchable(), programB.isSearchable()); |
| assertEquals(programA.getInternalProviderFlag1(), programB.getInternalProviderFlag1()); |
| assertEquals(programA.getInternalProviderFlag2(), programB.getInternalProviderFlag2()); |
| assertEquals(programA.getInternalProviderFlag3(), programB.getInternalProviderFlag3()); |
| assertEquals(programA.getInternalProviderFlag4(), programB.getInternalProviderFlag4()); |
| } |
| if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.N) { |
| assertTrue(Objects.equals(programA.getSeasonTitle(), programB.getSeasonTitle())); |
| assertTrue(Objects.equals(programA.isRecordingProhibited(), |
| programB.isRecordingProhibited())); |
| } |
| if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.O) { |
| assertEquals(programA.getReviewRatingStyle(), programB.getReviewRatingStyle()); |
| assertEquals(programA.getReviewRating(), programB.getReviewRating()); |
| } |
| if (includeIdAndProtectedFields) { |
| // Skip row ID since the one from system DB has the valid ID while the other does not. |
| assertEquals(programA.getId(), programB.getId()); |
| assertEquals(programA.getPackageName(), programB.getPackageName()); |
| assertEquals(programA.toContentValues(), programB.toContentValues()); |
| } |
| } |
| |
| private static MatrixCursor getProgramCursor(String[] projection, ContentValues contentValues) { |
| MatrixCursor cursor = new MatrixCursor(projection); |
| MatrixCursor.RowBuilder builder = cursor.newRow(); |
| for (String row : projection) { |
| if (row != null) { |
| builder.add(row, contentValues.get(row)); |
| } |
| } |
| cursor.moveToFirst(); |
| return cursor; |
| } |
| } |