Add tests for IMPORT.

python-processing-unit · python-processing-unit · commit e53d153aa711 · 2026-04-14T18:12:35.000-04:00
diff --git a/tests/cases/failing/import-alias-type-bool.pre b/tests/cases/failing/import-alias-type-bool.pre
@@ -0,0 +1 @@
+IMPORT(path, TRUE)
diff --git a/tests/cases/failing/import-alias-type-flt.pre b/tests/cases/failing/import-alias-type-flt.pre
@@ -0,0 +1 @@
+IMPORT(path, 0d1.0)
diff --git a/tests/cases/failing/import-alias-type-func.pre b/tests/cases/failing/import-alias-type-func.pre
@@ -0,0 +1 @@
+IMPORT(path, LAMBDA BOOL: (){})
diff --git a/tests/cases/failing/import-alias-type-int.pre b/tests/cases/failing/import-alias-type-int.pre
@@ -0,0 +1 @@
+IMPORT(path, 0d1)
diff --git a/tests/cases/failing/import-alias-type-map.pre b/tests/cases/failing/import-alias-type-map.pre
@@ -0,0 +1 @@
+IMPORT(path, <>)
diff --git a/tests/cases/failing/import-alias-type-str.pre b/tests/cases/failing/import-alias-type-str.pre
@@ -0,0 +1 @@
+IMPORT(path, "helper")
diff --git a/tests/cases/failing/import-alias-type-thr.pre b/tests/cases/failing/import-alias-type-thr.pre
@@ -0,0 +1 @@
+IMPORT(path, ASYNC{})
diff --git a/tests/cases/failing/import-alias-type-tns.pre b/tests/cases/failing/import-alias-type-tns.pre
@@ -0,0 +1 @@
+IMPORT(path, [0d1])
diff --git a/tests/cases/failing/import-arity-three.pre b/tests/cases/failing/import-arity-three.pre
@@ -0,0 +1 @@
+IMPORT(path, helper, extra)
diff --git a/tests/cases/failing/import-arity-zero.pre b/tests/cases/failing/import-arity-zero.pre
@@ -0,0 +1 @@
+IMPORT()
diff --git a/tests/cases/failing/import-missing-module.pre b/tests/cases/failing/import-missing-module.pre
@@ -0,0 +1 @@
+IMPORT(prefiximportmissingmoduletest)
diff --git a/tests/cases/failing/import-type-bool.pre b/tests/cases/failing/import-type-bool.pre
@@ -0,0 +1 @@
+IMPORT(TRUE)
diff --git a/tests/cases/failing/import-type-flt.pre b/tests/cases/failing/import-type-flt.pre
@@ -0,0 +1 @@
+IMPORT(0d1.0)
diff --git a/tests/cases/failing/import-type-func.pre b/tests/cases/failing/import-type-func.pre
@@ -0,0 +1 @@
+IMPORT(LAMBDA BOOL: (){})
diff --git a/tests/cases/failing/import-type-int.pre b/tests/cases/failing/import-type-int.pre
@@ -0,0 +1 @@
+IMPORT(0d1)
diff --git a/tests/cases/failing/import-type-map.pre b/tests/cases/failing/import-type-map.pre
@@ -0,0 +1 @@
+IMPORT(<>)
diff --git a/tests/cases/failing/import-type-str.pre b/tests/cases/failing/import-type-str.pre
@@ -0,0 +1 @@
+IMPORT("path")
diff --git a/tests/cases/failing/import-type-thr.pre b/tests/cases/failing/import-type-thr.pre
@@ -0,0 +1 @@
+IMPORT(ASYNC{})
diff --git a/tests/cases/failing/import-type-tns.pre b/tests/cases/failing/import-type-tns.pre
@@ -0,0 +1 @@
+IMPORT([0d1])
diff --git a/tests/cases/passing/import-alias-qualified-namespace.pre b/tests/cases/passing/import-alias-qualified-namespace.pre
@@ -0,0 +1,3 @@
+IMPORT(path, helper)
+
+ASSERT(EQ(TYPE(helper.TEMPFILE("import-alias-qualified-namespace.txt")), "STR"))
diff --git a/tests/cases/passing/import-cached-alias.ps1 b/tests/cases/passing/import-cached-alias.ps1
@@ -0,0 +1,43 @@
+$helperPath = Join-Path (Split-Path -Parent (Split-Path -Parent $PSScriptRoot)) 'helpers\prefix-input.ps1'
+. $helperPath
+
+$tempDir = New-PrefixTempDir
+$moduleName = 'prefiximport' + ([IO.Path]::GetRandomFileName().Replace('.', ''))
+
+try {
+  $markerPath = Join-Path $tempDir 'marker.txt'
+  $markerLiteral = $markerPath.Replace('\', '/')
+  $programPath = Join-Path $tempDir 'program.pre'
+  $modulePath = Join-Path $tempDir "$moduleName.pre"
+
+  Set-Content -Path $modulePath -Encoding Ascii -Value @"
+STR: marker = "$markerLiteral"
+
+IF(EXISTFILE(marker)){
+    WRITEFILE("reloaded", marker)
+} ELSE {
+    WRITEFILE("loaded", marker)
+}
+
+FUNC STR: marker_text(){
+    RETURN(READFILE(marker))
+}
+"@
+
+  Set-Content -Path $programPath -Encoding Ascii -Value @"
+IMPORT($moduleName)
+IMPORT($moduleName, again)
+
+ASSERT(EQ($moduleName.marker_text(), "loaded"))
+ASSERT(EQ(again.marker_text(), "loaded"))
+"@
+
+  $output = & (Get-PrefixExePath) $programPath 2>&1
+  if ($LASTEXITCODE -ne 0) {
+    $outputText = ($output | ForEach-Object { $_.ToString() }) -join [Environment]::NewLine
+    throw "Prefix exited with code $LASTEXITCODE`n$outputText"
+  }
+}
+finally {
+  Remove-Item -Path $tempDir -Recurse -Force -ErrorAction SilentlyContinue
+}
diff --git a/tests/cases/passing/import-isolated-top-level.ps1 b/tests/cases/passing/import-isolated-top-level.ps1
@@ -0,0 +1,36 @@
+$helperPath = Join-Path (Split-Path -Parent (Split-Path -Parent $PSScriptRoot)) 'helpers\prefix-input.ps1'
+. $helperPath
+
+$tempDir = New-PrefixTempDir
+$moduleName = 'prefiximport' + ([IO.Path]::GetRandomFileName().Replace('.', ''))
+
+try {
+  $programPath = Join-Path $tempDir 'program.pre'
+  $modulePath = Join-Path $tempDir "$moduleName.pre"
+
+  Set-Content -Path $modulePath -Encoding Ascii -Value @'
+INT: shared = 0d7
+
+FUNC INT: read_shared(){
+    RETURN(shared)
+}
+'@
+
+  Set-Content -Path $programPath -Encoding Ascii -Value @"
+INT: shared = 0d11
+
+IMPORT($moduleName, helper)
+
+ASSERT(EQ(helper.read_shared(), 0d7))
+ASSERT(EQ(shared, 0d11))
+"@
+
+  $output = & (Get-PrefixExePath) $programPath 2>&1
+  if ($LASTEXITCODE -ne 0) {
+    $outputText = ($output | ForEach-Object { $_.ToString() }) -join [Environment]::NewLine
+    throw "Prefix exited with code $LASTEXITCODE`n$outputText"
+  }
+}
+finally {
+  Remove-Item -Path $tempDir -Recurse -Force -ErrorAction SilentlyContinue
+}
diff --git a/tests/cases/passing/import-local-before-stdlib.ps1 b/tests/cases/passing/import-local-before-stdlib.ps1
@@ -0,0 +1,40 @@
+$helperPath = Join-Path (Split-Path -Parent (Split-Path -Parent $PSScriptRoot)) 'helpers\prefix-input.ps1'
+. $helperPath
+
+$prefixDir = Split-Path -Parent (Split-Path -Parent (Split-Path -Parent $PSScriptRoot))
+$tempDir = New-PrefixTempDir
+$moduleName = 'prefiximport' + ([IO.Path]::GetRandomFileName().Replace('.', ''))
+$stdlibPath = Join-Path $prefixDir "lib\std\$moduleName.pre"
+
+try {
+  $programPath = Join-Path $tempDir 'program.pre'
+  $localModulePath = Join-Path $tempDir "$moduleName.pre"
+
+  Set-Content -Path $localModulePath -Encoding Ascii -Value @'
+FUNC STR: identify(){
+    RETURN("local")
+}
+'@
+
+  Set-Content -Path $stdlibPath -Encoding Ascii -Value @'
+FUNC STR: identify(){
+    RETURN("stdlib")
+}
+'@
+
+  Set-Content -Path $programPath -Encoding Ascii -Value @"
+IMPORT($moduleName)
+
+ASSERT(EQ($moduleName.identify(), "local"))
+"@
+
+  $output = & (Get-PrefixExePath) $programPath 2>&1
+  if ($LASTEXITCODE -ne 0) {
+    $outputText = ($output | ForEach-Object { $_.ToString() }) -join [Environment]::NewLine
+    throw "Prefix exited with code $LASTEXITCODE`n$outputText"
+  }
+}
+finally {
+  Remove-Item -Path $tempDir -Recurse -Force -ErrorAction SilentlyContinue
+  Remove-Item -Path $stdlibPath -Force -ErrorAction SilentlyContinue
+}
diff --git a/tests/cases/passing/import-no-companion-prex.ps1 b/tests/cases/passing/import-no-companion-prex.ps1
@@ -0,0 +1,34 @@
+$helperPath = Join-Path (Split-Path -Parent (Split-Path -Parent $PSScriptRoot)) 'helpers\prefix-input.ps1'
+. $helperPath
+
+$tempDir = New-PrefixTempDir
+$moduleName = 'prefiximport' + ([IO.Path]::GetRandomFileName().Replace('.', ''))
+
+try {
+  $programPath = Join-Path $tempDir 'program.pre'
+  $modulePath = Join-Path $tempDir "$moduleName.pre"
+  $pointerPath = Join-Path $tempDir "$moduleName.prex"
+
+  Set-Content -Path $modulePath -Encoding Ascii -Value @'
+FUNC STR: identify(){
+    RETURN("source")
+}
+'@
+
+  Set-Content -Path $pointerPath -Encoding Ascii -Value 'this companion pointer file must be ignored by IMPORT'
+
+  Set-Content -Path $programPath -Encoding Ascii -Value @"
+IMPORT($moduleName)
+
+ASSERT(EQ($moduleName.identify(), "source"))
+"@
+
+  $output = & (Get-PrefixExePath) $programPath 2>&1
+  if ($LASTEXITCODE -ne 0) {
+    $outputText = ($output | ForEach-Object { $_.ToString() }) -join [Environment]::NewLine
+    throw "Prefix exited with code $LASTEXITCODE`n$outputText"
+  }
+}
+finally {
+  Remove-Item -Path $tempDir -Recurse -Force -ErrorAction SilentlyContinue
+}
diff --git a/tests/cases/passing/import-package-init.ps1 b/tests/cases/passing/import-package-init.ps1
@@ -0,0 +1,34 @@
+$helperPath = Join-Path (Split-Path -Parent (Split-Path -Parent $PSScriptRoot)) 'helpers\prefix-input.ps1'
+. $helperPath
+
+$tempDir = New-PrefixTempDir
+$packageName = 'prefixpkg' + ([IO.Path]::GetRandomFileName().Replace('.', ''))
+
+try {
+  $programPath = Join-Path $tempDir 'program.pre'
+  $packageDir = Join-Path $tempDir $packageName
+  $initPath = Join-Path $packageDir 'init.pre'
+
+  New-Item -ItemType Directory -Path $packageDir | Out-Null
+
+  Set-Content -Path $initPath -Encoding Ascii -Value @'
+FUNC STR: identify(){
+    RETURN("package")
+}
+'@
+
+  Set-Content -Path $programPath -Encoding Ascii -Value @"
+IMPORT($packageName)
+
+ASSERT(EQ($packageName.identify(), "package"))
+"@
+
+  $output = & (Get-PrefixExePath) $programPath 2>&1
+  if ($LASTEXITCODE -ne 0) {
+    $outputText = ($output | ForEach-Object { $_.ToString() }) -join [Environment]::NewLine
+    throw "Prefix exited with code $LASTEXITCODE`n$outputText"
+  }
+}
+finally {
+  Remove-Item -Path $tempDir -Recurse -Force -ErrorAction SilentlyContinue
+}
diff --git a/tests/cases/passing/import-package-missing-init-errors.ps1 b/tests/cases/passing/import-package-missing-init-errors.ps1
@@ -0,0 +1,32 @@
+$helperPath = Join-Path (Split-Path -Parent (Split-Path -Parent $PSScriptRoot)) 'helpers\prefix-input.ps1'
+. $helperPath
+
+$tempDir = New-PrefixTempDir
+$packageName = 'prefixpkg' + ([IO.Path]::GetRandomFileName().Replace('.', ''))
+
+try {
+  $programPath = Join-Path $tempDir 'program.pre'
+  $modulePath = Join-Path $tempDir "$packageName.pre"
+  $packageDir = Join-Path $tempDir $packageName
+
+  New-Item -ItemType Directory -Path $packageDir | Out-Null
+
+  Set-Content -Path $modulePath -Encoding Ascii -Value @'
+FUNC STR: identify(){
+    RETURN("module")
+}
+'@
+
+  Set-Content -Path $programPath -Encoding Ascii -Value @"
+IMPORT($packageName)
+"@
+
+  $result = Invoke-PrefixProgramWithInput -ProgramPath $programPath -InputText ''
+  if ($result.ExitCode -eq 0) {
+    $outputText = "STDOUT: $(Format-VisibleText $result.Stdout)`nSTDERR: $(Format-VisibleText $result.Stderr)"
+    throw "Expected IMPORT($packageName) to fail when the package directory has no init.pre`n$outputText"
+  }
+}
+finally {
+  Remove-Item -Path $tempDir -Recurse -Force -ErrorAction SilentlyContinue
+}
diff --git a/tests/cases/passing/import-package-precedence.ps1 b/tests/cases/passing/import-package-precedence.ps1
@@ -0,0 +1,41 @@
+$helperPath = Join-Path (Split-Path -Parent (Split-Path -Parent $PSScriptRoot)) 'helpers\prefix-input.ps1'
+. $helperPath
+
+$tempDir = New-PrefixTempDir
+$packageName = 'prefixpkg' + ([IO.Path]::GetRandomFileName().Replace('.', ''))
+
+try {
+  $programPath = Join-Path $tempDir 'program.pre'
+  $modulePath = Join-Path $tempDir "$packageName.pre"
+  $packageDir = Join-Path $tempDir $packageName
+  $initPath = Join-Path $packageDir 'init.pre'
+
+  New-Item -ItemType Directory -Path $packageDir | Out-Null
+
+  Set-Content -Path $modulePath -Encoding Ascii -Value @'
+FUNC STR: identify(){
+    RETURN("module")
+}
+'@
+
+  Set-Content -Path $initPath -Encoding Ascii -Value @'
+FUNC STR: identify(){
+    RETURN("package")
+}
+'@
+
+  Set-Content -Path $programPath -Encoding Ascii -Value @"
+IMPORT($packageName)
+
+ASSERT(EQ($packageName.identify(), "package"))
+"@
+
+  $output = & (Get-PrefixExePath) $programPath 2>&1
+  if ($LASTEXITCODE -ne 0) {
+    $outputText = ($output | ForEach-Object { $_.ToString() }) -join [Environment]::NewLine
+    throw "Prefix exited with code $LASTEXITCODE`n$outputText"
+  }
+}
+finally {
+  Remove-Item -Path $tempDir -Recurse -Force -ErrorAction SilentlyContinue
+}
diff --git a/tests/cases/passing/import-qualified-namespace.pre b/tests/cases/passing/import-qualified-namespace.pre
@@ -0,0 +1,3 @@
+IMPORT(path)
+
+ASSERT(EQ(TYPE(path.TEMPFILE("import-qualified-namespace.txt")), "STR"))
diff --git a/tests/cases/passing/import-qualified-only.ps1 b/tests/cases/passing/import-qualified-only.ps1
@@ -0,0 +1,29 @@
+$helperPath = Join-Path (Split-Path -Parent (Split-Path -Parent $PSScriptRoot)) 'helpers\prefix-input.ps1'
+. $helperPath
+
+$tempDir = New-PrefixTempDir
+$moduleName = 'prefiximport' + ([IO.Path]::GetRandomFileName().Replace('.', ''))
+
+try {
+  $programPath = Join-Path $tempDir 'program.pre'
+  $modulePath = Join-Path $tempDir "$moduleName.pre"
+
+  Set-Content -Path $modulePath -Encoding Ascii -Value @'
+INT: hidden = 0d7
+'@
+
+  Set-Content -Path $programPath -Encoding Ascii -Value @"
+IMPORT($moduleName)
+
+ASSERT(EQ(hidden, 0d7))
+"@
+
+  $result = Invoke-PrefixProgramWithInput -ProgramPath $programPath -InputText ''
+  if ($result.ExitCode -eq 0) {
+    $outputText = "STDOUT: $(Format-VisibleText $result.Stdout)`nSTDERR: $(Format-VisibleText $result.Stderr)"
+    throw "Expected unqualified module binding access to fail after IMPORT($moduleName)`n$outputText"
+  }
+}
+finally {
+  Remove-Item -Path $tempDir -Recurse -Force -ErrorAction SilentlyContinue
+}
diff --git a/tests/cases/passing/import-return-false.pre b/tests/cases/passing/import-return-false.pre
@@ -0,0 +1,2 @@
+ASSERT(EQ(TYPE(IMPORT(path)), "BOOL"))
+ASSERT(NOT(IMPORT(path)))
diff --git a/tests/cases/passing/import-stdlib-before-userlib.ps1 b/tests/cases/passing/import-stdlib-before-userlib.ps1
diff --git a/tests/helpers/prefix-input.ps1 b/tests/helpers/prefix-input.ps1
