def test_create(self, mock_creds, mock_epic_api, mock_uuid, mock_stdout): """Test that create CLI command invokes createHandle API method with the supplied mandatory params. """ mock_epic_api.return_value.createHandle.return_value = 'GREAT SUCCESS' mock_uuid.uuid1.return_value = 'UUID' self.args.location = 'http://www.testB2SafeCmd.com/1' self.args.checksum = '1cb285b' self.args.extratype = None self.args.loc10320 = None epicclient.create(self.args) mock_creds.assert_called_once mock_creds.assert_called_once_with(self.creds['store'], self.credpath) mock_creds.parse.assert_called_once mock_creds_instance = mock_creds.return_value mock_epic_api.assert_called_once mock_epic_api.assert_called_once_with(mock_creds_instance) mock_epic_api_instance = mock_epic_api.return_value mock_epic_api_instance.createHandle.assert_called_once mock_epic_api_instance.createHandle.assert_called_once_with( mock_creds_instance.prefix, 'http://www.testB2SafeCmd.com/1', '1cb285b', None, None, 'UUID') mock_stdout.write.assert_called_once mock_stdout.write.assert_called_once_with('GREAT SUCCESS') # Test None result mock_epic_api.return_value.createHandle.return_value = None mock_stdout.reset_mock() epicclient.create(self.args) mock_stdout.write.assert_called_once mock_stdout.write.assert_called_once_with('error')
def test_create_with_loc10320(self, mock_creds, mock_epic_api, mock_uuid): """Test that create CLI command invokes createHandle API method with the supplied mandatory params and the given 10320/LOC property. """ mock_epic_api.return_value.createHandle.return_value = 'GREAT SUCCESS' mock_uuid.uuid1.return_value = 'UUID' self.args.location = 'http://www.testB2SafeCmd.com/1' self.args.checksum = '1cb285b' self.args.extratype = None self.args.loc10320 = 'FOO_LOC' epicclient.create(self.args) mock_creds.assert_called_once mock_creds.assert_called_once_with(self.creds['store'], self.credpath) mock_creds.parse.assert_called_once mock_creds_instance = mock_creds.return_value mock_epic_api.assert_called_once mock_epic_api.assert_called_once_with(mock_creds_instance) mock_epic_api_instance = mock_epic_api.return_value mock_epic_api_instance.createHandle.assert_called_once mock_epic_api_instance.createHandle.assert_called_once_with( mock_creds_instance.prefix, 'http://www.testB2SafeCmd.com/1', '1cb285b', None, 'FOO_LOC'.split(';'), 'UUID')