github/sqoop
5
0
Fork 0
mirror of https://github.com/apache/sqoop.git synced 2025-05-06 06:12:26 +08:00
Code Issues Projects Releases Wiki Activity

SQOOP-2641: SQOOP2: Fix test case failure for repository-postgresql

Browse Source 
(Colin Ma via Jarek Jarcec Cecho)
This commit is contained in:
Jarek Jarcec Cecho 2015-10-27 10:01:53 -07:00
parent bc0de7c199
commit b09b897cf8
2 changed files with 27 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
repository/repository-postgresql/pom.xml
View File

@ -70,6 +70,12 @@ limitations under the License.
<type>test-jar</type>
<scope>test</scope>
</dependency>
<dependency>
<groupId>org.mockito</groupId>
<artifactId>mockito-all</artifactId>
<scope>test</scope>
</dependency>
</dependencies>
<build>

21
repository/repository-postgresql/src/test/java/org/apache/sqoop/integration/repository/postgresql/PostgresqlTestCase.java
View File

@ -21,12 +21,14 @@
import org.apache.sqoop.common.Direction;
import org.apache.sqoop.common.test.db.DatabaseProvider;
import org.apache.sqoop.common.test.db.PostgreSQLProvider;
import org.apache.sqoop.connector.ConnectorManager;
import org.apache.sqoop.json.DriverBean;
import org.apache.sqoop.model.*;
import org.apache.sqoop.repository.postgresql.PostgresqlRepositoryHandler;
import org.apache.sqoop.submission.SubmissionStatus;
import org.apache.sqoop.submission.counter.CounterGroup;
import org.apache.sqoop.submission.counter.Counters;
import org.mockito.Mockito;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.BeforeMethod;
@ -44,11 +46,17 @@ abstract public class PostgresqlTestCase {
public static DatabaseProvider provider;
public static PostgresqlTestUtils utils;
public PostgresqlRepositoryHandler handler;
private ConnectorManager mockConnectorManager;
@BeforeClass(alwaysRun = true)
public void setUpClass() {
provider = new PostgreSQLProvider();
utils = new PostgresqlTestUtils(provider);
mockConnectorManager = Mockito.mock(ConnectorManager.class);
Mockito.when(mockConnectorManager.getConnectorConfigurable("A")).thenReturn(getConnector(true, true, "A", "org.apache.sqoop.test.A"));
Mockito.when(mockConnectorManager.getConnectorConfigurable("B")).thenReturn(getConnector(true, true, "B", "org.apache.sqoop.test.B"));
ConnectorManager.setInstance(mockConnectorManager);
}
@BeforeMethod(alwaysRun = true)
@ -165,4 +173,17 @@ protected List<MConfig> getConfigs(String configName1, String configName2) {
return configs;
}
protected MConnector getConnector(boolean from, boolean to, String connectorName, String connectorClass) {
MFromConfig fromConfig = null;
MToConfig toConfig = null;
if (from) {
fromConfig = getFromConfig();
}
if (to) {
toConfig = getToConfig();
}
return new MConnector(connectorName, connectorClass, "1.0-test", getLinkConfig(), fromConfig,
toConfig);
}
}