27 lines
670 B
Go
27 lines
670 B
Go
package auth
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
)
|
|
|
|
func TestUserIDKey(t *testing.T) {
|
|
// Test that the UserIDKey constant is defined correctly
|
|
if UserIDKey != "user_id" {
|
|
t.Errorf("UserIDKey = %v, want %v", UserIDKey, "user_id")
|
|
}
|
|
|
|
// Test context with user ID
|
|
ctx := context.WithValue(context.Background(), UserIDKey, "test-user-123")
|
|
value := ctx.Value(UserIDKey)
|
|
if value != "test-user-123" {
|
|
t.Errorf("Context value = %v, want %v", value, "test-user-123")
|
|
}
|
|
|
|
// Test context without user ID
|
|
emptyCtx := context.Background()
|
|
emptyValue := emptyCtx.Value(UserIDKey)
|
|
if emptyValue != nil {
|
|
t.Errorf("Empty context value = %v, want nil", emptyValue)
|
|
}
|
|
}
|