javax.ws.rs.core.StreamingOutput#write

Source File: SupportUtil.java From syndesis with Apache License 2.0

6 votes

protected void addSource(String integrationName, String integrationId, ZipOutputStream os) throws IOException {
    StreamingOutput export = integrationSupportHandler.export(Arrays.asList(integrationId));
    ZipEntry ze = new ZipEntry(integrationName + ".src.zip");
    os.putNextEntry(ze);
    File file = null;
    try {
        file = File.createTempFile(integrationName, ".src.zip");
        export.write(FileUtils.openOutputStream(file));
        FileUtils.copyFile(file, os);
        os.closeEntry();
    } finally {
        if (file != null && !file.delete()) {
            file.deleteOnExit();
        }
    }

}

Source File: FunctionApiV2ResourceTest.java From pulsar with Apache License 2.0

6 votes

@Test
public void testDownloadFunctionFile() throws Exception {
    URL fileUrl = getClass().getClassLoader().getResource("test_worker_config.yml");
    File file = Paths.get(fileUrl.toURI()).toFile();
    String fileLocation = file.getAbsolutePath();
    String testDir = FunctionApiV2ResourceTest.class.getProtectionDomain().getCodeSource().getLocation().getPath();
    FunctionsImplV2 function = new FunctionsImplV2(() -> mockedWorkerService);
    StreamingOutput streamOutput = (StreamingOutput) function.downloadFunction("file://" + fileLocation, null).getEntity();
    File pkgFile = new File(testDir, UUID.randomUUID().toString());
    OutputStream output = new FileOutputStream(pkgFile);
    streamOutput.write(output);
    Assert.assertTrue(pkgFile.exists());
    if (pkgFile.exists()) {
        pkgFile.delete();
    }
}

Source File: FunctionApiV3ResourceTest.java From pulsar with Apache License 2.0

6 votes

@Test
public void testDownloadFunctionHttpUrl() throws Exception {
    String jarHttpUrl = "https://repo1.maven.org/maven2/org/apache/pulsar/pulsar-common/2.4.2/pulsar-common-2.4.2.jar";
    String testDir = FunctionApiV3ResourceTest.class.getProtectionDomain().getCodeSource().getLocation().getPath();
    WorkerService worker = mock(WorkerService.class);
    doReturn(true).when(worker).isInitialized();
    WorkerConfig config = mock(WorkerConfig.class);
    when(config.isAuthorizationEnabled()).thenReturn(false);
    when(worker.getWorkerConfig()).thenReturn(config);
    FunctionsImpl function = new FunctionsImpl(()-> worker);
    StreamingOutput streamOutput = function.downloadFunction(jarHttpUrl, null, null);
    File pkgFile = new File(testDir, UUID.randomUUID().toString());
    OutputStream output = new FileOutputStream(pkgFile);
    streamOutput.write(output);
    Assert.assertTrue(pkgFile.exists());
    if (pkgFile.exists()) {
        pkgFile.delete();
    }
}

Source File: FunctionApiV3ResourceTest.java From pulsar with Apache License 2.0

6 votes

@Test
public void testDownloadFunctionFile() throws Exception {
    URL fileUrl = getClass().getClassLoader().getResource("test_worker_config.yml");
    File file = Paths.get(fileUrl.toURI()).toFile();
    String fileLocation = file.getAbsolutePath();
    String testDir = FunctionApiV3ResourceTest.class.getProtectionDomain().getCodeSource().getLocation().getPath();
    WorkerService worker = mock(WorkerService.class);
    doReturn(true).when(worker).isInitialized();
    WorkerConfig config = mock(WorkerConfig.class);
    when(config.isAuthorizationEnabled()).thenReturn(false);
    when(worker.getWorkerConfig()).thenReturn(config);
    FunctionsImpl function = new FunctionsImpl(() -> worker);
    StreamingOutput streamOutput = function.downloadFunction("file://" + fileLocation, null, null);
    File pkgFile = new File(testDir, UUID.randomUUID().toString());
    OutputStream output = new FileOutputStream(pkgFile);
    streamOutput.write(output);
    Assert.assertTrue(pkgFile.exists());
    if (pkgFile.exists()) {
        pkgFile.delete();
    }
}

