Remove required entries when property is removed

CHANGELOG.md

@@ -1,5 +1,12 @@
 # Smithy Changelog
 
+## 1.0.4 (TBD)
+
+### Features
+
+* Ensure that when a property is removed from a JSON schema object, that a corresponding "required"
+  entry is also removed.
+
 ## 1.0.3 (2020-05-26)
 
 ### Bug Fixes

smithy-jsonschema/src/main/java/software/amazon/smithy/jsonschema/Schema.java

@@ -15,6 +15,7 @@
 
 package software.amazon.smithy.jsonschema;
 
+import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Collection;
 import java.util.Collections;
@@ -562,7 +563,7 @@ public static final class Builder implements SmithyBuilder<Schema> {
 
         private Integer maxProperties;
         private Integer minProperties;
-        private Collection<String> required = ListUtils.of();
+        private Collection<String> required = new ArrayList<>();
         private Map<String, Schema> properties = new HashMap<>();
         private Schema additionalProperties;
         private Schema propertyNames;
@@ -688,7 +689,11 @@ public Builder minProperties(Integer minProperties) {
         }
 
         public Builder required(Collection<String> required) {
-            this.required = required == null ? ListUtils.of() : required;
+            if (required == null) {
+                this.required.clear();
+            } else {
+                this.required = new ArrayList<>(required);
+            }
             return this;
         }
 
@@ -709,6 +714,7 @@ public Builder putProperty(String key, Schema value) {
 
         public Builder removeProperty(String key) {
             properties.remove(key);
+            required.remove(key);
             return this;
         }
 

smithy-jsonschema/src/test/java/software/amazon/smithy/jsonschema/SchemaTest.java

@@ -16,6 +16,7 @@
 package software.amazon.smithy.jsonschema;
 
 import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.contains;
 import static org.hamcrest.Matchers.equalTo;
 import static org.hamcrest.Matchers.not;
 
@@ -24,6 +25,7 @@
 import java.util.Set;
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.Test;
+import software.amazon.smithy.utils.ListUtils;
 import software.amazon.smithy.utils.SetUtils;
 
 public class SchemaTest {
@@ -180,4 +182,19 @@ public void ignoresNegativeNumericPositions() {
 
         assertThat(schema.selectSchema("allOf", "-1"), equalTo(Optional.empty()));
     }
+
+    @Test
+    public void removingPropertiesRemovesRequiredPropertiesToo() {
+        Schema schema = Schema.builder()
+                .removeProperty("notThere")
+                .required(null)
+                .putProperty("foo", Schema.builder().build())
+                .putProperty("bar", Schema.builder().build())
+                .required(ListUtils.of("foo", "bar"))
+                .removeProperty("foo")
+                .build();
+
+        assertThat(schema.getProperties().keySet(), contains("bar"));
+        assertThat(schema.getRequired(), contains("bar"));
+    }
 }