Search in sources :

Example 16 with VfsComponentContext

use of org.apache.commons.vfs2.provider.VfsComponentContext in project hop by apache.

the class S3NFileNameParserTest method testParseUri.

@Test
public void testParseUri() throws Exception {
    VfsComponentContext context = mock(VfsComponentContext.class);
    FileName fileName = mock(FileName.class);
    String uri = "s3n://bucket/file";
    FileName noBaseFile = parser.parseUri(context, null, uri);
    assertNotNull(noBaseFile);
    assertEquals("bucket", ((S3NFileName) noBaseFile).getBucketId());
    FileName withBaseFile = parser.parseUri(context, fileName, uri);
    assertNotNull(withBaseFile);
    assertEquals("bucket", ((S3NFileName) withBaseFile).getBucketId());
    // assumption is that the whole URL is valid until it comes time to resolve to S3 objects
    uri = "s3n://s3n/bucket/file";
    withBaseFile = parser.parseUri(context, fileName, uri);
    assertEquals("s3n", ((S3NFileName) withBaseFile).getBucketId());
}
Also used : VfsComponentContext(org.apache.commons.vfs2.provider.VfsComponentContext) FileName(org.apache.commons.vfs2.FileName) S3NFileName(org.apache.hop.vfs.s3.s3n.vfs.S3NFileName) Test(org.junit.Test)

Example 17 with VfsComponentContext

use of org.apache.commons.vfs2.provider.VfsComponentContext in project hop by apache.

the class S3NFileObjectTest method setUp.

@Before
public void setUp() throws Exception {
    s3ServiceMock = mock(AmazonS3.class);
    S3Object s3Object = new S3Object();
    s3Object.setKey(OBJECT_NAME);
    s3Object.setBucketName(BUCKET_NAME);
    filename = new S3NFileName(SCHEME, BUCKET_NAME, "/" + BUCKET_NAME, FileType.FOLDER);
    S3NFileName rootFileName = new S3NFileName(SCHEME, BUCKET_NAME, "", FileType.FOLDER);
    S3NFileSystem fileSystem = new S3NFileSystem(rootFileName, new FileSystemOptions());
    fileSystemSpy = spy(fileSystem);
    VfsComponentContext context = mock(VfsComponentContext.class);
    final DefaultFileSystemManager fsm = new DefaultFileSystemManager();
    FilesCache cache = mock(FilesCache.class);
    fsm.setFilesCache(cache);
    fsm.setCacheStrategy(CacheStrategy.ON_RESOLVE);
    when(context.getFileSystemManager()).thenReturn(fsm);
    fileSystemSpy.setContext(context);
    S3NFileObject s3FileObject = new S3NFileObject(filename, fileSystemSpy);
    s3FileObjectBucketSpy = spy(s3FileObject);
    s3FileObjectFileSpy = spy(new S3NFileObject(new S3NFileName(SCHEME, BUCKET_NAME, "/" + BUCKET_NAME + "/" + origKey, FileType.IMAGINARY), fileSystemSpy));
    S3NFileObject s3FileObjectRoot = new S3NFileObject(rootFileName, fileSystemSpy);
    s3FileObjectSpyRoot = spy(s3FileObjectRoot);
    // specify the behaviour of S3 Service
    // when( s3ServiceMock.getBucket( BUCKET_NAME ) ).thenReturn( testBucket );
    when(s3ServiceMock.getObject(BUCKET_NAME, OBJECT_NAME)).thenReturn(s3Object);
    when(s3ServiceMock.getObject(BUCKET_NAME, OBJECT_NAME)).thenReturn(s3Object);
    when(s3ServiceMock.listBuckets()).thenReturn(createBuckets());
    when(s3ServiceMock.doesBucketExistV2(BUCKET_NAME)).thenReturn(true);
    childObjectListing = mock(ObjectListing.class);
    when(childObjectListing.getObjectSummaries()).thenReturn(createObjectSummaries(0)).thenReturn(new ArrayList<>());
    when(childObjectListing.getCommonPrefixes()).thenReturn(new ArrayList<>()).thenReturn(createCommonPrefixes(3));
    when(childObjectListing.isTruncated()).thenReturn(true).thenReturn(false);
    when(s3ServiceMock.listObjects(any(ListObjectsRequest.class))).thenReturn(childObjectListing);
    when(s3ServiceMock.listObjects(anyString(), anyString())).thenReturn(childObjectListing);
    when(s3ServiceMock.listNextBatchOfObjects(any(ObjectListing.class))).thenReturn(childObjectListing);
    s3ObjectMock = mock(S3Object.class);
    s3ObjectInputStream = mock(S3ObjectInputStream.class);
    s3ObjectMetadata = mock(ObjectMetadata.class);
    when(s3ObjectMock.getObjectContent()).thenReturn(s3ObjectInputStream);
    when(s3ServiceMock.getObjectMetadata(anyString(), anyString())).thenReturn(s3ObjectMetadata);
    when(s3ObjectMetadata.getContentLength()).thenReturn(contentLength);
    when(s3ObjectMetadata.getLastModified()).thenReturn(testDate);
    when(s3ServiceMock.getObject(anyString(), anyString())).thenReturn(s3ObjectMock);
    when(fileSystemSpy.getS3Client()).thenReturn(s3ServiceMock);
}
Also used : AmazonS3(com.amazonaws.services.s3.AmazonS3) S3NFileObject(org.apache.hop.vfs.s3.s3n.vfs.S3NFileObject) ArrayList(java.util.ArrayList) S3NFileSystem(org.apache.hop.vfs.s3.s3n.vfs.S3NFileSystem) S3NFileName(org.apache.hop.vfs.s3.s3n.vfs.S3NFileName) VfsComponentContext(org.apache.commons.vfs2.provider.VfsComponentContext) DefaultFileSystemManager(org.apache.commons.vfs2.impl.DefaultFileSystemManager) Before(org.junit.Before)

