1 files changed, 8 insertions, 0 deletions
diff --git a/src/test/java/org/zwobble/couscous/tests/ValueObjectTests.java b/src/test/java/org/zwobble/couscous/tests/ValueObjectTests.java
index 0768ee7..022e7ed 100644
--- a/src/test/java/org/zwobble/couscous/tests/ValueObjectTests.java
+++ b/src/test/java/org/zwobble/couscous/tests/ValueObjectTests.java
@@ -3,7 +3,9 @@ package org.zwobble.couscous.tests;
 import java.lang.reflect.Field;
 import java.lang.reflect.Method;
 import java.lang.reflect.Modifier;
+import java.util.Collections;
 import java.util.List;
+import java.util.Set;
 import java.util.function.Function;
 import java.util.stream.IntStream;
 import java.util.stream.Stream;
@@ -127,6 +129,8 @@ public class ValueObjectTests {
             return true;
         } else if (type.equals(List.class)) {
             return asList("[list 1]");
+        } else if (type.equals(Set.class)) {
+            return Collections.singleton("[set 1]");
         } else if (type.equals(PrimitiveValue.class)) {
             return generateFirstInstance(StringValue.class);
         } else if (type.equals(ExpressionNode.class)) {
@@ -145,6 +149,8 @@ public class ValueObjectTests {
             return false;
         } else if (type.equals(List.class)) {
             return asList("[list 2]");
+        } else if (type.equals(Set.class)) {
+            return Collections.singleton("[set 2]");
         } else if (type.equals(PrimitiveValue.class)) {
             return generateSecondInstance(StringValue.class);
         } else if (type.equals(ExpressionNode.class)) {
@@ -163,6 +169,8 @@ public class ValueObjectTests {
             return true;
         } else if (type.equals(List.class)) {
             return asList("[list]");
+        } else if (type.equals(Set.class)) {
+            return Collections.singleton("[set]");
         } else if (type.equals(PrimitiveValue.class)) {
             return generateInstance(StringValue.class);
         } else if (type.equals(ExpressionNode.class)) {