Jo
92ac380621
* Authn: Anon session service (#63052) * add anon sessions package * add usage stat fn * implement count for cache * add anonservice to authn broker * lint * add tests for remote cache count * move anon service to services * wrap tagging in goroutine * make func used (cherry picked from commitff78103a24) * add local cache to protect multiple writes to DB cache (cherry picked from commitbdb084736b)
121 lines
3.1 KiB
Go
121 lines
3.1 KiB
Go
package remotecache
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/grafana/grafana/pkg/infra/db"
|
|
"github.com/grafana/grafana/pkg/infra/log"
|
|
)
|
|
|
|
func TestDatabaseStorageGarbageCollection(t *testing.T) {
|
|
sqlstore := db.InitTestDB(t)
|
|
|
|
db := &databaseCache{
|
|
SQLStore: sqlstore,
|
|
codec: &gobCodec{},
|
|
log: log.New("remotecache.database"),
|
|
}
|
|
|
|
obj := &CacheableStruct{String: "foolbar"}
|
|
|
|
// set time.now to 2 weeks ago
|
|
var err error
|
|
getTime = func() time.Time { return time.Now().AddDate(0, 0, -2) }
|
|
err = db.Set(context.Background(), "key1", obj, 1000*time.Second)
|
|
assert.Equal(t, err, nil)
|
|
|
|
err = db.Set(context.Background(), "key2", obj, 1000*time.Second)
|
|
assert.Equal(t, err, nil)
|
|
|
|
err = db.Set(context.Background(), "key3", obj, 1000*time.Second)
|
|
assert.Equal(t, err, nil)
|
|
|
|
// insert object that should never expire
|
|
err = db.Set(context.Background(), "key4", obj, 0)
|
|
assert.Equal(t, err, nil)
|
|
|
|
getTime = time.Now
|
|
err = db.Set(context.Background(), "key5", obj, 1000*time.Second)
|
|
assert.Equal(t, err, nil)
|
|
|
|
// run GC
|
|
db.internalRunGC()
|
|
|
|
// try to read values
|
|
_, err = db.Get(context.Background(), "key1")
|
|
assert.Equal(t, err, ErrCacheItemNotFound, "expected cache item not found. got: ", err)
|
|
_, err = db.Get(context.Background(), "key2")
|
|
assert.Equal(t, err, ErrCacheItemNotFound)
|
|
_, err = db.Get(context.Background(), "key3")
|
|
assert.Equal(t, err, ErrCacheItemNotFound)
|
|
|
|
_, err = db.Get(context.Background(), "key4")
|
|
assert.Equal(t, err, nil)
|
|
_, err = db.Get(context.Background(), "key5")
|
|
assert.Equal(t, err, nil)
|
|
}
|
|
|
|
func TestSecondSet(t *testing.T) {
|
|
var err error
|
|
sqlstore := db.InitTestDB(t)
|
|
|
|
db := &databaseCache{
|
|
SQLStore: sqlstore,
|
|
codec: &gobCodec{},
|
|
log: log.New("remotecache.database"),
|
|
}
|
|
|
|
obj := &CacheableStruct{String: "hey!"}
|
|
|
|
err = db.Set(context.Background(), "killa-gorilla", obj, 0)
|
|
assert.Equal(t, err, nil)
|
|
|
|
err = db.Set(context.Background(), "killa-gorilla", obj, 0)
|
|
assert.Equal(t, err, nil)
|
|
}
|
|
|
|
func TestDatabaseStorageCount(t *testing.T) {
|
|
sqlstore := db.InitTestDB(t)
|
|
|
|
db := &databaseCache{
|
|
SQLStore: sqlstore,
|
|
codec: &gobCodec{},
|
|
log: log.New("remotecache.database"),
|
|
}
|
|
|
|
obj := &CacheableStruct{String: "foolbar"}
|
|
|
|
// set time.now to 2 weeks ago
|
|
var err error
|
|
getTime = func() time.Time { return time.Now().AddDate(0, 0, -2) }
|
|
err = db.Set(context.Background(), "pref-key1", obj, 1000*time.Second)
|
|
require.NoError(t, err)
|
|
|
|
err = db.Set(context.Background(), "pref-key2", obj, 1000*time.Second)
|
|
require.NoError(t, err)
|
|
|
|
err = db.Set(context.Background(), "pref-key3", obj, 1000*time.Second)
|
|
require.NoError(t, err)
|
|
|
|
// insert object that should never expire
|
|
err = db.Set(context.Background(), "pref-key4", obj, 0)
|
|
require.NoError(t, err)
|
|
|
|
getTime = time.Now
|
|
err = db.Set(context.Background(), "pref-key5", obj, 1000*time.Second)
|
|
require.NoError(t, err)
|
|
|
|
// run GC
|
|
db.internalRunGC()
|
|
|
|
// try to read values
|
|
n, errC := db.Count(context.Background(), "pref-")
|
|
require.NoError(t, errC)
|
|
assert.Equal(t, int64(2), n)
|
|
}
|