@Test(expected = org.apache.sqoop.SqoopOptions.InvalidOptionsException.class) public void givenDirectImportMysqlAvroDataFileValidationThrows() throws SqoopOptions.InvalidOptionsException { SqoopOptions options = mock(SqoopOptions.class); when(options.getFileLayout()).thenReturn(SqoopOptions.FileLayout.AvroDataFile); when(options.getConnectString()).thenReturn(mysqlConnectionString); importTool.validateDirectMysqlOptions(options); verify(options, times(1)).getFileLayout(); verify(options, times(1)).getConnectString(); verifyNoMoreInteractions(options); }
@Test public void givenDirectImportMysqlTextFileValidationPasses() throws SqoopOptions.InvalidOptionsException { SqoopOptions options = mock(SqoopOptions.class); when(options.getFileLayout()).thenReturn(SqoopOptions.FileLayout.TextFile); when(options.getConnectString()).thenReturn(mysqlConnectionString); importTool.validateDirectMysqlOptions(options); verify(options, times(1)).getFileLayout(); }
@Test(expected = org.apache.sqoop.SqoopOptions.InvalidOptionsException.class) public void givenDirectImportMysqlSequenceFileValidationThrows() throws SqoopOptions.InvalidOptionsException { SqoopOptions options = mock(SqoopOptions.class); when(options.getFileLayout()).thenReturn(SqoopOptions.FileLayout.SequenceFile); when(options.getConnectString()).thenReturn(mysqlConnectionString); importTool.validateDirectMysqlOptions(options); verify(options, times(1)).getFileLayout(); verify(options, times(1)).getConnectString(); verifyNoMoreInteractions(options); }
@Test(expected = org.apache.sqoop.SqoopOptions.InvalidOptionsException.class) public void givenDirectImportMysqlParquetFileValidationThrows() throws SqoopOptions.InvalidOptionsException { SqoopOptions options = mock(SqoopOptions.class); when(options.getFileLayout()).thenReturn(SqoopOptions.FileLayout.ParquetFile); when(options.getConnectString()).thenReturn(mysqlConnectionString); importTool.validateDirectMysqlOptions(options); verify(options, times(1)).getFileLayout(); verify(options, times(1)).getConnectString(); verifyNoMoreInteractions(options); } @Test(expected = org.apache.sqoop.SqoopOptions.InvalidOptionsException.class)
@Test public void testFtpTransferModeAscii() throws ParseException, InvalidOptionsException { String[] args = new String[] { "--dataset", "mydatasetname", "--as-textfile" }; configureAndValidateOptions(args); assertEquals(SqoopOptions.FileLayout.TextFile,sqoopOption.getFileLayout()); } @Test
@Test public void testFtpTransferModeDefaultsToAscii() throws ParseException, InvalidOptionsException { String[] args = new String[] { "--dataset", "mydatasetname" }; configureAndValidateOptions(args); assertEquals(SqoopOptions.FileLayout.TextFile,sqoopOption.getFileLayout()); }
@Test public void testSetBufferSize() throws ParseException, InvalidOptionsException { final Integer EXPECTED_BUFFER = 1024; String[] args = new String[] { "--dataset", "mydatasetname", "--as-binaryfile", "--buffersize", EXPECTED_BUFFER.toString() }; configureAndValidateOptions(args); assertEquals(SqoopOptions.FileLayout.BinaryFile,sqoopOption.getFileLayout()); assertEquals(EXPECTED_BUFFER, sqoopOption.getBufferSize()); }
@Test public void testAsBinaryFileSetsCorrectFileLayoutAndDefaultBufferSize() throws ParseException, InvalidOptionsException { String[] args = new String[] { "--dataset", "mydatasetname", "--as-binaryfile" }; configureAndValidateOptions(args); assertEquals(SqoopOptions.FileLayout.BinaryFile,sqoopOption.getFileLayout()); assertEquals(MainframeConfiguration.MAINFRAME_FTP_TRANSFER_BINARY_DEFAULT_BUFFER_SIZE, sqoopOption.getBufferSize()); }