REST Data Panache dev mode fix #9510

quarkusio · Sep 29, 2020 · 3e1af27 · 3e1af27
1 parent c9ab9fe
commit 3e1af27
Show file tree

Hide file tree

Showing 4 changed files with 77 additions and 23 deletions.
diff --git a/.../test/java/io/quarkus/hibernate/orm/rest/data/panache/deployment/AbstractDevModeTest.java b/.../test/java/io/quarkus/hibernate/orm/rest/data/panache/deployment/AbstractDevModeTest.java
@@ -0,0 +1,28 @@
+package io.quarkus.hibernate.orm.rest.data.panache.deployment;
+
+import static io.restassured.RestAssured.given;
+import static io.restassured.RestAssured.when;
+import static org.assertj.core.api.Assertions.assertThat;
+
+import org.junit.jupiter.api.Test;
+
+import io.restassured.response.Response;
+
+public abstract class AbstractDevModeTest {
+
+    @Test
+    void testGet() {
+        when().get("/items/1")
+                .then().statusCode(200);
+    }
+
+    @Test
+    void testCreate() {
+        Response response = given().accept("application/json")
+                .and().contentType("application/json")
+                .and().body("{\"name\": \"test-simple\", \"collection\": {\"id\": \"full\"}}")
+                .when().post("/items")
+                .thenReturn();
+        assertThat(response.getStatusCode()).isEqualTo(201);
+    }
+}
diff --git a/...s/hibernate/orm/rest/data/panache/deployment/entity/PanacheEntityResourceDevModeTest.java b/...s/hibernate/orm/rest/data/panache/deployment/entity/PanacheEntityResourceDevModeTest.java
@@ -0,0 +1,19 @@
+package io.quarkus.hibernate.orm.rest.data.panache.deployment.entity;
+
+import org.jboss.shrinkwrap.api.ShrinkWrap;
+import org.jboss.shrinkwrap.api.spec.JavaArchive;
+import org.junit.jupiter.api.extension.RegisterExtension;
+
+import io.quarkus.hibernate.orm.rest.data.panache.deployment.AbstractDevModeTest;
+import io.quarkus.test.QuarkusDevModeTest;
+
+public class PanacheEntityResourceDevModeTest extends AbstractDevModeTest {
+
+    @RegisterExtension
+    static final QuarkusDevModeTest TEST = new QuarkusDevModeTest()
+            .setArchiveProducer(() -> ShrinkWrap.create(JavaArchive.class)
+                    .addClasses(Collection.class, AbstractItem.class, Item.class, ItemsController.class)
+                    .addAsResource("application.properties")
+                    .addAsResource("import.sql"));
+
+}
diff --git a/...ate/orm/rest/data/panache/deployment/repository/PanacheRepositoryResourceDevModeTest.java b/...ate/orm/rest/data/panache/deployment/repository/PanacheRepositoryResourceDevModeTest.java
@@ -0,0 +1,19 @@
+package io.quarkus.hibernate.orm.rest.data.panache.deployment.repository;
+
+import org.jboss.shrinkwrap.api.ShrinkWrap;
+import org.jboss.shrinkwrap.api.spec.JavaArchive;
+import org.junit.jupiter.api.extension.RegisterExtension;
+
+import io.quarkus.hibernate.orm.rest.data.panache.deployment.AbstractDevModeTest;
+import io.quarkus.test.QuarkusDevModeTest;
+
+public class PanacheRepositoryResourceDevModeTest extends AbstractDevModeTest {
+
+    @RegisterExtension
+    static final QuarkusDevModeTest TEST = new QuarkusDevModeTest()
+            .setArchiveProducer(() -> ShrinkWrap.create(JavaArchive.class)
+                    .addClasses(Collection.class, AbstractItem.class, Item.class, ItemsController.class, ItemsRepository.class)
+                    .addAsResource("application.properties")
+                    .addAsResource("import.sql"));
+
+}
diff --git a/...me/src/main/java/io/quarkus/rest/data/panache/runtime/resource/ResourceLinksProvider.java b/...me/src/main/java/io/quarkus/rest/data/panache/runtime/resource/ResourceLinksProvider.java
@@ -3,30 +3,30 @@
 import java.util.HashMap;
 import java.util.Map;
 
-import javax.ws.rs.core.UriInfo;
-
-import org.jboss.resteasy.core.ResourceMethodRegistry;
-import org.jboss.resteasy.core.ResteasyContext;
+import org.jboss.resteasy.links.LinksProvider;
 import org.jboss.resteasy.links.RESTServiceDiscovery;
-import org.jboss.resteasy.links.impl.ClassLinksProvider;
-import org.jboss.resteasy.links.impl.ObjectLinksProvider;
-import org.jboss.resteasy.spi.Registry;
 
 public final class ResourceLinksProvider {
 
     private static final String SELF_REF = "self";
 
     public Map<String, String> getClassLinks(Class<?> className) {
-        return linksToMap(getClassLinksProvider().getLinks(className));
+        RESTServiceDiscovery links = LinksProvider
+                .getClassLinksProvider()
+                .getLinks(className, Thread.currentThread().getContextClassLoader());
+        return linksToMap(links);
     }
 
     public Map<String, String> getInstanceLinks(Object instance) {
-        return linksToMap(getObjectLinksProvider().getLinks(instance));
+        RESTServiceDiscovery links = LinksProvider
+                .getObjectLinksProvider()
+                .getLinks(instance, Thread.currentThread().getContextClassLoader());
+        return linksToMap(links);
     }
 
     public String getSelfLink(Object instance) {
-        RESTServiceDiscovery.AtomLink link = getObjectLinksProvider()
-                .getLinks(instance)
+        RESTServiceDiscovery.AtomLink link = LinksProvider.getObjectLinksProvider()
+                .getLinks(instance, Thread.currentThread().getContextClassLoader())
                 .getLinkForRel(SELF_REF);
         return link == null ? null : link.getHref();
     }
@@ -38,16 +38,4 @@ private Map<String, String> linksToMap(RESTServiceDiscovery serviceDiscovery) {
         }
         return links;
     }
-
-    private ObjectLinksProvider getObjectLinksProvider() {
-        UriInfo uriInfo = ResteasyContext.getContextData(UriInfo.class);
-        ResourceMethodRegistry registry = (ResourceMethodRegistry) ResteasyContext.getContextData(Registry.class);
-        return new ObjectLinksProvider(uriInfo, registry);
-    }
-
-    private ClassLinksProvider getClassLinksProvider() {
-        UriInfo uriInfo = ResteasyContext.getContextData(UriInfo.class);
-        ResourceMethodRegistry registry = (ResourceMethodRegistry) ResteasyContext.getContextData(Registry.class);
-        return new ClassLinksProvider(uriInfo, registry);
-    }
 }