ComprehensiveCountryDetectorTest.java revision a58a8751b4c2ce457f0082a0baaee61312d56195 
1/*
2 * Copyright (C) 2010 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 com.android.server.location;
18
19import android.location.Country;
20import android.location.CountryListener;
21import android.test.AndroidTestCase;
22
23public class ComprehensiveCountryDetectorTest extends AndroidTestCase {
24    private class TestCountryDetector extends ComprehensiveCountryDetector {
25        public final static String COUNTRY_ISO = "us";
26        private boolean mLocationBasedDetectorStarted;
27        private boolean mLocationBasedDetectorStopped;
28        protected boolean mNotified;
29        private boolean listenerAdded = false;
30
31        private Country mNotifiedCountry;
32        public TestCountryDetector() {
33            super(getContext());
34        }
35
36        public void notifyLocationBasedListener(Country country) {
37            mNotified = true;
38            mNotifiedCountry = country;
39            mLocationBasedCountryDetector.notifyListener(country);
40        }
41
42        public boolean locationBasedDetectorStarted() {
43            return mLocationBasedCountryDetector != null && mLocationBasedDetectorStarted;
44        }
45
46        public boolean locationBasedDetectorStopped() {
47            return mLocationBasedCountryDetector == null && mLocationBasedDetectorStopped;
48        }
49
50        public boolean locationRefreshStarted() {
51            return mLocationRefreshTimer != null;
52        }
53
54        public boolean locationRefreshCancelled() {
55            return mLocationRefreshTimer == null;
56        }
57
58        @Override
59        protected CountryDetectorBase createLocationBasedCountryDetector() {
60            return new CountryDetectorBase(mContext) {
61                @Override
62                public Country detectCountry() {
63                    mLocationBasedDetectorStarted = true;
64                    return null;
65                }
66
67                @Override
68                public void stop() {
69                    mLocationBasedDetectorStopped = true;
70                }
71            };
72        }
73
74        @Override
75        protected Country getNetworkBasedCountry() {
76            return null;
77        }
78
79        @Override
80        protected Country getLastKnownLocationBasedCountry() {
81            return mNotifiedCountry;
82        }
83
84        @Override
85        protected Country getSimBasedCountry() {
86            return null;
87        }
88
89        @Override
90        protected Country getLocaleCountry() {
91            return null;
92        }
93
94        @Override
95        protected void runAfterDetectionAsync(final Country country, final Country detectedCountry,
96                final boolean notifyChange, final boolean startLocationBasedDetection) {
97            runAfterDetection(country, detectedCountry, notifyChange, startLocationBasedDetection);
98        };
99
100        @Override
101        protected boolean isAirplaneModeOff() {
102            return true;
103        }
104
105        @Override
106        protected synchronized void addPhoneStateListener() {
107            listenerAdded = true;
108        }
109
110        @Override
111        protected synchronized void removePhoneStateListener() {
112            listenerAdded = false;
113        }
114
115        @Override
116        protected boolean isGeoCoderImplemented() {
117            return true;
118        }
119
120        public boolean isPhoneStateListenerAdded() {
121            return listenerAdded;
122        }
123    }
124
125    private class CountryListenerImpl implements CountryListener {
126        private boolean mNotified;
127        private Country mCountry;
128
129        public void onCountryDetected(Country country) {
130            mNotified = true;
131            mCountry = country;
132        }
133
134        public boolean notified() {
135            return mNotified;
136        }
137
138        public Country getCountry() {
139            return mCountry;
140        }
141    }
142
143    public void testDetectNetworkBasedCountry() {
144        final Country resultCountry = new Country(
145                TestCountryDetector.COUNTRY_ISO, Country.COUNTRY_SOURCE_NETWORK);
146        TestCountryDetector countryDetector = new TestCountryDetector() {
147            @Override
148            protected Country getNetworkBasedCountry() {
149                return resultCountry;
150            }
151        };
152        CountryListenerImpl listener = new CountryListenerImpl();
153        countryDetector.setCountryListener(listener);
154        Country country = countryDetector.detectCountry();
155        assertTrue(sameCountry(country, resultCountry));
156        assertFalse(listener.notified());
157        assertFalse(countryDetector.locationBasedDetectorStarted());
158        assertFalse(countryDetector.locationRefreshStarted());
159        countryDetector.stop();
160    }
161
162    public void testDetectLocationBasedCountry() {
163        final Country resultCountry = new Country(
164                TestCountryDetector.COUNTRY_ISO, Country.COUNTRY_SOURCE_SIM);
165        final Country locationBasedCountry = new Country(
166                TestCountryDetector.COUNTRY_ISO, Country.COUNTRY_SOURCE_LOCATION);
167        TestCountryDetector countryDetector = new TestCountryDetector() {
168            @Override
169            protected Country getSimBasedCountry() {
170                return resultCountry;
171            }
172        };
173        CountryListenerImpl listener = new CountryListenerImpl();
174        countryDetector.setCountryListener(listener);
175        Country country = countryDetector.detectCountry();
176        assertTrue(sameCountry(country, resultCountry));
177        assertTrue(countryDetector.locationBasedDetectorStarted());
178        countryDetector.notifyLocationBasedListener(locationBasedCountry);
179        assertTrue(listener.notified());
180        assertTrue(sameCountry(listener.getCountry(), locationBasedCountry));
181        assertTrue(countryDetector.locationBasedDetectorStopped());
182        assertTrue(countryDetector.locationRefreshStarted());
183        countryDetector.stop();
184        assertTrue(countryDetector.locationRefreshCancelled());
185    }
186
187    public void testLocaleBasedCountry() {
188        final Country resultCountry = new Country(
189                TestCountryDetector.COUNTRY_ISO, Country.COUNTRY_SOURCE_LOCALE);
190        TestCountryDetector countryDetector = new TestCountryDetector() {
191            @Override
192            protected Country getLocaleCountry() {
193                return resultCountry;
194            }
195        };
196        CountryListenerImpl listener = new CountryListenerImpl();
197        countryDetector.setCountryListener(listener);
198        Country country = countryDetector.detectCountry();
199        assertTrue(sameCountry(country, resultCountry));
200        assertFalse(listener.notified());
201        assertTrue(countryDetector.locationBasedDetectorStarted());
202        assertTrue(countryDetector.locationRefreshStarted());
203        countryDetector.stop();
204        assertTrue(countryDetector.locationRefreshCancelled());
205    }
206
207    public void testStoppingDetector() {
208        // Test stopping detector when LocationBasedCountryDetector was started
209        final Country resultCountry = new Country(
210                TestCountryDetector.COUNTRY_ISO, Country.COUNTRY_SOURCE_SIM);
211        TestCountryDetector countryDetector = new TestCountryDetector() {
212            @Override
213            protected Country getSimBasedCountry() {
214                return resultCountry;
215            }
216        };
217        CountryListenerImpl listener = new CountryListenerImpl();
218        countryDetector.setCountryListener(listener);
219        Country country = countryDetector.detectCountry();
220        assertTrue(sameCountry(country, resultCountry));
221        assertTrue(countryDetector.locationBasedDetectorStarted());
222        countryDetector.stop();
223        // The LocationBasedDetector should be stopped.
224        assertTrue(countryDetector.locationBasedDetectorStopped());
225        // The location refresh should not running.
226        assertTrue(countryDetector.locationRefreshCancelled());
227    }
228
229    public void testLocationBasedCountryNotFound() {
230        final Country resultCountry = new Country(
231                TestCountryDetector.COUNTRY_ISO, Country.COUNTRY_SOURCE_SIM);
232        TestCountryDetector countryDetector = new TestCountryDetector() {
233            @Override
234            protected Country getSimBasedCountry() {
235                return resultCountry;
236            }
237        };
238        CountryListenerImpl listener = new CountryListenerImpl();
239        countryDetector.setCountryListener(listener);
240        Country country = countryDetector.detectCountry();
241        assertTrue(sameCountry(country, resultCountry));
242        assertTrue(countryDetector.locationBasedDetectorStarted());
243        countryDetector.notifyLocationBasedListener(null);
244        assertFalse(listener.notified());
245        assertTrue(sameCountry(listener.getCountry(), null));
246        assertTrue(countryDetector.locationBasedDetectorStopped());
247        assertTrue(countryDetector.locationRefreshStarted());
248        countryDetector.stop();
249        assertTrue(countryDetector.locationRefreshCancelled());
250    }
251
252    public void testNoCountryFound() {
253        TestCountryDetector countryDetector = new TestCountryDetector();
254        CountryListenerImpl listener = new CountryListenerImpl();
255        countryDetector.setCountryListener(listener);
256        Country country = countryDetector.detectCountry();
257        assertTrue(sameCountry(country, null));
258        assertTrue(countryDetector.locationBasedDetectorStarted());
259        countryDetector.notifyLocationBasedListener(null);
260        assertFalse(listener.notified());
261        assertTrue(sameCountry(listener.getCountry(), null));
262        assertTrue(countryDetector.locationBasedDetectorStopped());
263        assertTrue(countryDetector.locationRefreshStarted());
264        countryDetector.stop();
265        assertTrue(countryDetector.locationRefreshCancelled());
266    }
267
268    public void testAddRemoveListener() {
269        TestCountryDetector countryDetector = new TestCountryDetector();
270        CountryListenerImpl listener = new CountryListenerImpl();
271        countryDetector.setCountryListener(listener);
272        assertTrue(countryDetector.isPhoneStateListenerAdded());
273        assertTrue(countryDetector.locationBasedDetectorStarted());
274        countryDetector.setCountryListener(null);
275        assertFalse(countryDetector.isPhoneStateListenerAdded());
276        assertTrue(countryDetector.locationBasedDetectorStopped());
277    }
278
279    public void testGeocoderNotImplemented() {
280        TestCountryDetector countryDetector = new TestCountryDetector() {
281            @Override
282            protected boolean isGeoCoderImplemented() {
283                return false;
284            }
285        };
286        CountryListenerImpl listener = new CountryListenerImpl();
287        countryDetector.setCountryListener(listener);
288        assertTrue(countryDetector.isPhoneStateListenerAdded());
289        assertFalse(countryDetector.locationBasedDetectorStarted());
290        countryDetector.setCountryListener(null);
291        assertFalse(countryDetector.isPhoneStateListenerAdded());
292    }
293
294    private boolean sameCountry(Country country1, Country country2) {
295        return country1 == null && country2 == null || country1 != null && country2 != null &&
296        country1.getCountryIso().equalsIgnoreCase(country2.getCountryIso()) &&
297        country1.getSource() == country2.getSource();
298    }
299}
300