Fix more failing unit tests

lf-lang · Oct 13, 2023 · 23d65a5 · 23d65a5
1 parent 2c98113
commit 23d65a5
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/core/src/main/java/org/lflang/AbstractTargetProperty.java b/core/src/main/java/org/lflang/AbstractTargetProperty.java
@@ -167,4 +167,14 @@ public void update(TargetConfig config, Element node, MessageReporter reporter)
   public void update(TargetConfig config, String value, MessageReporter reporter) {
     this.update(config, fromString(value, reporter));
   }
+
+  @Override
+  public boolean equals(Object obj) {
+    return obj.getClass().getName().equals(this.getClass().getName());
+  }
+
+  @Override
+  public int hashCode() {
+    return this.getClass().getName().hashCode();
+  }
 }
diff --git a/core/src/main/java/org/lflang/federated/generator/FedGenerator.java b/core/src/main/java/org/lflang/federated/generator/FedGenerator.java
@@ -422,8 +422,11 @@ private void analyzeFederates(Reactor federation, LFGeneratorContext context) {
         && !federation.getHost().getAddr().equals("localhost")) {
       rtiConfig.setHost(federation.getHost().getAddr());
     }
+
+    var d = new DockerProperty();
+    var x = targetConfig.get(d);
     // If the federation is dockerized, use "rti" as the hostname.
-    if (rtiConfig.getHost().equals("localhost") && targetConfig.get(new DockerProperty()).enabled) {
+    if (rtiConfig.getHost().equals("localhost") && x.enabled) {
       rtiConfig.setHost("rti");
     }
 

diff --git a/core/src/main/java/org/lflang/target/TargetConfig.java b/core/src/main/java/org/lflang/target/TargetConfig.java
@@ -189,7 +189,7 @@ public <T, S extends TargetPropertyType> void override(
 
   /** Reset this target property to its initial value (and mark it as unset). */
   public void reset(AbstractTargetProperty<?, ?> property) {
-    this.properties.remove(property);
+    this.properties.put(property, property.initialValue());
     this.setProperties.remove(property);
   }