Source File: BufferedStreamingOutputTest.java From robe with GNU Lesser General Public License v3.0

6 votes

@Test
public void write() throws IOException {

    String testSentence = "robe";
    ByteArrayInputStream inputStream = new ByteArrayInputStream(testSentence.getBytes("UTF-8"));
    File file = Files.writeToTemp(inputStream);
    BufferedInputStream bufferedInputStream = new BufferedInputStream(new FileInputStream(file));
    StreamingOutput stream = new BufferedStreamingOutput(bufferedInputStream, 5);

    ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
    stream.write(outputStream);
    Assert.assertTrue(outputStream.size() == 4);


    bufferedInputStream = new BufferedInputStream(new FileInputStream(file));
    stream = new BufferedStreamingOutput(bufferedInputStream, 3);

    outputStream = new ByteArrayOutputStream();
    stream.write(outputStream);
    Assert.assertTrue(outputStream.size() == 4);

}

Source File: ConnectorHandlerTest.java From syndesis with Apache License 2.0

5 votes

@Test
public void connectorIconShouldHaveCorrectContentType() throws IOException {
    try (MockWebServer mockWebServer = new MockWebServer(); final Buffer resultBuffer = new Buffer()) {
        mockWebServer.start();

        resultBuffer.writeAll(Okio.source(getClass().getResourceAsStream("test-image.png")));

        mockWebServer.enqueue(new MockResponse().setHeader(CONTENT_TYPE, "image/png").setBody(resultBuffer));

        final Connector connector = new Connector.Builder().id("connector-id").icon(mockWebServer.url("/u/23079786").toString())
            .build();
        when(dataManager.fetch(Connector.class, "connector-id")).thenReturn(connector);
        when(dataManager.fetchAll(Integration.class)).thenReturn(ListResult.of(Collections.emptyList()));

        final Response response = handler.getConnectorIcon("connector-id").get();

        assertThat(response.getStatusInfo()).as("Wrong status code").isEqualTo(Response.Status.OK);
        assertThat(response.getHeaderString(CONTENT_TYPE)).as("Wrong content type").isEqualTo("image/png");
        assertThat(response.getHeaderString(CONTENT_LENGTH)).as("Wrong content length").isEqualTo("2018");

        final StreamingOutput so = (StreamingOutput) response.getEntity();
        final ByteArrayOutputStream bos = new ByteArrayOutputStream();
        try (BufferedSink sink = Okio.buffer(Okio.sink(bos)); BufferedSource source = new Buffer();
             ImageInputStream iis = ImageIO.createImageInputStream(source.inputStream());) {
            so.write(sink.outputStream());
            source.readAll(sink);
            final Iterator<ImageReader> readers = ImageIO.getImageReaders(iis);
            if (readers.hasNext()) {
                final ImageReader reader = readers.next();
                try {
                    reader.setInput(iis);
                    assertThat(reader.getHeight(0)).as("Wrong image height").isEqualTo(106d);
                    assertThat(reader.getWidth(0)).as("Wrong image width").isEqualTo(106d);
                } finally {
                    reader.dispose();
                }
            }
        }
    }
}

Source File: FunctionApiV2ResourceTest.java From pulsar with Apache License 2.0

5 votes

@Test
public void testDownloadFunctionHttpUrl() throws Exception {
    String jarHttpUrl = "https://repo1.maven.org/maven2/org/apache/pulsar/pulsar-common/2.4.2/pulsar-common-2.4.2.jar";
    String testDir = FunctionApiV2ResourceTest.class.getProtectionDomain().getCodeSource().getLocation().getPath();
    FunctionsImplV2 function = new FunctionsImplV2(() -> mockedWorkerService);
    StreamingOutput streamOutput = (StreamingOutput) function.downloadFunction(jarHttpUrl, null).getEntity();
    File pkgFile = new File(testDir, UUID.randomUUID().toString());
    OutputStream output = new FileOutputStream(pkgFile);
    streamOutput.write(output);
    Assert.assertTrue(pkgFile.exists());
    if (pkgFile.exists()) {
        pkgFile.delete();
    }
}