Example 18 with VfsComponentContext

use of org.apache.commons.vfs2.provider.VfsComponentContext in project hop by apache.

the class AzureFileNameParser method parseUri.

@Override
public FileName parseUri(final VfsComponentContext context, FileName base, String filename) throws FileSystemException {
    final StringBuilder name = new StringBuilder();
    Authority auth = null;
    String path = null;
    FileType fileType;
    int eidx = filename.indexOf("@/");
    if (eidx != -1)
        filename = filename.substring(0, eidx + 1) + "windowsazure.com" + filename.substring(eidx + 1);
    String scheme;
    try {
        auth = extractToPath(filename, name);
        if (auth.getUserName() == null) {
            scheme = UriParser.extractScheme(filename, name);
            UriParser.canonicalizePath(name, 0, name.length(), this);
            UriParser.fixSeparators(name);
        } else {
            scheme = auth.getScheme();
        }
        fileType = UriParser.normalisePath(name);
        path = name.toString();
        if (path.equals("")) {
            path = "/";
        }
    } catch (FileSystemException fse) {
        scheme = UriParser.extractScheme(filename, name);
        UriParser.canonicalizePath(name, 0, name.length(), this);
        UriParser.fixSeparators(name);
        fileType = UriParser.normalisePath(name);
        path = name.toString();
    }
    return new AzureFileName(scheme, path, fileType);
}
Also used : FileSystemException(org.apache.commons.vfs2.FileSystemException) FileType(org.apache.commons.vfs2.FileType)

Example 19 with VfsComponentContext

use of org.apache.commons.vfs2.provider.VfsComponentContext in project hop by apache.

the class S3FileNameParser method parseUri.

@Override
public FileName parseUri(VfsComponentContext context, FileName base, String uri) throws FileSystemException {
    StringBuilder name = new StringBuilder();
    String scheme = UriParser.extractScheme(uri, name);
    UriParser.canonicalizePath(name, 0, name.length(), this);
    // Normalize separators in the path
    UriParser.fixSeparators(name);
    // Normalise the path
    FileType fileType = UriParser.normalisePath(name);
    String fullPath = name.toString();
    // Extract bucket name
    final String bucketName = UriParser.extractFirstElement(name);
    return new S3FileName(scheme, bucketName, fullPath, fileType);
}
Also used : FileType(org.apache.commons.vfs2.FileType)

Example 20 with VfsComponentContext

use of org.apache.commons.vfs2.provider.VfsComponentContext in project hop by apache.

the class S3AFileNameParser method parseUri.

@Override
public FileName parseUri(VfsComponentContext context, FileName base, String uri) throws FileSystemException {
    StringBuilder name = new StringBuilder();
    String scheme = UriParser.extractScheme(uri, name);
    UriParser.canonicalizePath(name, 0, name.length(), this);
    // Normalize separators in the path
    UriParser.fixSeparators(name);
    // Normalise the path
    FileType fileType = UriParser.normalisePath(name);
    // Extract bucket name
    final String bucketName = UriParser.extractFirstElement(name);
    return new S3AFileName(scheme, bucketName, name.toString(), fileType);
}
Also used : FileType(org.apache.commons.vfs2.FileType)

Aggregations

FileType (org.apache.commons.vfs2.FileType)15 VfsComponentContext (org.apache.commons.vfs2.provider.VfsComponentContext)9 FileName (org.apache.commons.vfs2.FileName)6 Test (org.junit.Test)5 AmazonS3 (com.amazonaws.services.s3.AmazonS3)4 ArrayList (java.util.ArrayList)4 DefaultFileSystemManager (org.apache.commons.vfs2.impl.DefaultFileSystemManager)4 Before (org.junit.Before)4 FileSystemException (org.apache.commons.vfs2.FileSystemException)3 ListObjectsRequest (com.amazonaws.services.s3.model.ListObjectsRequest)2 ObjectListing (com.amazonaws.services.s3.model.ObjectListing)2 ObjectMetadata (com.amazonaws.services.s3.model.ObjectMetadata)2 S3Object (com.amazonaws.services.s3.model.S3Object)2 S3ObjectInputStream (com.amazonaws.services.s3.model.S3ObjectInputStream)2 FileSystemOptions (org.apache.commons.vfs2.FileSystemOptions)2 FilesCache (org.apache.commons.vfs2.FilesCache)2 S3FileName (org.apache.hop.vfs.s3.s3.vfs.S3FileName)2 S3NFileName (org.apache.hop.vfs.s3.s3n.vfs.S3NFileName)2 FileSystemManager (org.apache.commons.vfs2.FileSystemManager)1 Cryptor (org.apache.commons.vfs2.util.Cryptor)1