Search in sources :

Example 1 with S3NFileSystem

use of org.apache.hop.vfs.s3.s3n.vfs.S3NFileSystem in project hop by apache.

the class S3NFileSystemTest method setUp.

@Before
public void setUp() throws Exception {
    fileName = new S3NFileName(S3FileNameTest.SCHEME, "/", "", FileType.FOLDER);
    fileSystem = new S3NFileSystem(fileName, new FileSystemOptions());
}
Also used : S3NFileSystem(org.apache.hop.vfs.s3.s3n.vfs.S3NFileSystem) S3NFileName(org.apache.hop.vfs.s3.s3n.vfs.S3NFileName) FileSystemOptions(org.apache.commons.vfs2.FileSystemOptions) Before(org.junit.Before)

Example 2 with S3NFileSystem

use of org.apache.hop.vfs.s3.s3n.vfs.S3NFileSystem in project hop by apache.

the class S3NFileSystemTest method testGetS3Service.

@Test
public void testGetS3Service() throws Exception {
    assertNotNull(fileSystem.getS3Client());
    FileSystemOptions options = new FileSystemOptions();
    UserAuthenticator authenticator = mock(UserAuthenticator.class);
    UserAuthenticationData authData = mock(UserAuthenticationData.class);
    when(authenticator.requestAuthentication(S3FileProvider.AUTHENTICATOR_TYPES)).thenReturn(authData);
    when(authData.getData(UserAuthenticationData.USERNAME)).thenReturn("username".toCharArray());
    when(authData.getData(UserAuthenticationData.PASSWORD)).thenReturn("password".toCharArray());
    DefaultFileSystemConfigBuilder.getInstance().setUserAuthenticator(options, authenticator);
    fileSystem = new S3NFileSystem(fileName, options);
    assertNotNull(fileSystem.getS3Client());
}
Also used : S3NFileSystem(org.apache.hop.vfs.s3.s3n.vfs.S3NFileSystem) UserAuthenticationData(org.apache.commons.vfs2.UserAuthenticationData) UserAuthenticator(org.apache.commons.vfs2.UserAuthenticator) FileSystemOptions(org.apache.commons.vfs2.FileSystemOptions) Test(org.junit.Test)

Example 3 with S3NFileSystem

use of org.apache.hop.vfs.s3.s3n.vfs.S3NFileSystem 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)

Aggregations

S3NFileSystem (org.apache.hop.vfs.s3.s3n.vfs.S3NFileSystem)3 FileSystemOptions (org.apache.commons.vfs2.FileSystemOptions)2 S3NFileName (org.apache.hop.vfs.s3.s3n.vfs.S3NFileName)2 Before (org.junit.Before)2 AmazonS3 (com.amazonaws.services.s3.AmazonS3)1 ArrayList (java.util.ArrayList)1 UserAuthenticationData (org.apache.commons.vfs2.UserAuthenticationData)1 UserAuthenticator (org.apache.commons.vfs2.UserAuthenticator)1 DefaultFileSystemManager (org.apache.commons.vfs2.impl.DefaultFileSystemManager)1 VfsComponentContext (org.apache.commons.vfs2.provider.VfsComponentContext)1 S3NFileObject (org.apache.hop.vfs.s3.s3n.vfs.S3NFileObject)1 Test (org.junit.Test)1