d3e934d07a
* Added username to ChannelMemberHistory struct in anticipation of supporting GlobalRelay in Compliance Export * Removed translation from debug output - this makes it complicated to use utils functions from tests in the enterprise repo * Added an advanced email function that allows for greater control over message details. Updated MessageExport config to support GlobalRelay. Added attachment support to InBucket unit tests * Moving templates in from enterprise to solve test issues * Added export format to diagnostics * Changed email attachment code to use FileBackend so that S3 storage is properly supported
441 строка
14 KiB
Go
441 строка
14 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See License.txt for license information.
|
|
|
|
package storetest
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/mattermost/mattermost-server/model"
|
|
"github.com/mattermost/mattermost-server/store"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestComplianceStore(t *testing.T, ss store.Store) {
|
|
t.Run("", func(t *testing.T) { testComplianceStore(t, ss) })
|
|
t.Run("ComplianceExport", func(t *testing.T) { testComplianceExport(t, ss) })
|
|
t.Run("ComplianceExportDirectMessages", func(t *testing.T) { testComplianceExportDirectMessages(t, ss) })
|
|
t.Run("MessageExport", func(t *testing.T) { testComplianceMessageExport(t, ss) })
|
|
}
|
|
|
|
func testComplianceStore(t *testing.T, ss store.Store) {
|
|
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 store.Store) {
|
|
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}, -1))
|
|
|
|
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}, -1))
|
|
|
|
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, -1)).(*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 store.Store) {
|
|
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}, -1))
|
|
|
|
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}, -1))
|
|
|
|
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, -1)).(*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 testComplianceMessageExport(t *testing.T, ss store.Store) {
|
|
// get the starting number of message export entries
|
|
startTime := model.GetMillis()
|
|
var numMessageExports = 0
|
|
if r1 := <-ss.Compliance().MessageExport(startTime-10, 10); r1.Err != nil {
|
|
t.Fatal(r1.Err)
|
|
} else {
|
|
messages := r1.Data.([]*model.MessageExport)
|
|
numMessageExports = len(messages)
|
|
}
|
|
|
|
// need a team
|
|
team := &model.Team{
|
|
DisplayName: "DisplayName",
|
|
Name: "zz" + model.NewId() + "b",
|
|
Email: model.NewId() + "@nowhere.com",
|
|
Type: model.TEAM_OPEN,
|
|
}
|
|
team = store.Must(ss.Team().Save(team)).(*model.Team)
|
|
|
|
// and two users that are a part of that team
|
|
user1 := &model.User{
|
|
Email: model.NewId(),
|
|
Username: model.NewId(),
|
|
}
|
|
user1 = store.Must(ss.User().Save(user1)).(*model.User)
|
|
store.Must(ss.Team().SaveMember(&model.TeamMember{
|
|
TeamId: team.Id,
|
|
UserId: user1.Id,
|
|
}, -1))
|
|
|
|
user2 := &model.User{
|
|
Email: model.NewId(),
|
|
Username: model.NewId(),
|
|
}
|
|
user2 = store.Must(ss.User().Save(user2)).(*model.User)
|
|
store.Must(ss.Team().SaveMember(&model.TeamMember{
|
|
TeamId: team.Id,
|
|
UserId: user2.Id,
|
|
}, -1))
|
|
|
|
// need a public channel as well as a DM channel between the two users
|
|
channel := &model.Channel{
|
|
TeamId: team.Id,
|
|
Name: model.NewId(),
|
|
DisplayName: "Channel2",
|
|
Type: model.CHANNEL_OPEN,
|
|
}
|
|
channel = store.Must(ss.Channel().Save(channel, -1)).(*model.Channel)
|
|
directMessageChannel := store.Must(ss.Channel().CreateDirectChannel(user1.Id, user2.Id)).(*model.Channel)
|
|
|
|
// user1 posts twice in the public channel
|
|
post1 := &model.Post{
|
|
ChannelId: channel.Id,
|
|
UserId: user1.Id,
|
|
CreateAt: startTime,
|
|
Message: "zz" + model.NewId() + "a",
|
|
}
|
|
post1 = store.Must(ss.Post().Save(post1)).(*model.Post)
|
|
|
|
post2 := &model.Post{
|
|
ChannelId: channel.Id,
|
|
UserId: user1.Id,
|
|
CreateAt: startTime + 10,
|
|
Message: "zz" + model.NewId() + "b",
|
|
}
|
|
post2 = store.Must(ss.Post().Save(post2)).(*model.Post)
|
|
|
|
// user1 also sends a DM to user2
|
|
post3 := &model.Post{
|
|
ChannelId: directMessageChannel.Id,
|
|
UserId: user1.Id,
|
|
CreateAt: startTime + 20,
|
|
Message: "zz" + model.NewId() + "c",
|
|
}
|
|
post3 = store.Must(ss.Post().Save(post3)).(*model.Post)
|
|
|
|
// fetch the message exports for all three posts that user1 sent
|
|
messageExportMap := map[string]model.MessageExport{}
|
|
if r1 := <-ss.Compliance().MessageExport(startTime-10, 10); r1.Err != nil {
|
|
t.Fatal(r1.Err)
|
|
} else {
|
|
messages := r1.Data.([]*model.MessageExport)
|
|
assert.Equal(t, numMessageExports+3, len(messages))
|
|
|
|
for _, v := range messages {
|
|
messageExportMap[*v.PostId] = *v
|
|
}
|
|
}
|
|
|
|
// post1 was made by user1 in channel1 and team1
|
|
assert.Equal(t, post1.Id, *messageExportMap[post1.Id].PostId)
|
|
assert.Equal(t, post1.CreateAt, *messageExportMap[post1.Id].PostCreateAt)
|
|
assert.Equal(t, post1.Message, *messageExportMap[post1.Id].PostMessage)
|
|
assert.Equal(t, channel.Id, *messageExportMap[post1.Id].ChannelId)
|
|
assert.Equal(t, channel.DisplayName, *messageExportMap[post1.Id].ChannelDisplayName)
|
|
assert.Equal(t, user1.Id, *messageExportMap[post1.Id].UserId)
|
|
assert.Equal(t, user1.Email, *messageExportMap[post1.Id].UserEmail)
|
|
assert.Equal(t, user1.Username, *messageExportMap[post1.Id].Username)
|
|
|
|
// post2 was made by user1 in channel1 and team1
|
|
assert.Equal(t, post2.Id, *messageExportMap[post2.Id].PostId)
|
|
assert.Equal(t, post2.CreateAt, *messageExportMap[post2.Id].PostCreateAt)
|
|
assert.Equal(t, post2.Message, *messageExportMap[post2.Id].PostMessage)
|
|
assert.Equal(t, channel.Id, *messageExportMap[post2.Id].ChannelId)
|
|
assert.Equal(t, channel.DisplayName, *messageExportMap[post2.Id].ChannelDisplayName)
|
|
assert.Equal(t, user1.Id, *messageExportMap[post2.Id].UserId)
|
|
assert.Equal(t, user1.Email, *messageExportMap[post2.Id].UserEmail)
|
|
assert.Equal(t, user1.Username, *messageExportMap[post2.Id].Username)
|
|
|
|
// post3 is a DM between user1 and user2
|
|
assert.Equal(t, post3.Id, *messageExportMap[post3.Id].PostId)
|
|
assert.Equal(t, post3.CreateAt, *messageExportMap[post3.Id].PostCreateAt)
|
|
assert.Equal(t, post3.Message, *messageExportMap[post3.Id].PostMessage)
|
|
assert.Equal(t, directMessageChannel.Id, *messageExportMap[post3.Id].ChannelId)
|
|
assert.Equal(t, user1.Id, *messageExportMap[post3.Id].UserId)
|
|
assert.Equal(t, user1.Email, *messageExportMap[post3.Id].UserEmail)
|
|
assert.Equal(t, user1.Username, *messageExportMap[post3.Id].Username)
|
|
}
|