Function TestBuildCustomScriptFilesStep

pkg/workflow/safe_scripts_test.go:251–281

TestBuildCustomScriptFilesStep verifies the generated step writes scripts to files

(t *testing.T)

// TestBuildCustomScriptFilesStep verifies the generated step writes scripts to files
func TestBuildCustomScriptFilesStep(t *testing.T) {
	scripts := map[string]*SafeScriptConfig{
		"my-handler": {
			// Users write only the handler body — no function wrapper or boilerplate needed
			Script: "return { success: true };",
			Inputs: map[string]*InputDefinition{
				"channel": {Type: "string"},
			},
		},
	}

	steps, err := buildCustomScriptFilesStep(scripts)

	require.NoError(t, err, "Should not return error for valid scripts")
	require.NotEmpty(t, steps, "Should produce steps")

	fullYAML := strings.Join(steps, "")

	assert.Contains(t, fullYAML, "Configure Safe Outputs Custom Scripts", "Should have configure step name")
	assert.Contains(t, fullYAML, "safe_output_script_my_handler.cjs", "Should reference the output filename")
	// Verify heredoc delimiter follows the randomized GH_AW_SAFE_OUTPUT_SCRIPT_MY_HANDLER_<hex>_EOF format
	delimRE := regexp.MustCompile(`GH_AW_SAFE_OUTPUT_SCRIPT_MY_HANDLER_[0-9a-f]{16}_EOF`)
	assert.True(t, delimRE.MatchString(fullYAML), "Should use correct randomized heredoc delimiter")
	// Verify the compiler generates the full outer wrapper
	assert.Contains(t, fullYAML, "async function main(config = {}) {", "Should generate main function declaration")
	assert.Contains(t, fullYAML, "const { channel } = config;", "Should generate config input destructuring")
	assert.Contains(t, fullYAML, "return async function handleMyHandler(item, resolvedTemporaryIds, temporaryIdMap) {", "Should generate handler function")
	assert.Contains(t, fullYAML, "module.exports = { main };", "Should include module.exports wrapper")
	// User's handler body content should appear indented inside the handler
	assert.Contains(t, fullYAML, "return { success: true };", "Should include user's handler body")
}

// TestBuildCustomScriptFilesStepEmpty verifies nil return for empty scripts
func TestBuildCustomScriptFilesStepEmpty(t *testing.T) {