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 */ 16package androidx.work.impl.constraints.trackers; 17 18import static org.hamcrest.MatcherAssert.assertThat; 19import static org.hamcrest.Matchers.is; 20import static org.mockito.Mockito.mock; 21import static org.mockito.Mockito.verify; 22import static org.mockito.Mockito.verifyNoMoreInteractions; 23import static org.mockito.Mockito.when; 24 25import android.content.Context; 26import android.support.test.filters.SmallTest; 27import android.support.test.runner.AndroidJUnit4; 28 29import androidx.work.impl.constraints.ConstraintListener; 30 31import org.junit.After; 32import org.junit.Before; 33import org.junit.Test; 34import org.junit.runner.RunWith; 35 36@SmallTest 37@RunWith(AndroidJUnit4.class) 38public class ConstraintTrackerTest { 39 40 private TestConstraintTracker mTracker; 41 private Context mMockContext; 42 43 @Before 44 public void setUp() { 45 mMockContext = mock(Context.class); 46 when(mMockContext.getApplicationContext()).thenReturn(mMockContext); 47 mTracker = new TestConstraintTracker(mMockContext); 48 } 49 50 @After 51 public void tearDown() { 52 try { 53 mTracker.stopTracking(); 54 } catch (IllegalArgumentException e) { 55 // Ignore any exceptions if the receiver isn't registered. 56 } 57 } 58 59 @Test 60 public void testAddListener_manyListeners_doesNotCallGetInitialStateAndStartTrackingTwice() { 61 ConstraintListener<Boolean> constraintListener1 = mock(ConstraintListener.class); 62 ConstraintListener<Boolean> constraintListener2 = mock(ConstraintListener.class); 63 64 assertThat(mTracker.mIsTracking, is(false)); 65 assertThat(mTracker.mStartTrackingCount, is(0)); 66 assertThat(mTracker.mGetInitialStateCount, is(0)); 67 68 mTracker.addListener(constraintListener1); 69 assertThat(mTracker.mIsTracking, is(true)); 70 assertThat(mTracker.mStartTrackingCount, is(1)); 71 assertThat(mTracker.mGetInitialStateCount, is(1)); 72 73 mTracker.addListener(constraintListener2); 74 assertThat(mTracker.mIsTracking, is(true)); 75 assertThat(mTracker.mStartTrackingCount, is(1)); 76 assertThat(mTracker.mGetInitialStateCount, is(1)); 77 } 78 79 @Test 80 public void testAddListener_sameListener_doesNotCallGetInitialStateAndStartTrackingTwice() { 81 ConstraintListener<Boolean> constraintListener = mock(ConstraintListener.class); 82 83 assertThat(mTracker.mIsTracking, is(false)); 84 assertThat(mTracker.mStartTrackingCount, is(0)); 85 assertThat(mTracker.mGetInitialStateCount, is(0)); 86 87 mTracker.addListener(constraintListener); 88 assertThat(mTracker.mIsTracking, is(true)); 89 assertThat(mTracker.mStartTrackingCount, is(1)); 90 assertThat(mTracker.mGetInitialStateCount, is(1)); 91 92 mTracker.addListener(constraintListener); 93 assertThat(mTracker.mIsTracking, is(true)); 94 assertThat(mTracker.mStartTrackingCount, is(1)); 95 assertThat(mTracker.mGetInitialStateCount, is(1)); 96 } 97 98 @Test 99 public void testAddListener_notifiesAddedListener() { 100 ConstraintListener<Boolean> constraintListener1 = mock(ConstraintListener.class); 101 ConstraintListener<Boolean> constraintListener2 = mock(ConstraintListener.class); 102 103 mTracker.mInitialState = true; 104 mTracker.addListener(constraintListener1); 105 verify(constraintListener1).onConstraintChanged(true); 106 107 mTracker.addListener(constraintListener2); 108 verify(constraintListener2).onConstraintChanged(true); 109 verifyNoMoreInteractions(constraintListener1); 110 } 111 112 @Test 113 public void testAddListener_constraintNotSet() { 114 ConstraintListener<Boolean> constraintListener1 = mock(ConstraintListener.class); 115 ConstraintListener<Boolean> constraintListener2 = mock(ConstraintListener.class); 116 117 mTracker.addListener(constraintListener1); 118 mTracker.addListener(constraintListener2); 119 verify(constraintListener1).onConstraintChanged(null); 120 verify(constraintListener2).onConstraintChanged(null); 121 } 122 123 @Test 124 public void testSetState_newState_notifiesAllListeners() { 125 ConstraintListener<Boolean> constraintListener1 = mock(ConstraintListener.class); 126 ConstraintListener<Boolean> constraintListener2 = mock(ConstraintListener.class); 127 128 mTracker.addListener(constraintListener1); 129 mTracker.addListener(constraintListener2); 130 verify(constraintListener1).onConstraintChanged(null); 131 verify(constraintListener2).onConstraintChanged(null); 132 133 mTracker.setState(true); 134 verify(constraintListener1).onConstraintChanged(true); 135 verify(constraintListener2).onConstraintChanged(true); 136 } 137 138 @Test 139 public void testSetState_sameState_doesNotNotify() { 140 ConstraintListener<Boolean> constraintListener1 = mock(ConstraintListener.class); 141 ConstraintListener<Boolean> constraintListener2 = mock(ConstraintListener.class); 142 143 mTracker.addListener(constraintListener1); 144 mTracker.addListener(constraintListener2); 145 verify(constraintListener1).onConstraintChanged(null); 146 verify(constraintListener2).onConstraintChanged(null); 147 148 mTracker.setState(false); 149 verify(constraintListener1).onConstraintChanged(false); 150 verify(constraintListener2).onConstraintChanged(false); 151 152 mTracker.setState(false); 153 verifyNoMoreInteractions(constraintListener1); 154 verifyNoMoreInteractions(constraintListener2); 155 } 156 157 @Test 158 public void testRemoveListener() { 159 ConstraintListener<Boolean> constraintListener1 = mock(ConstraintListener.class); 160 ConstraintListener<Boolean> constraintListener2 = mock(ConstraintListener.class); 161 162 mTracker.addListener(constraintListener1); 163 mTracker.addListener(constraintListener2); 164 165 mTracker.removeListener(constraintListener1); 166 assertThat(mTracker.mIsTracking, is(true)); 167 assertThat(mTracker.mStopTrackingCount, is(0)); 168 169 mTracker.removeListener(constraintListener2); 170 assertThat(mTracker.mIsTracking, is(false)); 171 assertThat(mTracker.mStopTrackingCount, is(1)); 172 } 173 174 private static class TestConstraintTracker extends ConstraintTracker<Boolean> { 175 boolean mIsTracking; 176 int mGetInitialStateCount; 177 int mStartTrackingCount; 178 int mStopTrackingCount; 179 Boolean mInitialState = null; 180 181 TestConstraintTracker(Context context) { 182 super(context); 183 } 184 185 @Override 186 public Boolean getInitialState() { 187 mGetInitialStateCount++; 188 return mInitialState; 189 } 190 191 @Override 192 public void startTracking() { 193 mIsTracking = true; 194 mStartTrackingCount++; 195 } 196 197 @Override 198 public void stopTracking() { 199 mIsTracking = false; 200 mStopTrackingCount++; 201 } 202 } 203} 204