Fix bug, before unregistering an activity check whether is the correc…

…t one
aitorvs · Dec 7, 2015 · 1fdf813 · 1fdf813
1 parent 7d8100f
commit 1fdf813
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/library/src/main/java/com/aitorvs/android/allowme/AllowMe.java b/library/src/main/java/com/aitorvs/android/allowme/AllowMe.java
@@ -24,6 +24,7 @@
 import android.support.annotation.NonNull;
 import android.support.v4.app.ActivityCompat;
 import android.support.v4.content.ContextCompat;
+import android.util.Log;
 
 import java.security.InvalidParameterException;
 import java.util.ArrayList;
@@ -32,6 +33,7 @@
 
 public class AllowMe {
 
+    private static final String TAG = AllowMe.class.getSimpleName();
     private static AllowMe instance;
 
     private Activity mActivity;
@@ -56,7 +58,11 @@ public static void registerActivity(@NonNull Activity activity) {
     }
 
     public static void unregisterActivity(@NonNull Activity activity) {
-        getInstance().mActivity = null;
+        if (activity == getInstance().mActivity) {
+            getInstance().mActivity = null;
+        } else {
+            Log.w(TAG, "unregisterActivity: Old activity is trying to unregister");
+        }
 
     }