1 /* 2 * Copyright (C) 2017 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package android.arch.persistence.room.integration.testapp.paging; 18 19 import static android.test.MoreAsserts.assertEmpty; 20 21 import static org.hamcrest.CoreMatchers.is; 22 import static org.hamcrest.MatcherAssert.assertThat; 23 24 import android.arch.persistence.room.integration.testapp.test.TestDatabaseTest; 25 import android.arch.persistence.room.integration.testapp.test.TestUtil; 26 import android.arch.persistence.room.integration.testapp.vo.User; 27 import android.arch.persistence.room.paging.LimitOffsetDataSource; 28 import android.support.annotation.NonNull; 29 import android.support.test.filters.SmallTest; 30 import android.support.test.runner.AndroidJUnit4; 31 32 import org.junit.After; 33 import org.junit.Test; 34 import org.junit.runner.RunWith; 35 36 import java.util.ArrayList; 37 import java.util.List; 38 39 @SmallTest 40 @RunWith(AndroidJUnit4.class) 41 public class LimitOffsetDataSourceTest extends TestDatabaseTest { 42 43 @After teardown()44 public void teardown() { 45 mUserDao.deleteEverything(); 46 } 47 loadUsersByAgeDesc()48 private LimitOffsetDataSource<User> loadUsersByAgeDesc() { 49 return (LimitOffsetDataSource<User>) mUserDao.loadUsersByAgeDesc(); 50 } 51 52 @Test emptyPage()53 public void emptyPage() { 54 LimitOffsetDataSource<User> dataSource = loadUsersByAgeDesc(); 55 assertThat(dataSource.countItems(), is(0)); 56 } 57 58 @Test loadCount()59 public void loadCount() { 60 createUsers(6); 61 LimitOffsetDataSource<User> dataSource = loadUsersByAgeDesc(); 62 assertThat(dataSource.countItems(), is(6)); 63 } 64 65 @Test singleItem()66 public void singleItem() { 67 List<User> users = createUsers(1); 68 LimitOffsetDataSource<User> dataSource = loadUsersByAgeDesc(); 69 assertThat(dataSource.countItems(), is(1)); 70 List<User> initial = dataSource.loadRange(0, 10); 71 72 assertThat(initial.get(0), is(users.get(0))); 73 assertEmpty(dataSource.loadRange(1, 10)); 74 } 75 76 @Test initial()77 public void initial() { 78 List<User> users = createUsers(10); 79 LimitOffsetDataSource<User> dataSource = loadUsersByAgeDesc(); 80 assertThat(dataSource.countItems(), is(10)); 81 List<User> initial = dataSource.loadRange(0, 1); 82 assertThat(initial.get(0), is(users.get(0))); 83 List<User> second = dataSource.loadRange(1, 1); 84 assertThat(second.get(0), is(users.get(1))); 85 } 86 87 @Test loadAll()88 public void loadAll() { 89 List<User> users = createUsers(10); 90 91 LimitOffsetDataSource<User> dataSource = loadUsersByAgeDesc(); 92 List<User> all = dataSource.loadRange(0, 10); 93 assertThat(users, is(all)); 94 } 95 96 @Test loadAfter()97 public void loadAfter() { 98 List<User> users = createUsers(10); 99 LimitOffsetDataSource<User> dataSource = loadUsersByAgeDesc(); 100 List<User> result = dataSource.loadRange(4, 2); 101 assertThat(result, is(users.subList(4, 6))); 102 } 103 104 @NonNull createUsers(int count)105 private List<User> createUsers(int count) { 106 List<User> users = new ArrayList<>(); 107 for (int i = 0; i < count; i++) { 108 User user = TestUtil.createUser(i); 109 user.setAge(1); 110 mUserDao.insert(user); 111 users.add(user); 112 } 113 return users; 114 } 115 } 116