1 files changed, 119 insertions, 0 deletions
diff --git a/mobile/android/geckoview/src/androidTest/java/org/mozilla/geckoview/test/TrackingPermissionService.java b/mobile/android/geckoview/src/androidTest/java/org/mozilla/geckoview/test/TrackingPermissionService.java
new file mode 100644
index 0000000000..141849589e
--- /dev/null
+++ b/mobile/android/geckoview/src/androidTest/java/org/mozilla/geckoview/test/TrackingPermissionService.java
@@ -0,0 +1,119 @@
+package org.mozilla.geckoview.test;
+
+import android.content.Context;
+import android.content.Intent;
+import androidx.annotation.NonNull;
+import androidx.annotation.Nullable;
+import java.io.File;
+import java.util.List;
+import org.mozilla.gecko.util.GeckoBundle;
+import org.mozilla.geckoview.GeckoResult;
+import org.mozilla.geckoview.GeckoSession;
+import org.mozilla.geckoview.GeckoSession.PermissionDelegate;
+import org.mozilla.geckoview.GeckoSession.PermissionDelegate.ContentPermission;
+import org.mozilla.geckoview.GeckoSessionSettings;
+
+public class TrackingPermissionService extends TestRuntimeService {
+  public static final int MESSAGE_SET_TRACKING_PERMISSION = FIRST_SAFE_MESSAGE + 1;
+  public static final int MESSAGE_SET_PRIVATE_BROWSING_TRACKING_PERMISSION = FIRST_SAFE_MESSAGE + 2;
+  public static final int MESSAGE_GET_TRACKING_PERMISSION = FIRST_SAFE_MESSAGE + 3;
+
+  private ContentPermission mContentPermission;
+
+  @Override
+  protected GeckoSession createSession(final Intent intent) {
+    return new GeckoSession(
+        new GeckoSessionSettings.Builder()
+            .usePrivateMode(mTestData.getBoolean("privateMode"))
+            .build());
+  }
+
+  @Override
+  protected void onSessionReady(final GeckoSession session) {
+    session.setNavigationDelegate(
+        new GeckoSession.NavigationDelegate() {
+          @Override
+          public void onLocationChange(
+              final @NonNull GeckoSession session,
+              final @Nullable String url,
+              final @NonNull List<ContentPermission> perms) {
+            for (ContentPermission perm : perms) {
+              if (perm.permission == PermissionDelegate.PERMISSION_TRACKING) {
+                mContentPermission = perm;
+              }
+            }
+          }
+        });
+  }
+
+  @Override
+  protected GeckoResult<GeckoBundle> handleMessage(final int messageId, final GeckoBundle data) {
+    if (mContentPermission == null) {
+      throw new IllegalStateException("Content permission not received yet!");
+    }
+
+    switch (messageId) {
+      case MESSAGE_SET_TRACKING_PERMISSION:
+        {
+          final int permission = data.getInt("trackingPermission");
+          mRuntime.getStorageController().setPermission(mContentPermission, permission);
+          break;
+        }
+      case MESSAGE_SET_PRIVATE_BROWSING_TRACKING_PERMISSION:
+        {
+          final int permission = data.getInt("trackingPermission");
+          mRuntime
+              .getStorageController()
+              .setPrivateBrowsingPermanentPermission(mContentPermission, permission);
+          break;
+        }
+      case MESSAGE_GET_TRACKING_PERMISSION:
+        {
+          final GeckoBundle result = new GeckoBundle(1);
+          result.putInt("trackingPermission", mContentPermission.value);
+          return GeckoResult.fromValue(result);
+        }
+    }
+
+    return null;
+  }
+
+  public static class TrackingPermissionInstance
+      extends RuntimeInstance<TrackingPermissionService> {
+    public static GeckoBundle testData(boolean privateMode) {
+      GeckoBundle testData = new GeckoBundle(1);
+      testData.putBoolean("privateMode", privateMode);
+      return testData;
+    }
+
+    private TrackingPermissionInstance(
+        final Context context, final File profileFolder, final boolean privateMode) {
+      super(context, TrackingPermissionService.class, profileFolder, testData(privateMode));
+    }
+
+    public static TrackingPermissionInstance start(
+        final Context context, final File profileFolder, final boolean privateMode) {
+      TrackingPermissionInstance instance =
+          new TrackingPermissionInstance(context, profileFolder, privateMode);
+      instance.sendIntent();
+      return instance;
+    }
+
+    public GeckoResult<Integer> getTrackingPermission() {
+      return query(MESSAGE_GET_TRACKING_PERMISSION)
+          .map(bundle -> bundle.getInt("trackingPermission"));
+    }
+
+    public void setTrackingPermission(final int permission) {
+      final GeckoBundle bundle = new GeckoBundle(1);
+      bundle.putInt("trackingPermission", permission);
+      sendMessage(MESSAGE_SET_TRACKING_PERMISSION, bundle);
+    }
+
+    public void setPrivateBrowsingPermanentTrackingPermission(final int permission) {
+      final GeckoBundle bundle = new GeckoBundle(1);
+      bundle.putInt("trackingPermission", permission);
+      sendMessage(MESSAGE_SET_PRIVATE_BROWSING_TRACKING_PERMISSION, bundle);
+    }
+  }
+}