1 package com.xtremelabs.robolectric.shadows; 2 3 import android.app.Application; 4 import android.appwidget.AppWidgetManager; 5 import android.content.BroadcastReceiver; 6 import android.content.ComponentName; 7 import android.content.ContentResolver; 8 import android.content.Context; 9 import android.content.Intent; 10 import android.content.IntentFilter; 11 import android.content.ServiceConnection; 12 import android.content.res.Resources; 13 import android.os.IBinder; 14 import android.os.Looper; 15 import android.view.LayoutInflater; 16 import android.widget.Toast; 17 import com.xtremelabs.robolectric.Robolectric; 18 import com.xtremelabs.robolectric.internal.Implementation; 19 import com.xtremelabs.robolectric.internal.Implements; 20 import com.xtremelabs.robolectric.internal.RealObject; 21 import com.xtremelabs.robolectric.res.ResourceLoader; 22 import com.xtremelabs.robolectric.tester.org.apache.http.FakeHttpLayer; 23 import com.xtremelabs.robolectric.util.Scheduler; 24 25 import java.util.ArrayList; 26 import java.util.HashMap; 27 import java.util.Iterator; 28 import java.util.List; 29 import java.util.Map; 30 31 import static com.xtremelabs.robolectric.Robolectric.newInstanceOf; 32 import static com.xtremelabs.robolectric.Robolectric.shadowOf; 33 34 /** 35 * Shadows the {@code android.app.Application} class. 36 */ 37 @SuppressWarnings({"UnusedDeclaration"}) 38 @Implements(Application.class) 39 public class ShadowApplication extends ShadowContextWrapper { 40 private static final Map<String, String> SYSTEM_SERVICE_MAP = new HashMap<String, String>(); 41 42 static { 43 // note that these are different! 44 // They specify concrete classes within Robolectric for interfaces or abstract classes defined by Android SYSTEM_SERVICE_MAP.put(Context.WINDOW_SERVICE, "com.xtremelabs.robolectric.tester.android.view.TestWindowManager")45 SYSTEM_SERVICE_MAP.put(Context.WINDOW_SERVICE, "com.xtremelabs.robolectric.tester.android.view.TestWindowManager"); SYSTEM_SERVICE_MAP.put(Context.CLIPBOARD_SERVICE, "com.xtremelabs.robolectric.tester.android.text.TestClipboardManager")46 SYSTEM_SERVICE_MAP.put(Context.CLIPBOARD_SERVICE, "com.xtremelabs.robolectric.tester.android.text.TestClipboardManager"); SYSTEM_SERVICE_MAP.put(Context.SENSOR_SERVICE, "android.hardware.TestSensorManager")47 SYSTEM_SERVICE_MAP.put(Context.SENSOR_SERVICE, "android.hardware.TestSensorManager"); SYSTEM_SERVICE_MAP.put(Context.VIBRATOR_SERVICE, "android.os.TestVibrator")48 SYSTEM_SERVICE_MAP.put(Context.VIBRATOR_SERVICE, "android.os.TestVibrator"); 49 50 // the rest are as mapped in docs... SYSTEM_SERVICE_MAP.put(Context.LAYOUT_INFLATER_SERVICE, "android.view.LayoutInflater")51 SYSTEM_SERVICE_MAP.put(Context.LAYOUT_INFLATER_SERVICE, "android.view.LayoutInflater"); SYSTEM_SERVICE_MAP.put(Context.ACTIVITY_SERVICE, "android.app.ActivityManager")52 SYSTEM_SERVICE_MAP.put(Context.ACTIVITY_SERVICE, "android.app.ActivityManager"); SYSTEM_SERVICE_MAP.put(Context.POWER_SERVICE, "android.os.PowerManager")53 SYSTEM_SERVICE_MAP.put(Context.POWER_SERVICE, "android.os.PowerManager"); SYSTEM_SERVICE_MAP.put(Context.ALARM_SERVICE, "android.app.AlarmManager")54 SYSTEM_SERVICE_MAP.put(Context.ALARM_SERVICE, "android.app.AlarmManager"); SYSTEM_SERVICE_MAP.put(Context.NOTIFICATION_SERVICE, "android.app.NotificationManager")55 SYSTEM_SERVICE_MAP.put(Context.NOTIFICATION_SERVICE, "android.app.NotificationManager"); SYSTEM_SERVICE_MAP.put(Context.KEYGUARD_SERVICE, "android.app.KeyguardManager")56 SYSTEM_SERVICE_MAP.put(Context.KEYGUARD_SERVICE, "android.app.KeyguardManager"); SYSTEM_SERVICE_MAP.put(Context.LOCATION_SERVICE, "android.location.LocationManager")57 SYSTEM_SERVICE_MAP.put(Context.LOCATION_SERVICE, "android.location.LocationManager"); SYSTEM_SERVICE_MAP.put(Context.SEARCH_SERVICE, "android.app.SearchManager")58 SYSTEM_SERVICE_MAP.put(Context.SEARCH_SERVICE, "android.app.SearchManager"); SYSTEM_SERVICE_MAP.put(Context.STORAGE_SERVICE, "android.os.storage.StorageManager")59 SYSTEM_SERVICE_MAP.put(Context.STORAGE_SERVICE, "android.os.storage.StorageManager"); SYSTEM_SERVICE_MAP.put(Context.CONNECTIVITY_SERVICE, "android.net.ConnectivityManager")60 SYSTEM_SERVICE_MAP.put(Context.CONNECTIVITY_SERVICE, "android.net.ConnectivityManager"); SYSTEM_SERVICE_MAP.put(Context.WIFI_SERVICE, "android.net.wifi.WifiManager")61 SYSTEM_SERVICE_MAP.put(Context.WIFI_SERVICE, "android.net.wifi.WifiManager"); SYSTEM_SERVICE_MAP.put(Context.AUDIO_SERVICE, "android.media.AudioManager")62 SYSTEM_SERVICE_MAP.put(Context.AUDIO_SERVICE, "android.media.AudioManager"); SYSTEM_SERVICE_MAP.put(Context.TELEPHONY_SERVICE, "android.telephony.TelephonyManager")63 SYSTEM_SERVICE_MAP.put(Context.TELEPHONY_SERVICE, "android.telephony.TelephonyManager"); SYSTEM_SERVICE_MAP.put(Context.INPUT_METHOD_SERVICE, "android.view.inputmethod.InputMethodManager")64 SYSTEM_SERVICE_MAP.put(Context.INPUT_METHOD_SERVICE, "android.view.inputmethod.InputMethodManager"); SYSTEM_SERVICE_MAP.put(Context.UI_MODE_SERVICE, "android.app.UiModeManager")65 SYSTEM_SERVICE_MAP.put(Context.UI_MODE_SERVICE, "android.app.UiModeManager"); SYSTEM_SERVICE_MAP.put(Context.DOWNLOAD_SERVICE, "android.app.DownloadManager")66 SYSTEM_SERVICE_MAP.put(Context.DOWNLOAD_SERVICE, "android.app.DownloadManager"); 67 } 68 69 @RealObject private Application realApplication; 70 71 private ResourceLoader resourceLoader; 72 private ContentResolver contentResolver; 73 private Map<String, Object> systemServices = new HashMap<String, Object>(); 74 private List<Intent> startedActivities = new ArrayList<Intent>(); 75 private List<Intent> startedServices = new ArrayList<Intent>(); 76 private List<Intent> stoppedServies = new ArrayList<Intent>(); 77 private List<Intent> broadcastIntents = new ArrayList<Intent>(); 78 private List<ServiceConnection> unboundServiceConnections = new ArrayList<ServiceConnection>(); 79 private List<Wrapper> registeredReceivers = new ArrayList<Wrapper>(); 80 private Map<String, Intent> stickyIntents = new HashMap<String, Intent>(); 81 private FakeHttpLayer fakeHttpLayer = new FakeHttpLayer(); 82 private Looper mainLooper = ShadowLooper.myLooper(); 83 private Scheduler backgroundScheduler = new Scheduler(); 84 private Map<String, Map<String, Object>> sharedPreferenceMap = new HashMap<String, Map<String, Object>>(); 85 private ArrayList<Toast> shownToasts = new ArrayList<Toast>(); 86 private ShadowAlertDialog latestAlertDialog; 87 private ShadowDialog latestDialog; 88 private Object bluetoothAdapter = Robolectric.newInstanceOf("android.bluetooth.BluetoothAdapter"); 89 private Resources resources; 90 91 // these are managed by the AppSingletonizier... kinda gross, sorry [xw] 92 LayoutInflater layoutInflater; 93 AppWidgetManager appWidgetManager; 94 private ServiceConnection serviceConnection; 95 private ComponentName componentNameForBindService; 96 private IBinder serviceForBindService; 97 private List<String> unbindableActions = new ArrayList<String>(); 98 99 /** 100 * Associates a {@code ResourceLoader} with an {@code Application} instance 101 * 102 * @param application application 103 * @param resourceLoader resource loader 104 * @return the application 105 * todo: make this non-static? 106 */ bind(Application application, ResourceLoader resourceLoader)107 public static Application bind(Application application, ResourceLoader resourceLoader) { 108 ShadowApplication shadowApplication = shadowOf(application); 109 if (shadowApplication.resourceLoader != null) throw new RuntimeException("ResourceLoader already set!"); 110 shadowApplication.resourceLoader = resourceLoader; 111 shadowApplication.resources = ShadowResources.bind(new Resources(null, null, null), resourceLoader); 112 return application; 113 } 114 getShownToasts()115 public List<Toast> getShownToasts() { 116 return shownToasts; 117 } 118 getBackgroundScheduler()119 public Scheduler getBackgroundScheduler() { 120 return backgroundScheduler; 121 } 122 123 @Override 124 @Implementation getApplicationContext()125 public Context getApplicationContext() { 126 return realApplication; 127 } 128 129 @Override 130 @Implementation getResources()131 public Resources getResources() { 132 if (resources == null ) { 133 resources = ShadowResources.bind(new Resources(null, null, null), resourceLoader); 134 } 135 return resources; 136 } 137 138 @Implementation 139 @Override getContentResolver()140 public ContentResolver getContentResolver() { 141 if (contentResolver == null) { 142 contentResolver = new ContentResolver(realApplication) { 143 }; 144 } 145 return contentResolver; 146 } 147 148 @Implementation 149 @Override getSystemService(String name)150 public Object getSystemService(String name) { 151 if (name.equals(Context.LAYOUT_INFLATER_SERVICE)) { 152 return LayoutInflater.from(realApplication); 153 } else { 154 Object service = systemServices.get(name); 155 if (service == null) { 156 String serviceClassName = SYSTEM_SERVICE_MAP.get(name); 157 if (serviceClassName != null) { 158 try { 159 service = newInstanceOf(Class.forName(serviceClassName)); 160 } catch (ClassNotFoundException e) { 161 throw new RuntimeException(e); 162 } 163 systemServices.put(name, service); 164 } 165 } 166 return service; 167 } 168 } 169 170 @Implementation 171 @Override startActivity(Intent intent)172 public void startActivity(Intent intent) { 173 startedActivities.add(intent); 174 } 175 176 @Implementation 177 @Override startService(Intent intent)178 public ComponentName startService(Intent intent) { 179 startedServices.add(intent); 180 return new ComponentName("some.service.package", "SomeServiceName-FIXME"); 181 } 182 183 @Implementation 184 @Override stopService(Intent name)185 public boolean stopService(Intent name) { 186 stoppedServies.add(name); 187 188 return startedServices.contains(name); 189 } 190 setComponentNameAndServiceForBindService(ComponentName name, IBinder service)191 public void setComponentNameAndServiceForBindService(ComponentName name, IBinder service) { 192 this.componentNameForBindService = name; 193 this.serviceForBindService = service; 194 } 195 196 @Implementation bindService(Intent intent, final ServiceConnection serviceConnection, int i)197 public boolean bindService(Intent intent, final ServiceConnection serviceConnection, int i) { 198 if (unbindableActions.contains(intent.getAction())) { 199 return false; 200 } 201 startedServices.add(intent); 202 shadowOf(Looper.getMainLooper()).post(new Runnable() { 203 @Override 204 public void run() { 205 serviceConnection.onServiceConnected(componentNameForBindService, serviceForBindService); 206 } 207 }, 0); 208 return true; 209 } 210 211 @Implementation unbindService(final ServiceConnection serviceConnection)212 public void unbindService(final ServiceConnection serviceConnection) { 213 unboundServiceConnections.add(serviceConnection); 214 shadowOf(Looper.getMainLooper()).post(new Runnable() { 215 @Override 216 public void run() { 217 serviceConnection.onServiceDisconnected(componentNameForBindService); 218 } 219 }, 0); 220 } 221 getUnboundServiceConnections()222 public List<ServiceConnection> getUnboundServiceConnections() { 223 return unboundServiceConnections; 224 } 225 /** 226 * Consumes the most recent {@code Intent} started by {@link #startActivity(android.content.Intent)} and returns it. 227 * 228 * @return the most recently started {@code Intent} 229 */ 230 @Override getNextStartedActivity()231 public Intent getNextStartedActivity() { 232 if (startedActivities.isEmpty()) { 233 return null; 234 } else { 235 return startedActivities.remove(0); 236 } 237 } 238 239 /** 240 * Returns the most recent {@code Intent} started by {@link #startActivity(android.content.Intent)} without 241 * consuming it. 242 * 243 * @return the most recently started {@code Intent} 244 */ 245 @Override peekNextStartedActivity()246 public Intent peekNextStartedActivity() { 247 if (startedActivities.isEmpty()) { 248 return null; 249 } else { 250 return startedActivities.get(0); 251 } 252 } 253 254 /** 255 * Consumes the most recent {@code Intent} started by {@link #startService(android.content.Intent)} and returns it. 256 * 257 * @return the most recently started {@code Intent} 258 */ 259 @Override getNextStartedService()260 public Intent getNextStartedService() { 261 if (startedServices.isEmpty()) { 262 return null; 263 } else { 264 return startedServices.remove(0); 265 } 266 } 267 268 /** 269 * Returns the most recent {@code Intent} started by {@link #startService(android.content.Intent)} without 270 * consuming it. 271 * 272 * @return the most recently started {@code Intent} 273 */ 274 @Override peekNextStartedService()275 public Intent peekNextStartedService() { 276 if (startedServices.isEmpty()) { 277 return null; 278 } else { 279 return startedServices.get(0); 280 } 281 } 282 283 /** 284 * Clears all {@code Intent} started by {@link #startService(android.content.Intent)} 285 */ 286 @Override clearStartedServices()287 public void clearStartedServices() { 288 startedServices.clear(); 289 } 290 291 /** 292 * Consumes the {@code Intent} requested to stop a service by {@link #stopService(android.content.Intent)} 293 * from the bottom of the stack of stop requests. 294 */ 295 @Override getNextStoppedService()296 public Intent getNextStoppedService() { 297 if (stoppedServies.isEmpty()) { 298 return null; 299 } else { 300 return stoppedServies.remove(0); 301 } 302 } 303 304 /** 305 * Non-Android accessor (and a handy way to get a working {@code ResourceLoader} 306 * 307 * @return the {@code ResourceLoader} associated with this Application 308 */ getResourceLoader()309 public ResourceLoader getResourceLoader() { 310 return resourceLoader; 311 } 312 313 /** 314 * Broadcasts the {@code Intent} by iterating through the registered receivers, invoking their filters, and calling 315 * {@code onRecieve(Application, Intent)} as appropriate. Does not enqueue the {@code Intent} for later inspection. 316 * 317 * @param intent the {@code Intent} to broadcast 318 * todo: enqueue the Intent for later inspection 319 */ 320 @Override 321 @Implementation sendBroadcast(Intent intent)322 public void sendBroadcast(Intent intent) { 323 broadcastIntents.add(intent); 324 325 List<Wrapper> copy = new ArrayList<Wrapper>(); 326 copy.addAll(registeredReceivers); 327 for (Wrapper wrapper : copy) { 328 if (wrapper.intentFilter.matchAction(intent.getAction())) { 329 wrapper.broadcastReceiver.onReceive(realApplication, intent); 330 } 331 } 332 } 333 getBroadcastIntents()334 public List<Intent> getBroadcastIntents() { 335 return broadcastIntents; 336 } 337 338 @Implementation sendStickyBroadcast(Intent intent)339 public void sendStickyBroadcast(Intent intent) { 340 stickyIntents.put(intent.getAction(), intent); 341 sendBroadcast(intent); 342 } 343 344 /** 345 * Always returns {@code null} 346 * 347 * @return {@code null} 348 */ 349 @Override 350 @Implementation registerReceiver(BroadcastReceiver receiver, IntentFilter filter)351 public Intent registerReceiver(BroadcastReceiver receiver, IntentFilter filter) { 352 return registerReceiverWithContext(receiver, filter, realApplication); 353 } 354 registerReceiverWithContext(BroadcastReceiver receiver, IntentFilter filter, Context context)355 Intent registerReceiverWithContext(BroadcastReceiver receiver, IntentFilter filter, Context context) { 356 if (receiver != null) { 357 registeredReceivers.add(new Wrapper(receiver, filter, context)); 358 } 359 return getStickyIntent(filter); 360 } 361 getStickyIntent(IntentFilter filter)362 private Intent getStickyIntent(IntentFilter filter) { 363 for (Intent stickyIntent : stickyIntents.values()) { 364 String action = null; 365 for (int i = 0; i < filter.countActions(); i++) { 366 action = filter.getAction(i); 367 if (stickyIntent.getAction().equals(action)) { 368 return stickyIntent; 369 } 370 } 371 } 372 373 return null; 374 } 375 376 @Override 377 @Implementation unregisterReceiver(BroadcastReceiver broadcastReceiver)378 public void unregisterReceiver(BroadcastReceiver broadcastReceiver) { 379 boolean found = false; 380 Iterator<Wrapper> iterator = registeredReceivers.iterator(); 381 while (iterator.hasNext()) { 382 Wrapper wrapper = iterator.next(); 383 if (wrapper.broadcastReceiver == broadcastReceiver) { 384 iterator.remove(); 385 found = true; 386 } 387 } 388 if (!found) { 389 throw new IllegalArgumentException("Receiver not registered: " + broadcastReceiver); 390 } 391 } 392 393 /** 394 * Iterates through all of the registered receivers on this {@code Application} and if any of them match the given 395 * {@code Context} object throws a {@code RuntimeException} 396 * 397 * @param context the {@code Context} to check for on each of the remaining registered receivers 398 * @param type the type to report for the context if an exception is thrown 399 * @throws RuntimeException if there are any recievers registered with the given {@code Context} 400 */ assertNoBroadcastListenersRegistered(Context context, String type)401 public void assertNoBroadcastListenersRegistered(Context context, String type) { 402 for (Wrapper registeredReceiver : registeredReceivers) { 403 if (registeredReceiver.context == context) { 404 RuntimeException e = new IllegalStateException(type + " " + context + " leaked has leaked IntentReceiver " 405 + registeredReceiver.broadcastReceiver + " that was originally registered here. " + 406 "Are you missing a call to unregisterReceiver()?"); 407 e.setStackTrace(registeredReceiver.exception.getStackTrace()); 408 throw e; 409 } 410 } 411 } 412 assertNoBroadcastListenersOfActionRegistered(Context context, String action)413 public void assertNoBroadcastListenersOfActionRegistered(Context context, String action) { 414 for (Wrapper registeredReceiver : registeredReceivers) { 415 if (registeredReceiver.context == context) { 416 Iterator<String> actions = registeredReceiver.intentFilter.actionsIterator(); 417 while (actions.hasNext()) { 418 if (actions.next().equals(action)) { 419 RuntimeException e = new IllegalStateException("Unexpected BroadcastReceiver on " + context + 420 " with action " + action + " " 421 + registeredReceiver.broadcastReceiver + " that was originally registered here:"); 422 e.setStackTrace(registeredReceiver.exception.getStackTrace()); 423 throw e; 424 } 425 } 426 } 427 } 428 } 429 hasReceiverForIntent(Intent intent)430 public boolean hasReceiverForIntent(Intent intent) { 431 for (Wrapper wrapper : registeredReceivers) { 432 if (wrapper.intentFilter.matchAction(intent.getAction())) { 433 return true; 434 } 435 } 436 return false; 437 } 438 getReceiversForIntent(Intent intent)439 public List<BroadcastReceiver> getReceiversForIntent(Intent intent) { 440 ArrayList<BroadcastReceiver> broadcastReceivers = new ArrayList<BroadcastReceiver>(); 441 for (Wrapper wrapper : registeredReceivers) { 442 if (wrapper.intentFilter.matchAction(intent.getAction())) { 443 broadcastReceivers.add(wrapper.getBroadcastReceiver()); 444 } 445 } 446 return broadcastReceivers; 447 } 448 449 /** 450 * Non-Android accessor. 451 * 452 * @return list of {@link Wrapper}s for registered receivers 453 */ getRegisteredReceivers()454 public List<Wrapper> getRegisteredReceivers() { 455 return registeredReceivers; 456 } 457 458 /** 459 * Non-Android accessor. 460 * 461 * @return the layout inflater used by this {@code Application} 462 */ getLayoutInflater()463 public LayoutInflater getLayoutInflater() { 464 return layoutInflater; 465 } 466 467 /** 468 * Non-Android accessor. 469 * 470 * @return the app widget manager used by this {@code Application} 471 */ getAppWidgetManager()472 public AppWidgetManager getAppWidgetManager() { 473 return appWidgetManager; 474 } 475 getFakeHttpLayer()476 public FakeHttpLayer getFakeHttpLayer() { 477 return fakeHttpLayer; 478 } 479 480 @Override 481 @Implementation getMainLooper()482 public Looper getMainLooper() { 483 return mainLooper; 484 } 485 getSharedPreferenceMap()486 public Map<String, Map<String, Object>> getSharedPreferenceMap() { 487 return sharedPreferenceMap; 488 } 489 getLatestAlertDialog()490 public ShadowAlertDialog getLatestAlertDialog() { 491 return latestAlertDialog; 492 } 493 setLatestAlertDialog(ShadowAlertDialog latestAlertDialog)494 public void setLatestAlertDialog(ShadowAlertDialog latestAlertDialog) { 495 this.latestAlertDialog = latestAlertDialog; 496 } 497 getLatestDialog()498 public ShadowDialog getLatestDialog() { 499 return latestDialog; 500 } 501 setLatestDialog(ShadowDialog latestDialog)502 public void setLatestDialog(ShadowDialog latestDialog) { 503 this.latestDialog = latestDialog; 504 } 505 getBluetoothAdapter()506 public Object getBluetoothAdapter() { 507 return bluetoothAdapter; 508 } 509 declareActionUnbindable(String action)510 public void declareActionUnbindable(String action) { 511 unbindableActions.add(action); 512 } 513 setSystemService(String key, Object service)514 public void setSystemService(String key, Object service) { 515 systemServices.put(key, service); 516 } 517 518 public class Wrapper { 519 public BroadcastReceiver broadcastReceiver; 520 public IntentFilter intentFilter; 521 public Context context; 522 public Throwable exception; 523 Wrapper(BroadcastReceiver broadcastReceiver, IntentFilter intentFilter, Context context)524 public Wrapper(BroadcastReceiver broadcastReceiver, IntentFilter intentFilter, Context context) { 525 this.broadcastReceiver = broadcastReceiver; 526 this.intentFilter = intentFilter; 527 this.context = context; 528 exception = new Throwable(); 529 } 530 getBroadcastReceiver()531 public BroadcastReceiver getBroadcastReceiver() { 532 return broadcastReceiver; 533 } 534 getIntentFilter()535 public IntentFilter getIntentFilter() { 536 return intentFilter; 537 } 538 getContext()539 public Context getContext() { 540 return context; 541 } 542 } 543 } 544