Fix tests

This commit is contained in:
mcrakhman 2022-11-24 22:20:01 +01:00
parent f98d629a78
commit 9155a3cbf2
No known key found for this signature in database
GPG Key ID: DED12CFEF5B8396B
2 changed files with 42 additions and 6 deletions

View File

@ -4,7 +4,9 @@ import (
"github.com/anytypeio/go-anytype-infrastructure-experiments/common/commonspace/spacesyncproto"
spacestorage "github.com/anytypeio/go-anytype-infrastructure-experiments/common/commonspace/storage"
"github.com/anytypeio/go-anytype-infrastructure-experiments/common/pkg/acl/aclrecordproto"
"github.com/anytypeio/go-anytype-infrastructure-experiments/common/pkg/acl/treechangeproto"
"github.com/stretchr/testify/require"
"sort"
"strconv"
"testing"
)
@ -18,9 +20,14 @@ func spaceTestPayload() spacestorage.SpaceStorageCreatePayload {
Payload: []byte("aclRoot"),
Id: "aclRootId",
}
settings := &treechangeproto.RawTreeChangeWithId{
RawChange: []byte("settings"),
Id: "settingsId",
}
return spacestorage.SpaceStorageCreatePayload{
AclWithId: aclRoot,
SpaceHeaderWithId: header,
AclWithId: aclRoot,
SpaceHeaderWithId: header,
SpaceSettingsWithId: settings,
}
}
@ -68,7 +75,7 @@ func TestSpaceStorage_NewAndCreateTree(t *testing.T) {
require.NoError(t, err)
testSpace(t, store, payload)
t.Run("create tree and get tree", func(t *testing.T) {
t.Run("create tree, get tree and mark deleted", func(t *testing.T) {
payload := treeTestPayload()
treeStore, err := store.CreateTreeStorage(payload)
require.NoError(t, err)
@ -77,6 +84,14 @@ func TestSpaceStorage_NewAndCreateTree(t *testing.T) {
otherStore, err := store.TreeStorage(payload.RootRawChange.Id)
require.NoError(t, err)
testTreePayload(t, otherStore, payload)
initialStatus := "deleted"
err = store.SetTreeDeletedStatus(otherStore.Id(), initialStatus)
require.NoError(t, err)
status, err := store.TreeDeletedStatus(otherStore.Id())
require.NoError(t, err)
require.Equal(t, initialStatus, status)
})
}
@ -101,8 +116,11 @@ func TestSpaceStorage_StoredIds(t *testing.T) {
_, err := store.CreateTreeStorage(treePayload)
require.NoError(t, err)
}
ids = append(ids, payload.SpaceSettingsWithId.Id)
sort.Strings(ids)
storedIds, err := store.StoredIds()
require.NoError(t, err)
sort.Strings(storedIds)
require.Equal(t, ids, storedIds)
}

View File

@ -4,8 +4,10 @@ import (
"github.com/anytypeio/go-anytype-infrastructure-experiments/common/commonspace/spacesyncproto"
spacestorage "github.com/anytypeio/go-anytype-infrastructure-experiments/common/commonspace/storage"
"github.com/anytypeio/go-anytype-infrastructure-experiments/common/pkg/acl/aclrecordproto"
"github.com/anytypeio/go-anytype-infrastructure-experiments/common/pkg/acl/treechangeproto"
"github.com/stretchr/testify/require"
"os"
"sort"
"strconv"
"testing"
)
@ -19,9 +21,14 @@ func spaceTestPayload() spacestorage.SpaceStorageCreatePayload {
Payload: []byte("aclRoot"),
Id: "aclRootId",
}
settings := &treechangeproto.RawTreeChangeWithId{
RawChange: []byte("settings"),
Id: "settingsId",
}
return spacestorage.SpaceStorageCreatePayload{
AclWithId: aclRoot,
SpaceHeaderWithId: header,
AclWithId: aclRoot,
SpaceHeaderWithId: header,
SpaceSettingsWithId: settings,
}
}
@ -68,7 +75,7 @@ func TestSpaceStorage_NewAndCreateTree(t *testing.T) {
}()
testSpace(t, store, payload)
t.Run("create tree and get tree", func(t *testing.T) {
t.Run("create tree, get tree and mark deleted", func(t *testing.T) {
payload := treeTestPayload()
treeStore, err := store.CreateTreeStorage(payload)
require.NoError(t, err)
@ -77,6 +84,14 @@ func TestSpaceStorage_NewAndCreateTree(t *testing.T) {
otherStore, err := store.TreeStorage(payload.RootRawChange.Id)
require.NoError(t, err)
testTreePayload(t, otherStore, payload)
initialStatus := "deleted"
err = store.SetTreeDeletedStatus(otherStore.Id(), initialStatus)
require.NoError(t, err)
status, err := store.TreeDeletedStatus(otherStore.Id())
require.NoError(t, err)
require.Equal(t, initialStatus, status)
})
}
@ -100,8 +115,11 @@ func TestSpaceStorage_StoredIds(t *testing.T) {
_, err := store.CreateTreeStorage(treePayload)
require.NoError(t, err)
}
ids = append(ids, payload.SpaceSettingsWithId.Id)
sort.Strings(ids)
storedIds, err := store.StoredIds()
sort.Strings(storedIds)
require.NoError(t, err)
require.Equal(t, ids, storedIds)
}