Search in sources :

Example 1 with GoogleDriveExtractor

use of org.apache.gobblin.source.extractor.extract.google.GoogleDriveExtractor in project incubator-gobblin by apache.

the class GoogleDriveSourceTest method testGetExtractor.

public void testGetExtractor() throws IOException {
    @SuppressWarnings("rawtypes") GoogleDriveSource source = new GoogleDriveSource<>();
    GoogleDriveFsHelper fsHelper = mock(GoogleDriveFsHelper.class);
    source.fsHelper = fsHelper;
    Extractor extractor = source.getExtractor(new WorkUnitState());
    Assert.assertTrue(extractor instanceof GoogleDriveExtractor);
}
Also used : GoogleDriveFsHelper(org.apache.gobblin.source.extractor.extract.google.GoogleDriveFsHelper) WorkUnitState(org.apache.gobblin.configuration.WorkUnitState) GoogleDriveExtractor(org.apache.gobblin.source.extractor.extract.google.GoogleDriveExtractor) GoogleDriveExtractor(org.apache.gobblin.source.extractor.extract.google.GoogleDriveExtractor) Extractor(org.apache.gobblin.source.extractor.Extractor) GoogleDriveSource(org.apache.gobblin.source.extractor.extract.google.GoogleDriveSource)

Aggregations

WorkUnitState (org.apache.gobblin.configuration.WorkUnitState)1 Extractor (org.apache.gobblin.source.extractor.Extractor)1 GoogleDriveExtractor (org.apache.gobblin.source.extractor.extract.google.GoogleDriveExtractor)1 GoogleDriveFsHelper (org.apache.gobblin.source.extractor.extract.google.GoogleDriveFsHelper)1 GoogleDriveSource (org.apache.gobblin.source.extractor.extract.google.GoogleDriveSource)1