Source File: BinaryDataProviderTest.java From cxf with Apache License 2.0

5 votes

@SuppressWarnings({ "unchecked", "rawtypes" })
@Test
public void testReadFrom() throws Exception {
    MessageBodyReader p = new BinaryDataProvider();
    byte[] bytes = (byte[])p.readFrom(byte[].class, byte[].class, new Annotation[]{},
                                      MediaType.APPLICATION_OCTET_STREAM_TYPE,
                                      new MetadataMap<String, Object>(),
                                      new ByteArrayInputStream("hi".getBytes()));
    assertArrayEquals(new String("hi").getBytes(), bytes);

    InputStream is = (InputStream)p.readFrom(InputStream.class, InputStream.class, new Annotation[]{},
                                             MediaType.APPLICATION_OCTET_STREAM_TYPE,
                                             new MetadataMap<String, Object>(),
        new ByteArrayInputStream("hi".getBytes()));
    bytes = IOUtils.readBytesFromStream(is);
    assertArrayEquals(new String("hi").getBytes(), bytes);

    Reader r = (Reader)p.readFrom(Reader.class, Reader.class, new Annotation[]{},
                                  MediaType.APPLICATION_OCTET_STREAM_TYPE,
                                  new MetadataMap<String, Object>(),
                                  new ByteArrayInputStream("hi".getBytes()));
    assertEquals(IOUtils.toString(r), "hi");

    StreamingOutput so = (StreamingOutput)p.readFrom(StreamingOutput.class, StreamingOutput.class,
                                  new Annotation[]{},
                                  MediaType.APPLICATION_OCTET_STREAM_TYPE,
                                  new MetadataMap<String, Object>(),
                                  new ByteArrayInputStream("hi".getBytes()));
    ByteArrayOutputStream baos = new ByteArrayOutputStream();
    so.write(baos);
    bytes = baos.toByteArray();
    assertArrayEquals(new String("hi").getBytes(), bytes);
}

Source File: StreamOutputEntityProvider.java From everrest with Eclipse Public License 2.0

5 votes

@Override
public void writeTo(StreamingOutput streamingOutput,
                    Class<?> type,
                    Type genericType,
                    Annotation[] annotations,
                    MediaType mediaType,
                    MultivaluedMap<String, Object> httpHeaders,
                    OutputStream entityStream) throws IOException {
    streamingOutput.write(entityStream);
}

Source File: BulkExtractTest.java From secure-data-service with Apache License 2.0

5 votes

@Test
public void testGetSampleExtract() throws Exception {
    injector.setEducatorContext();

    ResponseImpl res = (ResponseImpl) bulkExtract.get(req);
    assertEquals(200, res.getStatus());
    MultivaluedMap<String, Object> headers = res.getMetadata();
    assertNotNull(headers);
    assertTrue(headers.containsKey("content-disposition"));
    assertTrue(headers.containsKey("last-modified"));
    String header = (String) headers.getFirst("content-disposition");
    assertNotNull(header);
    assertTrue(header.startsWith("attachment"));
    assertTrue(header.indexOf("sample-extract.tar") > 0);

    Object entity = res.getEntity();
    assertNotNull(entity);
    StreamingOutput out = (StreamingOutput) entity;
    File file = new File("out.zip");
    FileOutputStream os = new FileOutputStream(file);
    out.write(os);
    os.flush();
    assertTrue(file.exists());

    assertEquals(798669192L, FileUtils.checksumCRC32(file));
    FileUtils.deleteQuietly(file);
}

Source File: BulkExtractTest.java From secure-data-service with Apache License 2.0

5 votes

