Ensure field is set to correct value

1 files changed, 7 insertions, 1 deletions

diff --git a/src/main/java/org/zwobble/couscous/interpreter/values/ObjectInterpreterValue.java b/src/main/java/org/zwobble/couscous/interpreter/values/ObjectInterpreterValue.java
index 7ff8bb1..f0ecbd1 100644
--- a/src/main/java/org/zwobble/couscous/interpreter/values/ObjectInterpreterValue.java
+++ b/src/main/java/org/zwobble/couscous/interpreter/values/ObjectInterpreterValue.java
@@ -5,8 +5,11 @@ import java.util.Map;
 import java.util.Optional;
 
 import org.zwobble.couscous.interpreter.NoSuchField;
+import org.zwobble.couscous.interpreter.UnexpectedValueType;
 import org.zwobble.couscous.values.PrimitiveValue;
 
+import lombok.val;
+
 public class ObjectInterpreterValue implements InterpreterValue {
     private final ConcreteType type;
     private final Map<String, InterpreterValue> fields;
@@ -36,7 +39,10 @@ public class ObjectInterpreterValue implements InterpreterValue {
     }
 
     public void setField(String fieldName, InterpreterValue value) {
-        type.getField(fieldName).orElseThrow(() -> new NoSuchField(fieldName));
+        val field = type.getField(fieldName).orElseThrow(() -> new NoSuchField(fieldName));
+        if (!field.getType().equals(value.getType().getName())) {
+            throw new UnexpectedValueType(field.getType(), value.getType().getName());
+        }
         fields.put(fieldName, value);   
     }
 }