check unit test

goto · Nov 14, 2024 · 5878e3f · 5878e3f
1 parent caefacf
commit 5878e3f
Showing 1 changed file with 58 additions and 0 deletions.
diff --git a/plugins/internal/tengoutil/secure_script_test.go b/plugins/internal/tengoutil/secure_script_test.go
@@ -66,4 +66,62 @@ func TestNewSecureScript(t *testing.T) {
 		_, err = s.Compile()
 		assert.NoError(t, err)
 	})
+
+	t.Run("HTTP GET with headers", func(t *testing.T) {
+		s, err := NewSecureScript([]byte(heredoc.Doc(`
+			http := import("http")
+			headers := { "User-Agent": "test-agent", "Accept": "application/json" }
+			response := http.get("http://example.com", headers)
+			response.body
+		`)), nil)
+		assert.NoError(t, err)
+		_, err = s.Compile()
+		assert.NoError(t, err)
+	})
+
+	t.Run("HTTP GET with invalid URL argument type", func(t *testing.T) {
+		s, err := NewSecureScript([]byte(heredoc.Doc(`
+			http := import("http")
+			http.get(12345) 
+		`)), nil)
+		assert.NoError(t, err)
+		compiledScript, err := s.Compile()
+		assert.NoError(t, err)
+
+		_, err = compiledScript.Run()
+		assert.Error(t, err)
+		assert.Contains(t, err.Error(), "expected argument 1 (URL) to be a string")
+	})
+
+	t.Run("HTTP GET with invalid header value type", func(t *testing.T) {
+		s, err := NewSecureScript([]byte(heredoc.Doc(`
+			http := import("http")
+			headers := { "User-Agent": 12345 }
+			http.get("http://example.com", headers)
+		`)), nil)
+		assert.NoError(t, err)
+		compiledScript, err := s.Compile()
+		assert.NoError(t, err)
+
+		_, err = compiledScript.Run()
+		assert.Error(t, err)
+		assert.Contains(t, err.Error(), "header value for key 'User-Agent' must be a string")
+	})
+
+	t.Run("HTTP GET with timeout", func(t *testing.T) {
+		s, err := NewSecureScript([]byte(heredoc.Doc(`
+			http := import("http")
+			response := http.get("http://example.com")
+			response.body
+		`)), nil)
+		assert.NoError(t, err)
+
+		defaultTimeout = 1 * time.Millisecond
+		compiledScript, err := s.Compile()
+		assert.NoError(t, err)
+
+		_, err = compiledScript.Run()
+		assert.Error(t, err)
+		assert.Contains(t, err.Error(), "context deadline exceeded")
+	})
 }