mirror of
https://github.com/argoproj/argo-cd
synced 2026-05-24 09:50:08 +00:00
76bacfdea4
* fix: Add initial implementation for rate limiting failed logins * Trigger test build * Remove deprecated code and fix new project tests * move cache related code from sessionmanager to cache access wrapper * avoid using sleep in sessionmanager tests * mention SECONDS in session manager environment variables to make it easier to understand meaning of each variable * Login button should be disabled while user is waiting for login result * prevent timing-based user enumeration attack * reject too many failed attempts; always compute hash and introduce random delay * remove unused constants * fix linter errors Co-authored-by: Alexander Matyushentsev <amatyushentsev@gmail.com>
88 lines
2.5 KiB
Go
88 lines
2.5 KiB
Go
package cache
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
. "github.com/argoproj/argo-cd/pkg/apis/application/v1alpha1"
|
|
cacheutil "github.com/argoproj/argo-cd/util/cache"
|
|
appstatecache "github.com/argoproj/argo-cd/util/cache/appstate"
|
|
"github.com/argoproj/argo-cd/util/oidc"
|
|
)
|
|
|
|
type fixtures struct {
|
|
*Cache
|
|
}
|
|
|
|
func newFixtures() *fixtures {
|
|
return &fixtures{NewCache(
|
|
appstatecache.NewCache(
|
|
cacheutil.NewCache(cacheutil.NewInMemoryCache(1*time.Hour)),
|
|
1*time.Minute,
|
|
),
|
|
1*time.Minute,
|
|
1*time.Minute,
|
|
1*time.Minute,
|
|
)}
|
|
}
|
|
|
|
func TestCache_GetClusterConnectionState(t *testing.T) {
|
|
cache := newFixtures().Cache
|
|
// cache miss
|
|
_, err := cache.GetClusterInfo("my-server")
|
|
assert.Equal(t, ErrCacheMiss, err)
|
|
// populate cache
|
|
err = cache.SetClusterInfo("my-server", &ClusterInfo{ConnectionState: ConnectionState{Status: "my-state"}})
|
|
assert.NoError(t, err)
|
|
// cache miss
|
|
_, err = cache.GetClusterInfo("other-server")
|
|
assert.Equal(t, ErrCacheMiss, err)
|
|
// cache hit
|
|
value, err := cache.GetClusterInfo("my-server")
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, ClusterInfo{ConnectionState: ConnectionState{Status: "my-state"}}, value)
|
|
}
|
|
|
|
func TestCache_GetRepoConnectionState(t *testing.T) {
|
|
cache := newFixtures().Cache
|
|
// cache miss
|
|
_, err := cache.GetRepoConnectionState("my-repo")
|
|
assert.Equal(t, ErrCacheMiss, err)
|
|
// populate cache
|
|
err = cache.SetRepoConnectionState("my-repo", &ConnectionState{Status: "my-state"})
|
|
assert.NoError(t, err)
|
|
// cache miss
|
|
_, err = cache.GetRepoConnectionState("other-repo")
|
|
assert.Equal(t, ErrCacheMiss, err)
|
|
// cache hit
|
|
value, err := cache.GetRepoConnectionState("my-repo")
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, ConnectionState{Status: "my-state"}, value)
|
|
}
|
|
|
|
func TestCache_GetOIDCState(t *testing.T) {
|
|
cache := newFixtures().Cache
|
|
// cache miss
|
|
_, err := cache.GetOIDCState("my-key")
|
|
assert.Equal(t, ErrCacheMiss, err)
|
|
// populate cache
|
|
err = cache.SetOIDCState("my-key", &oidc.OIDCState{ReturnURL: "my-return-url"})
|
|
assert.NoError(t, err)
|
|
//cache miss
|
|
_, err = cache.GetOIDCState("other-key")
|
|
assert.Equal(t, ErrCacheMiss, err)
|
|
// cache hit
|
|
value, err := cache.GetOIDCState("my-key")
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, &oidc.OIDCState{ReturnURL: "my-return-url"}, value)
|
|
}
|
|
|
|
func TestAddCacheFlagsToCmd(t *testing.T) {
|
|
cache, err := AddCacheFlagsToCmd(&cobra.Command{})()
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 1*time.Hour, cache.connectionStatusCacheExpiration)
|
|
assert.Equal(t, 3*time.Minute, cache.oidcCacheExpiration)
|
|
}
|