Add tests for MAIN.

Author: python-processing-unit

tests/cases/failing/main-arity-one.pre

@@ -0,0 +1 @@
+MAIN(0d1)

tests/cases/passing/main-imported-function.ps1

@@ -0,0 +1,37 @@
+$prefixDir = Split-Path -Parent (Split-Path -Parent (Split-Path -Parent $PSScriptRoot))
+$exePath = Join-Path $prefixDir 'prefix.exe'
+
+if (-not (Test-Path $exePath)) {
+  throw "Interpreter executable not found at: $exePath"
+}
+
+$tempDir = Join-Path ([IO.Path]::GetTempPath()) ([IO.Path]::GetRandomFileName())
+New-Item -ItemType Directory -Path $tempDir | Out-Null
+
+try {
+  $modulePath = Join-Path $tempDir 'main_func_wrapper.pre'
+  $programPath = Join-Path $tempDir 'program.pre'
+
+  Set-Content -Path $modulePath -Encoding Ascii -Value @'
+FUNC BOOL: imported_main(){
+    RETURN(MAIN())
+}
+'@
+
+  $moduleLiteral = $modulePath.Replace('\', '\\')
+
+  Set-Content -Path $programPath -Encoding Ascii -Value @"
+IMPORT_PATH("$moduleLiteral", helper)
+
+ASSERT(NOT(helper.imported_main()))
+"@
+
+  $output = & $exePath $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
+}

tests/cases/passing/main-imported-top-level.ps1

@@ -0,0 +1,40 @@
+$prefixDir = Split-Path -Parent (Split-Path -Parent (Split-Path -Parent $PSScriptRoot))
+$exePath = Join-Path $prefixDir 'prefix.exe'
+
+if (-not (Test-Path $exePath)) {
+  throw "Interpreter executable not found at: $exePath"
+}
+
+$tempDir = Join-Path ([IO.Path]::GetTempPath()) ([IO.Path]::GetRandomFileName())
+New-Item -ItemType Directory -Path $tempDir | Out-Null
+
+try {
+  $modulePath = Join-Path $tempDir 'main_top_result.pre'
+  $programPath = Join-Path $tempDir 'program.pre'
+
+  Set-Content -Path $modulePath -Encoding Ascii -Value @'
+BOOL: top_result = MAIN()
+
+FUNC BOOL: read_top_result(){
+    RETURN(top_result)
+}
+'@
+
+  $moduleLiteral = $modulePath.Replace('\', '\\')
+
+  Set-Content -Path $programPath -Encoding Ascii -Value @"
+IMPORT_PATH("$moduleLiteral", helper)
+
+ASSERT(NOT(helper.read_top_result()))
+ASSERT(EQ(TYPE(helper.read_top_result()), "BOOL"))
+"@
+
+  $output = & $exePath $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
+}

tests/cases/passing/main-in-condition.pre

@@ -0,0 +1,9 @@
+BOOL: branch_taken = FALSE
+
+IF(MAIN()){
+    branch_taken = TRUE
+} ELSE {
+    ASSERT(FALSE)
+}
+
+ASSERT(branch_taken)

tests/cases/passing/main-primary-function.pre

@@ -0,0 +1,5 @@
+FUNC BOOL: from_primary_function(){
+    RETURN(MAIN())
+}
+
+ASSERT(from_primary_function())

tests/cases/passing/main-return-type.pre

@@ -0,0 +1 @@
+ASSERT(EQ(TYPE(MAIN()), "BOOL"))

tests/cases/passing/main-source-location.ps1

@@ -0,0 +1,50 @@
+$prefixDir = Split-Path -Parent (Split-Path -Parent (Split-Path -Parent $PSScriptRoot))
+$exePath = Join-Path $prefixDir 'prefix.exe'
+
+if (-not (Test-Path $exePath)) {
+  throw "Interpreter executable not found at: $exePath"
+}
+
+$tempDir = Join-Path ([IO.Path]::GetTempPath()) ([IO.Path]::GetRandomFileName())
+New-Item -ItemType Directory -Path $tempDir | Out-Null
+
+try {
+  $modulePath = Join-Path $tempDir 'main_func_wrapper.pre'
+  $programPath = Join-Path $tempDir 'program.pre'
+
+  Set-Content -Path $modulePath -Encoding Ascii -Value @'
+FUNC BOOL: imported_main(){
+    RETURN(MAIN())
+}
+
+FUNC BOOL: call_callback(FUNC: callback){
+    RETURN(callback())
+}
+
+FUNC BOOL: call_imported_main(){
+    RETURN(imported_main())
+}
+'@
+
+  $moduleLiteral = $modulePath.Replace('\', '\\')
+
+  Set-Content -Path $programPath -Encoding Ascii -Value @"
+IMPORT_PATH("$moduleLiteral", helper)
+
+FUNC BOOL: local_main(){
+    RETURN(MAIN())
+}
+
+ASSERT(helper.call_callback(local_main))
+ASSERT(NOT(helper.call_imported_main()))
+"@
+
+  $output = & $exePath $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
+}

tests/cases/passing/main-top-level.pre

@@ -0,0 +1 @@
+ASSERT(MAIN())