Add tests for INPUT.

python-processing-unit · python-processing-unit · commit 1556b963e28c · 2026-04-13T12:26:09.000-04:00
diff --git a/tests/cases/failing/input-arity-two.pre b/tests/cases/failing/input-arity-two.pre
@@ -0,0 +1 @@
+INPUT("left", "right")
diff --git a/tests/cases/failing/input-type-bool.pre b/tests/cases/failing/input-type-bool.pre
@@ -0,0 +1 @@
+INPUT(TRUE)
diff --git a/tests/cases/failing/input-type-flt.pre b/tests/cases/failing/input-type-flt.pre
@@ -0,0 +1 @@
+INPUT(0d1.0)
diff --git a/tests/cases/failing/input-type-func.pre b/tests/cases/failing/input-type-func.pre
@@ -0,0 +1 @@
+INPUT(LAMBDA BOOL: (){})
diff --git a/tests/cases/failing/input-type-int.pre b/tests/cases/failing/input-type-int.pre
@@ -0,0 +1 @@
+INPUT(0d1)
diff --git a/tests/cases/failing/input-type-map.pre b/tests/cases/failing/input-type-map.pre
@@ -0,0 +1 @@
+INPUT(<>)
diff --git a/tests/cases/failing/input-type-thr.pre b/tests/cases/failing/input-type-thr.pre
@@ -0,0 +1 @@
+INPUT(ASYNC{})
diff --git a/tests/cases/failing/input-type-tns.pre b/tests/cases/failing/input-type-tns.pre
@@ -0,0 +1 @@
+INPUT([0d1])
diff --git a/tests/cases/passing/input-empty-line.ps1 b/tests/cases/passing/input-empty-line.ps1
@@ -0,0 +1,27 @@
+$helperPath = Join-Path (Split-Path -Parent (Split-Path -Parent $PSScriptRoot)) 'helpers\prefix-input.ps1'
+. $helperPath
+
+$tempDir = New-PrefixTempDir
+
+try {
+  $programPath = Join-Path $tempDir 'program.pre'
+
+  Set-Content -Path $programPath -Encoding Ascii -Value @'
+STR: line = INPUT()
+ASSERT(ISSTR(line))
+ASSERT(EQ(line, ""))
+'@
+
+  $result = Invoke-PrefixProgramWithInput -ProgramPath $programPath -InputText "`r`ntrailing`r`n"
+
+  Assert-PrefixSuccess $result
+  Assert-NoErrorOutput $result
+
+  if ($result.Stdout.Length -ne 0) {
+    $stdout = Format-VisibleText $result.Stdout
+    throw "Expected no stdout output, got: $stdout"
+  }
+}
+finally {
+  Remove-Item -Path $tempDir -Recurse -Force -ErrorAction SilentlyContinue
+}
diff --git a/tests/cases/passing/input-prompt-output.ps1 b/tests/cases/passing/input-prompt-output.ps1
@@ -0,0 +1,27 @@
+$helperPath = Join-Path (Split-Path -Parent (Split-Path -Parent $PSScriptRoot)) 'helpers\prefix-input.ps1'
+. $helperPath
+
+$tempDir = New-PrefixTempDir
+
+try {
+  $programPath = Join-Path $tempDir 'program.pre'
+
+  Set-Content -Path $programPath -Encoding Ascii -Value @'
+STR: line = INPUT("PROMPT>")
+ASSERT(EQ(line, "alpha"))
+PRINT("DONE=", line)
+'@
+
+  $result = Invoke-PrefixProgramWithInput -ProgramPath $programPath -InputText "alpha`r`n"
+
+  Assert-PrefixSuccess $result
+  Assert-NoErrorOutput $result
+
+  if (-not [regex]::IsMatch($result.Stdout, '\APROMPT>DONE=alpha(?:\r\n|\n)\z')) {
+    $stdout = Format-VisibleText $result.Stdout
+    throw "Unexpected stdout: $stdout"
+  }
+}
+finally {
+  Remove-Item -Path $tempDir -Recurse -Force -ErrorAction SilentlyContinue
+}
diff --git a/tests/cases/passing/input-return-type.ps1 b/tests/cases/passing/input-return-type.ps1
@@ -0,0 +1,27 @@
+$helperPath = Join-Path (Split-Path -Parent (Split-Path -Parent $PSScriptRoot)) 'helpers\prefix-input.ps1'
+. $helperPath
+
+$tempDir = New-PrefixTempDir
+
+try {
+  $programPath = Join-Path $tempDir 'program.pre'
+
+  Set-Content -Path $programPath -Encoding Ascii -Value @'
+STR: line = INPUT()
+ASSERT(ISSTR(line))
+ASSERT(EQ(line, "Alpha beta 42"))
+'@
+
+  $result = Invoke-PrefixProgramWithInput -ProgramPath $programPath -InputText "Alpha beta 42`r`nignored`r`n"
+
+  Assert-PrefixSuccess $result
+  Assert-NoErrorOutput $result
+
+  if ($result.Stdout.Length -ne 0) {
+    $stdout = Format-VisibleText $result.Stdout
+    throw "Expected no stdout output, got: $stdout"
+  }
+}
+finally {
+  Remove-Item -Path $tempDir -Recurse -Force -ErrorAction SilentlyContinue
+}
diff --git a/tests/cases/passing/input-sequential-lines.ps1 b/tests/cases/passing/input-sequential-lines.ps1
@@ -0,0 +1,26 @@
+$helperPath = Join-Path (Split-Path -Parent (Split-Path -Parent $PSScriptRoot)) 'helpers\prefix-input.ps1'
+. $helperPath
+
+$tempDir = New-PrefixTempDir
+
+try {
+  $programPath = Join-Path $tempDir 'program.pre'
+
+  Set-Content -Path $programPath -Encoding Ascii -Value @'
+ASSERT(EQ(INPUT(), "alpha"))
+ASSERT(EQ(INPUT(), "beta"))
+'@
+
+  $result = Invoke-PrefixProgramWithInput -ProgramPath $programPath -InputText "alpha`r`nbeta`r`ngamma`r`n"
+
+  Assert-PrefixSuccess $result
+  Assert-NoErrorOutput $result
+
+  if ($result.Stdout.Length -ne 0) {
+    $stdout = Format-VisibleText $result.Stdout
+    throw "Expected no stdout output, got: $stdout"
+  }
+}
+finally {
+  Remove-Item -Path $tempDir -Recurse -Force -ErrorAction SilentlyContinue
+}
diff --git a/tests/helpers/prefix-input.ps1 b/tests/helpers/prefix-input.ps1
@@ -0,0 +1,87 @@
+function Get-PrefixExePath {
+  $testsDir = Split-Path -Parent $PSScriptRoot
+  $prefixDir = Split-Path -Parent $testsDir
+  $exePath = Join-Path $prefixDir 'prefix.exe'
+
+  if (-not (Test-Path $exePath)) {
+    throw "Interpreter executable not found at: $exePath"
+  }
+
+  return $exePath
+}
+
+function New-PrefixTempDir {
+  $tempDir = Join-Path ([IO.Path]::GetTempPath()) ([IO.Path]::GetRandomFileName())
+  New-Item -ItemType Directory -Path $tempDir | Out-Null
+  return $tempDir
+}
+
+function Start-PrefixProcess {
+  param(
+    [Parameter(Mandatory = $true)]
+    [string]$ProgramPath
+  )
+
+  if (-not (Test-Path $ProgramPath)) {
+    throw "Program not found at: $ProgramPath"
+  }
+
+  $startInfo = New-Object System.Diagnostics.ProcessStartInfo
+  $startInfo.FileName = Get-PrefixExePath
+  $startInfo.Arguments = ('"{0}"' -f $ProgramPath)
+  $startInfo.RedirectStandardInput = $true
+  $startInfo.RedirectStandardOutput = $true
+  $startInfo.RedirectStandardError = $true
+  $startInfo.UseShellExecute = $false
+  $startInfo.CreateNoWindow = $true
+
+  return [System.Diagnostics.Process]::Start($startInfo)
+}
+
+function Invoke-PrefixProgramWithInput {
+  param(
+    [Parameter(Mandatory = $true)]
+    [string]$ProgramPath,
+
+    [Parameter(Mandatory = $true)]
+    [string]$InputText
+  )
+
+  $process = Start-PrefixProcess -ProgramPath $ProgramPath
+  try {
+    $process.StandardInput.Write($InputText)
+    $process.StandardInput.Close()
+
+    $stdout = $process.StandardOutput.ReadToEnd()
+    $stderr = $process.StandardError.ReadToEnd()
+    $process.WaitForExit()
+
+    return [pscustomobject]@{
+      ExitCode = $process.ExitCode
+      Stdout = $stdout
+      Stderr = $stderr
+    }
+  }
+  finally {
+    $process.Dispose()
+  }
+}
+
+function Format-VisibleText([string]$Text) {
+  return $Text.Replace("`r", '\r').Replace("`n", '\n')
+}
+
+function Assert-PrefixSuccess($Result) {
+  if ($Result.ExitCode -ne 0) {
+    $stdout = Format-VisibleText $Result.Stdout
+    $stderr = Format-VisibleText $Result.Stderr
+    throw "Prefix exited with code $($Result.ExitCode)`nSTDOUT: $stdout`nSTDERR: $stderr"
+  }
+}
+
+function Assert-NoErrorOutput($Result) {
+  if ($Result.Stderr.Length -ne 0) {
+    $stderr = Format-VisibleText $Result.Stderr
+    throw "Expected no stderr output, got: $stderr"
+  }
+}
