test: Testing for null being passed to put()

Author: quintesse

src/main/java/org/codejive/properties/Properties.java

@@ -293,6 +293,9 @@ public String getRaw(String rawKey) {
 
     @Override
     public String put(String key, String value) {
+        if (key == null || value == null) {
+            throw new NullPointerException();
+        }
         String rawValue = escape(value, false);
         if (values.containsKey(key)) {
             replaceValue(key, rawValue, value);

src/test/java/org/codejive/properties/TestProperties.java

@@ -12,7 +12,6 @@
 import org.junit.jupiter.api.Test;
 
 public class TestProperties {
-
     @Test
     void testLoad() throws IOException, URISyntaxException {
         Properties p = Properties.loadProperties(getResource("/test.properties"));
@@ -303,6 +302,23 @@ void testPutFirstWithHeader() throws IOException, URISyntaxException {
         }
     }
 
+    @Test
+    void testPutNull() throws IOException, URISyntaxException {
+        Properties p = new Properties();
+        assertThatThrownBy(() -> {
+            p.put("one", null);
+        }).isInstanceOf(NullPointerException.class);
+        assertThatThrownBy(() -> {
+            p.setProperty("one", null);
+        }).isInstanceOf(NullPointerException.class);
+        assertThatThrownBy(() -> {
+            p.put(null, "value");
+        }).isInstanceOf(NullPointerException.class);
+        assertThatThrownBy(() -> {
+            p.setProperty(null, "value");
+        }).isInstanceOf(NullPointerException.class);
+    }
+
     @Test
     void testRemoveFirst() throws IOException, URISyntaxException {
         Properties p = Properties.loadProperties(getResource("/test.properties"));