24fec62d7f
Forked clean from public hackathon repo (Starfire-AgentTeam, BSL 1.1) with full rebrand to Molecule AI under github.com/Molecule-AI/molecule-monorepo. Brand: Starfire → Molecule AI. Slug: starfire / agent-molecule → molecule. Env vars: STARFIRE_* → MOLECULE_*. Go module: github.com/agent-molecule/platform → github.com/Molecule-AI/molecule-monorepo/platform. Python packages: starfire_plugin → molecule_plugin, starfire_agent → molecule_agent. DB: agentmolecule → molecule. History truncated; see public repo for prior commits and contributor attribution. Verified green: go test -race ./... (platform), pytest (workspace-template 1129 + sdk 132), vitest (canvas 352), build (mcp). Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
110 lines
2.9 KiB
Go
110 lines
2.9 KiB
Go
package handlers
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// ==================== GET /workspaces/:id/traces ====================
|
|
|
|
func TestTracesList_NoLangfuseConfig(t *testing.T) {
|
|
setupTestDB(t)
|
|
setupTestRedis(t)
|
|
handler := NewTracesHandler()
|
|
|
|
// Ensure Langfuse env vars are not set
|
|
os.Unsetenv("LANGFUSE_HOST")
|
|
os.Unsetenv("LANGFUSE_PUBLIC_KEY")
|
|
os.Unsetenv("LANGFUSE_SECRET_KEY")
|
|
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
c.Params = gin.Params{{Key: "id", Value: "ws-traces"}}
|
|
c.Request = httptest.NewRequest("GET", "/workspaces/ws-traces/traces", nil)
|
|
|
|
handler.List(c)
|
|
|
|
if w.Code != http.StatusOK {
|
|
t.Errorf("expected 200, got %d: %s", w.Code, w.Body.String())
|
|
}
|
|
|
|
// Should return empty array when Langfuse is not configured
|
|
var resp []interface{}
|
|
if err := json.Unmarshal(w.Body.Bytes(), &resp); err != nil {
|
|
t.Fatalf("failed to parse response: %v", err)
|
|
}
|
|
if len(resp) != 0 {
|
|
t.Errorf("expected empty list when Langfuse not configured, got %d items", len(resp))
|
|
}
|
|
}
|
|
|
|
func TestTracesList_PartialLangfuseConfig(t *testing.T) {
|
|
setupTestDB(t)
|
|
setupTestRedis(t)
|
|
handler := NewTracesHandler()
|
|
|
|
// Set only host, missing keys
|
|
os.Setenv("LANGFUSE_HOST", "http://localhost:3000")
|
|
os.Unsetenv("LANGFUSE_PUBLIC_KEY")
|
|
os.Unsetenv("LANGFUSE_SECRET_KEY")
|
|
defer func() {
|
|
os.Unsetenv("LANGFUSE_HOST")
|
|
}()
|
|
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
c.Params = gin.Params{{Key: "id", Value: "ws-traces-partial"}}
|
|
c.Request = httptest.NewRequest("GET", "/workspaces/ws-traces-partial/traces", nil)
|
|
|
|
handler.List(c)
|
|
|
|
if w.Code != http.StatusOK {
|
|
t.Errorf("expected 200, got %d: %s", w.Code, w.Body.String())
|
|
}
|
|
|
|
var resp []interface{}
|
|
json.Unmarshal(w.Body.Bytes(), &resp)
|
|
if len(resp) != 0 {
|
|
t.Errorf("expected empty list with partial config, got %d items", len(resp))
|
|
}
|
|
}
|
|
|
|
func TestTracesList_LangfuseUnreachable(t *testing.T) {
|
|
setupTestDB(t)
|
|
setupTestRedis(t)
|
|
handler := NewTracesHandler()
|
|
|
|
// Set all env vars but point to unreachable host
|
|
os.Setenv("LANGFUSE_HOST", "http://localhost:99999")
|
|
os.Setenv("LANGFUSE_PUBLIC_KEY", "pk-test")
|
|
os.Setenv("LANGFUSE_SECRET_KEY", "sk-test")
|
|
defer func() {
|
|
os.Unsetenv("LANGFUSE_HOST")
|
|
os.Unsetenv("LANGFUSE_PUBLIC_KEY")
|
|
os.Unsetenv("LANGFUSE_SECRET_KEY")
|
|
}()
|
|
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
c.Params = gin.Params{{Key: "id", Value: "ws-traces-down"}}
|
|
c.Request = httptest.NewRequest("GET", "/workspaces/ws-traces-down/traces", nil)
|
|
|
|
handler.List(c)
|
|
|
|
// Should gracefully return empty when Langfuse is unreachable
|
|
if w.Code != http.StatusOK {
|
|
t.Errorf("expected 200, got %d: %s", w.Code, w.Body.String())
|
|
}
|
|
|
|
var resp []interface{}
|
|
json.Unmarshal(w.Body.Bytes(), &resp)
|
|
if len(resp) != 0 {
|
|
t.Errorf("expected empty list when Langfuse unreachable, got %d items", len(resp))
|
|
}
|
|
}
|