wehub-resource-sync
f99010fae1
CI / lint (push) Failing after 1s
CI / frontend (push) Failing after 1s
CI / scripts (push) Failing after 1s
CI / Go Test (ubuntu-latest) (push) Failing after 0s
CI / frontend-node-25 (push) Failing after 1s
CI / docs (push) Failing after 0s
CI / coverage (push) Failing after 0s
CI / e2e (push) Failing after 0s
Docker / build-and-push (push) Failing after 1s
CI / integration (push) Failing after 4m43s
CI / Go Test (windows-latest) (push) Has been cancelled
CI / Desktop Unit Tests (Windows) (push) Has been cancelled
Desktop Artifacts / Desktop Build (Linux (arm64)) (push) Has been cancelled
Desktop Artifacts / Desktop Build (Linux) (push) Has been cancelled
Desktop Artifacts / Desktop Build (Windows) (push) Has been cancelled
Desktop Artifacts (macOS) / Desktop Build (macOS (aarch64)) (push) Has been cancelled
Desktop Artifacts (macOS) / Desktop Build (macOS (x86_64)) (push) Has been cancelled
32 lines
1.0 KiB
Go
32 lines
1.0 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestReplaceSessionContentAtomic(t *testing.T) {
|
|
d := testDB(t)
|
|
insertSession(t, d, "s1", "proj")
|
|
msgs := []Message{
|
|
{SessionID: "s1", Ordinal: 0, Role: "user", Content: "key AKIA7QHWN2DKR4FYPLJM"},
|
|
}
|
|
signals := SessionSignalUpdate{Outcome: "success", SecretLeakCount: 1,
|
|
SecretsRulesVersion: "rulesv1"}
|
|
findings := []SecretFinding{{
|
|
SessionID: "s1", RuleName: "aws-access-key", Confidence: "definite",
|
|
LocationKind: "message", MessageOrdinal: 0, MatchStart: 4, MatchEnd: 24,
|
|
MatchIndex: 0, RedactedMatch: "AKIA…MPLE", RulesVersion: "rulesv1",
|
|
}}
|
|
require.NoError(t, d.ReplaceSessionContent("s1", msgs, signals, findings))
|
|
got, _ := d.GetAllMessages(context.Background(), "s1")
|
|
require.Len(t, got, 1)
|
|
f, _ := d.SessionSecretFindings(context.Background(), "s1")
|
|
require.Len(t, f, 1)
|
|
s, _ := d.GetSession(context.Background(), "s1")
|
|
assert.Equal(t, 1, s.SecretLeakCount)
|
|
}
|