diff --git a/client/src/main/java/org/apache/sqoop/client/SqoopClient.java b/client/src/main/java/org/apache/sqoop/client/SqoopClient.java index f037399e..1cf549ea 100644 --- a/client/src/main/java/org/apache/sqoop/client/SqoopClient.java +++ b/client/src/main/java/org/apache/sqoop/client/SqoopClient.java @@ -313,8 +313,8 @@ public Status saveLink(MLink link) { * @param link link that should be updated * @return */ - public Status updateLink(MLink link) { - return applyLinkValidations(resourceRequests.updateLink(link), link); + public Status updateLink(MLink link, String oldLinkName) { + return applyLinkValidations(resourceRequests.updateLink(link, oldLinkName), link); } /** @@ -403,8 +403,8 @@ public Status saveJob(MJob job) { * @param job Job that should be updated * @return */ - public Status updateJob(MJob job) { - return applyJobValidations(resourceRequests.updateJob(job), job); + public Status updateJob(MJob job, String oldJobName) { + return applyJobValidations(resourceRequests.updateJob(job, oldJobName), job); } /** diff --git a/client/src/main/java/org/apache/sqoop/client/request/JobResourceRequest.java b/client/src/main/java/org/apache/sqoop/client/request/JobResourceRequest.java index ab9c93be..3a0abefd 100644 --- a/client/src/main/java/org/apache/sqoop/client/request/JobResourceRequest.java +++ b/client/src/main/java/org/apache/sqoop/client/request/JobResourceRequest.java @@ -86,11 +86,11 @@ public ValidationResultBean create(String serverUrl, MJob job) { return validationResultBean; } - public ValidationResultBean update(String serverUrl, MJob job) { + public ValidationResultBean update(String serverUrl, MJob job, String oldJobName) { JobBean jobBean = new JobBean(job); // Extract all config inputs including sensitive inputs JSONObject jobJson = jobBean.extract(false); - String response = super.put(serverUrl + RESOURCE + UrlSafeUtils.urlPathEncode(job.getName()), + String response = super.put(serverUrl + RESOURCE + UrlSafeUtils.urlPathEncode(oldJobName), jobJson.toJSONString()); ValidationResultBean validationBean = new ValidationResultBean(); validationBean.restore(JSONUtils.parse(response)); diff --git a/client/src/main/java/org/apache/sqoop/client/request/LinkResourceRequest.java b/client/src/main/java/org/apache/sqoop/client/request/LinkResourceRequest.java index b3262bdc..2b19dacd 100644 --- a/client/src/main/java/org/apache/sqoop/client/request/LinkResourceRequest.java +++ b/client/src/main/java/org/apache/sqoop/client/request/LinkResourceRequest.java @@ -72,11 +72,11 @@ public ValidationResultBean create(String serverUrl, MLink link) { return validationBean; } - public ValidationResultBean update(String serverUrl, MLink link) { + public ValidationResultBean update(String serverUrl, MLink link, String oldLinkName) { LinkBean linkBean = new LinkBean(link); // Extract all config inputs including sensitive inputs JSONObject linkJson = linkBean.extract(false); - String response = super.put(serverUrl + LINK_RESOURCE + UrlSafeUtils.urlPathEncode(link.getName()), + String response = super.put(serverUrl + LINK_RESOURCE + UrlSafeUtils.urlPathEncode(oldLinkName), linkJson.toJSONString()); ValidationResultBean validationBean = new ValidationResultBean(); validationBean.restore(JSONUtils.parse(response)); diff --git a/client/src/main/java/org/apache/sqoop/client/request/SqoopResourceRequests.java b/client/src/main/java/org/apache/sqoop/client/request/SqoopResourceRequests.java index 810b9133..3109718d 100644 --- a/client/src/main/java/org/apache/sqoop/client/request/SqoopResourceRequests.java +++ b/client/src/main/java/org/apache/sqoop/client/request/SqoopResourceRequests.java @@ -126,8 +126,8 @@ public LinkBean readLink(String linkArg) { return getLinkResourceRequest().read(serverUrl, linkArg); } - public ValidationResultBean updateLink(MLink link) { - return getLinkResourceRequest().update(serverUrl, link); + public ValidationResultBean updateLink(MLink link, String oldLinkName) { + return getLinkResourceRequest().update(serverUrl, link, oldLinkName); } public void enableLink(String lArg, Boolean enabled) { @@ -150,8 +150,8 @@ public JobBean readJobsByConnector(String cArg) { return getJobResourceRequest().readByConnector(serverUrl, cArg); } - public ValidationResultBean updateJob(MJob job) { - return getJobResourceRequest().update(serverUrl, job); + public ValidationResultBean updateJob(MJob job, String oldJobName) { + return getJobResourceRequest().update(serverUrl, job, oldJobName); } public void enableJob(String jArg, Boolean enabled) { diff --git a/core/src/main/java/org/apache/sqoop/repository/JdbcRepository.java b/core/src/main/java/org/apache/sqoop/repository/JdbcRepository.java index 7a9f1c65..1b7cd2eb 100644 --- a/core/src/main/java/org/apache/sqoop/repository/JdbcRepository.java +++ b/core/src/main/java/org/apache/sqoop/repository/JdbcRepository.java @@ -293,9 +293,8 @@ public Object doIt(Connection conn) { if (!link.hasPersistenceId()) { throw new SqoopException(RepositoryError.JDBCREPO_0016); } - if (!handler.existsLink(link.getName(), conn)) { - throw new SqoopException(RepositoryError.JDBCREPO_0017, "Invalid name: " - + link.getName()); + if (!handler.existsLink(link.getPersistenceId(), conn)) { + throw new SqoopException(RepositoryError.JDBCREPO_0016); } handler.updateLink(link, conn); @@ -438,9 +437,8 @@ public Object doIt(Connection conn) { if(!job.hasPersistenceId()) { throw new SqoopException(RepositoryError.JDBCREPO_0019); } - if(!handler.existsJob(job.getName(), conn)) { - throw new SqoopException(RepositoryError.JDBCREPO_0020, - "Invalid id: " + job.getPersistenceId()); + if(!handler.existsJob(job.getPersistenceId(), conn)) { + throw new SqoopException(RepositoryError.JDBCREPO_0019); } handler.updateJob(job, conn); diff --git a/core/src/main/java/org/apache/sqoop/repository/JdbcRepositoryHandler.java b/core/src/main/java/org/apache/sqoop/repository/JdbcRepositoryHandler.java index 82e29c5d..7047be96 100644 --- a/core/src/main/java/org/apache/sqoop/repository/JdbcRepositoryHandler.java +++ b/core/src/main/java/org/apache/sqoop/repository/JdbcRepositoryHandler.java @@ -232,6 +232,15 @@ public abstract class JdbcRepositoryHandler { */ public abstract boolean existsLink(String linkName, Connection conn); + /** + * Check if given link exists in repository. + * + * @param linkId Link id + * @param conn Connection to the repository + * @return True if the link exists + */ + public abstract boolean existsLink(long linkId, Connection conn); + /** * Check if given link is referenced somewhere and thus can't * be removed. @@ -322,6 +331,15 @@ public abstract class JdbcRepositoryHandler { */ public abstract boolean existsJob(String jobName, Connection conn); + /** + * Check if given job exists in the repository. + * + * @param jobId Job id + * @param conn Connection to the repository + * @return True if the job exists + */ + public abstract boolean existsJob(long jobId, Connection conn); + /** * Check if given job is referenced somewhere and thus can't * be removed. diff --git a/core/src/test/java/org/apache/sqoop/repository/TestJdbcRepository.java b/core/src/test/java/org/apache/sqoop/repository/TestJdbcRepository.java index 0ffc645b..d26ce719 100644 --- a/core/src/test/java/org/apache/sqoop/repository/TestJdbcRepository.java +++ b/core/src/test/java/org/apache/sqoop/repository/TestJdbcRepository.java @@ -558,8 +558,7 @@ public void testConnectorConfigUpgradeHandlerWithUpdateLinkError() { doNothing().when(repoHandlerMock).upgradeConnectorAndConfigs(any(MConnector.class), any(Connection.class)); doReturn(true).when(repoHandlerMock).existsLink(anyString(), any(Connection.class)); - SqoopException exception = new SqoopException(RepositoryError.JDBCREPO_0000, - "update link error."); + SqoopException exception = new SqoopException(RepositoryError.JDBCREPO_0016); doThrow(exception).when(repoHandlerMock).updateLink(any(MLink.class), any(Connection.class)); try { @@ -571,9 +570,6 @@ public void testConnectorConfigUpgradeHandlerWithUpdateLinkError() { verify(repoHandlerMock, times(2)).deleteJobInputs(anyString(), any(Connection.class)); verify(repoHandlerMock, times(2)).deleteLinkInputs(anyString(), any(Connection.class)); verify(repoHandlerMock, times(1)).upgradeConnectorAndConfigs(any(MConnector.class), any(Connection.class)); - verify(repoHandlerMock, times(1)).existsLink(anyString(), any(Connection.class)); - verify(repoHandlerMock, times(1)).updateLink(any(MLink.class), any(Connection.class)); - verifyNoMoreInteractions(repoHandlerMock); return ; } @@ -607,8 +603,7 @@ public void testConnectorConfigUpgradeHandlerWithUpdateJobError() { doReturn(true).when(repoHandlerMock).existsLink(anyString(), any(Connection.class)); doReturn(true).when(repoHandlerMock).existsJob(anyString(), any(Connection.class)); - SqoopException exception = new SqoopException(RepositoryError.JDBCREPO_0000, - "update job error."); + SqoopException exception = new SqoopException(RepositoryError.JDBCREPO_0016); doThrow(exception).when(repoHandlerMock).updateJob(any(MJob.class), any(Connection.class)); try { @@ -620,10 +615,7 @@ public void testConnectorConfigUpgradeHandlerWithUpdateJobError() { verify(repoHandlerMock, times(2)).deleteJobInputs(anyString(), any(Connection.class)); verify(repoHandlerMock, times(2)).deleteLinkInputs(anyString(), any(Connection.class)); verify(repoHandlerMock, times(1)).upgradeConnectorAndConfigs(any(MConnector.class), any(Connection.class)); - verify(repoHandlerMock, times(2)).existsLink(anyString(), any(Connection.class)); - verify(repoHandlerMock, times(2)).updateLink(any(MLink.class), any(Connection.class)); - verify(repoHandlerMock, times(1)).existsJob(anyString(), any(Connection.class)); - verify(repoHandlerMock, times(1)).updateJob(any(MJob.class), any(Connection.class)); + verify(repoHandlerMock, times(1)).existsLink(anyLong(), any(Connection.class)); verifyNoMoreInteractions(repoHandlerMock); return ; } @@ -740,8 +732,7 @@ public void testDriverConfigUpgradeHandlerWithUpdateJobError() { doNothing().when(repoHandlerMock).upgradeDriverAndConfigs(any(MDriver.class), any(Connection.class)); doReturn(true).when(repoHandlerMock).existsJob(anyString(), any(Connection.class)); - SqoopException exception = new SqoopException(RepositoryError.JDBCREPO_0000, - "update job error."); + SqoopException exception = new SqoopException(RepositoryError.JDBCREPO_0019); doThrow(exception).when(repoHandlerMock).updateJob(any(MJob.class), any(Connection.class)); try { @@ -751,9 +742,7 @@ public void testDriverConfigUpgradeHandlerWithUpdateJobError() { verify(repoHandlerMock, times(1)).findJobs(any(Connection.class)); verify(repoHandlerMock, times(2)).deleteJobInputs(anyString(), any(Connection.class)); verify(repoHandlerMock, times(1)).upgradeDriverAndConfigs(any(MDriver.class), any(Connection.class)); - verify(repoHandlerMock, times(1)).existsJob(anyString(), any(Connection.class)); - verify(repoHandlerMock, times(1)).updateJob(any(MJob.class), any(Connection.class)); - verifyNoMoreInteractions(repoHandlerMock); + verify(repoHandlerMock, times(1)).existsJob(anyLong(), any(Connection.class)); return ; } diff --git a/repository/repository-common/src/main/java/org/apache/sqoop/repository/common/CommonRepositoryHandler.java b/repository/repository-common/src/main/java/org/apache/sqoop/repository/common/CommonRepositoryHandler.java index efbd5331..5490324c 100644 --- a/repository/repository-common/src/main/java/org/apache/sqoop/repository/common/CommonRepositoryHandler.java +++ b/repository/repository-common/src/main/java/org/apache/sqoop/repository/common/CommonRepositoryHandler.java @@ -462,6 +462,26 @@ public boolean existsLink(String linkName, Connection conn) { } } + /** + * {@inheritDoc} + */ + @Override + public boolean existsLink(long linkId, Connection conn) { + try (PreparedStatement stmt = conn.prepareStatement(crudQueries.getStmtSelectLinkCheckById())) { + stmt.setLong(1, linkId); + try (ResultSet rs = stmt.executeQuery()) { + + // Should be always valid in query with count + rs.next(); + + return rs.getLong(1) == 1; + } + } catch (SQLException ex) { + logException(ex, linkId); + throw new SqoopException(CommonRepositoryError.COMMON_0022, ex); + } + } + /** * {@inheritDoc} */ @@ -751,6 +771,26 @@ public void updateJob(MJob job, Connection conn) { } } + /** + * {@inheritDoc} + */ + @Override + public boolean existsJob(long jobId, Connection conn) { + try (PreparedStatement stmt = conn.prepareStatement(crudQueries.getStmtSelectJobCheckById())) { + stmt.setLong(1, jobId); + try (ResultSet rs = stmt.executeQuery()) { + + // Should be always valid in query with count + rs.next(); + + return rs.getLong(1) == 1; + } + } catch (SQLException ex) { + logException(ex, jobId); + throw new SqoopException(CommonRepositoryError.COMMON_0026, ex); + } + } + /** * {@inheritDoc} */ diff --git a/repository/repository-common/src/main/java/org/apache/sqoop/repository/common/CommonRepositoryInsertUpdateDeleteSelectQuery.java b/repository/repository-common/src/main/java/org/apache/sqoop/repository/common/CommonRepositoryInsertUpdateDeleteSelectQuery.java index e1e23e27..58404d73 100644 --- a/repository/repository-common/src/main/java/org/apache/sqoop/repository/common/CommonRepositoryInsertUpdateDeleteSelectQuery.java +++ b/repository/repository-common/src/main/java/org/apache/sqoop/repository/common/CommonRepositoryInsertUpdateDeleteSelectQuery.java @@ -364,6 +364,11 @@ public class CommonRepositoryInsertUpdateDeleteSelectQuery { "SELECT count(*) FROM " + CommonRepoUtils.getTableName(SCHEMA_SQOOP, TABLE_SQ_LINK_NAME) + " WHERE " + CommonRepoUtils.escapeColumnName(COLUMN_SQ_LNK_NAME) + " = ?"; + // DML: Check if given link exists + private static final String STMT_SELECT_LINK_CHECK_BY_ID = + "SELECT count(*) FROM " + CommonRepoUtils.getTableName(SCHEMA_SQOOP, TABLE_SQ_LINK_NAME) + + " WHERE " + CommonRepoUtils.escapeColumnName(COLUMN_SQ_LNK_ID) + " = ?"; + /** * *******JOB TABLE ************* */ @@ -419,6 +424,11 @@ public class CommonRepositoryInsertUpdateDeleteSelectQuery { "SELECT count(*) FROM " + CommonRepoUtils.getTableName(SCHEMA_SQOOP, TABLE_SQ_JOB_NAME) + " WHERE " + CommonRepoUtils.escapeColumnName(COLUMN_SQB_NAME) + " = ?"; + // DML: Check if given job exists + private static final String STMT_SELECT_JOB_CHECK_BY_ID = + "SELECT count(*) FROM " + CommonRepoUtils.getTableName(SCHEMA_SQOOP, TABLE_SQ_JOB_NAME) + + " WHERE " + CommonRepoUtils.escapeColumnName(COLUMN_SQB_ID) + " = ?"; + // DML: Check if there are jobs for given link private static final String STMT_SELECT_JOBS_FOR_LINK_CHECK = "SELECT SUM(CNT) FROM (" @@ -841,6 +851,10 @@ public String getStmtSelectLinkCheckByName() { return STMT_SELECT_LINK_CHECK_BY_NAME; } + public String getStmtSelectLinkCheckById() { + return STMT_SELECT_LINK_CHECK_BY_ID; + } + public String getStmtInsertJob() { return STMT_INSERT_JOB; } @@ -869,6 +883,10 @@ public String getStmtSelectJobCheckByName() { return STMT_SELECT_JOB_CHECK_BY_NAME; } + public String getStmtSelectJobCheckById() { + return STMT_SELECT_JOB_CHECK_BY_ID; + } + public String getStmtSelectJobsForLinkCheck() { return STMT_SELECT_JOBS_FOR_LINK_CHECK; } diff --git a/server/src/main/java/org/apache/sqoop/handler/JobRequestHandler.java b/server/src/main/java/org/apache/sqoop/handler/JobRequestHandler.java index 7fa6a3b0..b2fce4f5 100644 --- a/server/src/main/java/org/apache/sqoop/handler/JobRequestHandler.java +++ b/server/src/main/java/org/apache/sqoop/handler/JobRequestHandler.java @@ -179,13 +179,14 @@ private JsonBean createUpdateJob(RequestContext ctx, boolean create) { // Job object MJob postedJob = jobs.get(0); + String oldJobName = ctx.getLastURLElement(); // Authorization check if (create) { AuthorizationEngine.createJob(ctx.getUserName(), postedJob.getFromLinkName(), postedJob.getToLinkName()); } else { AuthorizationEngine.updateJob(ctx.getUserName(), postedJob.getFromLinkName(), postedJob.getToLinkName(), - postedJob.getName()); + oldJobName); } // Verify that user is not trying to spoof us @@ -203,8 +204,7 @@ private JsonBean createUpdateJob(RequestContext ctx, boolean create) { // if update get the job id from the request URI if (!create) { - String jobIdentifier = ctx.getLastURLElement(); - MJob existingJob = HandlerUtils.getJobFromIdentifier(jobIdentifier); + MJob existingJob = HandlerUtils.getJobFromIdentifier(oldJobName); if (postedJob.getPersistenceId() == MPersistableEntity.PERSISTANCE_ID_DEFAULT) { postedJob.setPersistenceId(existingJob.getPersistenceId()); } diff --git a/server/src/main/java/org/apache/sqoop/handler/LinkRequestHandler.java b/server/src/main/java/org/apache/sqoop/handler/LinkRequestHandler.java index 5b1258f9..9f6ed901 100644 --- a/server/src/main/java/org/apache/sqoop/handler/LinkRequestHandler.java +++ b/server/src/main/java/org/apache/sqoop/handler/LinkRequestHandler.java @@ -131,6 +131,7 @@ private JsonBean createUpdateLink(RequestContext ctx, boolean create) { MLink postedLink = links.get(0); MConnector mConnector = HandlerUtils.getConnectorFromConnectorName(postedLink.getConnectorName()); + String oldLinkName = ctx.getLastURLElement(); // Authorization check if (create) { @@ -138,7 +139,7 @@ private JsonBean createUpdateLink(RequestContext ctx, boolean create) { mConnector.getUniqueName()); } else { AuthorizationEngine.updateLink(ctx.getUserName(), mConnector.getUniqueName(), - postedLink.getName()); + oldLinkName); } MLinkConfig linkConfig = ConnectorManager.getInstance() @@ -148,8 +149,7 @@ private JsonBean createUpdateLink(RequestContext ctx, boolean create) { } // if update get the link id from the request URI if (!create) { - String linkName = ctx.getLastURLElement(); - MLink existingLink = repository.findLink(linkName); + MLink existingLink = repository.findLink(oldLinkName); if (postedLink.getPersistenceId() == MPersistableEntity.PERSISTANCE_ID_DEFAULT) { postedLink.setPersistenceId(existingLink.getPersistenceId()); } diff --git a/shell/src/main/java/org/apache/sqoop/shell/UpdateJobFunction.java b/shell/src/main/java/org/apache/sqoop/shell/UpdateJobFunction.java index 0b6ff01d..3675445f 100644 --- a/shell/src/main/java/org/apache/sqoop/shell/UpdateJobFunction.java +++ b/shell/src/main/java/org/apache/sqoop/shell/UpdateJobFunction.java @@ -67,7 +67,7 @@ private Status updateJob(String jobArg, List args, boolean isInteractive // TODO(SQOOP-1634): using from/to and driver config id, this call can be avoided MJob job = client.getJob(jobArg); - + String oldJobName = job.getName(); ResourceBundle fromConnectorBundle = client.getConnectorConfigBundle( job.getFromConnectorName()); ResourceBundle toConnectorBundle = client.getConnectorConfigBundle( @@ -91,14 +91,14 @@ private Status updateJob(String jobArg, List args, boolean isInteractive } // Try to create - status = client.updateJob(job); + status = client.updateJob(job, oldJobName); } while(!status.canProceed()); } else { JobDynamicConfigOptions options = new JobDynamicConfigOptions(); options.prepareOptions(job); CommandLine line = ConfigOptions.parseOptions(options, 0, args, false); if (fillJob(line, job)) { - status = client.updateJob(job); + status = client.updateJob(job, oldJobName); if (!status.canProceed()) { printJobValidationMessages(job); return status; diff --git a/shell/src/main/java/org/apache/sqoop/shell/UpdateLinkFunction.java b/shell/src/main/java/org/apache/sqoop/shell/UpdateLinkFunction.java index a72b6ade..4119bd18 100644 --- a/shell/src/main/java/org/apache/sqoop/shell/UpdateLinkFunction.java +++ b/shell/src/main/java/org/apache/sqoop/shell/UpdateLinkFunction.java @@ -65,6 +65,7 @@ private Status updateLink(String linkArg, List args, boolean isInteracti // TODO(SQOOP-1634): using link config id, this call can be avoided MLink link = client.getLink(linkArg); + String oldLinkName = link.getName(); ResourceBundle connectorLinkConfigBundle = client.getConnectorConfigBundle(link.getConnectorName()); Status status = Status.OK; @@ -84,14 +85,14 @@ private Status updateLink(String linkArg, List args, boolean isInteracti } // Try to create - status = client.updateLink(link); + status = client.updateLink(link, oldLinkName); } while(!status.canProceed()); } else { LinkDynamicConfigOptions options = new LinkDynamicConfigOptions(); options.prepareOptions(link); CommandLine line = ConfigOptions.parseOptions(options, 0, args, false); if (fillLink(line, link)) { - status = client.updateLink(link); + status = client.updateLink(link, oldLinkName); if (!status.canProceed()) { printLinkValidationMessages(link); return null; diff --git a/shell/src/test/java/org/apache/sqoop/shell/TestUpdateCommand.java b/shell/src/test/java/org/apache/sqoop/shell/TestUpdateCommand.java index f5062da5..7fd39b9d 100644 --- a/shell/src/test/java/org/apache/sqoop/shell/TestUpdateCommand.java +++ b/shell/src/test/java/org/apache/sqoop/shell/TestUpdateCommand.java @@ -19,6 +19,7 @@ package org.apache.sqoop.shell; import static org.mockito.Matchers.any; +import static org.mockito.Matchers.anyString; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.when; import static org.testng.Assert.assertEquals; @@ -108,7 +109,7 @@ public void testUpdateLink() throws InterruptedException { MLink link = new MLink("connector_test", new MLinkConfig(new ArrayList(), new ArrayList())); when(client.getLink("link_test")).thenReturn(link); when(client.getConnectorConfigBundle("connector_test")).thenReturn(new MapResourceBundle(new HashMap())); - when(client.updateLink(any(MLink.class))).thenReturn(Status.OK); + when(client.updateLink(any(MLink.class), anyString())).thenReturn(Status.OK); // update link -name link_test Status status = (Status) updateCmd.execute(Arrays.asList(Constants.FN_LINK, "-name", "link_test")); @@ -140,7 +141,7 @@ public void testUpdateLinkInteractive() throws Exception { when(client.getConnector("connector_test")).thenReturn(new MConnector("", "", "", null, null, null)); MLink link = new MLink("connector_test", new MLinkConfig(getConfig("CONFIGFROMNAME"), new ArrayList())); when(client.getLink("link_test")).thenReturn(link); - when(client.updateLink(any(MLink.class))).thenReturn(Status.OK); + when(client.updateLink(any(MLink.class), anyString())).thenReturn(Status.OK); when(client.getConnectorConfigBundle(any(String.class))).thenReturn(resourceBundle); // update link -name link_test @@ -177,11 +178,12 @@ public void testUpdateJob() throws InterruptedException { new MFromConfig(new ArrayList(), new ArrayList()), new MToConfig(new ArrayList(), new ArrayList()), new MDriverConfig(new ArrayList(), new ArrayList())); + job.setName("job_test"); when(client.getJob("job_test")).thenReturn(job); when(client.getConnector(any(String.class))).thenReturn(new MConnector("connect_test", "", "", null, null, null)); when(client.getConnectorConfigBundle(any(String.class))).thenReturn(new MapResourceBundle(new HashMap())); when(client.getDriverConfigBundle()).thenReturn(new MapResourceBundle(new HashMap())); - when(client.updateJob(job)).thenReturn(Status.OK); + when(client.updateJob(job, "job_test")).thenReturn(Status.OK); // update job -name job_test Status status = (Status) updateCmd.execute(Arrays.asList(Constants.FN_JOB, "-name", "job_test")); @@ -218,10 +220,10 @@ public void testUpdateJobInteractive() { when(client.getConnector(any(String.class))).thenReturn(new MConnector("connect_test", "", "", null, null, null)); when(client.getConnectorConfigBundle(any(String.class))).thenReturn(resourceBundle); when(client.getDriverConfigBundle()).thenReturn(resourceBundle); - when(client.updateJob(job)).thenReturn(Status.OK); + when(client.updateJob(any(MJob.class), any(String.class))).thenReturn(Status.OK); // update job -name job_test - initData("jobname\r" + // job name + initData("job_test\r" + // job name // From job config "abc\r" + // for input with name "String" "12345\r" + // for input with name "Integer" @@ -253,7 +255,7 @@ public void testUpdateJobInteractive() { "7654321\r"); // for input with name "DateTime" Status status = (Status) updateCmd.execute(Arrays.asList(Constants.FN_JOB, "-name", "job_test")); assertTrue(status != null && status == Status.OK); - assertEquals(job.getName(), "jobname"); + assertEquals(job.getName(), "job_test"); // check from job config assertEquals(job.getFromJobConfig().getStringInput("fromJobConfig.String").getValue(), "abc"); assertEquals(job.getFromJobConfig().getIntegerInput("fromJobConfig.Integer").getValue().intValue(), 12345); diff --git a/test/src/test/java/org/apache/sqoop/integration/server/InformalObjectNameTest.java b/test/src/test/java/org/apache/sqoop/integration/server/InformalObjectNameTest.java index 1056f12c..5e204c83 100644 --- a/test/src/test/java/org/apache/sqoop/integration/server/InformalObjectNameTest.java +++ b/test/src/test/java/org/apache/sqoop/integration/server/InformalObjectNameTest.java @@ -93,7 +93,7 @@ private void verifyActionsForLink(String linkName) { assertEquals(rdbmsLink, repositoryLink); // update link - getClient().updateLink(rdbmsLink); + getClient().updateLink(rdbmsLink, rdbmsLink.getName()); // enable link getClient().enableLink(linkName, true); @@ -138,7 +138,7 @@ private void verifyActionsForJob(String jobName) throws Exception { assertEquals(job, repositoryJob); // update job - getClient().updateJob(job); + getClient().updateJob(job, job.getName()); // enable job getClient().enableJob(jobName, true);