MCPcopy Create free account
hub / github.com/chainreactors/EvilProxy / TestExtractTaskID

Function TestExtractTaskID

internal/toolinjection/injection_test.go:30–66  ·  view source on GitHub ↗
(t *testing.T)

Source from the content-addressed store, hash-verified

28}
29
30func TestExtractTaskID(t *testing.T) {
31 // Round-trip: encode and extract
32 id := GenerateOpenAIToolCallID(42)
33 taskID, ok := ExtractTaskID(id)
34 if !ok {
35 t.Fatalf("ExtractTaskID failed for %s", id)
36 }
37 if taskID != 42 {
38 t.Errorf("expected taskID 42, got %d", taskID)
39 }
40
41 // Claude format
42 id2 := GenerateClaudeToolUseID(0xDEAD)
43 taskID2, ok2 := ExtractTaskID(id2)
44 if !ok2 {
45 t.Fatalf("ExtractTaskID failed for %s", id2)
46 }
47 if taskID2 != 0xDEAD {
48 t.Errorf("expected taskID 0xDEAD, got %d", taskID2)
49 }
50
51 // Non-injected ID
52 _, ok3 := ExtractTaskID("call_abc123")
53 if ok3 {
54 t.Error("should not extract from non-injected ID")
55 }
56
57 // taskID 0 (global rules)
58 id4 := GenerateOpenAIToolCallID(0)
59 taskID4, ok4 := ExtractTaskID(id4)
60 if !ok4 {
61 t.Fatalf("ExtractTaskID failed for %s", id4)
62 }
63 if taskID4 != 0 {
64 t.Errorf("expected taskID 0, got %d", taskID4)
65 }
66}
67
68func TestIsInjectedID(t *testing.T) {
69 if IsInjectedID("call_abc123") {

Callers

nothing calls this directly

Calls 4

GenerateOpenAIToolCallIDFunction · 0.85
ExtractTaskIDFunction · 0.85
GenerateClaudeToolUseIDFunction · 0.85
ErrorMethod · 0.45

Tested by

no test coverage detected