fix: unittest

hrp/internal/build/main.go

@@ -23,10 +23,10 @@ const (
 	funppy                  = `import funppy`
 	fungo                   = `"github.com/httprunner/funplugin/fungo"`
 	regexPythonFunctionName = `def ([a-zA-Z_]\w*)\(.*\)`
-	regexGoImports          = `import\s*\(\n([\s\S]*)\n\)`
-	regexGoImport           = `import\s*(\"[\s\S]*\")\n`
+	regexGoImports          = `import \(([\s\S]*?)\)`
+	regexGoImport           = `import (\"[\s\S]*\")`
 	regexGoFunctionName     = `func ([A-Z][a-zA-Z_]\w*)\(.*\)`
-	regexGoFunctionContent  = `func [\s\S]*?\n}\n`
+	regexGoFunctionContent  = `func [\s\S]*?\n}`
 )
 
 //go:embed templates/debugtalkPythonTemplate

hrp/internal/build/main_test.go

@@ -1,30 +1,21 @@
 package build
 
 import (
-	"github.com/httprunner/httprunner/v4/hrp/internal/builtin"
 	"regexp"
 	"testing"
 
+	"github.com/httprunner/httprunner/v4/hrp/internal/builtin"
+
 	"github.com/stretchr/testify/assert"
 )
 
 func TestRun(t *testing.T) {
-	err := Run("../../../examples/demo-with-no-fungo/plugin/debugtalk.go", "")
+	err := Run("plugin/debugtalk.go", "./debugtalk_gen.bin")
 	if !assert.Nil(t, err) {
 		t.Fatal()
 	}
 
-	err = Run("../../../examples/demo-with-no-funppy/debugtalk.py", "")
-	if !assert.Nil(t, err) {
-		t.Fatal()
-	}
-
-	err = Run("../../../examples/demo-with-no-fungo/plugin/debugtalk.go", "./debugtalk_gen.bin")
-	if !assert.Nil(t, err) {
-		t.Fatal()
-	}
-
-	err = Run("../../../examples/demo-with-no-funppy/debugtalk.py", "./debugtalk_gen.py")
+	err = Run("plugin/debugtalk.py", "./debugtalk_gen.py")
 	if !assert.Nil(t, err) {
 		t.Fatal()
 	}

hrp/internal/build/plugin/debugtalk.go

@@ -0,0 +1,44 @@
+package noplugin
+
+import (
+	"fmt"
+)
+
+func SumTwoInt(a, b int) int {
+	return a + b
+}
+
+func SumInts(args ...int) int {
+	var sum int
+	for _, arg := range args {
+		sum += arg
+	}
+	return sum
+}
+
+func Sum(args ...interface{}) (interface{}, error) {
+	var sum float64
+	for _, arg := range args {
+		switch v := arg.(type) {
+		case int:
+			sum += float64(v)
+		case float64:
+			sum += v
+		default:
+			return nil, fmt.Errorf("unexpected type: %T", arg)
+		}
+	}
+	return sum, nil
+}
+
+func SetupHookExample(args string) string {
+	return fmt.Sprintf("step name: %v, setup...", args)
+}
+
+func TeardownHookExample(args string) string {
+	return fmt.Sprintf("step name: %v, teardown...", args)
+}
+
+func GetVersion() string {
+	return "v0.4"
+}

hrp/internal/build/plugin/debugtalk.py

@@ -0,0 +1,57 @@
+import logging
+import time
+from typing import List
+
+
+def get_version():
+    return "v0.4"
+
+
+def sleep(n_secs):
+    time.sleep(n_secs)
+
+
+def sum(*args):
+    result = 0
+    for arg in args:
+        result += arg
+    return result
+
+
+def sum_ints(*args: List[int]) -> int:
+    result = 0
+    for arg in args:
+        result += arg
+    return result
+
+
+def sum_two_int(a: int, b: int) -> int:
+    return a + b
+
+
+def sum_two_string(a: str, b: str) -> str:
+    return a + b
+
+
+def sum_strings(*args: List[str]) -> str:
+    result = ""
+    for arg in args:
+        result += arg
+    return result
+
+
+def concatenate(*args: List[str]) -> str:
+    result = ""
+    for arg in args:
+        result += str(arg)
+    return result
+
+
+def setup_hook_example(name):
+    logging.warning("setup_hook_example")
+    return f"setup_hook_example: {name}"
+
+
+def teardown_hook_example(name):
+    logging.warning("teardown_hook_example")
+    return f"teardown_hook_example: {name}"