2d261fde59
* Adding mysql support * Fixing other test cases * Tests passing on mysql * Fix the row number generation * Adding blocks_history table * merging the migrations in one single set of files * Passing mysql tests * Fixing default encoding on mysql * Simplifying things * Removing from the blocks table all deleted blocks * Better indentation * Moving db types to constants to make it less error prone * reducing duplicated code * Removing log line * Now sql tests are running properly in mysql, sqlite and postgres
453 lines
11 KiB
Go
453 lines
11 KiB
Go
package storetests
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/mattermost/focalboard/server/model"
|
|
"github.com/mattermost/focalboard/server/services/store"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func StoreTestBlocksStore(t *testing.T, setup func(t *testing.T) (store.Store, func())) {
|
|
container := store.Container{
|
|
WorkspaceID: "0",
|
|
}
|
|
|
|
t.Run("InsertBlock", func(t *testing.T) {
|
|
store, tearDown := setup(t)
|
|
defer tearDown()
|
|
testInsertBlock(t, store, container)
|
|
})
|
|
t.Run("DeleteBlock", func(t *testing.T) {
|
|
store, tearDown := setup(t)
|
|
defer tearDown()
|
|
testDeleteBlock(t, store, container)
|
|
})
|
|
t.Run("GetSubTree2", func(t *testing.T) {
|
|
store, tearDown := setup(t)
|
|
defer tearDown()
|
|
testGetSubTree2(t, store, container)
|
|
})
|
|
t.Run("GetSubTree3", func(t *testing.T) {
|
|
store, tearDown := setup(t)
|
|
defer tearDown()
|
|
testGetSubTree3(t, store, container)
|
|
})
|
|
t.Run("GetParentID", func(t *testing.T) {
|
|
store, tearDown := setup(t)
|
|
defer tearDown()
|
|
testGetParentID(t, store, container)
|
|
})
|
|
t.Run("GetRootID", func(t *testing.T) {
|
|
store, tearDown := setup(t)
|
|
defer tearDown()
|
|
testGetRootID(t, store, container)
|
|
})
|
|
}
|
|
|
|
func testInsertBlock(t *testing.T, store store.Store, container store.Container) {
|
|
userID := "user-id"
|
|
|
|
blocks, err := store.GetAllBlocks(container)
|
|
require.NoError(t, err)
|
|
initialCount := len(blocks)
|
|
|
|
t.Run("valid block", func(t *testing.T) {
|
|
block := model.Block{
|
|
ID: "id-test",
|
|
RootID: "id-test",
|
|
ModifiedBy: userID,
|
|
}
|
|
|
|
err := store.InsertBlock(container, block)
|
|
require.NoError(t, err)
|
|
|
|
blocks, err := store.GetAllBlocks(container)
|
|
require.NoError(t, err)
|
|
require.Len(t, blocks, initialCount+1)
|
|
})
|
|
|
|
t.Run("invalid rootid", func(t *testing.T) {
|
|
block := model.Block{
|
|
ID: "id-test",
|
|
RootID: "",
|
|
ModifiedBy: userID,
|
|
}
|
|
|
|
err := store.InsertBlock(container, block)
|
|
require.Error(t, err)
|
|
|
|
blocks, err := store.GetAllBlocks(container)
|
|
require.NoError(t, err)
|
|
require.Len(t, blocks, initialCount+1)
|
|
})
|
|
|
|
t.Run("invalid fields data", func(t *testing.T) {
|
|
block := model.Block{
|
|
ID: "id-test",
|
|
RootID: "id-test",
|
|
ModifiedBy: userID,
|
|
Fields: map[string]interface{}{"no-serialiable-value": t.Run},
|
|
}
|
|
|
|
err := store.InsertBlock(container, block)
|
|
require.Error(t, err)
|
|
|
|
blocks, err := store.GetAllBlocks(container)
|
|
require.NoError(t, err)
|
|
require.Len(t, blocks, initialCount+1)
|
|
})
|
|
}
|
|
|
|
func testGetSubTree2(t *testing.T, store store.Store, container store.Container) {
|
|
userID := "user-id"
|
|
|
|
blocks, err := store.GetAllBlocks(container)
|
|
require.NoError(t, err)
|
|
initialCount := len(blocks)
|
|
|
|
blocksToInsert := []model.Block{
|
|
{
|
|
ID: "parent",
|
|
RootID: "parent",
|
|
ModifiedBy: userID,
|
|
},
|
|
{
|
|
ID: "child1",
|
|
RootID: "parent",
|
|
ParentID: "parent",
|
|
ModifiedBy: userID,
|
|
},
|
|
{
|
|
ID: "child2",
|
|
RootID: "parent",
|
|
ParentID: "parent",
|
|
ModifiedBy: userID,
|
|
},
|
|
{
|
|
ID: "grandchild1",
|
|
RootID: "parent",
|
|
ParentID: "child1",
|
|
ModifiedBy: userID,
|
|
},
|
|
{
|
|
ID: "grandchild2",
|
|
RootID: "parent",
|
|
ParentID: "child2",
|
|
ModifiedBy: userID,
|
|
},
|
|
{
|
|
ID: "greatgrandchild1",
|
|
RootID: "parent",
|
|
ParentID: "grandchild1",
|
|
ModifiedBy: userID,
|
|
},
|
|
}
|
|
|
|
InsertBlocks(t, store, container, blocksToInsert)
|
|
defer DeleteBlocks(t, store, container, blocksToInsert, "test")
|
|
|
|
blocks, err = store.GetAllBlocks(container)
|
|
require.NoError(t, err)
|
|
require.Len(t, blocks, initialCount+6)
|
|
|
|
t.Run("from root id", func(t *testing.T) {
|
|
blocks, err = store.GetSubTree2(container, "parent")
|
|
require.NoError(t, err)
|
|
require.Len(t, blocks, 3)
|
|
require.True(t, ContainsBlockWithID(blocks, "parent"))
|
|
require.True(t, ContainsBlockWithID(blocks, "child1"))
|
|
require.True(t, ContainsBlockWithID(blocks, "child2"))
|
|
})
|
|
|
|
t.Run("from child id", func(t *testing.T) {
|
|
blocks, err = store.GetSubTree2(container, "child1")
|
|
require.NoError(t, err)
|
|
require.Len(t, blocks, 2)
|
|
require.True(t, ContainsBlockWithID(blocks, "child1"))
|
|
require.True(t, ContainsBlockWithID(blocks, "grandchild1"))
|
|
})
|
|
|
|
t.Run("from not existing id", func(t *testing.T) {
|
|
blocks, err = store.GetSubTree2(container, "not-exists")
|
|
require.NoError(t, err)
|
|
require.Len(t, blocks, 0)
|
|
})
|
|
}
|
|
|
|
func testGetSubTree3(t *testing.T, store store.Store, container store.Container) {
|
|
userID := "user-id"
|
|
|
|
blocks, err := store.GetAllBlocks(container)
|
|
require.NoError(t, err)
|
|
initialCount := len(blocks)
|
|
|
|
blocksToInsert := []model.Block{
|
|
{
|
|
ID: "parent",
|
|
RootID: "parent",
|
|
ModifiedBy: userID,
|
|
},
|
|
{
|
|
ID: "child1",
|
|
RootID: "parent",
|
|
ParentID: "parent",
|
|
ModifiedBy: userID,
|
|
},
|
|
{
|
|
ID: "child2",
|
|
RootID: "parent",
|
|
ParentID: "parent",
|
|
ModifiedBy: userID,
|
|
},
|
|
{
|
|
ID: "grandchild1",
|
|
RootID: "parent",
|
|
ParentID: "child1",
|
|
ModifiedBy: userID,
|
|
},
|
|
{
|
|
ID: "grandchild2",
|
|
RootID: "parent",
|
|
ParentID: "child2",
|
|
ModifiedBy: userID,
|
|
},
|
|
{
|
|
ID: "greatgrandchild1",
|
|
RootID: "parent",
|
|
ParentID: "grandchild1",
|
|
ModifiedBy: userID,
|
|
},
|
|
}
|
|
|
|
InsertBlocks(t, store, container, blocksToInsert)
|
|
defer DeleteBlocks(t, store, container, blocksToInsert, "test")
|
|
|
|
blocks, err = store.GetAllBlocks(container)
|
|
require.NoError(t, err)
|
|
require.Len(t, blocks, initialCount+6)
|
|
|
|
t.Run("from root id", func(t *testing.T) {
|
|
blocks, err = store.GetSubTree3(container, "parent")
|
|
require.NoError(t, err)
|
|
require.Len(t, blocks, 5)
|
|
require.True(t, ContainsBlockWithID(blocks, "parent"))
|
|
require.True(t, ContainsBlockWithID(blocks, "child1"))
|
|
require.True(t, ContainsBlockWithID(blocks, "child2"))
|
|
require.True(t, ContainsBlockWithID(blocks, "grandchild1"))
|
|
require.True(t, ContainsBlockWithID(blocks, "grandchild2"))
|
|
})
|
|
|
|
t.Run("from child id", func(t *testing.T) {
|
|
blocks, err = store.GetSubTree3(container, "child1")
|
|
require.NoError(t, err)
|
|
require.Len(t, blocks, 3)
|
|
require.True(t, ContainsBlockWithID(blocks, "child1"))
|
|
require.True(t, ContainsBlockWithID(blocks, "grandchild1"))
|
|
require.True(t, ContainsBlockWithID(blocks, "greatgrandchild1"))
|
|
})
|
|
|
|
t.Run("from not existing id", func(t *testing.T) {
|
|
blocks, err = store.GetSubTree3(container, "not-exists")
|
|
require.NoError(t, err)
|
|
require.Len(t, blocks, 0)
|
|
})
|
|
}
|
|
|
|
func testGetRootID(t *testing.T, store store.Store, container store.Container) {
|
|
userID := "user-id"
|
|
|
|
blocks, err := store.GetAllBlocks(container)
|
|
require.NoError(t, err)
|
|
initialCount := len(blocks)
|
|
|
|
blocksToInsert := []model.Block{
|
|
{
|
|
ID: "parent",
|
|
RootID: "parent",
|
|
ModifiedBy: userID,
|
|
},
|
|
{
|
|
ID: "child1",
|
|
RootID: "parent",
|
|
ParentID: "parent",
|
|
ModifiedBy: userID,
|
|
},
|
|
{
|
|
ID: "child2",
|
|
RootID: "parent",
|
|
ParentID: "parent",
|
|
ModifiedBy: userID,
|
|
},
|
|
{
|
|
ID: "grandchild1",
|
|
RootID: "parent",
|
|
ParentID: "child1",
|
|
ModifiedBy: userID,
|
|
},
|
|
{
|
|
ID: "grandchild2",
|
|
RootID: "parent",
|
|
ParentID: "child2",
|
|
ModifiedBy: userID,
|
|
},
|
|
{
|
|
ID: "greatgrandchild1",
|
|
RootID: "parent",
|
|
ParentID: "grandchild1",
|
|
ModifiedBy: userID,
|
|
},
|
|
}
|
|
|
|
InsertBlocks(t, store, container, blocksToInsert)
|
|
defer DeleteBlocks(t, store, container, blocksToInsert, "test")
|
|
|
|
blocks, err = store.GetAllBlocks(container)
|
|
require.NoError(t, err)
|
|
require.Len(t, blocks, initialCount+6)
|
|
|
|
t.Run("from root id", func(t *testing.T) {
|
|
rootID, err := store.GetRootID(container, "parent")
|
|
require.NoError(t, err)
|
|
require.Equal(t, "parent", rootID)
|
|
})
|
|
|
|
t.Run("from child id", func(t *testing.T) {
|
|
rootID, err := store.GetRootID(container, "child1")
|
|
require.NoError(t, err)
|
|
require.Equal(t, "parent", rootID)
|
|
})
|
|
|
|
t.Run("from not existing id", func(t *testing.T) {
|
|
_, err := store.GetRootID(container, "not-exists")
|
|
require.Error(t, err)
|
|
})
|
|
}
|
|
|
|
func testGetParentID(t *testing.T, store store.Store, container store.Container) {
|
|
userID := "user-id"
|
|
|
|
blocks, err := store.GetAllBlocks(container)
|
|
require.NoError(t, err)
|
|
initialCount := len(blocks)
|
|
|
|
blocksToInsert := []model.Block{
|
|
{
|
|
ID: "parent",
|
|
RootID: "parent",
|
|
ModifiedBy: userID,
|
|
},
|
|
{
|
|
ID: "child1",
|
|
RootID: "parent",
|
|
ParentID: "parent",
|
|
ModifiedBy: userID,
|
|
},
|
|
{
|
|
ID: "child2",
|
|
RootID: "parent",
|
|
ParentID: "parent",
|
|
ModifiedBy: userID,
|
|
},
|
|
{
|
|
ID: "grandchild1",
|
|
RootID: "parent",
|
|
ParentID: "child1",
|
|
ModifiedBy: userID,
|
|
},
|
|
{
|
|
ID: "grandchild2",
|
|
RootID: "parent",
|
|
ParentID: "child2",
|
|
ModifiedBy: userID,
|
|
},
|
|
{
|
|
ID: "greatgrandchild1",
|
|
RootID: "parent",
|
|
ParentID: "grandchild1",
|
|
ModifiedBy: userID,
|
|
},
|
|
}
|
|
|
|
InsertBlocks(t, store, container, blocksToInsert)
|
|
defer DeleteBlocks(t, store, container, blocksToInsert, "test")
|
|
|
|
blocks, err = store.GetAllBlocks(container)
|
|
require.NoError(t, err)
|
|
require.Len(t, blocks, initialCount+6)
|
|
|
|
t.Run("from root id", func(t *testing.T) {
|
|
parentID, err := store.GetParentID(container, "parent")
|
|
require.NoError(t, err)
|
|
require.Equal(t, "", parentID)
|
|
})
|
|
|
|
t.Run("from child id", func(t *testing.T) {
|
|
parentID, err := store.GetParentID(container, "grandchild1")
|
|
require.NoError(t, err)
|
|
require.Equal(t, "child1", parentID)
|
|
})
|
|
|
|
t.Run("from not existing id", func(t *testing.T) {
|
|
_, err := store.GetParentID(container, "not-exists")
|
|
require.Error(t, err)
|
|
})
|
|
}
|
|
|
|
func testDeleteBlock(t *testing.T, store store.Store, container store.Container) {
|
|
userID := "user-id"
|
|
|
|
blocks, err := store.GetAllBlocks(container)
|
|
require.NoError(t, err)
|
|
initialCount := len(blocks)
|
|
|
|
blocksToInsert := []model.Block{
|
|
{
|
|
ID: "block1",
|
|
RootID: "block1",
|
|
ModifiedBy: userID,
|
|
},
|
|
{
|
|
ID: "block2",
|
|
RootID: "block2",
|
|
ModifiedBy: userID,
|
|
},
|
|
{
|
|
ID: "block3",
|
|
RootID: "block3",
|
|
ModifiedBy: userID,
|
|
},
|
|
}
|
|
InsertBlocks(t, store, container, blocksToInsert)
|
|
defer DeleteBlocks(t, store, container, blocksToInsert, "test")
|
|
|
|
blocks, err = store.GetAllBlocks(container)
|
|
require.NoError(t, err)
|
|
require.Len(t, blocks, initialCount+3)
|
|
|
|
t.Run("exiting id", func(t *testing.T) {
|
|
// Wait for not colliding the ID+insert_at key
|
|
time.Sleep(1 * time.Millisecond)
|
|
err := store.DeleteBlock(container, "block1", userID)
|
|
require.NoError(t, err)
|
|
})
|
|
|
|
t.Run("exiting id multiple times", func(t *testing.T) {
|
|
// Wait for not colliding the ID+insert_at key
|
|
time.Sleep(1 * time.Millisecond)
|
|
err := store.DeleteBlock(container, "block1", userID)
|
|
require.NoError(t, err)
|
|
// Wait for not colliding the ID+insert_at key
|
|
time.Sleep(1 * time.Millisecond)
|
|
err = store.DeleteBlock(container, "block1", userID)
|
|
require.NoError(t, err)
|
|
})
|
|
|
|
t.Run("from not existing id", func(t *testing.T) {
|
|
// Wait for not colliding the ID+insert_at key
|
|
time.Sleep(1 * time.Millisecond)
|
|
err := store.DeleteBlock(container, "not-exists", userID)
|
|
require.NoError(t, err)
|
|
})
|
|
}
|