Remove auth token when it is not found (#4050)
This commit is contained in:
1 parent
9aab26e91b
commit
8974606dea
2 files changed
+26
-2
No files matched your search
@@ -106,6 +106,7 @@ func TestCleanAndPickTokens(t *testing.T) {
|
||||
|
||||
boundaryAuthTokens := []*authtokens.AuthToken{at1a, keyringAuthToken1, at1b, keyringAuthToken2}
|
||||
unauthorizedAuthTokens := []*authtokens.AuthToken{}
|
||||
notFoundAuthTokens := []*authtokens.AuthToken{}
|
||||
randomErrorAuthTokens := []*authtokens.AuthToken{}
|
||||
fakeBoundaryLookupFn := func(ctx context.Context, addr, at string) (*authtokens.AuthToken, error) {
|
||||
for _, v := range randomErrorAuthTokens {
|
||||
@@ -113,6 +114,11 @@ func TestCleanAndPickTokens(t *testing.T) {
|
||||
return nil, errors.New("test error")
|
||||
}
|
||||
}
|
||||
for _, v := range notFoundAuthTokens {
|
||||
if at == v.Token {
|
||||
return nil, api.ErrNotFound
|
||||
}
|
||||
}
|
||||
for _, v := range unauthorizedAuthTokens {
|
||||
if at == v.Token {
|
||||
return nil, api.ErrUnauthorized
|
||||
@@ -188,6 +194,24 @@ func TestCleanAndPickTokens(t *testing.T) {
|
||||
assert.ElementsMatch(t, maps.Values(got), []string{at1a.Token})
|
||||
})
|
||||
|
||||
t.Run("boundary in memory auth token not found", func(t *testing.T) {
|
||||
require.NoError(t, r.AddRawToken(ctx, boundaryAddr, at1a.Token))
|
||||
require.NoError(t, r.AddRawToken(ctx, boundaryAddr, at1b.Token))
|
||||
|
||||
got, err := rs.cleanAndPickAuthTokens(ctx, u1)
|
||||
assert.NoError(t, err)
|
||||
assert.ElementsMatch(t, maps.Values(got), []string{at1a.Token, at1b.Token})
|
||||
|
||||
t.Cleanup(func() {
|
||||
notFoundAuthTokens = nil
|
||||
})
|
||||
|
||||
notFoundAuthTokens = []*authtokens.AuthToken{at1b}
|
||||
got, err = rs.cleanAndPickAuthTokens(ctx, u1)
|
||||
assert.NoError(t, err)
|
||||
assert.ElementsMatch(t, maps.Values(got), []string{at1a.Token})
|
||||
})
|
||||
|
||||
t.Run("boundary keyring auths token expires", func(t *testing.T) {
|
||||
key1 := ringToken{"k1", "t1"}
|
||||
atMap[key1] = keyringAuthToken1
|
||||
|
||||
Reference in new issue
Block a user