@Test
public void testGetExtractResponse() throws Exception {
    injector.setOauthAuthenticationWithEducationRole();
    mockApplicationEntity();
    mockBulkExtractEntity(null);

    HttpRequestContext context = new HttpRequestContextAdapter() {
        @Override
        public String getMethod() {
            return "GET";
        }
    };

    Response res = bulkExtract.getEdOrgExtractResponse(context, null, null);
    assertEquals(200, res.getStatus());
    MultivaluedMap<String, Object> headers = res.getMetadata();
    assertNotNull(headers);
    assertTrue(headers.containsKey("content-disposition"));
    assertTrue(headers.containsKey("last-modified"));
    String header = (String) headers.getFirst("content-disposition");
    assertNotNull(header);
    assertTrue(header.startsWith("attachment"));
    assertTrue(header.indexOf(INPUT_FILE_NAME) > 0);

    Object entity = res.getEntity();
    assertNotNull(entity);

    StreamingOutput out = (StreamingOutput) entity;
    ByteArrayOutputStream os = new ByteArrayOutputStream();
    out.write(os);
    os.flush();
    byte[] responseData = os.toByteArray();
    String s = new String(responseData);

    assertEquals(BULK_DATA, s);
}

Source File: BulkExtractTest.java From secure-data-service with Apache License 2.0

5 votes

@Test
public void testEdOrgFullExtract() throws IOException, ParseException {
    injector.setOauthAuthenticationWithEducationRole();
    mockApplicationEntity();
    Entity mockedEntity = mockBulkExtractEntity(null);
    Mockito.when(edOrgHelper.byId(eq("ONE"))).thenReturn(mockedEntity);

    Map<String, Object> authBody = new HashMap<String, Object>();
    authBody.put("applicationId", "App1");
    authBody.put(ApplicationAuthorizationResource.EDORG_IDS, ApplicationAuthorizationResourceTest.getAuthList("ONE"));
    Entity mockAppAuth = Mockito.mock(Entity.class);
    Mockito.when(mockAppAuth.getBody()).thenReturn(authBody);
    Mockito.when(mockMongoEntityRepository.findOne(eq("applicationAuthorization"), Mockito.any(NeutralQuery.class)))
            .thenReturn(mockAppAuth);

    Response res = bulkExtract.getEdOrgExtract(CONTEXT, req, "ONE");

    assertEquals(200, res.getStatus());
    MultivaluedMap<String, Object> headers = res.getMetadata();
    assertNotNull(headers);
    assertTrue(headers.containsKey("content-disposition"));
    assertTrue(headers.containsKey("last-modified"));
    String header = (String) headers.getFirst("content-disposition");
    assertNotNull(header);
    assertTrue(header.startsWith("attachment"));
    assertTrue(header.indexOf(INPUT_FILE_NAME) > 0);

    Object entity = res.getEntity();
    assertNotNull(entity);

    StreamingOutput out = (StreamingOutput) entity;
    ByteArrayOutputStream os = new ByteArrayOutputStream();
    out.write(os);
    os.flush();
    byte[] responseData = os.toByteArray();
    String s = new String(responseData);

    assertEquals(BULK_DATA, s);
}

Source File: BulkExtractTest.java From secure-data-service with Apache License 2.0

5 votes

@Test
public void testPublicExtract() throws IOException, ParseException {
    injector.setOauthAuthenticationWithEducationRole();
    mockApplicationEntity();
    Entity mockedEntity = mockBulkExtractEntity(null);
    Mockito.when(edOrgHelper.byId(eq("ONE"))).thenReturn(mockedEntity);

    Response res = bulkExtract.getPublicExtract(CONTEXT, req);

    assertEquals(200, res.getStatus());
    MultivaluedMap<String, Object> headers = res.getMetadata();
    assertNotNull(headers);
    assertTrue(headers.containsKey("content-disposition"));
    assertTrue(headers.containsKey("last-modified"));
    String header = (String) headers.getFirst("content-disposition");
    assertNotNull(header);
    assertTrue(header.startsWith("attachment"));
    assertTrue(header.indexOf(INPUT_FILE_NAME) > 0);

    Object entity = res.getEntity();
    assertNotNull(entity);

    StreamingOutput out = (StreamingOutput) entity;
    ByteArrayOutputStream os = new ByteArrayOutputStream();
    out.write(os);
    os.flush();
    byte[] responseData = os.toByteArray();
    String s = new String(responseData);

    assertEquals(BULK_DATA, s);
}

Java Code Examples for javax.ws.rs.core.StreamingOutput#write()