def test_get_user(self): get_response = 'Success' self._init_filer(get_response=get_response) ret = users.Users(self._filer).get(self._username) self._filer.get.assert_called_once_with('/config/auth/users/' + self._username) self.assertEqual(ret, get_response)
def test_modify_user_not_found(self): self._filer.get = mock.MagicMock( side_effect=exception.CTERAException()) with self.assertRaises(exception.CTERAException) as error: users.Users(self._filer).modify(self._username) self._filer.get.assert_called_once_with('/config/auth/users/' + self._username) self.assertEqual('Failed to get the user', error.exception.message)
def test_delete_user(self): user = self._get_user_object() self._init_filer(delete_response=user) ret = users.Users(self._filer).delete(self._username) self._filer.delete.assert_called_once_with('/config/auth/users/' + self._username) self.assertEqual(user.username, ret.username)
def test_first_user_already_exists(self): info = Object() info.isfirstlogin = False self._init_filer(get_response=info) users.Users(self._filer).add_first_user(self._username, self._password) self._filer.get.assert_called_once_with('/nosession/logininfo') self._filer.post.assert_not_called() self._login_mock.assert_called_once_with(self._username, self._password)
def test_add_user(self): add_response = '/config/auth/users/' + self._username self._init_filer(add_response=add_response) ret = users.Users(self._filer).add(self._username, self._password) self._filer.add.assert_called_once_with('/config/auth/users', mock.ANY) expected_param = self._get_user_object() actual_param = self._filer.add.call_args[0][1] self._assert_equal_objects(actual_param, expected_param) self.assertEqual(ret, add_response)
def test_add_first_user(self): info = Object() info.isfirstlogin = True self._init_filer(get_response=info) users.Users(self._filer).add_first_user(self._username, self._password) self._filer.get.assert_called_once_with('/nosession/logininfo') self._filer.post.assert_called_once_with('/nosession/createfirstuser', mock.ANY) expected_param = self._get_first_user_object('') actual_param = self._filer.post.call_args[0][1] self._assert_equal_objects(actual_param, expected_param) self._login_mock.assert_called_once_with(self._username, self._password)
def test_modify_user(self): get_response = Object() get_response.username = self._username put_response = 'Success' self._init_filer(get_response=get_response, put_response=put_response) ret = users.Users(self._filer).modify(self._username, self._password, self._full_name, self._email, self._uid) self._filer.get.assert_called_once_with('/config/auth/users/' + self._username) self._filer.put.assert_called_once_with( '/config/auth/users/' + self._username, mock.ANY) expected_param = self._get_user_object(self._full_name, self._email, self._uid) actual_param = self._filer.put.call_args[0][1] self._assert_equal_objects(actual_param, expected_param) self.assertEqual(ret, put_response)
def test_modify_user_update_failed(self): get_response = Object() get_response.username = self._username self._init_filer(get_response=get_response) self._filer.put = mock.MagicMock( side_effect=exception.CTERAException()) with self.assertRaises(exception.CTERAException) as error: users.Users(self._filer).modify(self._username, self._password, self._full_name, self._email, self._uid) self._filer.get.assert_called_once_with('/config/auth/users/' + self._username) self._filer.put.assert_called_once_with( '/config/auth/users/' + self._username, mock.ANY) expected_param = self._get_user_object(self._full_name, self._email, self._uid) actual_param = self._filer.put.call_args[0][1] self._assert_equal_objects(actual_param, expected_param) self.assertEqual('Failed to modify user', error.exception.message)
def test_add_user_raise(self): expected_exception = exception.CTERAException() self._filer.add = mock.MagicMock(side_effect=expected_exception) with self.assertRaises(exception.CTERAException) as error: users.Users(self._filer).add(self._username, self._password) self.assertEqual('User creation failed', error.exception.message)
def test_delete_user_raise(self): expected_exception = exception.CTERAException() self._filer.delete = mock.MagicMock(side_effect=expected_exception) with self.assertRaises(exception.CTERAException) as error: users.Users(self._filer).delete(self._username) self.assertEqual('User deletion failed', error.exception.message)