opencode/picoclaw
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: update PII redactor tests to match new signature and expectations

Browse source
This commit is contained in:
stevef 2026-03-30 19:40:24 +02:00
parent 4cff032f75
commit 67f3c4a91f
1 changed files with 7 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
pkg/security/pii/redactor_test.go
View file

@ -17,14 +17,15 @@ func TestRedactor_Redact(t *testing.T) {
input string input string
expected string expected string
}{ }{
{"Hello, contact me at steve@example.com", "Hello, contact me at [EMAIL]"}, {"Hello, contact me at steve@example.com", "Hello, contact me at [EMAIL_1]"},
{"My IP is 192.168.1.1", "My IP is [IP]"}, {"My IP is 192.168.1.1", "My IP is [IP_1]"},
{"Call me at +1 555-123-4567", "Call me at [PHONE]"}, {"Call me at +1 555-123-4567", "Call me at [PHONE_1]"},
{"Nothing sensitive here", "Nothing sensitive here"}, {"Nothing sensitive here", "Nothing sensitive here"},
} }
mapping := r.getMapping("test")
for _, tt := range tests { for _, tt := range tests {
assert.Equal(t, tt.expected, r.redact(tt.input)) assert.Equal(t, tt.expected, r.redact(tt.input, mapping))
} }
} }
@ -43,7 +44,7 @@ func TestRedactor_BeforeLLM(t *testing.T) {
require.NoError(t, err) require.NoError(t, err)
assert.Equal(t, agent.HookActionContinue, decision.Action) assert.Equal(t, agent.HookActionContinue, decision.Action)
assert.Equal(t, "My email is [EMAIL]", next.Messages[0].Content) assert.Equal(t, "My email is [EMAIL_1]", next.Messages[0].Content)
assert.Equal(t, "Keep 127.0.0.1", next.Messages[1].Content) assert.Equal(t, "Keep 127.0.0.1", next.Messages[1].Content)
} }
@ -61,5 +62,5 @@ func TestRedactor_AfterLLM(t *testing.T) {
require.NoError(t, err) require.NoError(t, err)
assert.Equal(t, agent.HookActionContinue, decision.Action) assert.Equal(t, agent.HookActionContinue, decision.Action)
assert.Equal(t, "The user's email was [EMAIL]", next.Response.Content) assert.Equal(t, "The user's email was user@foo.com", next.Response.Content)
} }