From 8baa85fd4c3607a7b3cebf2518a930a3b8a924bf Mon Sep 17 00:00:00 2001
From: Jan Tuomi <jans.tuomi@gmail.com>
Date: Fri, 9 Dec 2022 23:52:58 +0200
Subject: Add atom data type

---
 test/Spec.hs | 11 +++++++++++
 1 file changed, 11 insertions(+)

(limited to 'test/Spec.hs')

diff --git a/test/Spec.hs b/test/Spec.hs
index 4839ace..68e575c 100644
--- a/test/Spec.hs
+++ b/test/Spec.hs
@@ -126,7 +126,18 @@ e2eTests = testGroup "e2e" [
 
         let expectedLastAST = astRecord "Result/Ex" $
                 M.fromList [("value", astString "failed to read file: does-not-exist.txt")]
+        assertEqual "" expectedLastAST (last gotASTs),
+
+     do let env = builtinEnv
+        script1 <- readFile "test/scripts/atom1.milch"
+        (gotASTs, gotState) <- expectSuccessL env $ runInlineScript "<test>" script1
+
+        let expectedLastAST = astInteger 15
         assertEqual "" expectedLastAST (last gotASTs)
+
+        let expectedAtomMap = M.singleton (LAtomRef 1) (astInteger 15)
+        let gotAtomMap = stateAtomMap gotState
+        assertEqual "" expectedAtomMap gotAtomMap
     ]
 
 testGroup label xs = TestLabel label $ TestList $ map TestCase xs
-- 
cgit v1.3