mirror of
https://github.com/vxcontrol/langchaingo.git
synced 2026-07-25 08:36:24 -04:00
Travis Cline
77504b877f
* scripts: cleanup test scripts * lint: add test pattern and architectural linting * internal/httprr: expand httprr testing * test: add comprehensive unit test coverage * docs: add comprehensive testing guide to CONTRIBUTING.md * examples: standardize module versions and cleanup dependencies * all: re-record several httprr recordings
80 lines
1.9 KiB
Go
80 lines
1.9 KiB
Go
package serpapi
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/tmc/langchaingo/internal/httprr"
|
|
)
|
|
|
|
func TestSerpAPITool(t *testing.T) {
|
|
|
|
httprr.SkipIfNoCredentialsAndRecordingMissing(t, "SERPAPI_API_KEY")
|
|
|
|
rr := httprr.OpenForTest(t, http.DefaultTransport)
|
|
|
|
// Scrub the API key from requests to use test key in recordings
|
|
rr.ScrubReq(func(req *http.Request) error {
|
|
if req.URL != nil {
|
|
q := req.URL.Query()
|
|
q.Set("api_key", "test-api-key")
|
|
req.URL.RawQuery = q.Encode()
|
|
}
|
|
return nil
|
|
})
|
|
|
|
var opts []Option
|
|
opts = append(opts, WithHTTPClient(rr.Client()))
|
|
|
|
// Use test key when replaying, environment key when recording
|
|
if rr.Replaying() {
|
|
opts = append(opts, WithAPIKey("test-api-key"))
|
|
}
|
|
|
|
tool, err := New(opts...)
|
|
if err != nil {
|
|
t.Fatalf("Failed to create tool: %v", err)
|
|
}
|
|
|
|
// Test search functionality with a stable question
|
|
result, err := tool.Call(context.Background(), "What year was Unix first released at Bell Labs?")
|
|
if err != nil {
|
|
t.Fatalf("Tool call failed: %v", err)
|
|
}
|
|
if result == "" {
|
|
t.Fatal("Result should not be empty")
|
|
}
|
|
|
|
// Basic validation - result should contain meaningful content
|
|
if len(result) <= 3 {
|
|
t.Errorf("Result should contain more than 3 characters, got: %s", result)
|
|
}
|
|
|
|
// For debugging test failures, log the result
|
|
t.Logf("SerpAPI result: %s", result)
|
|
}
|
|
|
|
func TestSerpAPIToolError(t *testing.T) {
|
|
|
|
// Save original environment variable
|
|
originalKey := os.Getenv("SERPAPI_API_KEY")
|
|
// Temporarily unset the environment variable
|
|
os.Setenv("SERPAPI_API_KEY", "")
|
|
// Restore it after the test
|
|
t.Cleanup(func() {
|
|
os.Setenv("SERPAPI_API_KEY", originalKey)
|
|
})
|
|
|
|
// Test error handling when no API key is provided
|
|
_, err := New()
|
|
if err == nil {
|
|
t.Fatal("Expected error when no API key is provided")
|
|
}
|
|
if !strings.Contains(err.Error(), "missing the serpapi API key") {
|
|
t.Errorf("Expected missing API key error, got: %v", err)
|
|
}
|
|
}
|