fixed test for header value type

goto · Nov 14, 2024 · 50f3c8e · 50f3c8e
1 parent f351ff9
commit 50f3c8e
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/plugins/internal/tengoutil/secure_script_test.go b/plugins/internal/tengoutil/secure_script_test.go
@@ -96,20 +96,26 @@ func TestNewSecureScript(t *testing.T) {
 		assert.Contains(t, err.Error(), "unsupported protocol scheme")
 	})
 
-	t.Run("HTTP GET with invalid header value type", func(t *testing.T) {
-		s, err := NewSecureScript([]byte(`
+	t.Run("HTTP GET with valid header value type (int)", func(t *testing.T) {
+		s, err := NewSecureScript(([]byte)(heredoc.Doc(`
 			http := import("http")
 			headers := { "User-Agent": 12345 }
 			http.get("http://example.com", headers)
-		`), nil)
+		`)), nil)
 		assert.NoError(t, err)
 
 		_, err = s.Compile()
 		assert.NoError(t, err)
 
-		_, err = s.Run()
-		assert.Error(t, err)
-		assert.Contains(t, err.Error(), "header value for key 'User-Agent' must be a string, got int")
+		result, err := s.Run()
+		assert.NoError(t, err)
+
+		resultMap, ok := result.(*tengo.Map)
+		assert.True(t, ok)
+
+		userAgent, ok := resultMap.Value["User-Agent"].(*tengo.String)
+		assert.True(t, ok)
+		assert.Equal(t, "12345", userAgent.Value)
 	})
 
 	t.Run("HTTP GET with timeout", func(t *testing.T) {