def test_create_no_password(self):
user = ctera_filer_user.CteraFilerUser()
user.parameters = dict(username='******',
full_name='Admin',
email='*****@*****.**',
uid='uid')
self.assertRaises(CTERAException, user._ensure_present, None)
def _test_ensure_absent(is_present):
username = '******'
user = ctera_filer_user.CteraFilerUser()
user.parameters = dict(username=username)
user._ensure_absent(user.parameters if is_present else None)
if is_present:
user._ctera_filer.users.delete.assert_called_once_with(username)
else:
user._ctera_filer.users.delete.assert_not_called()
def test_get_user_exists(self):
expected_user_dict = dict(username='******',
password='******',
full_name='Admin',
email='*****@*****.**',
uid='uid')
user_object_dict = copy.deepcopy(expected_user_dict)
user_object_dict['fullName'] = user_object_dict.pop('full_name')
user = ctera_filer_user.CteraFilerUser()
user.parameters = dict(username='******')
user._ctera_filer.users.get = mock.MagicMock(
return_value=munch.Munch(user_object_dict))
self.assertDictEqual(expected_user_dict, user._get_user())
def _test__execute(is_present):
user = ctera_filer_user.CteraFilerUser()
user.parameters = dict(state='present' if is_present else 'absent')
user._get_user = mock.MagicMock(return_value=dict())
user._ensure_present = mock.MagicMock()
user._ensure_absent = mock.MagicMock()
user._execute()
if is_present:
user._ensure_present.assert_called_once_with(mock.ANY)
user._ensure_absent.assert_not_called()
else:
user._ensure_absent.assert_called_once_with(mock.ANY)
user._ensure_present.assert_not_called()
def _test_ensure_present(is_present, change_attributes):
current_attributes = dict(username='******',
password='******',
full_name='Admin',
email='*****@*****.**',
uid='uid')
desired_attributes = copy.deepcopy(current_attributes)
if change_attributes:
desired_attributes['full_name'] = 'Administrator'
user = ctera_filer_user.CteraFilerUser()
user.parameters = desired_attributes
user._ensure_present(current_attributes if is_present else None)
if is_present:
if change_attributes:
user._ctera_filer.users.modify.assert_called_with(
desired_attributes['username'],
full_name=desired_attributes['full_name'])
else:
user._ctera_filer.users.modify.assert_not_called()
else:
user._ctera_filer.users.add.assert_called_with(
**desired_attributes)
def test__get_user_failed(self):
user = ctera_filer_user.CteraFilerUser()
user.parameters = dict(username='******')
user._ctera_filer.users.get = mock.MagicMock(
side_effect=CTERAException(response=munch.Munch(code=401)))
self.assertRaises(CTERAException, user._get_user)
def test__get_user_doesnt_exist(self):
user = ctera_filer_user.CteraFilerUser()
user.parameters = dict(username='******')
user._ctera_filer.users.get = mock.MagicMock(
side_effect=CTERAException(response=munch.Munch(code=404)))
self.assertIsNone(user._get_user())