IdGeneratorTest.java revision 22a8129c2b3313100f851460f7da9e56ca98bd8f 
1/*
2 * Copyright 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
17package androidx.work.impl.utils;
18
19import static androidx.work.impl.utils.IdGenerator.INITIAL_ID;
20
21import static org.hamcrest.CoreMatchers.is;
22import static org.hamcrest.MatcherAssert.assertThat;
23import static org.mockito.ArgumentMatchers.anyInt;
24import static org.mockito.ArgumentMatchers.eq;
25import static org.mockito.Mockito.mock;
26import static org.mockito.Mockito.when;
27
28import static java.lang.Integer.MAX_VALUE;
29
30import android.content.Context;
31import android.content.SharedPreferences;
32import android.support.test.filters.SdkSuppress;
33import android.support.test.filters.SmallTest;
34import android.support.test.runner.AndroidJUnit4;
35
36import androidx.work.impl.WorkManagerImpl;
37
38import org.junit.Before;
39import org.junit.Test;
40import org.junit.runner.RunWith;
41import org.mockito.invocation.InvocationOnMock;
42import org.mockito.stubbing.Answer;
43
44@SmallTest
45@RunWith(AndroidJUnit4.class)
46@SdkSuppress(minSdkVersion = WorkManagerImpl.MIN_JOB_SCHEDULER_API_LEVEL)
47public class IdGeneratorTest {
48    private Integer mMockSharedPrefsNextId;
49    private IdGenerator mIdGenerator;
50
51    @Before
52    public void setUp() {
53        Context mMockContext = mock(Context.class);
54        SharedPreferences.Editor mockEditor = createMockSharedPreferencesEditor();
55        SharedPreferences mockSharedPrefs = createMockSharedPreferences(mockEditor);
56        when(mMockContext.getSharedPreferences(
57                eq(IdGenerator.PREFERENCE_FILE_KEY), anyInt()))
58                .thenReturn(mockSharedPrefs);
59        mIdGenerator = new IdGenerator(mMockContext);
60    }
61
62    @Test
63    public void testNextId_returnsInitialIdWhenNoStoredNextId() {
64        int nextId = mIdGenerator.nextJobSchedulerIdWithRange(INITIAL_ID, MAX_VALUE);
65        assertThat(nextId, is(INITIAL_ID));
66    }
67
68    @Test
69    public void testNextId_returnsStoredNextId() {
70        int expectedId = 100;
71        storeNextIdInSharedPrefs(expectedId);
72        int nextId = mIdGenerator.nextJobSchedulerIdWithRange(INITIAL_ID, MAX_VALUE);
73        assertThat(nextId, is(expectedId));
74    }
75
76    @Test
77    public void testNextId_returnsInitialIdAfterReturningMaxInteger() {
78        int expectedId = MAX_VALUE;
79        storeNextIdInSharedPrefs(expectedId);
80        int nextId = mIdGenerator.nextJobSchedulerIdWithRange(INITIAL_ID, MAX_VALUE);
81        assertThat(nextId, is(MAX_VALUE));
82        nextId = mIdGenerator.nextJobSchedulerIdWithRange(INITIAL_ID, MAX_VALUE);
83        assertThat(nextId, is(INITIAL_ID));
84    }
85
86    @Test
87    public void testNextId_belowMinRange() {
88        storeNextIdInSharedPrefs(2);
89        assertThat(mIdGenerator.nextJobSchedulerIdWithRange(10, 100), is(10));
90    }
91
92    @Test
93    public void testNextId_aboveMaxRange() {
94        storeNextIdInSharedPrefs(100);
95        assertThat(mIdGenerator.nextJobSchedulerIdWithRange(10, 100), is(100));
96    }
97
98    @Test
99    public void testNextId_aboveMaxRange2() {
100        storeNextIdInSharedPrefs(110);
101        assertThat(mIdGenerator.nextJobSchedulerIdWithRange(10, 100), is(10));
102    }
103
104    @Test
105    public void testNextId_withinRange() {
106        storeNextIdInSharedPrefs(20);
107        assertThat(mIdGenerator.nextJobSchedulerIdWithRange(10, 100), is(20));
108    }
109
110    /**
111     * Mocks setting a stored value in {@link SharedPreferences} for the next ID.
112     *
113     * @param nextId The next ID to store in {@link SharedPreferences}.
114     */
115    private void storeNextIdInSharedPrefs(int nextId) {
116        mMockSharedPrefsNextId = nextId;
117    }
118
119    private SharedPreferences createMockSharedPreferences(SharedPreferences.Editor mockEditor) {
120        final SharedPreferences mockSharedPreferences = mock(SharedPreferences.class);
121        when(mockSharedPreferences.edit()).thenReturn(mockEditor);
122        when(mockSharedPreferences.getInt(eq(IdGenerator.NEXT_JOB_SCHEDULER_ID_KEY), anyInt()))
123                .thenAnswer(new Answer<Integer>() {
124                    @Override
125                    public Integer answer(InvocationOnMock invocation) throws Throwable {
126                        int defValue = invocation.getArgument(1);
127                        return (mMockSharedPrefsNextId == null) ? defValue : mMockSharedPrefsNextId;
128                    }
129                });
130        return mockSharedPreferences;
131    }
132
133    private SharedPreferences.Editor createMockSharedPreferencesEditor() {
134        final SharedPreferences.Editor mockEditor = mock(SharedPreferences.Editor.class);
135        when(mockEditor.putInt(eq(IdGenerator.NEXT_JOB_SCHEDULER_ID_KEY), anyInt())).thenAnswer(
136                new Answer<SharedPreferences.Editor>() {
137                    @Override
138                    public SharedPreferences.Editor answer(InvocationOnMock invocation)
139                            throws Throwable {
140                        mMockSharedPrefsNextId = invocation.getArgument(1);
141                        return mockEditor;
142                    }
143                });
144        return mockEditor;
145    }
146}
147