Adding isDemoUser method in userManager

Change-Id: I0ad2c0e7bfac5eff5e5c95c3e533840921a67dc3

1 files changed, 11 insertions, 7 deletions

diff --git a/src/com/android/launcher3/compat/UserManagerCompatVN.java b/src/com/android/launcher3/compat/UserManagerCompatVN.java
index 771d141c3..be3232302 100644
--- a/src/com/android/launcher3/compat/UserManagerCompatVN.java
+++ b/src/com/android/launcher3/compat/UserManagerCompatVN.java
@@ -20,11 +20,11 @@ import android.annotation.TargetApi;
 import android.content.Context;
 import android.os.Build;
 
+import com.android.launcher3.Utilities;
+
 @TargetApi(Build.VERSION_CODES.N)
 public class UserManagerCompatVN extends UserManagerCompatVL {
 
-    private static final String TAG = "UserManagerCompatVN";
-
     UserManagerCompatVN(Context context) {
         super(context);
     }
@@ -36,11 +36,15 @@ public class UserManagerCompatVN extends UserManagerCompatVL {
 
     @Override
     public boolean isUserUnlocked(UserHandleCompat user) {
-        // TODO: Remove the try-catch block when the API permission has been relaxed (b/30475753)
-        try {
-            return mUserManager.isUserUnlocked(user.getUser());
-        } catch (RuntimeException e) {
-            return !isQuietModeEnabled(user);
+        return mUserManager.isUserUnlocked(user.getUser());
+    }
+
+    @Override
+    public boolean isDemoUser() {
+        if (Utilities.isNycMR1OrAbove()) {
+            return mUserManager.isDemoUser();
+        } else {
+            return super.isDemoUser();
         }
     }
 }