Peter Štibraný
7fd9ab9481
* Replace check for integration tests. * Revert changes in pkg/tsdb/mysql packages. * Fix formatting of few tests.
83 lines
2.1 KiB
Go
83 lines
2.1 KiB
Go
package remotecache
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/grafana/grafana/pkg/infra/db"
|
|
"github.com/grafana/grafana/pkg/infra/log"
|
|
"github.com/grafana/grafana/pkg/util/testutil"
|
|
)
|
|
|
|
func TestIntegrationDatabaseStorageGarbageCollection(t *testing.T) {
|
|
testutil.SkipIntegrationTestInShortMode(t)
|
|
|
|
sqlstore := db.InitTestDB(t)
|
|
|
|
db := &databaseCache{
|
|
SQLStore: sqlstore,
|
|
log: log.New("remotecache.database"),
|
|
}
|
|
|
|
obj := []byte("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 TestIntegrationSecondSet(t *testing.T) {
|
|
testutil.SkipIntegrationTestInShortMode(t)
|
|
|
|
var err error
|
|
sqlstore := db.InitTestDB(t)
|
|
|
|
db := &databaseCache{
|
|
SQLStore: sqlstore,
|
|
log: log.New("remotecache.database"),
|
|
}
|
|
|
|
obj := []byte("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)
|
|
}
|