Add tests for EXTEND.

Author: python-processing-unit

tests/cases/failing/extend-asmodule-not-qualified.pre

@@ -0,0 +1,8 @@
+IMPORT(path)
+
+STR: tests_dir = path.BASEPATH(path.BASEPATH(path.script_dir))
+STR: helper_path = JOIN(tests_dir, "/helpers/extend_core_once.pre")
+
+IMPORT_PATH(helper_path, helper)
+
+helper.extend_core.TST_EXT_ASMODULE_ONLY()

tests/cases/failing/extend-missing-file.pre

@@ -0,0 +1 @@
+EXTEND(prefix_missing_extension_target)

tests/cases/failing/extend-missing-init-symbol.pre

@@ -0,0 +1,6 @@
+IMPORT(path)
+
+STR: tests_dir = path.BASEPATH(path.BASEPATH(path.script_dir))
+STR: helper_path = JOIN(tests_dir, "/helpers/extend_missing_init_loader.pre")
+
+IMPORT_PATH(helper_path)

tests/cases/failing/extend-package-missing-init.pre

@@ -0,0 +1,6 @@
+IMPORT(path)
+
+STR: tests_dir = path.BASEPATH(path.BASEPATH(path.script_dir))
+STR: helper_path = JOIN(tests_dir, "/helpers/extend_missing_package_loader.pre")
+
+IMPORT_PATH(helper_path)

tests/cases/failing/extend-prex-rejected.pre

@@ -0,0 +1,6 @@
+IMPORT(path)
+
+STR: tests_dir = path.BASEPATH(path.BASEPATH(path.script_dir))
+STR: helper_path = JOIN(tests_dir, "/helpers/extend_prex_loader.pre")
+
+IMPORT_PATH(helper_path)

tests/cases/failing/extend-restricted-not-global.pre

@@ -0,0 +1,8 @@
+IMPORT(path)
+
+STR: tests_dir = path.BASEPATH(path.BASEPATH(path.script_dir))
+STR: helper_path = JOIN(tests_dir, "/helpers/extend_core_once.pre")
+
+IMPORT_PATH(helper_path, helper)
+
+TST_EXT_RESTRICTED()

tests/cases/passing/extend-alias-qualified-namespace.pre

@@ -0,0 +1,10 @@
+IMPORT(path)
+
+STR: tests_dir = path.BASEPATH(path.BASEPATH(path.script_dir))
+STR: helper_path = JOIN(tests_dir, "/helpers/extend_core_once.pre")
+
+ASSERT(NOT(IMPORT_PATH(helper_path, first)))
+ASSERT(NOT(IMPORT_PATH(helper_path, second)))
+ASSERT(EQ(first.extend_core.TST_EXT_RESTRICTED(), "core-restricted"))
+ASSERT(EQ(second.extend_core.TST_EXT_RESTRICTED(), "core-restricted"))
+ASSERT(EQ(TST_EXT_INIT_COUNT(), 0d1))

tests/cases/passing/extend-asmodule-global.pre

@@ -0,0 +1,7 @@
+IMPORT(path)
+
+STR: tests_dir = path.BASEPATH(path.BASEPATH(path.script_dir))
+STR: helper_path = JOIN(tests_dir, "/helpers/extend_core_once.pre")
+
+ASSERT(NOT(IMPORT_PATH(helper_path, helper)))
+ASSERT(EQ(TST_EXT_ASMODULE_ONLY(), "asmodule-global"))

tests/cases/passing/extend-cli-multiple.ps1

@@ -0,0 +1,27 @@
+$helperPath = Join-Path (Split-Path -Parent (Split-Path -Parent $PSScriptRoot)) 'helpers\prefix-input.ps1'
+. $helperPath
+
+$tempDir = New-PrefixTempDir
+$coreExtension = Get-PrefixHelperExtensionPath 'extend_core'
+$packageExtension = Get-PrefixHelperExtensionPath 'extend_package\init'
+
+try {
+  $programPath = Join-Path $tempDir 'program.pre'
+
+  Set-Content -Path $programPath -Encoding Ascii -Value @'
+ASSERT(EQ(TST_EXT_GLOBAL(), "core-global"))
+ASSERT(EQ(TST_EXT_PACKAGE_INIT(), "package-init"))
+'@
+
+  $result = Invoke-PrefixWithArguments -Arguments @($coreExtension, $packageExtension, $programPath)
+  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
+}

tests/cases/passing/extend-cli-prex-rejected.ps1

@@ -0,0 +1,23 @@
+$helperPath = Join-Path (Split-Path -Parent (Split-Path -Parent $PSScriptRoot)) 'helpers\prefix-input.ps1'
+. $helperPath
+
+$tempDir = New-PrefixTempDir
+$prexPath = Join-Path (Split-Path -Parent $helperPath) 'extend_prex_only.prex'
+
+try {
+  $programPath = Join-Path $tempDir 'program.pre'
+
+  Set-Content -Path $programPath -Encoding Ascii -Value @'
+ASSERT(TRUE)
+'@
+
+  $result = Invoke-PrefixWithArguments -Arguments @($prexPath, $programPath)
+  if ($result.ExitCode -eq 0) {
+    $stdout = Format-VisibleText $result.Stdout
+    $stderr = Format-VisibleText $result.Stderr
+    throw "Expected the interpreter to reject a .prex CLI extension input`nSTDOUT: $stdout`nSTDERR: $stderr"
+  }
+}
+finally {
+  Remove-Item -Path $tempDir -Recurse -Force -ErrorAction SilentlyContinue
+}