store/storetest package (#7588)

* prerequisites

* storetest package

1 files changed, 3 insertions, 307 deletions

diff --git a/store/sqlstore/compliance_store_test.go b/store/sqlstore/compliance_store_test.go
index 16d5e18b3..c4e0b480c 100644
--- a/store/sqlstore/compliance_store_test.go
+++ b/store/sqlstore/compliance_store_test.go
@@ -5,314 +5,10 @@ package sqlstore
 
 import (
 	"testing"
-	"time"
 
-	"github.com/mattermost/mattermost-server/model"
-	"github.com/mattermost/mattermost-server/store"
+	"github.com/mattermost/mattermost-server/store/storetest"
 )
 
-func TestSqlComplianceStore(t *testing.T) {
-	ss := Setup()
-
-	compliance1 := &model.Compliance{Desc: "Audit for federal subpoena case #22443", UserId: model.NewId(), Status: model.COMPLIANCE_STATUS_FAILED, StartAt: model.GetMillis() - 1, EndAt: model.GetMillis() + 1, Type: model.COMPLIANCE_TYPE_ADHOC}
-	store.Must(ss.Compliance().Save(compliance1))
-	time.Sleep(100 * time.Millisecond)
-
-	compliance2 := &model.Compliance{Desc: "Audit for federal subpoena case #11458", UserId: model.NewId(), Status: model.COMPLIANCE_STATUS_RUNNING, StartAt: model.GetMillis() - 1, EndAt: model.GetMillis() + 1, Type: model.COMPLIANCE_TYPE_ADHOC}
-	store.Must(ss.Compliance().Save(compliance2))
-	time.Sleep(100 * time.Millisecond)
-
-	c := ss.Compliance().GetAll(0, 1000)
-	result := <-c
-	compliances := result.Data.(model.Compliances)
-
-	if compliances[0].Status != model.COMPLIANCE_STATUS_RUNNING && compliance2.Id != compliances[0].Id {
-		t.Fatal()
-	}
-
-	compliance2.Status = model.COMPLIANCE_STATUS_FAILED
-	store.Must(ss.Compliance().Update(compliance2))
-
-	c = ss.Compliance().GetAll(0, 1000)
-	result = <-c
-	compliances = result.Data.(model.Compliances)
-
-	if compliances[0].Status != model.COMPLIANCE_STATUS_FAILED && compliance2.Id != compliances[0].Id {
-		t.Fatal()
-	}
-
-	c = ss.Compliance().GetAll(0, 1)
-	result = <-c
-	compliances = result.Data.(model.Compliances)
-
-	if len(compliances) != 1 {
-		t.Fatal("should only have returned 1")
-	}
-
-	c = ss.Compliance().GetAll(1, 1)
-	result = <-c
-	compliances = result.Data.(model.Compliances)
-
-	if len(compliances) != 1 {
-		t.Fatal("should only have returned 1")
-	}
-
-	rc2 := (<-ss.Compliance().Get(compliance2.Id)).Data.(*model.Compliance)
-	if rc2.Status != compliance2.Status {
-		t.Fatal()
-	}
-}
-
-func TestComplianceExport(t *testing.T) {
-	ss := Setup()
-
-	time.Sleep(100 * time.Millisecond)
-
-	t1 := &model.Team{}
-	t1.DisplayName = "DisplayName"
-	t1.Name = "zz" + model.NewId() + "b"
-	t1.Email = model.NewId() + "@nowhere.com"
-	t1.Type = model.TEAM_OPEN
-	t1 = store.Must(ss.Team().Save(t1)).(*model.Team)
-
-	u1 := &model.User{}
-	u1.Email = model.NewId()
-	u1.Username = model.NewId()
-	u1 = store.Must(ss.User().Save(u1)).(*model.User)
-	store.Must(ss.Team().SaveMember(&model.TeamMember{TeamId: t1.Id, UserId: u1.Id}))
-
-	u2 := &model.User{}
-	u2.Email = model.NewId()
-	u2.Username = model.NewId()
-	u2 = store.Must(ss.User().Save(u2)).(*model.User)
-	store.Must(ss.Team().SaveMember(&model.TeamMember{TeamId: t1.Id, UserId: u2.Id}))
-
-	c1 := &model.Channel{}
-	c1.TeamId = t1.Id
-	c1.DisplayName = "Channel2"
-	c1.Name = "zz" + model.NewId() + "b"
-	c1.Type = model.CHANNEL_OPEN
-	c1 = store.Must(ss.Channel().Save(c1)).(*model.Channel)
-
-	o1 := &model.Post{}
-	o1.ChannelId = c1.Id
-	o1.UserId = u1.Id
-	o1.CreateAt = model.GetMillis()
-	o1.Message = "zz" + model.NewId() + "b"
-	o1 = store.Must(ss.Post().Save(o1)).(*model.Post)
-
-	o1a := &model.Post{}
-	o1a.ChannelId = c1.Id
-	o1a.UserId = u1.Id
-	o1a.CreateAt = o1.CreateAt + 10
-	o1a.Message = "zz" + model.NewId() + "b"
-	o1a = store.Must(ss.Post().Save(o1a)).(*model.Post)
-
-	o2 := &model.Post{}
-	o2.ChannelId = c1.Id
-	o2.UserId = u1.Id
-	o2.CreateAt = o1.CreateAt + 20
-	o2.Message = "zz" + model.NewId() + "b"
-	o2 = store.Must(ss.Post().Save(o2)).(*model.Post)
-
-	o2a := &model.Post{}
-	o2a.ChannelId = c1.Id
-	o2a.UserId = u2.Id
-	o2a.CreateAt = o1.CreateAt + 30
-	o2a.Message = "zz" + model.NewId() + "b"
-	o2a = store.Must(ss.Post().Save(o2a)).(*model.Post)
-
-	time.Sleep(100 * time.Millisecond)
-
-	cr1 := &model.Compliance{Desc: "test" + model.NewId(), StartAt: o1.CreateAt - 1, EndAt: o2a.CreateAt + 1}
-	if r1 := <-ss.Compliance().ComplianceExport(cr1); r1.Err != nil {
-		t.Fatal(r1.Err)
-	} else {
-		cposts := r1.Data.([]*model.CompliancePost)
-
-		if len(cposts) != 4 {
-			t.Fatal("return wrong results length")
-		}
-
-		if cposts[0].PostId != o1.Id {
-			t.Fatal("Wrong sort")
-		}
-
-		if cposts[3].PostId != o2a.Id {
-			t.Fatal("Wrong sort")
-		}
-	}
-
-	cr2 := &model.Compliance{Desc: "test" + model.NewId(), StartAt: o1.CreateAt - 1, EndAt: o2a.CreateAt + 1, Emails: u2.Email}
-	if r1 := <-ss.Compliance().ComplianceExport(cr2); r1.Err != nil {
-		t.Fatal(r1.Err)
-	} else {
-		cposts := r1.Data.([]*model.CompliancePost)
-
-		if len(cposts) != 1 {
-			t.Fatal("return wrong results length")
-		}
-
-		if cposts[0].PostId != o2a.Id {
-			t.Fatal("Wrong sort")
-		}
-	}
-
-	cr3 := &model.Compliance{Desc: "test" + model.NewId(), StartAt: o1.CreateAt - 1, EndAt: o2a.CreateAt + 1, Emails: u2.Email + ", " + u1.Email}
-	if r1 := <-ss.Compliance().ComplianceExport(cr3); r1.Err != nil {
-		t.Fatal(r1.Err)
-	} else {
-		cposts := r1.Data.([]*model.CompliancePost)
-
-		if len(cposts) != 4 {
-			t.Fatal("return wrong results length")
-		}
-
-		if cposts[0].PostId != o1.Id {
-			t.Fatal("Wrong sort")
-		}
-
-		if cposts[3].PostId != o2a.Id {
-			t.Fatal("Wrong sort")
-		}
-	}
-
-	cr4 := &model.Compliance{Desc: "test" + model.NewId(), StartAt: o1.CreateAt - 1, EndAt: o2a.CreateAt + 1, Keywords: o2a.Message}
-	if r1 := <-ss.Compliance().ComplianceExport(cr4); r1.Err != nil {
-		t.Fatal(r1.Err)
-	} else {
-		cposts := r1.Data.([]*model.CompliancePost)
-
-		if len(cposts) != 1 {
-			t.Fatal("return wrong results length")
-		}
-
-		if cposts[0].PostId != o2a.Id {
-			t.Fatal("Wrong sort")
-		}
-	}
-
-	cr5 := &model.Compliance{Desc: "test" + model.NewId(), StartAt: o1.CreateAt - 1, EndAt: o2a.CreateAt + 1, Keywords: o2a.Message + " " + o1.Message}
-	if r1 := <-ss.Compliance().ComplianceExport(cr5); r1.Err != nil {
-		t.Fatal(r1.Err)
-	} else {
-		cposts := r1.Data.([]*model.CompliancePost)
-
-		if len(cposts) != 2 {
-			t.Fatal("return wrong results length")
-		}
-
-		if cposts[0].PostId != o1.Id {
-			t.Fatal("Wrong sort")
-		}
-	}
-
-	cr6 := &model.Compliance{Desc: "test" + model.NewId(), StartAt: o1.CreateAt - 1, EndAt: o2a.CreateAt + 1, Emails: u2.Email + ", " + u1.Email, Keywords: o2a.Message + " " + o1.Message}
-	if r1 := <-ss.Compliance().ComplianceExport(cr6); r1.Err != nil {
-		t.Fatal(r1.Err)
-	} else {
-		cposts := r1.Data.([]*model.CompliancePost)
-
-		if len(cposts) != 2 {
-			t.Fatal("return wrong results length")
-		}
-
-		if cposts[0].PostId != o1.Id {
-			t.Fatal("Wrong sort")
-		}
-
-		if cposts[1].PostId != o2a.Id {
-			t.Fatal("Wrong sort")
-		}
-	}
-}
-
-func TestComplianceExportDirectMessages(t *testing.T) {
-	ss := Setup()
-
-	time.Sleep(100 * time.Millisecond)
-
-	t1 := &model.Team{}
-	t1.DisplayName = "DisplayName"
-	t1.Name = "zz" + model.NewId() + "b"
-	t1.Email = model.NewId() + "@nowhere.com"
-	t1.Type = model.TEAM_OPEN
-	t1 = store.Must(ss.Team().Save(t1)).(*model.Team)
-
-	u1 := &model.User{}
-	u1.Email = model.NewId()
-	u1.Username = model.NewId()
-	u1 = store.Must(ss.User().Save(u1)).(*model.User)
-	store.Must(ss.Team().SaveMember(&model.TeamMember{TeamId: t1.Id, UserId: u1.Id}))
-
-	u2 := &model.User{}
-	u2.Email = model.NewId()
-	u2.Username = model.NewId()
-	u2 = store.Must(ss.User().Save(u2)).(*model.User)
-	store.Must(ss.Team().SaveMember(&model.TeamMember{TeamId: t1.Id, UserId: u2.Id}))
-
-	c1 := &model.Channel{}
-	c1.TeamId = t1.Id
-	c1.DisplayName = "Channel2"
-	c1.Name = "zz" + model.NewId() + "b"
-	c1.Type = model.CHANNEL_OPEN
-	c1 = store.Must(ss.Channel().Save(c1)).(*model.Channel)
-
-	cDM := store.Must(ss.Channel().CreateDirectChannel(u1.Id, u2.Id)).(*model.Channel)
-
-	o1 := &model.Post{}
-	o1.ChannelId = c1.Id
-	o1.UserId = u1.Id
-	o1.CreateAt = model.GetMillis()
-	o1.Message = "zz" + model.NewId() + "b"
-	o1 = store.Must(ss.Post().Save(o1)).(*model.Post)
-
-	o1a := &model.Post{}
-	o1a.ChannelId = c1.Id
-	o1a.UserId = u1.Id
-	o1a.CreateAt = o1.CreateAt + 10
-	o1a.Message = "zz" + model.NewId() + "b"
-	o1a = store.Must(ss.Post().Save(o1a)).(*model.Post)
-
-	o2 := &model.Post{}
-	o2.ChannelId = c1.Id
-	o2.UserId = u1.Id
-	o2.CreateAt = o1.CreateAt + 20
-	o2.Message = "zz" + model.NewId() + "b"
-	o2 = store.Must(ss.Post().Save(o2)).(*model.Post)
-
-	o2a := &model.Post{}
-	o2a.ChannelId = c1.Id
-	o2a.UserId = u2.Id
-	o2a.CreateAt = o1.CreateAt + 30
-	o2a.Message = "zz" + model.NewId() + "b"
-	o2a = store.Must(ss.Post().Save(o2a)).(*model.Post)
-
-	o3 := &model.Post{}
-	o3.ChannelId = cDM.Id
-	o3.UserId = u1.Id
-	o3.CreateAt = o1.CreateAt + 40
-	o3.Message = "zz" + model.NewId() + "b"
-	o3 = store.Must(ss.Post().Save(o3)).(*model.Post)
-
-	time.Sleep(100 * time.Millisecond)
-
-	cr1 := &model.Compliance{Desc: "test" + model.NewId(), StartAt: o1.CreateAt - 1, EndAt: o3.CreateAt + 1, Emails: u1.Email}
-	if r1 := <-ss.Compliance().ComplianceExport(cr1); r1.Err != nil {
-		t.Fatal(r1.Err)
-	} else {
-		cposts := r1.Data.([]*model.CompliancePost)
-
-		if len(cposts) != 4 {
-			t.Fatal("return wrong results length")
-		}
-
-		if cposts[0].PostId != o1.Id {
-			t.Fatal("Wrong sort")
-		}
-
-		if cposts[len(cposts)-1].PostId != o3.Id {
-			t.Fatal("Wrong sort")
-		}
-	}
+func TestComplianceStore(t *testing.T) {
+	StoreTest(t, storetest.TestComplianceStore)
 }