Something went wrong. Try again.
This repository has no description
Something went wrong. Try again.
Go
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465package daemon
import ( "testing")
func TestParsePythonIntent(t *testing.T) { cases := []struct { code string expected *ToolIntent }{ { code: "from pathlib import Path\nPath('demo.py').write_text('hello')", expected: &ToolIntent{Kind: "write", Target: "demo.py"}, }, { code: "read('second.py')", expected: &ToolIntent{Kind: "read", Target: "second.py"}, }, { code: "edit('main.go')", expected: &ToolIntent{Kind: "edit", Target: "main.go"}, }, { code: "sh('ls -la')", expected: &ToolIntent{Kind: "run", Target: "ls -la"}, }, { code: "open('test.txt', 'r')", expected: &ToolIntent{Kind: "read", Target: "test.txt"}, }, { code: "open('test.txt', 'w')", expected: &ToolIntent{Kind: "write", Target: "test.txt"}, }, { code: "p = Path('data.csv')\np.write_text('a,b,c')", expected: &ToolIntent{Kind: "write", Target: "data.csv"}, }, }
for _, tc := range cases { intent := ParsePythonIntent(tc.code) if intent == nil { t.Fatalf("expected intent for %s, got nil", tc.code) } if intent.Kind != tc.expected.Kind || intent.Target != tc.expected.Target { t.Fatalf("for %s: expected %+v, got %+v", tc.code, tc.expected, intent) } }}
func TestExtractPartialJSONCode(t *testing.T) { code1 := ExtractPartialJSONCode(`{"code":"read('second.`) if code1 != "read('second." { t.Fatalf("expected read('second., got %s", code1) }
code2 := ExtractPartialJSONCode(`{"code":"from pathlib import Path\nPath('demo.py').write_text('hello')"}`) expected2 := "from pathlib import Path\nPath('demo.py').write_text('hello')" if code2 != expected2 { t.Fatalf("expected %s, got %s", expected2, code2) }}