def testIOProblem(self):
"""should raise an IOError"""
# mock user arguments
mock_helper = MockHelper()
mock_helper.lastname = ['prodon', 'tsetung']
# mock the zebra manager and its return values
zebra_manager = mock_helper.get_zebra_manager()
zebra_manager.get_all_users = MagicMock(return_value=[])
command = RetrieveUserIdCommand()
command.get_zebra_manager = MagicMock(return_value=zebra_manager)
# run the command
command._output = Mock()
self.assertRaises(IOError, command.run, mock_helper)
def testIOProblem(self):
"""should raise an IOError"""
# mock user arguments
mock_helper = MockHelper()
mock_helper.lastname = ['prodon', 'tsetung']
# mock the zebra manager and its return values
zebra_manager = mock_helper.get_zebra_manager()
zebra_manager.get_all_users = MagicMock(return_value=[])
command = RetrieveUserIdCommand()
command.get_zebra_manager = MagicMock(return_value=zebra_manager)
# run the command
command._output = Mock()
self.assertRaises(IOError, command.run, mock_helper)
def testMultipleUsers(self):
"""should return the correct user ids"""
# mock user arguments
mock_helper = MockHelper()
mock_helper.lastname = ['prodon', 'tsetung']
# mock the zebra manager and its return values
zebra_manager = mock_helper.get_zebra_manager()
zebra_manager.get_all_users = MagicMock(return_value=self.mock_users)
command = RetrieveUserIdCommand()
command.get_zebra_manager = MagicMock(return_value=zebra_manager)
# run the command
output_mock = command._output = Mock()
command.run(mock_helper)
expected_result = [self.mock_users[0], self.mock_users[2]]
self.assertEquals((expected_result, ['prodon', 'tsetung']), output_mock.call_args[0])
def testMultipleUsers(self):
"""should return the correct user ids"""
# mock user arguments
mock_helper = MockHelper()
mock_helper.lastname = ['prodon', 'tsetung']
# mock the zebra manager and its return values
zebra_manager = mock_helper.get_zebra_manager()
zebra_manager.get_all_users = MagicMock(return_value=self.mock_users)
command = RetrieveUserIdCommand()
command.get_zebra_manager = MagicMock(return_value=zebra_manager)
# run the command
output_mock = command._output = Mock()
command.run(mock_helper)
expected_result = [self.mock_users[0], self.mock_users[2]]
self.assertEquals((expected_result, ['prodon', 'tsetung']),
output_mock.call_args[0])
