new: test 1 undefined Local TyRef

Author: cstml

lambda-buffers-compiler/test/Test/Utils/CompilerInput.hs

@@ -1,8 +1,13 @@
-module Test.Utils.CompilerInput (compilerInput'incoherent, compilerInput'maybe, compilerInput'undefinedVariable) where
+module Test.Utils.CompilerInput (
+  compilerInput'incoherent,
+  compilerInput'maybe,
+  compilerInput'undefinedVariable,
+  compilerInput'undefinedLocalTyRef,
+) where
 
 import LambdaBuffers.Compiler.ProtoCompat qualified as P
 import Test.Utils.Constructors (_CompilerInput)
-import Test.Utils.Module (module'incoherent, module'maybe, module'undefinedVar)
+import Test.Utils.Module (module'incoherent, module'maybe, module'undefinedVar, module'undefinedlocalTyRef)
 
 -- | Compiler Input containing 1 module with 1 definition - Maybe.
 compilerInput'maybe :: P.CompilerInput
@@ -15,3 +20,7 @@ compilerInput'incoherent = _CompilerInput [module'maybe, module'incoherent]
 -- | Contains 1 undefined variable.
 compilerInput'undefinedVariable :: P.CompilerInput
 compilerInput'undefinedVariable = _CompilerInput [module'undefinedVar]
+
+-- | Contains 1 undefined Local TyRef.
+compilerInput'undefinedLocalTyRef :: P.CompilerInput
+compilerInput'undefinedLocalTyRef = _CompilerInput [module'undefinedlocalTyRef]

lambda-buffers-compiler/test/Test/Utils/Constructors.hs

@@ -19,6 +19,8 @@ module Test.Utils.Constructors (
   _ModuleNamePart,
   _TyVarI',
   _SourceInfo,
+  _LocalRef,
+  _LocalRef',
 ) where
 
 import Control.Lens ((^.))
@@ -133,10 +135,15 @@ _TyDef :: P.TyName -> P.TyAbs -> P.TyDef
 _TyDef name ab = P.TyDef {P.tyName = name, P.tyAbs = ab, sourceInfo = P.defSourceInfo}
 
 _TyRefILocal :: Text -> P.Ty
-_TyRefILocal x =
-  P.TyRefI $
-    P.LocalI $
-      P.LocalRef
-        { P.tyName = P.TyName {P.name = x, sourceInfo = P.defSourceInfo}
-        , P.sourceInfo = P.defSourceInfo
-        }
+_TyRefILocal x = P.TyRefI $ P.LocalI $ _LocalRef x
+
+_LocalRef :: Text -> P.LocalRef
+_LocalRef = flip _LocalRef' P.defSourceInfo
+
+-- | LocalRef with Source Info - for error precision testing.
+_LocalRef' :: Text -> P.SourceInfo -> P.LocalRef
+_LocalRef' x s =
+  P.LocalRef
+    { P.tyName = P.TyName {P.name = x, sourceInfo = s}
+    , P.sourceInfo = s
+    }

lambda-buffers-compiler/test/Test/Utils/Module.hs

@@ -1,8 +1,8 @@
-module Test.Utils.Module (module'maybe, module'incoherent, module'undefinedVar) where
+module Test.Utils.Module (module'maybe, module'incoherent, module'undefinedVar, module'undefinedlocalTyRef) where
 
 import LambdaBuffers.Compiler.ProtoCompat qualified as P
 import Test.Utils.Constructors (_Module, _ModuleName)
-import Test.Utils.TyDef (tyDef'incoherent, tyDef'maybe, tyDef'undefinedVar)
+import Test.Utils.TyDef (tyDef'incoherent, tyDef'maybe, tyDef'undefinedVar, tyDef'undefinedlocalTyRef)
 
 -- _Module mn tds cds ins =
 
@@ -28,3 +28,11 @@ module'incoherent = _Module (_ModuleName ["Module"]) [tyDef'maybe, tyDef'incoher
 -}
 module'undefinedVar :: P.Module
 module'undefinedVar = _Module (_ModuleName ["Module"]) [tyDef'undefinedVar] mempty mempty
+
+{- | 1 Module containing
+  Foo b = Bar Baz b
+              ^^^
+  Should fail as Baz is a local undefined Ty Ref.
+-}
+module'undefinedlocalTyRef :: P.Module
+module'undefinedlocalTyRef = _Module (_ModuleName ["Module"]) [tyDef'undefinedlocalTyRef] mempty mempty

lambda-buffers-compiler/test/Test/Utils/TyDef.hs

@@ -1,8 +1,16 @@
-module Test.Utils.TyDef (tyDef'maybe, tyDef'incoherent, tyDef'undefinedVar, tyDef'undefinedVar'var) where
+module Test.Utils.TyDef (
+  tyDef'maybe,
+  tyDef'incoherent,
+  tyDef'undefinedVar,
+  tyDef'undefinedVar'var,
+  tyDef'undefinedlocalTyRef,
+  tyDef'undefinedlocalTyRef'localTyRef,
+) where
 
-import LambdaBuffers.Compiler.ProtoCompat qualified as P
 import LambdaBuffers.Compiler.ProtoCompat.Types (Ty (TyVarI))
+import LambdaBuffers.Compiler.ProtoCompat.Types qualified as P
 import Test.Utils.Constructors (
+  _LocalRef',
   _SourceInfo,
   _TupleI,
   _TyAbs,
@@ -45,6 +53,31 @@ tyDef'undefinedVar =
     (_TyName "Foo")
     (_TyAbs [] [("Bar", _TupleI [TyVarI tyDef'undefinedVar'var])])
 
--- | The undefined var in tyDef'undefinedVar - exported to see if the test identifies it correctly.
+{- | The undefined var (i.e. "b") in tyDef'undefinedVar.
+ Exported to see if the test identifies it correctly.
+-}
 tyDef'undefinedVar'var :: P.TyVar
 tyDef'undefinedVar'var = _TyVar' "b" (_SourceInfo 1 2)
+
+-- | Foo a = Foo Baz b
+tyDef'undefinedlocalTyRef :: P.TyDef
+tyDef'undefinedlocalTyRef =
+  _TyDef
+    (_TyName "Foo")
+    ( _TyAbs
+        [("a", _Type)]
+        [
+          ( "Foo"
+          , _TupleI
+              [ P.TyRefI tyDef'undefinedlocalTyRef'localTyRef
+              , _TyVarI "a"
+              ]
+          )
+        ]
+    )
+
+{- | The undefined Local TyRef (i.e. "Baz") in tyDef'undefinedlocalTyRef.
+ Exported to see if the test identifies it correctly.
+-}
+tyDef'undefinedlocalTyRef'localTyRef :: P.TyRef
+tyDef'undefinedlocalTyRef'localTyRef = P.LocalI $ _LocalRef' "Baz" (_SourceInfo 1 2)