| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| */ |
| package org.apache.johnzon.jsonb.jaxrs; |
| |
| import org.apache.cxf.endpoint.Server; |
| import org.apache.cxf.jaxrs.JAXRSServerFactoryBean; |
| import org.apache.cxf.jaxrs.client.WebClient; |
| import org.apache.cxf.jaxrs.lifecycle.SingletonResourceProvider; |
| import org.apache.cxf.transport.local.LocalConduit; |
| import org.apache.johnzon.jaxrs.jsonb.jaxrs.JsonbJaxrsProvider; |
| import org.junit.AfterClass; |
| import org.junit.Assert; |
| import org.junit.BeforeClass; |
| import org.junit.Test; |
| |
| import jakarta.ws.rs.GET; |
| import jakarta.ws.rs.POST; |
| import jakarta.ws.rs.Path; |
| import jakarta.ws.rs.Produces; |
| import jakarta.ws.rs.WebApplicationException; |
| import jakarta.ws.rs.core.GenericType; |
| import jakarta.ws.rs.core.MediaType; |
| import jakarta.ws.rs.core.StreamingOutput; |
| import java.io.IOException; |
| import java.io.OutputStream; |
| import java.lang.reflect.ParameterizedType; |
| import java.lang.reflect.Type; |
| import java.util.Arrays; |
| import java.util.List; |
| |
| import static java.util.Arrays.asList; |
| import static java.util.Collections.singletonList; |
| import jakarta.json.bind.Jsonb; |
| import jakarta.json.bind.JsonbBuilder; |
| import jakarta.ws.rs.ext.ContextResolver; |
| import static org.junit.Assert.assertEquals; |
| import static org.junit.Assert.assertTrue; |
| |
| public class JsonbJaxRsTest { |
| private final static String ENDPOINT_ADDRESS = "local://johnzon"; |
| private final static String ENDPOINT_ADDRESS_JSONB_CONFIG = "local://johnzonjsonbconfig"; |
| private static Server server; |
| private static Server serverJsonbConfig; |
| |
| public static class JsonBindingProvider implements ContextResolver<Jsonb> { |
| |
| private final Jsonb jsonb = JsonbBuilder.create(); |
| public static Boolean called = Boolean.FALSE; |
| |
| @Override |
| public Jsonb getContext(Class<?> type) { |
| called = Boolean.TRUE; |
| return jsonb; |
| } |
| |
| } |
| |
| @BeforeClass |
| public static void bindEndpoint() throws Exception { |
| JAXRSServerFactoryBean sf = new JAXRSServerFactoryBean(); |
| sf.setResourceClasses(JohnzonResource.class); |
| sf.setProviders(singletonList(new JsonbJaxrsProvider<>())); |
| sf.setResourceProvider(JohnzonResource.class, new SingletonResourceProvider(new JohnzonResource(), false)); |
| sf.setAddress(ENDPOINT_ADDRESS); |
| server = sf.create(); |
| sf = new JAXRSServerFactoryBean(); |
| sf.setResourceClasses(JohnzonResource.class); |
| sf.setProviders(Arrays.asList(new JsonbJaxrsProvider<>(), new JsonBindingProvider())); |
| sf.setResourceProvider(JohnzonResource.class, new SingletonResourceProvider(new JohnzonResource(), false)); |
| sf.setAddress(ENDPOINT_ADDRESS_JSONB_CONFIG); |
| serverJsonbConfig = sf.create(); |
| } |
| |
| @AfterClass |
| public static void unbind() throws Exception { |
| server.stop(); |
| server.destroy(); |
| serverJsonbConfig.stop(); |
| serverJsonbConfig.destroy(); |
| } |
| |
| @Test |
| public void asParam() { |
| final String result = client().path("johnzon").type(MediaType.APPLICATION_JSON_TYPE).post(new Johnzon("client")).readEntity(String.class); |
| assertTrue(Boolean.parseBoolean(result)); |
| } |
| |
| @Test |
| public void jsonbconfigProvider() { |
| client(ENDPOINT_ADDRESS_JSONB_CONFIG, MediaType.APPLICATION_JSON_TYPE).path("johnzon").get(String.class); |
| assertEquals(JsonBindingProvider.called, Boolean.TRUE); |
| } |
| |
| @Test |
| public void object() { |
| final String output = client().path("johnzon").get(String.class); |
| assertEquals("{\"name\":\"johnzon\"}", output); |
| final Johnzon johnzon = client().path("johnzon").get(Johnzon.class); |
| assertEquals("johnzon", johnzon.getName()); |
| } |
| |
| @Test |
| public void streamOutput() { |
| final String stream = client().path("johnzon/stream").get(String.class); |
| assertEquals("ok", stream); |
| } |
| |
| @Test |
| public void primitive() { |
| final String val = client(MediaType.TEXT_PLAIN_TYPE).path("johnzon/primitive").get(String.class); |
| assertEquals("1986", val); |
| } |
| |
| @Test |
| public void array() { |
| final String output = client().path("johnzon/all1").get(String.class); |
| assertTrue(output.contains("[{\"name\":\"johnzon")); // just check it is actually a JSON and we didnt get abused by an internal optim of CXF |
| final Johnzon[] johnzon = client().path("johnzon/all1").get(Johnzon[].class); |
| assertEquals(2, johnzon.length); |
| for (int i = 0; i < johnzon.length; i++) { |
| assertEquals("johnzon" + (i + 1), johnzon[i].getName()); |
| } |
| } |
| |
| @Test |
| public void testBinaryDownload() { |
| byte[] content = client(MediaType.TEXT_PLAIN_TYPE) |
| .path("johnzon/mybinary") |
| .get(byte[].class); |
| |
| Assert.assertEquals(100, content.length); |
| for (int i=0; i < 100; i++) { |
| Assert.assertEquals((byte) i, content[i]); |
| } |
| } |
| |
| @Test |
| public void list() { |
| final ParameterizedType list = new ParameterizedType() { |
| @Override |
| public Type[] getActualTypeArguments() { |
| return new Type[]{Johnzon.class}; |
| } |
| |
| @Override |
| public Type getRawType() { |
| return List.class; |
| } |
| |
| @Override |
| public Type getOwnerType() { |
| return null; |
| } |
| }; |
| final List<Johnzon> johnzons = client().path("johnzon/all2").get(new GenericType<List<Johnzon>>(list) {}); |
| assertEquals(2, johnzons.size()); |
| int i = 1; |
| for (final Johnzon f : johnzons) { |
| assertEquals("johnzon" + i, f.getName()); |
| i++; |
| } |
| } |
| |
| private static WebClient client() { |
| return client(MediaType.APPLICATION_JSON_TYPE); |
| } |
| |
| private static WebClient client(final MediaType mediaType) { |
| return client(ENDPOINT_ADDRESS, mediaType); |
| } |
| |
| private static WebClient client(final String endpoint, final MediaType mediaType) { |
| final WebClient client = WebClient.create(endpoint, singletonList(new JsonbJaxrsProvider<>())).accept(mediaType); |
| WebClient.getConfig(client).getRequestContext().put(LocalConduit.DIRECT_DISPATCH, Boolean.TRUE); |
| return client; |
| } |
| |
| public static class Johnzon { |
| private String name; |
| |
| public Johnzon(final String name) { |
| this.name = name; |
| } |
| |
| public Johnzon() { |
| // no-op |
| } |
| |
| public String getName() { |
| return name; |
| } |
| |
| public void setName(final String name) { |
| this.name = name; |
| } |
| } |
| |
| @Path("johnzon") |
| public static class JohnzonResource { |
| @GET |
| public Johnzon johnzon() { |
| return new Johnzon("johnzon"); |
| } |
| |
| @GET |
| @Path("all1") |
| public Johnzon[] johnzons1() { |
| return new Johnzon[] { new Johnzon("johnzon1"), new Johnzon("johnzon2") }; |
| } |
| |
| @GET |
| @Path("all2") |
| public List<Johnzon> johnzons2() { |
| return asList(johnzons1()); |
| } |
| |
| @POST |
| public String asParam(final Johnzon f) { |
| return Boolean.toString("client".equals(f.getName())); |
| } |
| |
| @GET |
| @Path("stream") |
| public StreamingOutput out() { |
| return new StreamingOutput() { |
| @Override |
| public void write(OutputStream outputStream) throws IOException, WebApplicationException { |
| outputStream.write("ok".getBytes()); |
| } |
| }; |
| } |
| |
| @GET |
| @Produces(MediaType.TEXT_PLAIN) |
| @Path("primitive") |
| public Integer primitive() { |
| return 1986; |
| } |
| |
| @GET |
| @Produces(MediaType.TEXT_PLAIN) |
| @Path("mybinary") |
| public byte[] binary() { |
| byte[] content = new byte[100]; |
| for (int i=0; i < 100; i++) { |
| content[i] = (byte) i; |
| } |
| |
| return content; |
| } |
| } |
| } |