setting the location of ReferenceSchema instances (the location where the $ref occured)

Author: erosb

core/src/main/java/org/everit/json/schema/loader/ReferenceLookup.java

@@ -166,6 +166,7 @@ Schema.Builder<?> lookup(String relPointerString, JsonObject ctx) {
                 .schemaJson(result.getQueryResult())
                 .rootSchemaJson(result.getContainingDocument()).build();
         Schema referredSchema = childLoader.load().build();
+        refBuilder.schemaLocation(SchemaLocation.parseURI(absPointerString));
         refBuilder.build().setReferredSchema(referredSchema);
         return refBuilder;
     }

core/src/test/java/org/everit/json/schema/loader/ReferenceLookupTest.java

@@ -32,12 +32,23 @@ public void before() {
     }
 
     private Schema performLookup(String pointerToRef) {
+        ReferenceSchema ref = obtainReferenceSchema(pointerToRef);
+        return ref.getReferredSchema();
+    }
+
+    private ReferenceSchema obtainReferenceSchema(String pointerToRef) {
         JsonObject jsonValue = query(pointerToRef).requireObject();
         ReferenceLookup subject = new ReferenceLookup(jsonValue.ls);
         String refPointer = jsonValue.require("$ref").requireString();
         Schema.Builder<?> actual = subject.lookup(refPointer, jsonValue);
-        ReferenceSchema ref = (ReferenceSchema) actual.build();
-        return ref.getReferredSchema();
+        return (ReferenceSchema) actual.build();
+    }
+
+    @Test
+    public void referenceSchemaLocationIsSet() {
+        when(httpClient.get("http://localhost/child-ref")).thenReturn(asStream(v4Subschema));
+        ReferenceSchema ref = obtainReferenceSchema("#/properties/definitionInRemote");
+        assertEquals("http://localhost/child-ref#/definitions/SubSchema", ref.getSchemaLocation());
     }
 
     @Test