1 package com.xtremelabs.robolectric.matchers; 2 3 import android.app.Service; 4 import android.content.Context; 5 import android.content.ContextWrapper; 6 import android.content.Intent; 7 import com.xtremelabs.robolectric.shadows.ShadowIntent; 8 import org.hamcrest.Description; 9 import org.junit.internal.matchers.TypeSafeMatcher; 10 11 import java.util.Set; 12 13 import static com.xtremelabs.robolectric.Robolectric.getShadowApplication; 14 import static com.xtremelabs.robolectric.Robolectric.shadowOf; 15 16 public class StartedServiceMatcher extends TypeSafeMatcher<Context> { 17 private final Intent expectedIntent; 18 19 private String message; 20 StartedServiceMatcher(Intent expectedIntent)21 public StartedServiceMatcher(Intent expectedIntent) { 22 this.expectedIntent = expectedIntent; 23 } 24 StartedServiceMatcher(String packageName, Class<? extends Service> expectedServiceClass)25 public StartedServiceMatcher(String packageName, Class<? extends Service> expectedServiceClass) { 26 this(createIntent(packageName, expectedServiceClass)); 27 } 28 StartedServiceMatcher(Class<? extends Service> expectedServiceClass)29 public StartedServiceMatcher(Class<? extends Service> expectedServiceClass) { 30 this(createIntent(expectedServiceClass)); 31 } 32 StartedServiceMatcher(Class<? extends Service> expectedServiceClass, String expectedAction)33 public StartedServiceMatcher(Class<? extends Service> expectedServiceClass, String expectedAction) { 34 this(createIntent(expectedServiceClass)); 35 36 expectedIntent.setAction(expectedAction); 37 } 38 39 /** 40 * Check if the class of the intent and the keys of the intent's extras match 41 * 42 * @param actualContext 43 * @return 44 */ 45 @Override matchesSafely(Context actualContext)46 public boolean matchesSafely(Context actualContext) { 47 if (expectedIntent == null) { 48 message = "null intent (did you mean to expect null?)"; 49 return false; 50 } 51 52 String expected = expectedIntent.toString(); 53 message = "to start " + expected + ", but "; 54 55 Intent actualStartedIntent = shadowOf((ContextWrapper) actualContext).getNextStartedService(); 56 57 if (actualStartedIntent == null) { 58 message += "didn't start anything"; 59 return false; 60 } 61 62 ShadowIntent shadowIntent = shadowOf(actualStartedIntent); 63 64 //boolean intentsMatch = shadowOf(expectedIntent).realIntentEquals(shadowIntent); 65 // Test only that we are sending intent to the right service class 66 boolean intentsMatch = shadowOf(expectedIntent).getIntentClass().equals(shadowIntent.getIntentClass()); 67 if (!intentsMatch) { 68 message += "started " + actualStartedIntent; 69 } else { 70 // Test that both intent extras have the same keys 71 Set<String> keys = shadowIntent.getExtras().keySet(); 72 Set<String> expectedKeys = shadowOf(expectedIntent).getExtras().keySet(); 73 intentsMatch = keys.equals(expectedKeys); 74 if(!intentsMatch){ 75 message += "did not get the same extras keys"; 76 } 77 } 78 79 return intentsMatch; 80 } 81 82 @Override describeTo(Description description)83 public void describeTo(Description description) { 84 description.appendText(message); 85 } 86 createIntent(Class<? extends Service> serviceClass, String extraKey, String extraValue)87 public static Intent createIntent(Class<? extends Service> serviceClass, String extraKey, String extraValue) { 88 Intent intent = createIntent(serviceClass); 89 intent.putExtra(extraKey, extraValue); 90 return intent; 91 } 92 createIntent(Class<? extends Service> serviceClass, String action)93 public static Intent createIntent(Class<? extends Service> serviceClass, String action) { 94 Intent intent = createIntent(serviceClass); 95 intent.setAction(action); 96 return intent; 97 } 98 createIntent(Class<? extends Service> serviceClass)99 public static Intent createIntent(Class<? extends Service> serviceClass) { 100 Package pack = serviceClass.getPackage(); 101 String packageName = "android.service"; 102 // getPackage is returning null when run from tests 103 if(pack != null) { 104 pack.getName(); 105 } 106 return createIntent(packageName, serviceClass); 107 } 108 createIntent(String packageName, Class<? extends Service> serviceClass)109 public static Intent createIntent(String packageName, Class<? extends Service> serviceClass) { 110 Intent intent = new Intent(); 111 intent.setClassName(packageName, serviceClass.getName()); 112 intent.setClass(getShadowApplication().getApplicationContext(), serviceClass); 113 return intent; 114 } 115 } 116