mirror of
https://github.com/grafana/grafana.git
synced 2025-01-13 09:32:12 -06:00
6750e881e3
* Add search index table * Stab a test * Add more tests * Add basic index * Switch to UID and add a test for the index * Improve tests coverage * Remove redundant whitespaces * Load all data source APIs when query history is loaded * Fix column type * Fix migration * Clean-up the index * Fix linting * Fix migrations * Fix migrations * Fix migrations * Rename index to details
72 lines
2.5 KiB
Go
72 lines
2.5 KiB
Go
package queryhistory
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/grafana/grafana/pkg/infra/db"
|
|
"github.com/grafana/grafana/pkg/web"
|
|
)
|
|
|
|
func TestIntegrationDeleteStaleQueryFromQueryHistory(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("skipping integration test")
|
|
}
|
|
testScenarioWithQueryInQueryHistory(t, "Stale query history can be deleted",
|
|
func(t *testing.T, sc scenarioContext) {
|
|
olderThan := sc.service.now().Unix() + 60
|
|
rowsDeleted, err := sc.service.DeleteStaleQueriesInQueryHistory(context.Background(), olderThan)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, rowsDeleted)
|
|
})
|
|
|
|
testScenarioWithQueryInQueryHistory(t, "Stale single starred query history can not be deleted",
|
|
func(t *testing.T, sc scenarioContext) {
|
|
sc.ctx.Req = web.SetURLParams(sc.ctx.Req, map[string]string{":uid": sc.initialResult.Result.UID})
|
|
resp := sc.service.starHandler(sc.reqContext)
|
|
require.Equal(t, 200, resp.Status())
|
|
|
|
olderThan := sc.service.now().Unix() + 60
|
|
rowsDeleted, err := sc.service.DeleteStaleQueriesInQueryHistory(context.Background(), olderThan)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 0, rowsDeleted)
|
|
})
|
|
|
|
testScenarioWithQueryInQueryHistory(t, "Not stale query history is not deleted",
|
|
func(t *testing.T, sc scenarioContext) {
|
|
olderThan := sc.service.now().Unix() - 60
|
|
rowsDeleted, err := sc.service.DeleteStaleQueriesInQueryHistory(context.Background(), olderThan)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 0, rowsDeleted)
|
|
})
|
|
|
|
// In this scenario we have 2 starred queries and 1 not starred query
|
|
testScenarioWithMultipleQueriesInQueryHistory(t, "Stale starred query history can not be deleted",
|
|
func(t *testing.T, sc scenarioContext) {
|
|
// all indices are added
|
|
err := sc.sqlStore.WithDbSession(context.Background(), func(dbSession *db.Session) error {
|
|
count, err := dbSession.Table("query_history_details").Count()
|
|
require.NoError(t, err)
|
|
require.Equal(t, int64(3), count)
|
|
return err
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
olderThan := sc.service.now().Unix() + 60
|
|
rowsDeleted, err := sc.service.DeleteStaleQueriesInQueryHistory(context.Background(), olderThan)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, rowsDeleted)
|
|
|
|
// only one details row is removed
|
|
err = sc.sqlStore.WithDbSession(context.Background(), func(dbSession *db.Session) error {
|
|
count, err := dbSession.Table("query_history_details").Count()
|
|
require.NoError(t, err)
|
|
require.Equal(t, int64(2), count)
|
|
return err
|
|
})
|
|
require.NoError(t, err)
|
|
})
|
|
}
|