From 7ab2ed2ec14fd46e2ce366bcfacdf0fd83cb42ab Mon Sep 17 00:00:00 2001
From: Jaroslav Tulach <jaroslav.tulach@enso.org>
Date: Mon, 27 Nov 2023 16:23:47 +0100
Subject: [PATCH] BindingsMap and IR can be set to null

---
 .../enso/interpreter/runtime/ModuleTest.java  | 79 ++++++++++++++++++-
 1 file changed, 77 insertions(+), 2 deletions(-)

diff --git a/engine/runtime/src/test/java/org/enso/interpreter/runtime/ModuleTest.java b/engine/runtime/src/test/java/org/enso/interpreter/runtime/ModuleTest.java
index 4f287157e0e6..6564aa2ef461 100644
--- a/engine/runtime/src/test/java/org/enso/interpreter/runtime/ModuleTest.java
+++ b/engine/runtime/src/test/java/org/enso/interpreter/runtime/ModuleTest.java
@@ -1,22 +1,31 @@
 package org.enso.interpreter.runtime;
 
-import static org.junit.Assert.assertTrue;
-
 import java.io.File;
 import java.io.IOException;
 import java.nio.file.Paths;
 import java.util.logging.Level;
+
+import org.enso.compiler.data.BindingsMap;
+import org.enso.compiler.data.BindingsMap$ModuleReference$Concrete;
+import org.enso.interpreter.runtime.Module;
+import static org.enso.interpreter.util.ScalaConversions.nil;
 import org.enso.pkg.QualifiedName;
 import org.enso.polyglot.LanguageInfo;
 import org.enso.polyglot.MethodNames;
 import org.enso.polyglot.RuntimeOptions;
 import org.graalvm.polyglot.Context;
 import org.graalvm.polyglot.Engine;
+import org.graalvm.polyglot.Source;
 import org.graalvm.polyglot.io.IOAccess;
 import org.junit.After;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNull;
+import static org.junit.Assert.assertTrue;
 import org.junit.Before;
 import org.junit.Test;
 
+import scala.Option;
+
 public class ModuleTest {
 
   private File f;
@@ -60,4 +69,70 @@ public void moduleKeepsFileRefAfterSourceUnset() {
     assertTrue(
         "getPath is non-null", tFile.getPath() != null && module.getPath() == tFile.getPath());
   }
+
+  @Test
+  public void updaterCanNullTheBindings() throws Exception {
+    var name = QualifiedName.simpleName("SimpleExample");
+    var ensoContext =
+        (EnsoContext)
+            ctx.getBindings(LanguageInfo.ID)
+                .invokeMember(MethodNames.TopScope.LEAK_CONTEXT)
+                .asHostObject();
+    var tFile = ensoContext.getTruffleFile(f);
+
+    var code = Source.newBuilder("enso", """
+    main = 42
+    """, name.toString()).build();
+
+    ctx.eval(code);
+    var module = ensoContext.getTopScope().getModule(name.toString()).get().asCompilerModule();
+    ctx.enter();
+    var compilerContext = ensoContext.getCompiler().context();
+
+    assertNull("No bindings map by default", module.getBindingsMap());
+
+    var bindings = new BindingsMap(nil(), new BindingsMap$ModuleReference$Concrete(module));
+    compilerContext.updateModule(module, (u) -> {
+      u.bindingsMap(bindings);
+    });
+    assertEquals("Bindings map has changed", bindings, module.getBindingsMap());
+
+    compilerContext.updateModule(module, (u) -> {
+      u.bindingsMap(null);
+    });
+    assertNull("No bindings map again", module.getBindingsMap());
+  }
+
+  @Test
+  public void updaterCanNullTheIR() throws Exception {
+    var name = QualifiedName.simpleName("AnotherSimpleExample");
+    var ensoContext =
+        (EnsoContext)
+            ctx.getBindings(LanguageInfo.ID)
+                .invokeMember(MethodNames.TopScope.LEAK_CONTEXT)
+                .asHostObject();
+    var tFile = ensoContext.getTruffleFile(f);
+
+    var code = Source.newBuilder("enso", """
+    main = 42
+    """, name.toString()).build();
+
+    ctx.eval(code);
+    var module = ensoContext.getTopScope().getModule(name.toString()).get().asCompilerModule();
+    ctx.enter();
+    var compilerContext = ensoContext.getCompiler().context();
+
+    assertNull("No IR by default", module.getIr());
+
+    var ir = new org.enso.compiler.core.ir.Module(nil(), nil(), nil(), false, Option.empty(), null, null);
+    compilerContext.updateModule(module, (u) -> {
+      u.ir(ir);
+    });
+    assertEquals("IR has changed", ir, module.getIr());
+
+    compilerContext.updateModule(module, (u) -> {
+      u.ir(null);
+    });
+    assertNull("No IR again", module.getIr());
+  }
 }