UserManagerTest.java revision 05191053545065c9c71afae173d3ab42a2d947fa 
1/*
2 * Copyright (C) 2011 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17package com.android.server.pm;
18
19import android.content.BroadcastReceiver;
20import android.content.Context;
21import android.content.Intent;
22import android.content.IntentFilter;
23import android.content.pm.UserInfo;
24import android.os.Bundle;
25import android.os.UserHandle;
26import android.os.UserManager;
27import android.test.AndroidTestCase;
28
29import java.util.ArrayList;
30import java.util.List;
31
32/** Test {@link UserManager} functionality. */
33public class UserManagerTest extends AndroidTestCase {
34
35    UserManager mUserManager = null;
36    Object mUserLock = new Object();
37
38    @Override
39    public void setUp() throws Exception {
40        mUserManager = (UserManager) getContext().getSystemService(Context.USER_SERVICE);
41        IntentFilter filter = new IntentFilter(Intent.ACTION_USER_REMOVED);
42        getContext().registerReceiver(new BroadcastReceiver() {
43            @Override
44            public void onReceive(Context context, Intent intent) {
45                synchronized (mUserLock) {
46                    mUserLock.notifyAll();
47                }
48            }
49        }, filter);
50    }
51
52    public void testHasPrimary() throws Exception {
53        assertTrue(findUser(0));
54    }
55
56    public void testAddUser() throws Exception {
57        UserInfo userInfo = mUserManager.createUser("Guest 1", UserInfo.FLAG_GUEST);
58        assertTrue(userInfo != null);
59
60        List<UserInfo> list = mUserManager.getUsers();
61        boolean found = false;
62        for (UserInfo user : list) {
63            if (user.id == userInfo.id && user.name.equals("Guest 1")
64                    && user.isGuest()
65                    && !user.isAdmin()
66                    && !user.isPrimary()) {
67                found = true;
68                Bundle restrictions = mUserManager.getUserRestrictions(user.getUserHandle());
69                assertFalse("New user should have DISALLOW_CONFIG_WIFI =false by default",
70                        restrictions.getBoolean(UserManager.DISALLOW_CONFIG_WIFI));
71            }
72        }
73        assertTrue(found);
74        removeUser(userInfo.id);
75    }
76
77    public void testAdd2Users() throws Exception {
78        UserInfo user1 = mUserManager.createUser("Guest 1", UserInfo.FLAG_GUEST);
79        UserInfo user2 = mUserManager.createUser("User 2", UserInfo.FLAG_ADMIN);
80
81        assertTrue(user1 != null);
82        assertTrue(user2 != null);
83
84        assertTrue(findUser(0));
85        assertTrue(findUser(user1.id));
86        assertTrue(findUser(user2.id));
87        removeUser(user1.id);
88        removeUser(user2.id);
89    }
90
91    public void testRemoveUser() throws Exception {
92        UserInfo userInfo = mUserManager.createUser("Guest 1", UserInfo.FLAG_GUEST);
93        removeUser(userInfo.id);
94
95        assertFalse(findUser(userInfo.id));
96    }
97
98    private boolean findUser(int id) {
99        List<UserInfo> list = mUserManager.getUsers();
100
101        for (UserInfo user : list) {
102            if (user.id == id) {
103                return true;
104            }
105        }
106        return false;
107    }
108
109    public void testSerialNumber() {
110        UserInfo user1 = mUserManager.createUser("User 1", UserInfo.FLAG_RESTRICTED);
111        int serialNumber1 = user1.serialNumber;
112        assertEquals(serialNumber1, mUserManager.getUserSerialNumber(user1.id));
113        assertEquals(user1.id, mUserManager.getUserHandle(serialNumber1));
114        removeUser(user1.id);
115        UserInfo user2 = mUserManager.createUser("User 2", UserInfo.FLAG_RESTRICTED);
116        int serialNumber2 = user2.serialNumber;
117        assertFalse(serialNumber1 == serialNumber2);
118        assertEquals(serialNumber2, mUserManager.getUserSerialNumber(user2.id));
119        assertEquals(user2.id, mUserManager.getUserHandle(serialNumber2));
120        removeUser(user2.id);
121    }
122
123    public void testMaxUsers() {
124        int N = UserManager.getMaxSupportedUsers();
125        int count = mUserManager.getUsers().size();
126        List<UserInfo> created = new ArrayList<UserInfo>();
127        // Create as many users as permitted and make sure creation passes
128        while (count < N) {
129            UserInfo ui = mUserManager.createUser("User " + count, 0);
130            assertNotNull(ui);
131            created.add(ui);
132            count++;
133        }
134        // Try to create one more user and make sure it fails
135        UserInfo extra = null;
136        assertNull(extra = mUserManager.createUser("One more", 0));
137        if (extra != null) {
138            removeUser(extra.id);
139        }
140        while (!created.isEmpty()) {
141            UserInfo user = created.remove(0);
142            removeUser(user.id);
143        }
144    }
145
146    public void testRestrictions() {
147        List<UserInfo> users = mUserManager.getUsers();
148        if (users.size() > 1) {
149            Bundle restrictions = new Bundle();
150            restrictions.putBoolean(UserManager.DISALLOW_INSTALL_APPS, true);
151            restrictions.putBoolean(UserManager.DISALLOW_CONFIG_WIFI, false);
152            mUserManager.setUserRestrictions(restrictions, new UserHandle(users.get(1).id));
153            Bundle stored = mUserManager.getUserRestrictions(new UserHandle(users.get(1).id));
154            assertEquals(stored.getBoolean(UserManager.DISALLOW_CONFIG_WIFI), false);
155            assertEquals(stored.getBoolean(UserManager.DISALLOW_UNINSTALL_APPS), false);
156            assertEquals(stored.getBoolean(UserManager.DISALLOW_INSTALL_APPS), true);
157        }
158    }
159
160    private void removeUser(int userId) {
161        synchronized (mUserLock) {
162            mUserManager.removeUser(userId);
163            while (mUserManager.getUserInfo(userId) != null) {
164                try {
165                    mUserLock.wait(1000);
166                } catch (InterruptedException ie) {
167                }
168            }
169        }
170    }
171
172}
173