Fix notification handling from inside a test apk.

The context of the target apk must be used for starting the
notification in this situation.

1 files changed, 3 insertions, 5 deletions

diff --git a/third_party/sl4a/src/main/java/com/google/android/mobly/snippet/SnippetRunner.java b/third_party/sl4a/src/main/java/com/google/android/mobly/snippet/SnippetRunner.java
index b3f5e8a..2276d2d 100644
--- a/third_party/sl4a/src/main/java/com/google/android/mobly/snippet/SnippetRunner.java
+++ b/third_party/sl4a/src/main/java/com/google/android/mobly/snippet/SnippetRunner.java
@@ -37,16 +37,14 @@ public class SnippetRunner extends AndroidJUnitRunner {
 
     private static final int NOTIFICATION_ID = NotificationIdFactory.create();
 
-    private Context mContext;
     private NotificationManager mNotificationManager;
     private int mServicePort;
     private Notification mNotification;
 
     @Override
     public void onCreate(Bundle arguments) {
-        mContext = getContext();
         mNotificationManager = (NotificationManager)
-                mContext.getSystemService(Context.NOTIFICATION_SERVICE);
+                getTargetContext().getSystemService(Context.NOTIFICATION_SERVICE);
         String servicePort = arguments.getString(ARG_PORT);
         if (servicePort == null) {
             throw new IllegalArgumentException("\"--e port <port>\" was not specified");
@@ -70,7 +68,7 @@ public class SnippetRunner extends AndroidJUnitRunner {
     }
 
     private void startServer() {
-        AndroidProxy androidProxy = new AndroidProxy(mContext);
+        AndroidProxy androidProxy = new AndroidProxy(getContext());
         if (androidProxy.startLocal(mServicePort) == null) {
             throw new RuntimeException("Failed to start server on port " + mServicePort);
         }
@@ -79,7 +77,7 @@ public class SnippetRunner extends AndroidJUnitRunner {
     }
 
     private void createNotification() {
-        Notification.Builder builder = new Notification.Builder(mContext);
+        Notification.Builder builder = new Notification.Builder(getTargetContext());
         builder.setSmallIcon(android.R.drawable.btn_star)
                 .setTicker(null)
                 .setWhen(System.currentTimeMillis())