1 files changed, 38 insertions, 0 deletions

diff --git a/store/sql_file_info_store_test.go b/store/sql_file_info_store_test.go
index daec06269..fecd862c0 100644
--- a/store/sql_file_info_store_test.go
+++ b/store/sql_file_info_store_test.go
@@ -25,6 +25,9 @@ func TestFileInfoSaveGet(t *testing.T) {
 	} else {
 		info = returned
 	}
+	defer func() {
+		<-store.FileInfo().PermanentDelete(info.Id)
+	}()
 
 	if result := <-store.FileInfo().Get(info.Id); result.Err != nil {
 		t.Fatal(result.Err)
@@ -43,6 +46,9 @@ func TestFileInfoSaveGet(t *testing.T) {
 	if result := <-store.FileInfo().Get(info2.Id); result.Err == nil {
 		t.Fatal("shouldn't have gotten deleted file")
 	}
+	defer func() {
+		<-store.FileInfo().PermanentDelete(info2.Id)
+	}()
 }
 
 func TestFileInfoSaveGetByPath(t *testing.T) {
@@ -60,6 +66,9 @@ func TestFileInfoSaveGetByPath(t *testing.T) {
 	} else {
 		info = returned
 	}
+	defer func() {
+		<-store.FileInfo().PermanentDelete(info.Id)
+	}()
 
 	if result := <-store.FileInfo().GetByPath(info.Path); result.Err != nil {
 		t.Fatal(result.Err)
@@ -78,6 +87,9 @@ func TestFileInfoSaveGetByPath(t *testing.T) {
 	if result := <-store.FileInfo().GetByPath(info2.Id); result.Err == nil {
 		t.Fatal("shouldn't have gotten deleted file")
 	}
+	defer func() {
+		<-store.FileInfo().PermanentDelete(info2.Id)
+	}()
 }
 
 func TestFileInfoGetForPost(t *testing.T) {
@@ -112,6 +124,9 @@ func TestFileInfoGetForPost(t *testing.T) {
 
 	for i, info := range infos {
 		infos[i] = Must(store.FileInfo().Save(info)).(*model.FileInfo)
+		defer func(id string) {
+			<-store.FileInfo().PermanentDelete(id)
+		}(infos[i].Id)
 	}
 
 	if result := <-store.FileInfo().GetForPost(postId, true, false); result.Err != nil {
@@ -143,6 +158,9 @@ func TestFileInfoAttachToPost(t *testing.T) {
 		CreatorId: userId,
 		Path:      "file.txt",
 	})).(*model.FileInfo)
+	defer func() {
+		<-store.FileInfo().PermanentDelete(info1.Id)
+	}()
 
 	if len(info1.PostId) != 0 {
 		t.Fatal("file shouldn't have a PostId")
@@ -162,6 +180,9 @@ func TestFileInfoAttachToPost(t *testing.T) {
 		CreatorId: userId,
 		Path:      "file.txt",
 	})).(*model.FileInfo)
+	defer func() {
+		<-store.FileInfo().PermanentDelete(info2.Id)
+	}()
 
 	if result := <-store.FileInfo().AttachToPost(info2.Id, postId); result.Err != nil {
 		t.Fatal(result.Err)
@@ -208,6 +229,9 @@ func TestFileInfoDeleteForPost(t *testing.T) {
 
 	for i, info := range infos {
 		infos[i] = Must(store.FileInfo().Save(info)).(*model.FileInfo)
+		defer func(id string) {
+			<-store.FileInfo().PermanentDelete(id)
+		}(infos[i].Id)
 	}
 
 	if result := <-store.FileInfo().DeleteForPost(postId); result.Err != nil {
@@ -218,3 +242,17 @@ func TestFileInfoDeleteForPost(t *testing.T) {
 		t.Fatal("shouldn't have returned any file infos")
 	}
 }
+
+func TestFileInfoPermanentDelete(t *testing.T) {
+	Setup()
+
+	info := Must(store.FileInfo().Save(&model.FileInfo{
+		PostId:    model.NewId(),
+		CreatorId: model.NewId(),
+		Path:      "file.txt",
+	})).(*model.FileInfo)
+
+	if result := <-store.FileInfo().PermanentDelete(info.Id); result.Err != nil {
+		t.Fatal(result.Err)
+	}
+}