mirror of
https://github.com/coder/coder.git
synced 2025-07-03 16:13:58 +00:00
fix: add requester IP to workspace build audit logs (#10242)
This commit is contained in:
33
coderd/audit/request_test.go
Normal file
33
coderd/audit/request_test.go
Normal file
@ -0,0 +1,33 @@
|
||||
package audit_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/require"
|
||||
"go.opentelemetry.io/otel/propagation"
|
||||
|
||||
"github.com/coder/coder/v2/coderd/audit"
|
||||
)
|
||||
|
||||
func TestBaggage(t *testing.T) {
|
||||
t.Parallel()
|
||||
prop := propagation.NewCompositeTextMapPropagator(
|
||||
propagation.TraceContext{},
|
||||
propagation.Baggage{},
|
||||
)
|
||||
|
||||
expected := audit.WorkspaceBuildBaggage{
|
||||
IP: "127.0.0.1",
|
||||
}
|
||||
|
||||
ctx, err := audit.BaggageToContext(context.Background(), expected)
|
||||
require.NoError(t, err)
|
||||
|
||||
carrier := propagation.MapCarrier{}
|
||||
prop.Inject(ctx, carrier)
|
||||
bCtx := prop.Extract(ctx, carrier)
|
||||
got := audit.BaggageFromContext(bCtx)
|
||||
|
||||
require.Equal(t, expected, got)
|
||||
}
|
Reference in New Issue
Block a user