store/storetest package (#7588)

* prerequisites

* storetest package

1 files changed, 4 insertions, 290 deletions

diff --git a/store/sqlstore/file_info_store_test.go b/store/sqlstore/file_info_store_test.go
index 8ddf34e19..c1295ee54 100644
--- a/store/sqlstore/file_info_store_test.go
+++ b/store/sqlstore/file_info_store_test.go
@@ -1,300 +1,14 @@
-// Copyright (c) 2016-present Mattermost, Inc. All Rights Reserved.
+// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
 // See License.txt for license information.
 
 package sqlstore
 
 import (
-	"fmt"
 	"testing"
 
-	"github.com/mattermost/mattermost-server/model"
-	"github.com/mattermost/mattermost-server/store"
+	"github.com/mattermost/mattermost-server/store/storetest"
 )
 
-func TestFileInfoSaveGet(t *testing.T) {
-	ss := Setup()
-
-	info := &model.FileInfo{
-		CreatorId: model.NewId(),
-		Path:      "file.txt",
-	}
-
-	if result := <-ss.FileInfo().Save(info); result.Err != nil {
-		t.Fatal(result.Err)
-	} else if returned := result.Data.(*model.FileInfo); len(returned.Id) == 0 {
-		t.Fatal("should've assigned an id to FileInfo")
-	} else {
-		info = returned
-	}
-	defer func() {
-		<-ss.FileInfo().PermanentDelete(info.Id)
-	}()
-
-	if result := <-ss.FileInfo().Get(info.Id); result.Err != nil {
-		t.Fatal(result.Err)
-	} else if returned := result.Data.(*model.FileInfo); returned.Id != info.Id {
-		t.Log(info)
-		t.Log(returned)
-		t.Fatal("should've returned correct FileInfo")
-	}
-
-	info2 := store.Must(ss.FileInfo().Save(&model.FileInfo{
-		CreatorId: model.NewId(),
-		Path:      "file.txt",
-		DeleteAt:  123,
-	})).(*model.FileInfo)
-
-	if result := <-ss.FileInfo().Get(info2.Id); result.Err == nil {
-		t.Fatal("shouldn't have gotten deleted file")
-	}
-	defer func() {
-		<-ss.FileInfo().PermanentDelete(info2.Id)
-	}()
-}
-
-func TestFileInfoSaveGetByPath(t *testing.T) {
-	ss := Setup()
-
-	info := &model.FileInfo{
-		CreatorId: model.NewId(),
-		Path:      fmt.Sprintf("%v/file.txt", model.NewId()),
-	}
-
-	if result := <-ss.FileInfo().Save(info); result.Err != nil {
-		t.Fatal(result.Err)
-	} else if returned := result.Data.(*model.FileInfo); len(returned.Id) == 0 {
-		t.Fatal("should've assigned an id to FileInfo")
-	} else {
-		info = returned
-	}
-	defer func() {
-		<-ss.FileInfo().PermanentDelete(info.Id)
-	}()
-
-	if result := <-ss.FileInfo().GetByPath(info.Path); result.Err != nil {
-		t.Fatal(result.Err)
-	} else if returned := result.Data.(*model.FileInfo); returned.Id != info.Id {
-		t.Log(info)
-		t.Log(returned)
-		t.Fatal("should've returned correct FileInfo")
-	}
-
-	info2 := store.Must(ss.FileInfo().Save(&model.FileInfo{
-		CreatorId: model.NewId(),
-		Path:      "file.txt",
-		DeleteAt:  123,
-	})).(*model.FileInfo)
-
-	if result := <-ss.FileInfo().GetByPath(info2.Id); result.Err == nil {
-		t.Fatal("shouldn't have gotten deleted file")
-	}
-	defer func() {
-		<-ss.FileInfo().PermanentDelete(info2.Id)
-	}()
-}
-
-func TestFileInfoGetForPost(t *testing.T) {
-	ss := Setup()
-
-	userId := model.NewId()
-	postId := model.NewId()
-
-	infos := []*model.FileInfo{
-		{
-			PostId:    postId,
-			CreatorId: userId,
-			Path:      "file.txt",
-		},
-		{
-			PostId:    postId,
-			CreatorId: userId,
-			Path:      "file.txt",
-		},
-		{
-			PostId:    postId,
-			CreatorId: userId,
-			Path:      "file.txt",
-			DeleteAt:  123,
-		},
-		{
-			PostId:    model.NewId(),
-			CreatorId: userId,
-			Path:      "file.txt",
-		},
-	}
-
-	for i, info := range infos {
-		infos[i] = store.Must(ss.FileInfo().Save(info)).(*model.FileInfo)
-		defer func(id string) {
-			<-ss.FileInfo().PermanentDelete(id)
-		}(infos[i].Id)
-	}
-
-	if result := <-ss.FileInfo().GetForPost(postId, true, false); result.Err != nil {
-		t.Fatal(result.Err)
-	} else if returned := result.Data.([]*model.FileInfo); len(returned) != 2 {
-		t.Fatal("should've returned exactly 2 file infos")
-	}
-
-	if result := <-ss.FileInfo().GetForPost(postId, false, false); result.Err != nil {
-		t.Fatal(result.Err)
-	} else if returned := result.Data.([]*model.FileInfo); len(returned) != 2 {
-		t.Fatal("should've returned exactly 2 file infos")
-	}
-
-	if result := <-ss.FileInfo().GetForPost(postId, true, true); result.Err != nil {
-		t.Fatal(result.Err)
-	} else if returned := result.Data.([]*model.FileInfo); len(returned) != 2 {
-		t.Fatal("should've returned exactly 2 file infos")
-	}
-}
-
-func TestFileInfoAttachToPost(t *testing.T) {
-	ss := Setup()
-
-	userId := model.NewId()
-	postId := model.NewId()
-
-	info1 := store.Must(ss.FileInfo().Save(&model.FileInfo{
-		CreatorId: userId,
-		Path:      "file.txt",
-	})).(*model.FileInfo)
-	defer func() {
-		<-ss.FileInfo().PermanentDelete(info1.Id)
-	}()
-
-	if len(info1.PostId) != 0 {
-		t.Fatal("file shouldn't have a PostId")
-	}
-
-	if result := <-ss.FileInfo().AttachToPost(info1.Id, postId); result.Err != nil {
-		t.Fatal(result.Err)
-	} else {
-		info1 = store.Must(ss.FileInfo().Get(info1.Id)).(*model.FileInfo)
-	}
-
-	if len(info1.PostId) == 0 {
-		t.Fatal("file should now have a PostId")
-	}
-
-	info2 := store.Must(ss.FileInfo().Save(&model.FileInfo{
-		CreatorId: userId,
-		Path:      "file.txt",
-	})).(*model.FileInfo)
-	defer func() {
-		<-ss.FileInfo().PermanentDelete(info2.Id)
-	}()
-
-	if result := <-ss.FileInfo().AttachToPost(info2.Id, postId); result.Err != nil {
-		t.Fatal(result.Err)
-	} else {
-		info2 = store.Must(ss.FileInfo().Get(info2.Id)).(*model.FileInfo)
-	}
-
-	if result := <-ss.FileInfo().GetForPost(postId, true, false); result.Err != nil {
-		t.Fatal(result.Err)
-	} else if infos := result.Data.([]*model.FileInfo); len(infos) != 2 {
-		t.Fatal("should've returned exactly 2 file infos")
-	}
-}
-
-func TestFileInfoDeleteForPost(t *testing.T) {
-	ss := Setup()
-
-	userId := model.NewId()
-	postId := model.NewId()
-
-	infos := []*model.FileInfo{
-		{
-			PostId:    postId,
-			CreatorId: userId,
-			Path:      "file.txt",
-		},
-		{
-			PostId:    postId,
-			CreatorId: userId,
-			Path:      "file.txt",
-		},
-		{
-			PostId:    postId,
-			CreatorId: userId,
-			Path:      "file.txt",
-			DeleteAt:  123,
-		},
-		{
-			PostId:    model.NewId(),
-			CreatorId: userId,
-			Path:      "file.txt",
-		},
-	}
-
-	for i, info := range infos {
-		infos[i] = store.Must(ss.FileInfo().Save(info)).(*model.FileInfo)
-		defer func(id string) {
-			<-ss.FileInfo().PermanentDelete(id)
-		}(infos[i].Id)
-	}
-
-	if result := <-ss.FileInfo().DeleteForPost(postId); result.Err != nil {
-		t.Fatal(result.Err)
-	}
-
-	if infos := store.Must(ss.FileInfo().GetForPost(postId, true, false)).([]*model.FileInfo); len(infos) != 0 {
-		t.Fatal("shouldn't have returned any file infos")
-	}
-}
-
-func TestFileInfoPermanentDelete(t *testing.T) {
-	ss := Setup()
-
-	info := store.Must(ss.FileInfo().Save(&model.FileInfo{
-		PostId:    model.NewId(),
-		CreatorId: model.NewId(),
-		Path:      "file.txt",
-	})).(*model.FileInfo)
-
-	if result := <-ss.FileInfo().PermanentDelete(info.Id); result.Err != nil {
-		t.Fatal(result.Err)
-	}
-}
-
-func TestFileInfoPermanentDeleteBatch(t *testing.T) {
-	ss := Setup()
-
-	postId := model.NewId()
-
-	store.Must(ss.FileInfo().Save(&model.FileInfo{
-		PostId:    postId,
-		CreatorId: model.NewId(),
-		Path:      "file.txt",
-		CreateAt:  1000,
-	}))
-
-	store.Must(ss.FileInfo().Save(&model.FileInfo{
-		PostId:    postId,
-		CreatorId: model.NewId(),
-		Path:      "file.txt",
-		CreateAt:  1200,
-	}))
-
-	store.Must(ss.FileInfo().Save(&model.FileInfo{
-		PostId:    postId,
-		CreatorId: model.NewId(),
-		Path:      "file.txt",
-		CreateAt:  2000,
-	}))
-
-	if result := <-ss.FileInfo().GetForPost(postId, true, false); result.Err != nil {
-		t.Fatal(result.Err)
-	} else if len(result.Data.([]*model.FileInfo)) != 3 {
-		t.Fatal("Expected 3 fileInfos")
-	}
-
-	store.Must(ss.FileInfo().PermanentDeleteBatch(1500, 1000))
-
-	if result := <-ss.FileInfo().GetForPost(postId, true, false); result.Err != nil {
-		t.Fatal(result.Err)
-	} else if len(result.Data.([]*model.FileInfo)) != 1 {
-		t.Fatal("Expected 3 fileInfos")
-	}
+func TestFileInfoStore(t *testing.T) {
+	StoreTest(t, storetest.TestFileInfoStore)
 }