@Test public void testStreaming() throws Exception { WSExecutionFactory ef = new WSExecutionFactory(); MetadataFactory mf = new MetadataFactory( "vdb", 1, "x", SystemMetadata.getInstance().getRuntimeTypeMap(), new Properties(), null); ef.getMetadata(mf, null); Procedure p = mf.getSchema().getProcedure(WSExecutionFactory.INVOKE_HTTP); assertEquals(7, p.getParameters().size()); TransformationMetadata tm = RealMetadataFactory.createTransformationMetadata(mf.asMetadataStore(), "vdb"); RuntimeMetadataImpl rm = new RuntimeMetadataImpl(tm); WSConnection mockConnection = Mockito.mock(WSConnection.class); Dispatch<Object> mockDispatch = mockDispatch(); DataSource mock = Mockito.mock(DataSource.class); ByteArrayInputStream baos = new ByteArrayInputStream(new byte[100]); Mockito.stub(mock.getInputStream()).toReturn(baos); Mockito.stub(mockDispatch.invoke(Mockito.any(DataSource.class))).toReturn(mock); Mockito.stub( mockConnection.createDispatch( Mockito.any(String.class), Mockito.any(String.class), Mockito.any(Class.class), Mockito.any(Service.Mode.class))) .toReturn(mockDispatch); CommandBuilder cb = new CommandBuilder(tm); Call call = (Call) cb.getCommand("call invokeHttp('GET', null, null, true)"); BinaryWSProcedureExecution pe = new BinaryWSProcedureExecution( call, rm, Mockito.mock(ExecutionContext.class), ef, mockConnection); pe.execute(); List<?> result = pe.getOutputParameterValues(); Blob b = (Blob) result.get(0); assertEquals(100, ObjectConverterUtil.convertToByteArray(b.getBinaryStream()).length); try { ObjectConverterUtil.convertToByteArray(b.getBinaryStream()); fail(); } catch (SQLException e) { // should only be able to read once } }
@Test public void testHeaders() throws Exception { WSExecutionFactory ef = new WSExecutionFactory(); MetadataFactory mf = new MetadataFactory( "vdb", 1, "x", SystemMetadata.getInstance().getRuntimeTypeMap(), new Properties(), null); ef.getMetadata(mf, null); Procedure p = mf.getSchema().getProcedure(WSExecutionFactory.INVOKE_HTTP); TransformationMetadata tm = RealMetadataFactory.createTransformationMetadata(mf.asMetadataStore(), "vdb"); RuntimeMetadataImpl rm = new RuntimeMetadataImpl(tm); WSConnection mockConnection = Mockito.mock(WSConnection.class); Dispatch<Object> mockDispatch = mockDispatch(); DataSource mock = Mockito.mock(DataSource.class); ByteArrayInputStream baos = new ByteArrayInputStream(new byte[100]); Mockito.stub(mock.getInputStream()).toReturn(baos); Mockito.stub(mockDispatch.invoke(Mockito.any(DataSource.class))).toReturn(mock); Mockito.stub( mockConnection.createDispatch( Mockito.any(String.class), Mockito.any(String.class), Mockito.any(Class.class), Mockito.any(Service.Mode.class))) .toReturn(mockDispatch); CommandBuilder cb = new CommandBuilder(tm); Call call = (Call) cb.getCommand( "call invokeHttp('GET', null, null, false, '{\"ContentType\":\"application/json\"}')"); BinaryWSProcedureExecution pe = new BinaryWSProcedureExecution( call, rm, Mockito.mock(ExecutionContext.class), ef, mockConnection); pe.execute(); Map<String, List<String>> headers = (Map<String, List<String>>) mockDispatch.getRequestContext().get(MessageContext.HTTP_REQUEST_HEADERS); assertEquals(Arrays.asList("application/json"), headers.get("ContentType")); }
@Test public void testPre81Procedure() throws Exception { WSExecutionFactory ef = new WSExecutionFactory(); MetadataFactory mf = new MetadataFactory( "vdb", 1, "x", SystemMetadata.getInstance().getRuntimeTypeMap(), new Properties(), null); ef.getMetadata(mf, null); Procedure p = mf.getSchema().getProcedure(WSExecutionFactory.INVOKE_HTTP); assertEquals(7, p.getParameters().size()); p.getParameters().remove(4); p.getParameters().remove(5); p = mf.getSchema().getProcedure("invoke"); assertEquals(6, p.getParameters().size()); p.getParameters().remove(5); // designer treated the result as an out parameter p.getParameters().get(0).setType(Type.Out); p.getParameters().add(p.getParameters().remove(0)); for (int i = 0; i < p.getParameters().size(); i++) { p.getParameters().get(i).setPosition(i + 1); } TransformationMetadata tm = RealMetadataFactory.createTransformationMetadata(mf.asMetadataStore(), "vdb"); RuntimeMetadataImpl rm = new RuntimeMetadataImpl(tm); WSConnection mockConnection = Mockito.mock(WSConnection.class); Dispatch<Object> mockDispatch = mockDispatch(); DataSource source = Mockito.mock(DataSource.class); Mockito.stub(mockDispatch.invoke(Mockito.any(DataSource.class))).toReturn(source); Mockito.stub( mockConnection.createDispatch( Mockito.any(String.class), Mockito.any(String.class), Mockito.any(Class.class), Mockito.any(Service.Mode.class))) .toReturn(mockDispatch); CommandBuilder cb = new CommandBuilder(tm); Call call = (Call) cb.getCommand("call invokeHttp('GET', null, null)"); BinaryWSProcedureExecution pe = new BinaryWSProcedureExecution( call, rm, Mockito.mock(ExecutionContext.class), ef, mockConnection); pe.execute(); pe.getOutputParameterValues(); mockConnection = Mockito.mock(WSConnection.class); mockDispatch = Mockito.mock(Dispatch.class); StAXSource ssource = Mockito.mock(StAXSource.class); Mockito.stub(mockDispatch.invoke(Mockito.any(StAXSource.class))).toReturn(ssource); Mockito.stub( mockConnection.createDispatch( Mockito.any(String.class), Mockito.any(String.class), Mockito.any(Class.class), Mockito.any(Service.Mode.class))) .toReturn(mockDispatch); call = (Call) cb.getCommand("call invoke()"); WSProcedureExecution wpe = new WSProcedureExecution( call, rm, Mockito.mock(ExecutionContext.class), ef, mockConnection); wpe.execute(); wpe.getOutputParameterValues(); }