1 files changed, 12 insertions, 0 deletions
diff --git a/app/src/test/java/se/leap/bitmaskclient/testutils/MockHelper.java b/app/src/test/java/se/leap/bitmaskclient/testutils/MockHelper.java
index fa9f9252..24801b58 100644
--- a/app/src/test/java/se/leap/bitmaskclient/testutils/MockHelper.java
+++ b/app/src/test/java/se/leap/bitmaskclient/testutils/MockHelper.java
@@ -1,7 +1,10 @@
 package se.leap.bitmaskclient.testutils;
 
+import android.content.Context;
 import android.content.Intent;
 import android.content.SharedPreferences;
+import android.content.pm.PackageInfo;
+import android.content.pm.PackageManager;
 import android.content.res.Resources;
 import android.os.Bundle;
 import android.os.Parcelable;
@@ -460,4 +463,13 @@ public class MockHelper {
                 thenReturn("Bitmask");
         return mockedResources;
     }
+
+    public static Context mockContext() throws PackageManager.NameNotFoundException {
+        Context context = mock(Context.class, RETURNS_DEEP_STUBS);
+        when(context.getPackageName()).thenReturn("se.leap.bitmaskclient");
+        PackageInfo mockPackageInfo = new PackageInfo();
+        mockPackageInfo.versionName = "0.9.10";
+        when(context.getPackageManager().getPackageInfo(anyString(), anyInt())).thenReturn(mockPackageInfo);
+        return context;
+    }
 }