Contents

package org.embulk.input.s3.explorer;

import com.amazonaws.services.s3.AmazonS3;
import com.amazonaws.services.s3.model.GetObjectMetadataRequest;
import com.amazonaws.services.s3.model.ObjectMetadata;
import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
import org.embulk.EmbulkTestRuntime;
import org.embulk.input.s3.FileList;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.runners.MockitoJUnitRunner;

import static org.mockito.Matchers.any;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

@RunWith(MockitoJUnitRunner.class)
public class TestS3SingleFileExplorer
{
    private static final String PATH = "path";
    private static final String BUCKET_NAME = "bucket_name";

    @SuppressFBWarnings("URF_UNREAD_PUBLIC_OR_PROTECTED_FIELD")
    @Rule
    public EmbulkTestRuntime embulkTestRuntime = new EmbulkTestRuntime();

    @Mock
    private AmazonS3 s3Client;

    @Mock
    private FileList.Builder builder;

    @Mock
    private ObjectMetadata metadata;

    private S3SingleFileExplorer s3SingleFileExplorer;

    @Before
    public void setUp()
    {
        s3SingleFileExplorer = new S3SingleFileExplorer(BUCKET_NAME, s3Client, null, PATH);
    }

    @Test
    public void addToBuilder_should_request_single_object_metadata()
    {
        when(s3Client.getObjectMetadata(any(GetObjectMetadataRequest.class))).thenReturn(metadata);
        when(metadata.getContentLength()).thenReturn(1L);
        s3SingleFileExplorer.addToBuilder(builder);

        verify(builder).add(PATH, 1);
    }
}

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
embulk-input-s3-0.3.5	src/test/java/org/embulk/input/s3/explorer/TestS3SingleFileExplorer.java
embulk-input-s3-0.3.4	src/test/java/org/embulk/input/s3/explorer/TestS3SingleFileExplorer.java