Allow schema exclusion to be package specific

pom.xml

@@ -7,7 +7,7 @@
   <modelVersion>4.0.0</modelVersion>
   <groupId>com.github.trickl</groupId>
   <artifactId>assertj-json-serialize</artifactId>
-  <version>0.1.8-SNAPSHOT</version>
+  <version>0.1.9-SNAPSHOT</version>
   <packaging>jar</packaging>
   <developers>
     <developer>

src/main/java/com/trickl/assertj/core/api/json/serialize/AbstractJsonObjectAssert.java

@@ -16,6 +16,8 @@
 import java.nio.file.Path;
 import java.nio.file.Paths;
 import java.security.CodeSource;
+import java.util.ArrayList;
+import java.util.List;
 
 import org.assertj.core.api.AbstractAssert;
 
@@ -45,7 +47,7 @@ public abstract class AbstractJsonObjectAssert<S extends AbstractJsonObjectAsser
 
   private String projectDir = null;
 
-  private boolean noInlineSchemas = false;
+  private List<String> excludeInlineSchemaPackages = new ArrayList<>();
 
   private String jsonDataFileExtension = DEFAULT_JSON_DATA_FILE_EXT;
 
@@ -184,8 +186,8 @@ public S doNotCreateExpectedIfAbsent() {
     return myself;
   }
 
-  public S withNoInlineSchemas() {
-    noInlineSchemas = true;
+  public S excludeInlineSchemaPackage(String packageName) {
+    excludeInlineSchemaPackages.add(packageName);
     return myself;
   }
 
@@ -218,8 +220,9 @@ protected String serialize(Object obj) {
   protected String schema(Object obj) {
     try {
       SchemaFactoryWrapper visitor = new SchemaFactoryWrapper();
-      if (noInlineSchemas) {
-        visitor.setVisitorContext(new NoInlineSchemaVisitorContext());
+      if (excludeInlineSchemaPackages.size() > 0) {
+        visitor.setVisitorContext(
+            new ExcludeInlineSchemaVisitorContext(excludeInlineSchemaPackages));
       }
       objectMapper.acceptJsonFormatVisitor(actual.getObject().getClass(), visitor);
       JsonSchema schema = visitor.finalSchema();

src/main/java/com/trickl/assertj/core/api/json/serialize/ExcludeInlineSchemaVisitorContext.java

@@ -0,0 +1,28 @@
+package com.trickl.assertj.core.api.json.serialize;
+
+import com.fasterxml.jackson.databind.JavaType;
+import com.fasterxml.jackson.module.jsonSchema.factories.VisitorContext;
+
+import java.util.List;
+
+import lombok.RequiredArgsConstructor;
+
+@RequiredArgsConstructor
+public class ExcludeInlineSchemaVisitorContext extends VisitorContext {
+  private final List<String> excludePackages;
+
+  @Override
+  public String getSeenSchemaUri(JavaType javaType) {
+    if (javaType != null && !javaType.isPrimitive() && isInExcludedPackage(javaType)) {
+      return javaTypeToUrn(javaType);
+    }
+    return null;
+  }
+
+  private boolean isInExcludedPackage(JavaType javaType) {
+    String packageName = javaType.getRawClass().getPackage().getName();
+    return excludePackages.stream().anyMatch((exclude -> 
+      packageName.equals(exclude) || packageName.startsWith(exclude + ".")
+    ));
+  }
+}

src/test/java/com/trickl/assertj/core/api/json/serialize/JsonObjectAssert_serializationAsExpected_Test.java

@@ -7,6 +7,7 @@
 import java.util.Map;
 
 import com.trickl.assertj.examples.Example;
+import com.trickl.assertj.examples.NestedComplexExample;
 import com.trickl.assertj.examples.NestedExample;
 
 import org.junit.Test;
@@ -72,10 +73,10 @@ public void should_pass_on_schema_match() {
   }
 
   @Test
-  public void should_respect_no_inlining() {         
+  public void should_respect_no_inlining_simple() {         
     NestedExample nested = new NestedExample();        
     assertThat(nested)
-        .withNoInlineSchemas()
+        .excludeInlineSchemaPackage("com.trickl.assertj.examples")
         .schemaAsExpected();     
   }
 
@@ -86,4 +87,12 @@ public void should_respect_allow_inlining() {
         .withSchemaFileExtension(".schema2.json")               
         .schemaAsExpected();     
   }
+
+  @Test
+  public void should_respect_no_inlining_complex() {         
+    NestedComplexExample nested = new NestedComplexExample();        
+    assertThat(nested)           
+        .excludeInlineSchemaPackage("com.trickl.assertj.examples")
+        .schemaAsExpected();     
+  }
 }

src/test/java/com/trickl/assertj/examples/NestedComplexExample.java

@@ -0,0 +1,17 @@
+package com.trickl.assertj.examples;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+
+import java.util.List;
+import java.util.Map;
+
+import lombok.Data;
+
+@Data
+public class NestedComplexExample {
+  @JsonProperty("list-example")    
+  private List<Example> firstExample;
+
+  @JsonProperty("map-example")
+  private Map<String, Example> secondExample;
+}

src/test/resources/com/trickl/assertj/examples/NestedComplexExample.schema.json

@@ -0,0 +1,20 @@
+{
+  "id": "urn:jsonschema:com:trickl:assertj:examples:NestedComplexExample",
+  "type": "object",
+  "properties": {
+    "list-example": {
+      "type": "array",
+      "items": {
+        "type": "object",
+        "$ref": "urn:jsonschema:com:trickl:assertj:examples:Example"
+      }
+    },
+    "map-example": {
+      "additionalProperties": {
+        "type": "object",
+        "$ref": "urn:jsonschema:com:trickl:assertj:examples:Example"
+      },
+      "type": "object"
+    }
+  }
+}