3
0

Prepare store/actionlog tests

This commit is contained in:
Tomaž Jerman 2020-11-12 09:11:01 +01:00
parent 979acf0a84
commit 544ce76c74
2 changed files with 108 additions and 8 deletions

View File

@ -2,26 +2,27 @@ package tests
import (
"context"
"strconv"
"strings"
"testing"
"time"
"github.com/cortezaproject/corteza-server/pkg/actionlog"
"github.com/cortezaproject/corteza-server/pkg/id"
"github.com/cortezaproject/corteza-server/pkg/rand"
"github.com/cortezaproject/corteza-server/store"
_ "github.com/joho/godotenv/autoload"
"github.com/stretchr/testify/require"
"strings"
"testing"
"time"
)
func testActionlog(t *testing.T, s store.Actionlogs) {
var (
ctx = context.Background()
makeNew = func(dd ...string) *actionlog.Action {
makeNew = func(id uint64, dd ...string) *actionlog.Action {
// minimum data set for new user
desc := strings.Join(dd, "")
return &actionlog.Action{
ID: id.Next(),
ID: id,
Timestamp: *now(),
Action: "test-action",
Resource: "test-resource",
@ -29,6 +30,20 @@ func testActionlog(t *testing.T, s store.Actionlogs) {
}
}
stringifySetRange = func(set actionlog.ActionSet) string {
if len(set) == 0 {
return ""
}
var out = strconv.FormatUint(set[0].ID, 10)
if len(set) > 1 {
out += ".." + strconv.FormatUint(set[len(set)-1].ID, 10)
}
return out
}
truncAndFill = func(t *testing.T, l int) (*require.Assertions, actionlog.ActionSet) {
req := require.New(t)
req.NoError(s.TruncateActionlogs(ctx))
@ -36,7 +51,7 @@ func testActionlog(t *testing.T, s store.Actionlogs) {
set := make([]*actionlog.Action, l)
for i := 0; i < l; i++ {
set[i] = makeNew(string(rand.Bytes(10)))
set[i] = makeNew(uint64(i), string(rand.Bytes(10)))
}
req.NoError(s.CreateActionlog(ctx, set...))
@ -74,4 +89,89 @@ func testActionlog(t *testing.T, s store.Actionlogs) {
})
})
t.Run("with keyed paging", func(t *testing.T) {
req := require.New(t)
req.NoError(s.TruncateActionlogs(ctx))
set := []*actionlog.Action{
makeNew(1, "01"),
makeNew(2, "02"),
makeNew(3, "03"),
makeNew(4, "04"),
makeNew(5, "05"),
makeNew(6, "06"),
makeNew(7, "07"),
makeNew(8, "08"),
makeNew(9, "09"),
makeNew(10, "10"),
}
req.NoError(s.CreateActionlog(ctx, set...))
f := actionlog.Filter{}
// Fetch first page
f.Limit = 3
set, f, err := store.SearchActionlogs(ctx, s, f)
req.NoError(err)
req.Len(set, 3)
req.NotNil(f.NextPage)
req.Nil(f.PrevPage)
req.Equal("10..8", stringifySetRange(set))
// 2nd page
f.Limit = 6
f.PageCursor = f.NextPage
set, f, err = store.SearchActionlogs(ctx, s, f)
req.NoError(err)
req.Len(set, 6)
req.NotNil(f.NextPage)
req.NotNil(f.PrevPage)
req.Equal("7..2", stringifySetRange(set))
// 3rd, last page (1 item left)
f.Limit = 2
f.PageCursor = f.NextPage
set, f, err = store.SearchActionlogs(ctx, s, f)
req.NoError(err)
req.Len(set, 1)
req.NotNil(f.NextPage)
req.NotNil(f.PrevPage)
req.Equal("1", stringifySetRange(set))
// try and go pass the last page
f.PageCursor = f.NextPage
set, _, err = store.SearchActionlogs(ctx, s, f)
req.NoError(err)
req.Len(set, 0)
// now, in reverse, last 3 items
f.Limit = 3
f.PageCursor = f.PrevPage
set, f, err = store.SearchActionlogs(ctx, s, f)
req.NoError(err)
req.Len(set, 3)
req.NotNil(f.NextPage)
req.NotNil(f.PrevPage)
req.Equal("3..1", stringifySetRange(set))
// still in reverse, next 6 items
f.Limit = 5
f.PageCursor = f.PrevPage
set, f, err = store.SearchActionlogs(ctx, s, f)
req.NoError(err)
req.Len(set, 5)
req.NotNil(f.NextPage)
req.NotNil(f.PrevPage)
req.Equal("9..4", stringifySetRange(set))
// still in reverse, last 5 items (actually, we'll only get 1)
f.Limit = 5
f.PageCursor = f.PrevPage
set, f, err = store.SearchActionlogs(ctx, s, f)
req.NoError(err)
req.Len(set, 1)
req.Nil(f.PrevPage)
req.NotNil(f.NextPage)
req.Equal("10", stringifySetRange(set))
})
}

View File

@ -261,7 +261,7 @@ func testUsers(t *testing.T, s store.Users) {
req.NotNil(f.PrevPage)
req.Equal("07..09", stringifySetRange(set))
// still in reverse, next 6 items
// still in reverse, next 5 items
f.Limit = 5
f.PageCursor = f.PrevPage
set, f, err = store.SearchUsers(ctx, s, f)