fix unit tests

lf-lang · Mar 23, 2022 · f9dd4ec · f9dd4ec
1 parent a5a073b
commit f9dd4ec
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/org.lflang.tests/src/org/lflang/tests/compiler/LinguaFrancaValidationTest.java b/org.lflang.tests/src/org/lflang/tests/compiler/LinguaFrancaValidationTest.java
@@ -906,8 +906,8 @@ public void nonzeroAfterMustHaveUnits() throws Exception {
             "    b = new X()",
             "    a.y -> b.x after 1",
             "}");
-        validator.assertError(parseWithoutError(testCase), LfPackage.eINSTANCE.getTime(),
-            null, "Missing or invalid time unit.");
+        validator.assertError(parseWithoutError(testCase), LfPackage.eINSTANCE.getConnection(),
+            null, "Missing time unit.");
     }
 
 

diff --git a/org.lflang/src/org/lflang/validation/LFValidator.java b/org.lflang/src/org/lflang/validation/LFValidator.java
@@ -363,10 +363,10 @@ public void checkConnection(Connection connection) {
         // Check the after delay
         if (connection.getDelay() != null) {
             final var delay = connection.getDelay();
-            if (delay instanceof ParameterReference || delay instanceof Time) {
+            if (delay instanceof ParameterReference || delay instanceof Time || delay instanceof Literal) {
                 checkExpressionAsTime(delay, Literals.CONNECTION__DELAY);
             } else {
-                error("After delays can only be given by time literals or paramters.",
+                error("After delays can only be given by time literals or parameters.",
                       Literals.CONNECTION__DELAY);
             }
         }