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])