def test_user_replace_avatar(self):
user_id = 1
old_avatar_file_id = 1
new_avatar_file_id = 2
user = MagicMock(spec=dir(User))
old_avatar = MagicMock(spec=dir(File))
expected_avatar = MagicMock(spec=dir(File))
user.id = user_id
user.avatar_file_id = old_avatar_file_id
old_avatar.id = old_avatar_file_id
expected_avatar.id = new_avatar_file_id
user_repository_mock.find_by_id.return_value = user
file_service_mock.add_file.return_value = expected_avatar
file_service_mock.get_file_by_id.return_value = old_avatar
user_service.set_avatar(user_id, avatar_file_data)
self.assertEqual(user.avatar_file_id, new_avatar_file_id)
file_service_mock.add_file.assert_called_once_with(
FileCategory.USER_AVATAR,
avatar_file_data, avatar_file_data.filename)
file_service_mock.get_file_by_id.assert_called_once_with(
old_avatar_file_id)
file_service_mock.delete_file.assert_called_once_with(old_avatar)
user_repository_mock.save.assert_called_once_with(user)
def test_user_has_no_avatar(self):
user_id = 1
avatar_file_id = None
user = MagicMock(spec=dir(User))
user.id = user_id
user.avatar_file_id = avatar_file_id
user_repository_mock.find_by_id.return_value = user
has_avatar = user_service.user_has_avatar(user_id)
self.assertEqual(has_avatar, False)
def test_user_remove_avatar(self):
user_id = 1
avatar_file_id = 1
user = MagicMock(spec=dir(User))
avatar = MagicMock(spec=dir(File))
user.id = user_id
user.avatar_file_id = avatar_file_id
avatar.id = avatar_file_id
user_repository_mock.find_by_id.return_value = user
file_service_mock.get_file_by_id.return_value = avatar
user_service.remove_avatar(user_id)
self.assertEqual(user.avatar_file_id, None)
file_service_mock.get_file_by_id.assert_called_once_with(
avatar_file_id)
file_service_mock.delete_file.assert_called_once_with(avatar)
user_repository_mock.save.assert_called_once_with(user)
