mirror of
https://github.com/lukaszraczylo/traefikoidc.git
synced 2026-11-03 07:44:13 +00:00
The verified-token cache is one process-wide cache keyed by the raw token. A verdict stored by one middleware instance satisfied any other instance, so a token valid for one client or audience skipped the issuer, audience and signature checks on another route. Each instance now prefixes keys with its provider, client ID and audience. Dynamic client registration updates the prefix.
57 lines
1.7 KiB
Go
57 lines
1.7 KiB
Go
package traefikoidc
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func newNSCache(t *testing.T) *UniversalCache {
|
|
t.Helper()
|
|
c := NewUniversalCache(UniversalCacheConfig{Type: CacheTypeToken, MaxSize: 100, DefaultTTL: time.Hour, CleanupInterval: time.Hour})
|
|
t.Cleanup(func() { _ = c.Close() })
|
|
return c
|
|
}
|
|
|
|
// A verdict stored by one middleware instance must not satisfy another
|
|
// instance that verifies against a different client or audience.
|
|
func TestTokenCacheIsolatedPerInstanceNamespace(t *testing.T) {
|
|
shared := newNSCache(t)
|
|
a := &TokenCache{cache: shared}
|
|
b := &TokenCache{cache: shared}
|
|
a.SetNamespace("https://idp|app-a|app-a")
|
|
b.SetNamespace("https://idp|app-b|app-b")
|
|
|
|
a.Set("tok", map[string]interface{}{"sub": "u"}, time.Minute)
|
|
if _, ok := b.Get("tok"); ok {
|
|
t.Fatal("instance B hit a verdict cached by instance A")
|
|
}
|
|
if _, ok := a.Get("tok"); !ok {
|
|
t.Fatal("instance A lost its own verdict")
|
|
}
|
|
a.Delete("tok")
|
|
if _, ok := a.Get("tok"); ok {
|
|
t.Fatal("Delete did not remove the namespaced entry")
|
|
}
|
|
}
|
|
|
|
func TestTokenCacheNamespaceFollowsDCR(t *testing.T) {
|
|
shared := newNSCache(t)
|
|
tc := &TokenCache{cache: shared}
|
|
tc.SetNamespace("p|old|old")
|
|
tc.Set("tok", map[string]interface{}{"sub": "u"}, time.Minute)
|
|
tc.SetNamespace("p|new|new")
|
|
if _, ok := tc.Get("tok"); ok {
|
|
t.Fatal("verdict cached under the old client survived a namespace change")
|
|
}
|
|
}
|
|
|
|
func TestTokenCacheEmptyNamespaceKeepsLegacyKey(t *testing.T) {
|
|
shared := newNSCache(t)
|
|
a := &TokenCache{cache: shared}
|
|
b := &TokenCache{cache: shared}
|
|
a.Set("tok", map[string]interface{}{"sub": "u"}, time.Minute)
|
|
if _, ok := b.Get("tok"); !ok {
|
|
t.Fatal("namespace-less caches must keep sharing the legacy key")
|
|
}
|
|
}
|