[MM-15275] Migrate "Team.Save" to Sync (#10695)

* [MM-15275] Migrate "Team.Save" to Sync

* review comments

* resolve merge conflicts

* fix ci

* merge error

* merge error

* fix failed test

* address code review

* review comments
Этот коммит содержится в:
Siyuan Liu
2019-04-30 01:19:06 -07:00
коммит произвёл Miguel de la Cruz
родитель eb3923c835
Коммит 503b26a904
14 изменённых файлов: 222 добавлений и 170 удалений

Просмотреть файл

@@ -8,6 +8,7 @@ import (
"testing" "testing"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/mattermost/mattermost-server/model" "github.com/mattermost/mattermost-server/model"
) )
@@ -295,16 +296,15 @@ func TestGetTeamsForScheme(t *testing.T) {
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
} }
result1 := <-th.App.Srv.Store.Team().Save(team1) team1, err := th.App.Srv.Store.Team().Save(team1)
assert.Nil(t, result1.Err) require.Nil(t, err)
team1 = result1.Data.(*model.Team)
l2, r2 := th.SystemAdminClient.GetTeamsForScheme(scheme1.Id, 0, 100) l2, r2 := th.SystemAdminClient.GetTeamsForScheme(scheme1.Id, 0, 100)
CheckNoError(t, r2) CheckNoError(t, r2)
assert.Zero(t, len(l2)) assert.Zero(t, len(l2))
team1.SchemeId = &scheme1.Id team1.SchemeId = &scheme1.Id
team1, err := th.App.Srv.Store.Team().Update(team1) team1, err = th.App.Srv.Store.Team().Update(team1)
assert.Nil(t, err) assert.Nil(t, err)
l3, r3 := th.SystemAdminClient.GetTeamsForScheme(scheme1.Id, 0, 100) l3, r3 := th.SystemAdminClient.GetTeamsForScheme(scheme1.Id, 0, 100)
@@ -318,9 +318,8 @@ func TestGetTeamsForScheme(t *testing.T) {
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
SchemeId: &scheme1.Id, SchemeId: &scheme1.Id,
} }
result3 := <-th.App.Srv.Store.Team().Save(team2) team2, err = th.App.Srv.Store.Team().Save(team2)
assert.Nil(t, result3.Err) require.Nil(t, err)
team2 = result3.Data.(*model.Team)
l4, r4 := th.SystemAdminClient.GetTeamsForScheme(scheme1.Id, 0, 100) l4, r4 := th.SystemAdminClient.GetTeamsForScheme(scheme1.Id, 0, 100)
CheckNoError(t, r4) CheckNoError(t, r4)
@@ -604,15 +603,14 @@ func TestDeleteScheme(t *testing.T) {
assert.Zero(t, role4.DeleteAt) assert.Zero(t, role4.DeleteAt)
// Make sure this scheme is in use by a team. // Make sure this scheme is in use by a team.
res := <-th.App.Srv.Store.Team().Save(&model.Team{ team, err := th.App.Srv.Store.Team().Save(&model.Team{
Name: model.NewId(), Name: model.NewId(),
DisplayName: model.NewId(), DisplayName: model.NewId(),
Email: model.NewId() + "@nowhere.com", Email: model.NewId() + "@nowhere.com",
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
SchemeId: &s1.Id, SchemeId: &s1.Id,
}) })
assert.Nil(t, res.Err) require.Nil(t, err)
team := res.Data.(*model.Team)
// Delete the Scheme. // Delete the Scheme.
_, r3 := th.SystemAdminClient.DeleteScheme(s1.Id) _, r3 := th.SystemAdminClient.DeleteScheme(s1.Id)

Просмотреть файл

@@ -24,11 +24,10 @@ import (
func (a *App) CreateTeam(team *model.Team) (*model.Team, *model.AppError) { func (a *App) CreateTeam(team *model.Team) (*model.Team, *model.AppError) {
team.InviteId = "" team.InviteId = ""
result := <-a.Srv.Store.Team().Save(team) rteam, err := a.Srv.Store.Team().Save(team)
if result.Err != nil { if err != nil {
return nil, result.Err return nil, err
} }
rteam := result.Data.(*model.Team)
if _, err := a.CreateDefaultChannels(rteam.Id); err != nil { if _, err := a.CreateDefaultChannels(rteam.Id); err != nil {
return nil, err return nil, err

Просмотреть файл

@@ -74,13 +74,10 @@ func manualTest(c *web.Context, w http.ResponseWriter, r *http.Request) {
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
} }
if result := <-c.App.Srv.Store.Team().Save(team); result.Err != nil { if createdTeam, err := c.App.Srv.Store.Team().Save(team); err != nil {
c.Err = result.Err c.Err = err
return return
} else { } else {
createdTeam := result.Data.(*model.Team)
channel := &model.Channel{DisplayName: "Town Square", Name: "town-square", Type: model.CHANNEL_OPEN, TeamId: createdTeam.Id} channel := &model.Channel{DisplayName: "Town Square", Name: "town-square", Type: model.CHANNEL_OPEN, TeamId: createdTeam.Id}
if _, err := c.App.CreateChannel(channel, false); err != nil { if _, err := c.App.CreateChannel(channel, false); err != nil {
c.Err = err c.Err = err

Просмотреть файл

@@ -178,30 +178,25 @@ func (s SqlTeamStore) CreateIndexesIfNotExists() {
s.CreateIndexIfNotExists("idx_teammembers_delete_at", "TeamMembers", "DeleteAt") s.CreateIndexIfNotExists("idx_teammembers_delete_at", "TeamMembers", "DeleteAt")
} }
func (s SqlTeamStore) Save(team *model.Team) store.StoreChannel { func (s SqlTeamStore) Save(team *model.Team) (*model.Team, *model.AppError) {
return store.Do(func(result *store.StoreResult) { if len(team.Id) > 0 {
if len(team.Id) > 0 { return nil, model.NewAppError("SqlTeamStore.Save",
result.Err = model.NewAppError("SqlTeamStore.Save", "store.sql_team.save.existing.app_error", nil, "id="+team.Id, http.StatusBadRequest)
"store.sql_team.save.existing.app_error", nil, "id="+team.Id, http.StatusBadRequest) }
return
}
team.PreSave() team.PreSave()
if result.Err = team.IsValid(); result.Err != nil { if err := team.IsValid(); err != nil {
return return nil, err
} }
if err := s.GetMaster().Insert(team); err != nil { if err := s.GetMaster().Insert(team); err != nil {
if IsUniqueConstraintError(err, []string{"Name", "teams_name_key"}) { if IsUniqueConstraintError(err, []string{"Name", "teams_name_key"}) {
result.Err = model.NewAppError("SqlTeamStore.Save", "store.sql_team.save.domain_exists.app_error", nil, "id="+team.Id+", "+err.Error(), http.StatusBadRequest) return nil, model.NewAppError("SqlTeamStore.Save", "store.sql_team.save.domain_exists.app_error", nil, "id="+team.Id+", "+err.Error(), http.StatusBadRequest)
return
}
result.Err = model.NewAppError("SqlTeamStore.Save", "store.sql_team.save.app_error", nil, "id="+team.Id+", "+err.Error(), http.StatusInternalServerError)
return
} }
result.Data = team return nil, model.NewAppError("SqlTeamStore.Save", "store.sql_team.save.app_error", nil, "id="+team.Id+", "+err.Error(), http.StatusInternalServerError)
}) }
return team, nil
} }
func (s SqlTeamStore) Update(team *model.Team) (*model.Team, *model.AppError) { func (s SqlTeamStore) Update(team *model.Team) (*model.Team, *model.AppError) {

Просмотреть файл

@@ -81,7 +81,7 @@ type Store interface {
} }
type TeamStore interface { type TeamStore interface {
Save(team *model.Team) StoreChannel Save(team *model.Team) (*model.Team, *model.AppError)
Update(team *model.Team) (*model.Team, *model.AppError) Update(team *model.Team) (*model.Team, *model.AppError)
UpdateDisplayName(name string, teamId string) StoreChannel UpdateDisplayName(name string, teamId string) StoreChannel
Get(id string) (*model.Team, *model.AppError) Get(id string) (*model.Team, *model.AppError)

Просмотреть файл

@@ -1047,14 +1047,16 @@ func testChannelStoreGetAllChannels(t *testing.T, ss store.Store, s SqlSupplier)
t1.Name = model.NewId() t1.Name = model.NewId()
t1.Email = MakeEmail() t1.Email = MakeEmail()
t1.Type = model.TEAM_OPEN t1.Type = model.TEAM_OPEN
store.Must(ss.Team().Save(&t1)) _, err := ss.Team().Save(&t1)
require.Nil(t, err)
t2 := model.Team{} t2 := model.Team{}
t2.DisplayName = "Name2" t2.DisplayName = "Name2"
t2.Name = model.NewId() t2.Name = model.NewId()
t2.Email = MakeEmail() t2.Email = MakeEmail()
t2.Type = model.TEAM_OPEN t2.Type = model.TEAM_OPEN
store.Must(ss.Team().Save(&t2)) _, err = ss.Team().Save(&t2)
require.Nil(t, err)
c1 := model.Channel{} c1 := model.Channel{}
c1.TeamId = t1.Id c1.TeamId = t1.Id
@@ -1463,7 +1465,8 @@ func testChannelStoreGetMembersForUser(t *testing.T, ss store.Store) {
t1.Name = model.NewId() t1.Name = model.NewId()
t1.Email = MakeEmail() t1.Email = MakeEmail()
t1.Type = model.TEAM_OPEN t1.Type = model.TEAM_OPEN
store.Must(ss.Team().Save(&t1)) _, err := ss.Team().Save(&t1)
require.Nil(t, err)
o1 := model.Channel{} o1 := model.Channel{}
o1.TeamId = t1.Id o1.TeamId = t1.Id
@@ -1506,7 +1509,8 @@ func testChannelStoreGetMembersForUserWithPagination(t *testing.T, ss store.Stor
t1.Name = model.NewId() t1.Name = model.NewId()
t1.Email = MakeEmail() t1.Email = MakeEmail()
t1.Type = model.TEAM_OPEN t1.Type = model.TEAM_OPEN
store.Must(ss.Team().Save(&t1)) _, err := ss.Team().Save(&t1)
require.Nil(t, err)
o1 := model.Channel{} o1 := model.Channel{}
o1.TeamId = t1.Id o1.TeamId = t1.Id
@@ -2218,14 +2222,16 @@ func testChannelStoreSearchAllChannels(t *testing.T, ss store.Store) {
t1.Name = model.NewId() t1.Name = model.NewId()
t1.Email = MakeEmail() t1.Email = MakeEmail()
t1.Type = model.TEAM_OPEN t1.Type = model.TEAM_OPEN
store.Must(ss.Team().Save(&t1)) _, err := ss.Team().Save(&t1)
require.Nil(t, err)
t2 := model.Team{} t2 := model.Team{}
t2.DisplayName = "Name2" t2.DisplayName = "Name2"
t2.Name = model.NewId() t2.Name = model.NewId()
t2.Email = MakeEmail() t2.Email = MakeEmail()
t2.Type = model.TEAM_OPEN t2.Type = model.TEAM_OPEN
store.Must(ss.Team().Save(&t2)) _, err = ss.Team().Save(&t2)
require.Nil(t, err)
o1 := model.Channel{ o1 := model.Channel{
TeamId: t1.Id, TeamId: t1.Id,
@@ -3093,7 +3099,8 @@ func testChannelStoreGetAllChannelsForExportAfter(t *testing.T, ss store.Store)
t1.Name = model.NewId() t1.Name = model.NewId()
t1.Email = MakeEmail() t1.Email = MakeEmail()
t1.Type = model.TEAM_OPEN t1.Type = model.TEAM_OPEN
store.Must(ss.Team().Save(&t1)) _, err := ss.Team().Save(&t1)
require.Nil(t, err)
c1 := model.Channel{} c1 := model.Channel{}
c1.TeamId = t1.Id c1.TeamId = t1.Id
@@ -3124,7 +3131,8 @@ func testChannelStoreGetChannelMembersForExport(t *testing.T, ss store.Store) {
t1.Name = model.NewId() t1.Name = model.NewId()
t1.Email = MakeEmail() t1.Email = MakeEmail()
t1.Type = model.TEAM_OPEN t1.Type = model.TEAM_OPEN
store.Must(ss.Team().Save(&t1)) _, err := ss.Team().Save(&t1)
require.Nil(t, err)
c1 := model.Channel{} c1 := model.Channel{}
c1.TeamId = t1.Id c1.TeamId = t1.Id
@@ -3176,7 +3184,8 @@ func testChannelStoreRemoveAllDeactivatedMembers(t *testing.T, ss store.Store) {
t1.Name = model.NewId() t1.Name = model.NewId()
t1.Email = MakeEmail() t1.Email = MakeEmail()
t1.Type = model.TEAM_OPEN t1.Type = model.TEAM_OPEN
store.Must(ss.Team().Save(&t1)) _, err := ss.Team().Save(&t1)
require.Nil(t, err)
c1 := model.Channel{} c1 := model.Channel{}
c1.TeamId = t1.Id c1.TeamId = t1.Id

Просмотреть файл

@@ -70,7 +70,8 @@ func testComplianceExport(t *testing.T, ss store.Store) {
t1.Name = "zz" + model.NewId() + "b" t1.Name = "zz" + model.NewId() + "b"
t1.Email = MakeEmail() t1.Email = MakeEmail()
t1.Type = model.TEAM_OPEN t1.Type = model.TEAM_OPEN
t1 = store.Must(ss.Team().Save(t1)).(*model.Team) t1, err := ss.Team().Save(t1)
require.Nil(t, err)
u1 := &model.User{} u1 := &model.User{}
u1.Email = MakeEmail() u1.Email = MakeEmail()
@@ -169,7 +170,8 @@ func testComplianceExportDirectMessages(t *testing.T, ss store.Store) {
t1.Name = "zz" + model.NewId() + "b" t1.Name = "zz" + model.NewId() + "b"
t1.Email = MakeEmail() t1.Email = MakeEmail()
t1.Type = model.TEAM_OPEN t1.Type = model.TEAM_OPEN
t1 = store.Must(ss.Team().Save(t1)).(*model.Team) t1, err := ss.Team().Save(t1)
require.Nil(t, err)
u1 := &model.User{} u1 := &model.User{}
u1.Email = MakeEmail() u1.Email = MakeEmail()
@@ -251,7 +253,8 @@ func testMessageExportPublicChannel(t *testing.T, ss store.Store) {
Email: MakeEmail(), Email: MakeEmail(),
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
} }
team = store.Must(ss.Team().Save(team)).(*model.Team) team, err = ss.Team().Save(team)
require.Nil(t, err)
// and two users that are a part of that team // and two users that are a part of that team
user1 := &model.User{ user1 := &model.User{
@@ -345,7 +348,8 @@ func testMessageExportPrivateChannel(t *testing.T, ss store.Store) {
Email: MakeEmail(), Email: MakeEmail(),
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
} }
team = store.Must(ss.Team().Save(team)).(*model.Team) team, err = ss.Team().Save(team)
require.Nil(t, err)
// and two users that are a part of that team // and two users that are a part of that team
user1 := &model.User{ user1 := &model.User{
@@ -441,7 +445,8 @@ func testMessageExportDirectMessageChannel(t *testing.T, ss store.Store) {
Email: MakeEmail(), Email: MakeEmail(),
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
} }
team = store.Must(ss.Team().Save(team)).(*model.Team) team, err = ss.Team().Save(team)
require.Nil(t, err)
// and two users that are a part of that team // and two users that are a part of that team
user1 := &model.User{ user1 := &model.User{
@@ -513,7 +518,8 @@ func testMessageExportGroupMessageChannel(t *testing.T, ss store.Store) {
Email: MakeEmail(), Email: MakeEmail(),
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
} }
team = store.Must(ss.Team().Save(team)).(*model.Team) team, err = ss.Team().Save(team)
require.Nil(t, err)
// and three users that are a part of that team // and three users that are a part of that team
user1 := &model.User{ user1 := &model.User{

Просмотреть файл

@@ -647,9 +647,8 @@ func testCreateGroupSyncable(t *testing.T, ss store.Store) {
Email: "success+" + model.NewId() + "@simulator.amazonses.com", Email: "success+" + model.NewId() + "@simulator.amazonses.com",
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
} }
res5 := <-ss.Team().Save(t1) team, err := ss.Team().Save(t1)
require.Nil(t, res5.Err) require.Nil(t, err)
team := res5.Data.(*model.Team)
// New GroupSyncable, happy path // New GroupSyncable, happy path
gt1 := model.NewGroupTeam(group.Id, team.Id, false) gt1 := model.NewGroupTeam(group.Id, team.Id, false)
@@ -687,9 +686,8 @@ func testGetGroupSyncable(t *testing.T, ss store.Store) {
Email: "success+" + model.NewId() + "@simulator.amazonses.com", Email: "success+" + model.NewId() + "@simulator.amazonses.com",
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
} }
res2 := <-ss.Team().Save(t1) team, err := ss.Team().Save(t1)
require.Nil(t, res2.Err) require.Nil(t, err)
team := res2.Data.(*model.Team)
// Create GroupSyncable // Create GroupSyncable
gt1 := model.NewGroupTeam(group.Id, team.Id, false) gt1 := model.NewGroupTeam(group.Id, team.Id, false)
@@ -739,9 +737,8 @@ func testGetAllGroupSyncablesByGroup(t *testing.T, ss store.Store) {
Email: "success+" + model.NewId() + "@simulator.amazonses.com", Email: "success+" + model.NewId() + "@simulator.amazonses.com",
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
} }
res2 := <-ss.Team().Save(t1) team, err := ss.Team().Save(t1)
require.Nil(t, res2.Err) require.Nil(t, err)
team := res2.Data.(*model.Team)
// create groupteam // create groupteam
res3 := <-ss.Group().CreateGroupSyncable(model.NewGroupTeam(group.Id, team.Id, false)) res3 := <-ss.Group().CreateGroupSyncable(model.NewGroupTeam(group.Id, team.Id, false))
@@ -789,9 +786,8 @@ func testUpdateGroupSyncable(t *testing.T, ss store.Store) {
Email: "success+" + model.NewId() + "@simulator.amazonses.com", Email: "success+" + model.NewId() + "@simulator.amazonses.com",
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
} }
res5 := <-ss.Team().Save(t1) team, err := ss.Team().Save(t1)
require.Nil(t, res5.Err) require.Nil(t, err)
team := res5.Data.(*model.Team)
// New GroupSyncable, happy path // New GroupSyncable, happy path
gt1 := model.NewGroupTeam(group.Id, team.Id, false) gt1 := model.NewGroupTeam(group.Id, team.Id, false)
@@ -861,9 +857,8 @@ func testDeleteGroupSyncable(t *testing.T, ss store.Store) {
Email: "success+" + model.NewId() + "@simulator.amazonses.com", Email: "success+" + model.NewId() + "@simulator.amazonses.com",
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
} }
res2 := <-ss.Team().Save(t1) team, err := ss.Team().Save(t1)
require.Nil(t, res2.Err) require.Nil(t, err)
team := res2.Data.(*model.Team)
// Create GroupSyncable // Create GroupSyncable
gt1 := model.NewGroupTeam(group.Id, team.Id, false) gt1 := model.NewGroupTeam(group.Id, team.Id, false)
@@ -931,9 +926,8 @@ func testPendingAutoAddTeamMembers(t *testing.T, ss store.Store) {
Email: "success+" + model.NewId() + "@simulator.amazonses.com", Email: "success+" + model.NewId() + "@simulator.amazonses.com",
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
} }
res = <-ss.Team().Save(team) team, err := ss.Team().Save(team)
require.Nil(t, res.Err) require.Nil(t, err)
team = res.Data.(*model.Team)
// Create GroupTeam // Create GroupTeam
res = <-ss.Group().CreateGroupSyncable(model.NewGroupTeam(group.Id, team.Id, true)) res = <-ss.Group().CreateGroupSyncable(model.NewGroupTeam(group.Id, team.Id, true))
@@ -1011,7 +1005,7 @@ func testPendingAutoAddTeamMembers(t *testing.T, ss store.Store) {
// No result if Team deleted // No result if Team deleted
team.DeleteAt = model.GetMillis() team.DeleteAt = model.GetMillis()
team, err := ss.Team().Update(team) team, err = ss.Team().Update(team)
require.Nil(t, err) require.Nil(t, err)
res = <-ss.Group().TeamMembersToAdd(0) res = <-ss.Group().TeamMembersToAdd(0)
require.Nil(t, res.Err) require.Nil(t, res.Err)
@@ -1430,9 +1424,8 @@ func pendingMemberRemovalsDataSetup(t *testing.T, ss store.Store) *removalsData
Type: model.TEAM_INVITE, Type: model.TEAM_INVITE,
GroupConstrained: model.NewBool(true), GroupConstrained: model.NewBool(true),
} }
res = <-ss.Team().Save(teamConstrained) teamConstrained, err := ss.Team().Save(teamConstrained)
require.Nil(t, res.Err) require.Nil(t, err)
teamConstrained = res.Data.(*model.Team)
teamUnconstrained := &model.Team{ teamUnconstrained := &model.Team{
DisplayName: "Name", DisplayName: "Name",
@@ -1444,9 +1437,8 @@ func pendingMemberRemovalsDataSetup(t *testing.T, ss store.Store) *removalsData
Email: "success+" + model.NewId() + "@simulator.amazonses.com", Email: "success+" + model.NewId() + "@simulator.amazonses.com",
Type: model.TEAM_INVITE, Type: model.TEAM_INVITE,
} }
res = <-ss.Team().Save(teamUnconstrained) teamUnconstrained, err = ss.Team().Save(teamUnconstrained)
require.Nil(t, res.Err) require.Nil(t, err)
teamUnconstrained = res.Data.(*model.Team)
// create groupteams // create groupteams
res = <-ss.Group().CreateGroupSyncable(model.NewGroupTeam(group.Id, teamConstrained.Id, true)) res = <-ss.Group().CreateGroupSyncable(model.NewGroupTeam(group.Id, teamConstrained.Id, true))
@@ -1648,12 +1640,11 @@ func testGetGroupsByTeam(t *testing.T, ss store.Store) {
Email: "success+" + model.NewId() + "@simulator.amazonses.com", Email: "success+" + model.NewId() + "@simulator.amazonses.com",
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
} }
res := <-ss.Team().Save(team1) team1, err := ss.Team().Save(team1)
require.Nil(t, res.Err) require.Nil(t, err)
team1 = res.Data.(*model.Team)
// Create Groups 1 and 2 // Create Groups 1 and 2
res = <-ss.Group().Create(&model.Group{ res := <-ss.Group().Create(&model.Group{
Name: model.NewId(), Name: model.NewId(),
DisplayName: "group-1", DisplayName: "group-1",
RemoteId: model.NewId(), RemoteId: model.NewId(),
@@ -1693,9 +1684,8 @@ func testGetGroupsByTeam(t *testing.T, ss store.Store) {
Email: "success+" + model.NewId() + "@simulator.amazonses.com", Email: "success+" + model.NewId() + "@simulator.amazonses.com",
Type: model.TEAM_INVITE, Type: model.TEAM_INVITE,
} }
res = <-ss.Team().Save(team2) team2, err = ss.Team().Save(team2)
require.Nil(t, res.Err) require.Nil(t, err)
team2 = res.Data.(*model.Team)
// Create Group3 // Create Group3
res = <-ss.Group().Create(&model.Group{ res = <-ss.Group().Create(&model.Group{

Просмотреть файл

@@ -545,19 +545,28 @@ func (_m *TeamStore) ResetAllTeamSchemes() store.StoreChannel {
} }
// Save provides a mock function with given fields: team // Save provides a mock function with given fields: team
func (_m *TeamStore) Save(team *model.Team) store.StoreChannel { func (_m *TeamStore) Save(team *model.Team) (*model.Team, *model.AppError) {
ret := _m.Called(team) ret := _m.Called(team)
var r0 store.StoreChannel var r0 *model.Team
if rf, ok := ret.Get(0).(func(*model.Team) store.StoreChannel); ok { if rf, ok := ret.Get(0).(func(*model.Team) *model.Team); ok {
r0 = rf(team) r0 = rf(team)
} else { } else {
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).(store.StoreChannel) r0 = ret.Get(0).(*model.Team)
} }
} }
return r0 var r1 *model.AppError
if rf, ok := ret.Get(1).(func(*model.Team) *model.AppError); ok {
r1 = rf(team)
} else {
if ret.Get(1) != nil {
r1 = ret.Get(1).(*model.AppError)
}
}
return r0, r1
} }
// SaveMember provides a mock function with given fields: member, maxUsersPerTeam // SaveMember provides a mock function with given fields: member, maxUsersPerTeam

Просмотреть файл

@@ -1073,7 +1073,8 @@ func testUserCountsWithPostsByDay(t *testing.T, ss store.Store) {
t1.Name = "zz" + model.NewId() + "b" t1.Name = "zz" + model.NewId() + "b"
t1.Email = MakeEmail() t1.Email = MakeEmail()
t1.Type = model.TEAM_OPEN t1.Type = model.TEAM_OPEN
t1 = store.Must(ss.Team().Save(t1)).(*model.Team) t1, err := ss.Team().Save(t1)
require.Nil(t, err)
c1 := &model.Channel{} c1 := &model.Channel{}
c1.TeamId = t1.Id c1.TeamId = t1.Id
@@ -1131,7 +1132,8 @@ func testPostCountsByDay(t *testing.T, ss store.Store) {
t1.Name = "zz" + model.NewId() + "b" t1.Name = "zz" + model.NewId() + "b"
t1.Email = MakeEmail() t1.Email = MakeEmail()
t1.Type = model.TEAM_OPEN t1.Type = model.TEAM_OPEN
t1 = store.Must(ss.Team().Save(t1)).(*model.Team) t1, err := ss.Team().Save(t1)
require.Nil(t, err)
c1 := &model.Channel{} c1 := &model.Channel{}
c1.TeamId = t1.Id c1.TeamId = t1.Id
@@ -1873,7 +1875,8 @@ func testPostStoreGetParentsForExportAfter(t *testing.T, ss store.Store) {
t1.Name = model.NewId() t1.Name = model.NewId()
t1.Email = MakeEmail() t1.Email = MakeEmail()
t1.Type = model.TEAM_OPEN t1.Type = model.TEAM_OPEN
store.Must(ss.Team().Save(&t1)) _, err := ss.Team().Save(&t1)
require.Nil(t, err)
c1 := model.Channel{} c1 := model.Channel{}
c1.TeamId = t1.Id c1.TeamId = t1.Id
@@ -1919,7 +1922,8 @@ func testPostStoreGetRepliesForExport(t *testing.T, ss store.Store) {
t1.Name = model.NewId() t1.Name = model.NewId()
t1.Email = MakeEmail() t1.Email = MakeEmail()
t1.Type = model.TEAM_OPEN t1.Type = model.TEAM_OPEN
store.Must(ss.Team().Save(&t1)) _, err := ss.Team().Save(&t1)
require.Nil(t, err)
c1 := model.Channel{} c1 := model.Channel{}
c1.TeamId = t1.Id c1.TeamId = t1.Id

Просмотреть файл

@@ -4,10 +4,10 @@
package storetest package storetest
import ( import (
"github.com/stretchr/testify/require"
"testing" "testing"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/mattermost/mattermost-server/model" "github.com/mattermost/mattermost-server/model"
"github.com/mattermost/mattermost-server/store" "github.com/mattermost/mattermost-server/store"
@@ -386,9 +386,8 @@ func testSchemeStoreDelete(t *testing.T, ss store.Store) {
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
SchemeId: &d4.Id, SchemeId: &d4.Id,
} }
tres4 := <-ss.Team().Save(t4) t4, err := ss.Team().Save(t4)
assert.Nil(t, tres4.Err) require.Nil(t, err)
t4 = tres4.Data.(*model.Team)
sres4 := <-ss.Scheme().Delete(d4.Id) sres4 := <-ss.Scheme().Delete(d4.Id)
assert.Nil(t, sres4.Err) assert.Nil(t, sres4.Err)

Просмотреть файл

@@ -127,7 +127,7 @@ func testGetAllFromTeam(t *testing.T, ss store.Store) {
team1.Email = MakeEmail() team1.Email = MakeEmail()
team1.Type = model.TEAM_OPEN team1.Type = model.TEAM_OPEN
if err := (<-ss.Team().Save(&team1)).Err; err != nil { if _, err := ss.Team().Save(&team1); err != nil {
t.Fatal("couldn't save team", err) t.Fatal("couldn't save team", err)
} }
@@ -137,7 +137,7 @@ func testGetAllFromTeam(t *testing.T, ss store.Store) {
team2.Email = MakeEmail() team2.Email = MakeEmail()
team2.Type = model.TEAM_OPEN team2.Type = model.TEAM_OPEN
if err := (<-ss.Team().Save(&team2)).Err; err != nil { if _, err := ss.Team().Save(&team2); err != nil {
t.Fatal("couldn't save team", err) t.Fatal("couldn't save team", err)
} }

Просмотреть файл

@@ -60,16 +60,16 @@ func testTeamStoreSave(t *testing.T, ss store.Store) {
o1.Email = MakeEmail() o1.Email = MakeEmail()
o1.Type = model.TEAM_OPEN o1.Type = model.TEAM_OPEN
if err := (<-ss.Team().Save(&o1)).Err; err != nil { if _, err := ss.Team().Save(&o1); err != nil {
t.Fatal("couldn't save item", err) t.Fatal("couldn't save item", err)
} }
if err := (<-ss.Team().Save(&o1)).Err; err == nil { if _, err := ss.Team().Save(&o1); err == nil {
t.Fatal("shouldn't be able to update from save") t.Fatal("shouldn't be able to update from save")
} }
o1.Id = "" o1.Id = ""
if err := (<-ss.Team().Save(&o1)).Err; err == nil { if _, err := ss.Team().Save(&o1); err == nil {
t.Fatal("should be unique domain") t.Fatal("should be unique domain")
} }
} }
@@ -80,7 +80,7 @@ func testTeamStoreUpdate(t *testing.T, ss store.Store) {
o1.Name = "z-z-z" + model.NewId() + "b" o1.Name = "z-z-z" + model.NewId() + "b"
o1.Email = MakeEmail() o1.Email = MakeEmail()
o1.Type = model.TEAM_OPEN o1.Type = model.TEAM_OPEN
if err := (<-ss.Team().Save(&o1)).Err; err != nil { if _, err := ss.Team().Save(&o1); err != nil {
t.Fatal(err) t.Fatal(err)
} }
@@ -107,11 +107,12 @@ func testTeamStoreUpdateDisplayName(t *testing.T, ss store.Store) {
o1.Name = "z-z-z" + model.NewId() + "b" o1.Name = "z-z-z" + model.NewId() + "b"
o1.Email = MakeEmail() o1.Email = MakeEmail()
o1.Type = model.TEAM_OPEN o1.Type = model.TEAM_OPEN
o1 = (<-ss.Team().Save(o1)).Data.(*model.Team) o1, err := ss.Team().Save(o1)
require.Nil(t, err)
newDisplayName := "NewDisplayName" newDisplayName := "NewDisplayName"
if err := (<-ss.Team().UpdateDisplayName(newDisplayName, o1.Id)).Err; err != nil { if err = (<-ss.Team().UpdateDisplayName(newDisplayName, o1.Id)).Err; err != nil {
t.Fatal(err) t.Fatal(err)
} }
@@ -126,7 +127,8 @@ func testTeamStoreGet(t *testing.T, ss store.Store) {
o1.Name = "z-z-z" + model.NewId() + "b" o1.Name = "z-z-z" + model.NewId() + "b"
o1.Email = MakeEmail() o1.Email = MakeEmail()
o1.Type = model.TEAM_OPEN o1.Type = model.TEAM_OPEN
store.Must(ss.Team().Save(&o1)) _, err := ss.Team().Save(&o1)
require.Nil(t, err)
r1, err := ss.Team().Get(o1.Id) r1, err := ss.Team().Get(o1.Id)
require.Nil(t, err) require.Nil(t, err)
@@ -143,7 +145,7 @@ func testTeamStoreGetByName(t *testing.T, ss store.Store) {
o1.Email = MakeEmail() o1.Email = MakeEmail()
o1.Type = model.TEAM_OPEN o1.Type = model.TEAM_OPEN
if err := (<-ss.Team().Save(&o1)).Err; err != nil { if _, err := ss.Team().Save(&o1); err != nil {
t.Fatal(err) t.Fatal(err)
} }
@@ -168,7 +170,7 @@ func testTeamStoreSearchByName(t *testing.T, ss store.Store) {
o1.Email = MakeEmail() o1.Email = MakeEmail()
o1.Type = model.TEAM_OPEN o1.Type = model.TEAM_OPEN
if err := (<-ss.Team().Save(&o1)).Err; err != nil { if _, err := ss.Team().Save(&o1); err != nil {
t.Fatal(err) t.Fatal(err)
} }
@@ -189,7 +191,8 @@ func testTeamStoreSearchAll(t *testing.T, ss store.Store) {
o.Type = model.TEAM_OPEN o.Type = model.TEAM_OPEN
o.AllowOpenInvite = true o.AllowOpenInvite = true
require.Nil(t, (<-ss.Team().Save(&o)).Err) _, err := ss.Team().Save(&o)
require.Nil(t, err)
p := model.Team{} p := model.Team{}
p.DisplayName = "ADisplayName" + model.NewId() p.DisplayName = "ADisplayName" + model.NewId()
@@ -198,7 +201,8 @@ func testTeamStoreSearchAll(t *testing.T, ss store.Store) {
p.Type = model.TEAM_OPEN p.Type = model.TEAM_OPEN
p.AllowOpenInvite = false p.AllowOpenInvite = false
require.Nil(t, (<-ss.Team().Save(&p)).Err) _, err = ss.Team().Save(&p)
require.Nil(t, err)
testCases := []struct { testCases := []struct {
Name string Name string
@@ -259,7 +263,8 @@ func testTeamStoreSearchOpen(t *testing.T, ss store.Store) {
o.Type = model.TEAM_OPEN o.Type = model.TEAM_OPEN
o.AllowOpenInvite = true o.AllowOpenInvite = true
require.Nil(t, (<-ss.Team().Save(&o)).Err) _, err := ss.Team().Save(&o)
require.Nil(t, err)
p := model.Team{} p := model.Team{}
p.DisplayName = "ADisplayName" + model.NewId() p.DisplayName = "ADisplayName" + model.NewId()
@@ -268,7 +273,8 @@ func testTeamStoreSearchOpen(t *testing.T, ss store.Store) {
p.Type = model.TEAM_OPEN p.Type = model.TEAM_OPEN
p.AllowOpenInvite = false p.AllowOpenInvite = false
require.Nil(t, (<-ss.Team().Save(&p)).Err) _, err = ss.Team().Save(&p)
require.Nil(t, err)
testCases := []struct { testCases := []struct {
Name string Name string
@@ -323,7 +329,8 @@ func testTeamStoreSearchPrivate(t *testing.T, ss store.Store) {
o.Type = model.TEAM_OPEN o.Type = model.TEAM_OPEN
o.AllowOpenInvite = true o.AllowOpenInvite = true
require.Nil(t, (<-ss.Team().Save(&o)).Err) _, err := ss.Team().Save(&o)
require.Nil(t, err)
p := model.Team{} p := model.Team{}
p.DisplayName = "ADisplayName" + model.NewId() p.DisplayName = "ADisplayName" + model.NewId()
@@ -332,7 +339,8 @@ func testTeamStoreSearchPrivate(t *testing.T, ss store.Store) {
p.Type = model.TEAM_OPEN p.Type = model.TEAM_OPEN
p.AllowOpenInvite = false p.AllowOpenInvite = false
require.Nil(t, (<-ss.Team().Save(&p)).Err) _, err = ss.Team().Save(&p)
require.Nil(t, err)
testCases := []struct { testCases := []struct {
Name string Name string
@@ -387,9 +395,8 @@ func testTeamStoreGetByInviteId(t *testing.T, ss store.Store) {
o1.Type = model.TEAM_OPEN o1.Type = model.TEAM_OPEN
o1.InviteId = model.NewId() o1.InviteId = model.NewId()
save1 := <-ss.Team().Save(&o1) save1, err := ss.Team().Save(&o1)
if err != nil {
if err := save1.Err; err != nil {
t.Fatal(err) t.Fatal(err)
} }
@@ -399,7 +406,7 @@ func testTeamStoreGetByInviteId(t *testing.T, ss store.Store) {
o2.Email = MakeEmail() o2.Email = MakeEmail()
o2.Type = model.TEAM_OPEN o2.Type = model.TEAM_OPEN
if r1 := <-ss.Team().GetByInviteId(save1.Data.(*model.Team).InviteId); r1.Err != nil { if r1 := <-ss.Team().GetByInviteId(save1.InviteId); r1.Err != nil {
t.Fatal(r1.Err) t.Fatal(r1.Err)
} else { } else {
if r1.Data.(*model.Team).ToJson() != o1.ToJson() { if r1.Data.(*model.Team).ToJson() != o1.ToJson() {
@@ -419,7 +426,8 @@ func testTeamStoreByUserId(t *testing.T, ss store.Store) {
o1.Email = MakeEmail() o1.Email = MakeEmail()
o1.Type = model.TEAM_OPEN o1.Type = model.TEAM_OPEN
o1.InviteId = model.NewId() o1.InviteId = model.NewId()
o1 = store.Must(ss.Team().Save(o1)).(*model.Team) o1, err := ss.Team().Save(o1)
require.Nil(t, err)
m1 := &model.TeamMember{TeamId: o1.Id, UserId: model.NewId()} m1 := &model.TeamMember{TeamId: o1.Id, UserId: model.NewId()}
store.Must(ss.Team().SaveMember(m1, -1)) store.Must(ss.Team().SaveMember(m1, -1))
@@ -446,14 +454,16 @@ func testGetAllTeamListing(t *testing.T, ss store.Store) {
o1.Email = MakeEmail() o1.Email = MakeEmail()
o1.Type = model.TEAM_OPEN o1.Type = model.TEAM_OPEN
o1.AllowOpenInvite = true o1.AllowOpenInvite = true
store.Must(ss.Team().Save(&o1)) _, err := ss.Team().Save(&o1)
require.Nil(t, err)
o2 := model.Team{} o2 := model.Team{}
o2.DisplayName = "DisplayName" o2.DisplayName = "DisplayName"
o2.Name = "zz" + model.NewId() + "b" o2.Name = "zz" + model.NewId() + "b"
o2.Email = MakeEmail() o2.Email = MakeEmail()
o2.Type = model.TEAM_OPEN o2.Type = model.TEAM_OPEN
store.Must(ss.Team().Save(&o2)) _, err = ss.Team().Save(&o2)
require.Nil(t, err)
o3 := model.Team{} o3 := model.Team{}
o3.DisplayName = "DisplayName" o3.DisplayName = "DisplayName"
@@ -461,14 +471,16 @@ func testGetAllTeamListing(t *testing.T, ss store.Store) {
o3.Email = MakeEmail() o3.Email = MakeEmail()
o3.Type = model.TEAM_INVITE o3.Type = model.TEAM_INVITE
o3.AllowOpenInvite = true o3.AllowOpenInvite = true
store.Must(ss.Team().Save(&o3)) _, err = ss.Team().Save(&o3)
require.Nil(t, err)
o4 := model.Team{} o4 := model.Team{}
o4.DisplayName = "DisplayName" o4.DisplayName = "DisplayName"
o4.Name = "zz" + model.NewId() + "b" o4.Name = "zz" + model.NewId() + "b"
o4.Email = MakeEmail() o4.Email = MakeEmail()
o4.Type = model.TEAM_INVITE o4.Type = model.TEAM_INVITE
store.Must(ss.Team().Save(&o4)) _, err = ss.Team().Save(&o4)
require.Nil(t, err)
if r1 := <-ss.Team().GetAllTeamListing(); r1.Err != nil { if r1 := <-ss.Team().GetAllTeamListing(); r1.Err != nil {
t.Fatal(r1.Err) t.Fatal(r1.Err)
@@ -494,7 +506,8 @@ func testGetAllTeamPageListing(t *testing.T, ss store.Store) {
o1.Email = MakeEmail() o1.Email = MakeEmail()
o1.Type = model.TEAM_OPEN o1.Type = model.TEAM_OPEN
o1.AllowOpenInvite = true o1.AllowOpenInvite = true
store.Must(ss.Team().Save(&o1)) _, err := ss.Team().Save(&o1)
require.Nil(t, err)
o2 := model.Team{} o2 := model.Team{}
o2.DisplayName = "DisplayName" o2.DisplayName = "DisplayName"
@@ -502,7 +515,8 @@ func testGetAllTeamPageListing(t *testing.T, ss store.Store) {
o2.Email = MakeEmail() o2.Email = MakeEmail()
o2.Type = model.TEAM_OPEN o2.Type = model.TEAM_OPEN
o2.AllowOpenInvite = false o2.AllowOpenInvite = false
store.Must(ss.Team().Save(&o2)) _, err = ss.Team().Save(&o2)
require.Nil(t, err)
o3 := model.Team{} o3 := model.Team{}
o3.DisplayName = "DisplayName" o3.DisplayName = "DisplayName"
@@ -510,7 +524,8 @@ func testGetAllTeamPageListing(t *testing.T, ss store.Store) {
o3.Email = MakeEmail() o3.Email = MakeEmail()
o3.Type = model.TEAM_INVITE o3.Type = model.TEAM_INVITE
o3.AllowOpenInvite = true o3.AllowOpenInvite = true
store.Must(ss.Team().Save(&o3)) _, err = ss.Team().Save(&o3)
require.Nil(t, err)
o4 := model.Team{} o4 := model.Team{}
o4.DisplayName = "DisplayName" o4.DisplayName = "DisplayName"
@@ -518,7 +533,8 @@ func testGetAllTeamPageListing(t *testing.T, ss store.Store) {
o4.Email = MakeEmail() o4.Email = MakeEmail()
o4.Type = model.TEAM_INVITE o4.Type = model.TEAM_INVITE
o4.AllowOpenInvite = false o4.AllowOpenInvite = false
store.Must(ss.Team().Save(&o4)) _, err = ss.Team().Save(&o4)
require.Nil(t, err)
if r1 := <-ss.Team().GetAllTeamPageListing(0, 10); r1.Err != nil { if r1 := <-ss.Team().GetAllTeamPageListing(0, 10); r1.Err != nil {
t.Fatal(r1.Err) t.Fatal(r1.Err)
@@ -542,7 +558,8 @@ func testGetAllTeamPageListing(t *testing.T, ss store.Store) {
o5.Email = MakeEmail() o5.Email = MakeEmail()
o5.Type = model.TEAM_OPEN o5.Type = model.TEAM_OPEN
o5.AllowOpenInvite = true o5.AllowOpenInvite = true
store.Must(ss.Team().Save(&o5)) _, err = ss.Team().Save(&o5)
require.Nil(t, err)
if r1 := <-ss.Team().GetAllTeamPageListing(0, 4); r1.Err != nil { if r1 := <-ss.Team().GetAllTeamPageListing(0, 4); r1.Err != nil {
t.Fatal(r1.Err) t.Fatal(r1.Err)
@@ -584,14 +601,16 @@ func testGetAllPrivateTeamListing(t *testing.T, ss store.Store) {
o1.Email = MakeEmail() o1.Email = MakeEmail()
o1.Type = model.TEAM_OPEN o1.Type = model.TEAM_OPEN
o1.AllowOpenInvite = true o1.AllowOpenInvite = true
store.Must(ss.Team().Save(&o1)) _, err := ss.Team().Save(&o1)
require.Nil(t, err)
o2 := model.Team{} o2 := model.Team{}
o2.DisplayName = "DisplayName" o2.DisplayName = "DisplayName"
o2.Name = "zz" + model.NewId() + "b" o2.Name = "zz" + model.NewId() + "b"
o2.Email = MakeEmail() o2.Email = MakeEmail()
o2.Type = model.TEAM_OPEN o2.Type = model.TEAM_OPEN
store.Must(ss.Team().Save(&o2)) _, err = ss.Team().Save(&o2)
require.Nil(t, err)
o3 := model.Team{} o3 := model.Team{}
o3.DisplayName = "DisplayName" o3.DisplayName = "DisplayName"
@@ -599,14 +618,16 @@ func testGetAllPrivateTeamListing(t *testing.T, ss store.Store) {
o3.Email = MakeEmail() o3.Email = MakeEmail()
o3.Type = model.TEAM_INVITE o3.Type = model.TEAM_INVITE
o3.AllowOpenInvite = true o3.AllowOpenInvite = true
store.Must(ss.Team().Save(&o3)) _, err = ss.Team().Save(&o3)
require.Nil(t, err)
o4 := model.Team{} o4 := model.Team{}
o4.DisplayName = "DisplayName" o4.DisplayName = "DisplayName"
o4.Name = "zz" + model.NewId() + "b" o4.Name = "zz" + model.NewId() + "b"
o4.Email = MakeEmail() o4.Email = MakeEmail()
o4.Type = model.TEAM_INVITE o4.Type = model.TEAM_INVITE
store.Must(ss.Team().Save(&o4)) _, err = ss.Team().Save(&o4)
require.Nil(t, err)
if r1 := <-ss.Team().GetAllPrivateTeamListing(); r1.Err != nil { if r1 := <-ss.Team().GetAllPrivateTeamListing(); r1.Err != nil {
t.Fatal(r1.Err) t.Fatal(r1.Err)
@@ -632,7 +653,8 @@ func testGetAllPrivateTeamPageListing(t *testing.T, ss store.Store) {
o1.Email = MakeEmail() o1.Email = MakeEmail()
o1.Type = model.TEAM_OPEN o1.Type = model.TEAM_OPEN
o1.AllowOpenInvite = true o1.AllowOpenInvite = true
store.Must(ss.Team().Save(&o1)) _, err := ss.Team().Save(&o1)
require.Nil(t, err)
o2 := model.Team{} o2 := model.Team{}
o2.DisplayName = "DisplayName" o2.DisplayName = "DisplayName"
@@ -640,7 +662,8 @@ func testGetAllPrivateTeamPageListing(t *testing.T, ss store.Store) {
o2.Email = MakeEmail() o2.Email = MakeEmail()
o2.Type = model.TEAM_OPEN o2.Type = model.TEAM_OPEN
o2.AllowOpenInvite = false o2.AllowOpenInvite = false
store.Must(ss.Team().Save(&o2)) _, err = ss.Team().Save(&o2)
require.Nil(t, err)
o3 := model.Team{} o3 := model.Team{}
o3.DisplayName = "DisplayName" o3.DisplayName = "DisplayName"
@@ -648,7 +671,8 @@ func testGetAllPrivateTeamPageListing(t *testing.T, ss store.Store) {
o3.Email = MakeEmail() o3.Email = MakeEmail()
o3.Type = model.TEAM_INVITE o3.Type = model.TEAM_INVITE
o3.AllowOpenInvite = true o3.AllowOpenInvite = true
store.Must(ss.Team().Save(&o3)) _, err = ss.Team().Save(&o3)
require.Nil(t, err)
o4 := model.Team{} o4 := model.Team{}
o4.DisplayName = "DisplayName" o4.DisplayName = "DisplayName"
@@ -656,7 +680,8 @@ func testGetAllPrivateTeamPageListing(t *testing.T, ss store.Store) {
o4.Email = MakeEmail() o4.Email = MakeEmail()
o4.Type = model.TEAM_INVITE o4.Type = model.TEAM_INVITE
o4.AllowOpenInvite = false o4.AllowOpenInvite = false
store.Must(ss.Team().Save(&o4)) _, err = ss.Team().Save(&o4)
require.Nil(t, err)
if r1 := <-ss.Team().GetAllPrivateTeamPageListing(0, 10); r1.Err != nil { if r1 := <-ss.Team().GetAllPrivateTeamPageListing(0, 10); r1.Err != nil {
t.Fatal(r1.Err) t.Fatal(r1.Err)
@@ -680,7 +705,8 @@ func testGetAllPrivateTeamPageListing(t *testing.T, ss store.Store) {
o5.Email = MakeEmail() o5.Email = MakeEmail()
o5.Type = model.TEAM_OPEN o5.Type = model.TEAM_OPEN
o5.AllowOpenInvite = true o5.AllowOpenInvite = true
store.Must(ss.Team().Save(&o5)) _, err = ss.Team().Save(&o5)
require.Nil(t, err)
if r1 := <-ss.Team().GetAllPrivateTeamPageListing(0, 4); r1.Err != nil { if r1 := <-ss.Team().GetAllPrivateTeamPageListing(0, 4); r1.Err != nil {
t.Fatal(r1.Err) t.Fatal(r1.Err)
@@ -722,14 +748,16 @@ func testDelete(t *testing.T, ss store.Store) {
o1.Email = MakeEmail() o1.Email = MakeEmail()
o1.Type = model.TEAM_OPEN o1.Type = model.TEAM_OPEN
o1.AllowOpenInvite = true o1.AllowOpenInvite = true
store.Must(ss.Team().Save(&o1)) _, err := ss.Team().Save(&o1)
require.Nil(t, err)
o2 := model.Team{} o2 := model.Team{}
o2.DisplayName = "DisplayName" o2.DisplayName = "DisplayName"
o2.Name = "zz" + model.NewId() + "b" o2.Name = "zz" + model.NewId() + "b"
o2.Email = MakeEmail() o2.Email = MakeEmail()
o2.Type = model.TEAM_OPEN o2.Type = model.TEAM_OPEN
store.Must(ss.Team().Save(&o2)) _, err = ss.Team().Save(&o2)
require.Nil(t, err)
if r1 := <-ss.Team().PermanentDelete(o1.Id); r1.Err != nil { if r1 := <-ss.Team().PermanentDelete(o1.Id); r1.Err != nil {
t.Fatal(r1.Err) t.Fatal(r1.Err)
@@ -743,7 +771,8 @@ func testTeamCount(t *testing.T, ss store.Store) {
o1.Email = MakeEmail() o1.Email = MakeEmail()
o1.Type = model.TEAM_OPEN o1.Type = model.TEAM_OPEN
o1.AllowOpenInvite = true o1.AllowOpenInvite = true
store.Must(ss.Team().Save(&o1)) _, err := ss.Team().Save(&o1)
require.Nil(t, err)
if r1 := <-ss.Team().AnalyticsTeamCount(); r1.Err != nil { if r1 := <-ss.Team().AnalyticsTeamCount(); r1.Err != nil {
t.Fatal(r1.Err) t.Fatal(r1.Err)
@@ -907,11 +936,12 @@ func testTeamMembersWithPagination(t *testing.T, ss store.Store) {
func testSaveTeamMemberMaxMembers(t *testing.T, ss store.Store) { func testSaveTeamMemberMaxMembers(t *testing.T, ss store.Store) {
maxUsersPerTeam := 5 maxUsersPerTeam := 5
team := store.Must(ss.Team().Save(&model.Team{ team, err := ss.Team().Save(&model.Team{
DisplayName: "DisplayName", DisplayName: "DisplayName",
Name: "z-z-z" + model.NewId() + "b", Name: "z-z-z" + model.NewId() + "b",
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
})).(*model.Team) })
require.Nil(t, err)
defer func() { defer func() {
<-ss.Team().PermanentDelete(team.Id) <-ss.Team().PermanentDelete(team.Id)
}() }()
@@ -1049,12 +1079,13 @@ func testGetTeamMember(t *testing.T, ss store.Store) {
s2 = (<-ss.Scheme().Save(s2)).Data.(*model.Scheme) s2 = (<-ss.Scheme().Save(s2)).Data.(*model.Scheme)
t.Log(s2) t.Log(s2)
t2 := store.Must(ss.Team().Save(&model.Team{ t2, err := ss.Team().Save(&model.Team{
DisplayName: "DisplayName", DisplayName: "DisplayName",
Name: "z-z-z" + model.NewId() + "b", Name: "z-z-z" + model.NewId() + "b",
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
SchemeId: &s2.Id, SchemeId: &s2.Id,
})).(*model.Team) })
require.Nil(t, err)
defer func() { defer func() {
<-ss.Team().PermanentDelete(t2.Id) <-ss.Team().PermanentDelete(t2.Id)
@@ -1270,11 +1301,12 @@ func testUpdateLastTeamIconUpdate(t *testing.T, ss store.Store) {
o1.Email = MakeEmail() o1.Email = MakeEmail()
o1.Type = model.TEAM_OPEN o1.Type = model.TEAM_OPEN
o1.LastTeamIconUpdate = lastTeamIconUpdateInitial o1.LastTeamIconUpdate = lastTeamIconUpdateInitial
o1 = (<-ss.Team().Save(o1)).Data.(*model.Team) o1, err := ss.Team().Save(o1)
require.Nil(t, err)
curTime := model.GetMillis() curTime := model.GetMillis()
if err := (<-ss.Team().UpdateLastTeamIconUpdate(o1.Id, curTime)).Err; err != nil { if err = (<-ss.Team().UpdateLastTeamIconUpdate(o1.Id, curTime)).Err; err != nil {
t.Fatal(err) t.Fatal(err)
} }
@@ -1329,9 +1361,14 @@ func testGetTeamsByScheme(t *testing.T, ss store.Store) {
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
} }
_ = (<-ss.Team().Save(t1)).Data.(*model.Team) _, err := ss.Team().Save(t1)
_ = (<-ss.Team().Save(t2)).Data.(*model.Team) require.Nil(t, err)
_ = (<-ss.Team().Save(t3)).Data.(*model.Team)
_, err = ss.Team().Save(t2)
require.Nil(t, err)
_, err = ss.Team().Save(t3)
require.Nil(t, err)
// Get the teams by a valid Scheme ID. // Get the teams by a valid Scheme ID.
res1 := <-ss.Team().GetTeamsByScheme(s1.Id, 0, 100) res1 := <-ss.Team().GetTeamsByScheme(s1.Id, 0, 100)
@@ -1362,7 +1399,8 @@ func testTeamStoreMigrateTeamMembers(t *testing.T, ss store.Store) {
InviteId: model.NewId(), InviteId: model.NewId(),
SchemeId: &s1, SchemeId: &s1,
} }
t1 = store.Must(ss.Team().Save(t1)).(*model.Team) t1, err := ss.Team().Save(t1)
require.Nil(t, err)
tm1 := &model.TeamMember{ tm1 := &model.TeamMember{
TeamId: t1.Id, TeamId: t1.Id,
@@ -1446,8 +1484,10 @@ func testResetAllTeamSchemes(t *testing.T, ss store.Store) {
SchemeId: &s1.Id, SchemeId: &s1.Id,
} }
t1 = (<-ss.Team().Save(t1)).Data.(*model.Team) t1, err := ss.Team().Save(t1)
t2 = (<-ss.Team().Save(t2)).Data.(*model.Team) require.Nil(t, err)
t2, err = ss.Team().Save(t2)
require.Nil(t, err)
assert.Equal(t, s1.Id, *t1.SchemeId) assert.Equal(t, s1.Id, *t1.SchemeId)
assert.Equal(t, s1.Id, *t2.SchemeId) assert.Equal(t, s1.Id, *t2.SchemeId)
@@ -1455,7 +1495,7 @@ func testResetAllTeamSchemes(t *testing.T, ss store.Store) {
res := <-ss.Team().ResetAllTeamSchemes() res := <-ss.Team().ResetAllTeamSchemes()
assert.Nil(t, res.Err) assert.Nil(t, res.Err)
t1, err := ss.Team().Get(t1.Id) t1, err = ss.Team().Get(t1.Id)
require.Nil(t, err) require.Nil(t, err)
t2, err = ss.Team().Get(t2.Id) t2, err = ss.Team().Get(t2.Id)
@@ -1530,7 +1570,8 @@ func testTeamStoreAnalyticsGetTeamCountForScheme(t *testing.T, ss store.Store) {
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
SchemeId: &s1.Id, SchemeId: &s1.Id,
} }
_ = (<-ss.Team().Save(t1)).Data.(*model.Team) _, err := ss.Team().Save(t1)
require.Nil(t, err)
count2 := (<-ss.Team().AnalyticsGetTeamCountForScheme(s1.Id)).Data.(int64) count2 := (<-ss.Team().AnalyticsGetTeamCountForScheme(s1.Id)).Data.(int64)
assert.Equal(t, int64(1), count2) assert.Equal(t, int64(1), count2)
@@ -1542,7 +1583,8 @@ func testTeamStoreAnalyticsGetTeamCountForScheme(t *testing.T, ss store.Store) {
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
SchemeId: &s1.Id, SchemeId: &s1.Id,
} }
_ = (<-ss.Team().Save(t2)).Data.(*model.Team) _, err = ss.Team().Save(t2)
require.Nil(t, err)
count3 := (<-ss.Team().AnalyticsGetTeamCountForScheme(s1.Id)).Data.(int64) count3 := (<-ss.Team().AnalyticsGetTeamCountForScheme(s1.Id)).Data.(int64)
assert.Equal(t, int64(2), count3) assert.Equal(t, int64(2), count3)
@@ -1553,7 +1595,8 @@ func testTeamStoreAnalyticsGetTeamCountForScheme(t *testing.T, ss store.Store) {
Email: MakeEmail(), Email: MakeEmail(),
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
} }
_ = (<-ss.Team().Save(t3)).Data.(*model.Team) _, err = ss.Team().Save(t3)
require.Nil(t, err)
count4 := (<-ss.Team().AnalyticsGetTeamCountForScheme(s1.Id)).Data.(int64) count4 := (<-ss.Team().AnalyticsGetTeamCountForScheme(s1.Id)).Data.(int64)
assert.Equal(t, int64(2), count4) assert.Equal(t, int64(2), count4)
@@ -1566,7 +1609,8 @@ func testTeamStoreAnalyticsGetTeamCountForScheme(t *testing.T, ss store.Store) {
SchemeId: &s1.Id, SchemeId: &s1.Id,
DeleteAt: model.GetMillis(), DeleteAt: model.GetMillis(),
} }
_ = (<-ss.Team().Save(t4)).Data.(*model.Team) _, err = ss.Team().Save(t4)
require.Nil(t, err)
count5 := (<-ss.Team().AnalyticsGetTeamCountForScheme(s1.Id)).Data.(int64) count5 := (<-ss.Team().AnalyticsGetTeamCountForScheme(s1.Id)).Data.(int64)
assert.Equal(t, int64(2), count5) assert.Equal(t, int64(2), count5)
@@ -1578,7 +1622,8 @@ func testTeamStoreGetAllForExportAfter(t *testing.T, ss store.Store) {
t1.Name = model.NewId() t1.Name = model.NewId()
t1.Email = MakeEmail() t1.Email = MakeEmail()
t1.Type = model.TEAM_OPEN t1.Type = model.TEAM_OPEN
store.Must(ss.Team().Save(&t1)) _, err := ss.Team().Save(&t1)
require.Nil(t, err)
r1 := <-ss.Team().GetAllForExportAfter(10000, strings.Repeat("0", 26)) r1 := <-ss.Team().GetAllForExportAfter(10000, strings.Repeat("0", 26))
assert.Nil(t, r1.Err) assert.Nil(t, r1.Err)
@@ -1602,7 +1647,8 @@ func testTeamStoreGetTeamMembersForExport(t *testing.T, ss store.Store) {
t1.Name = model.NewId() t1.Name = model.NewId()
t1.Email = MakeEmail() t1.Email = MakeEmail()
t1.Type = model.TEAM_OPEN t1.Type = model.TEAM_OPEN
store.Must(ss.Team().Save(&t1)) _, err := ss.Team().Save(&t1)
require.Nil(t, err)
u1 := model.User{} u1 := model.User{}
u1.Email = MakeEmail() u1.Email = MakeEmail()

Просмотреть файл

@@ -3325,11 +3325,12 @@ func testUserStoreGetAllAfter(t *testing.T, ss store.Store) {
func testUserStoreGetUsersBatchForIndexing(t *testing.T, ss store.Store) { func testUserStoreGetUsersBatchForIndexing(t *testing.T, ss store.Store) {
// Set up all the objects needed // Set up all the objects needed
t1 := store.Must(ss.Team().Save(&model.Team{ t1, err := ss.Team().Save(&model.Team{
DisplayName: "Team1", DisplayName: "Team1",
Name: model.NewId(), Name: model.NewId(),
Type: model.TEAM_OPEN, Type: model.TEAM_OPEN,
})).(*model.Team) })
require.Nil(t, err)
cPub1 := store.Must(ss.Channel().Save(&model.Channel{ cPub1 := store.Must(ss.Channel().Save(&model.Channel{
Name: model.NewId(), Name: model.NewId(),
Type: model.CHANNEL_OPEN, Type: model.CHANNEL_OPEN,
@@ -3445,21 +3446,20 @@ func testUserStoreGetUsersBatchForIndexing(t *testing.T, ss store.Store) {
func testUserStoreGetTeamGroupUsers(t *testing.T, ss store.Store) { func testUserStoreGetTeamGroupUsers(t *testing.T, ss store.Store) {
// create team // create team
id := model.NewId() id := model.NewId()
res := <-ss.Team().Save(&model.Team{ team, err := ss.Team().Save(&model.Team{
DisplayName: "dn_" + id, DisplayName: "dn_" + id,
Name: "n-" + id, Name: "n-" + id,
Email: id + "@test.com", Email: id + "@test.com",
Type: model.TEAM_INVITE, Type: model.TEAM_INVITE,
}) })
require.Nil(t, res.Err) require.Nil(t, err)
team := res.Data.(*model.Team)
require.NotNil(t, team) require.NotNil(t, team)
// create users // create users
var testUsers []*model.User var testUsers []*model.User
for i := 0; i < 3; i++ { for i := 0; i < 3; i++ {
id = model.NewId() id = model.NewId()
res = <-ss.User().Save(&model.User{ res := <-ss.User().Save(&model.User{
Email: id + "@test.com", Email: id + "@test.com",
Username: "un_" + id, Username: "un_" + id,
Nickname: "nn_" + id, Nickname: "nn_" + id,
@@ -3477,7 +3477,7 @@ func testUserStoreGetTeamGroupUsers(t *testing.T, ss store.Store) {
userNoGroup := testUsers[2] userNoGroup := testUsers[2]
// add non-group-member to the team (to prove that the query isn't just returning all members) // add non-group-member to the team (to prove that the query isn't just returning all members)
res = <-ss.Team().SaveMember(&model.TeamMember{ res := <-ss.Team().SaveMember(&model.TeamMember{
TeamId: team.Id, TeamId: team.Id,
UserId: userNoGroup.Id, UserId: userNoGroup.Id,
}, 999) }, 999)
@@ -3530,7 +3530,7 @@ func testUserStoreGetTeamGroupUsers(t *testing.T, ss store.Store) {
// update team to be group-constrained // update team to be group-constrained
team.GroupConstrained = model.NewBool(true) team.GroupConstrained = model.NewBool(true)
team, err := ss.Team().Update(team) team, err = ss.Team().Update(team)
require.Nil(t, err) require.Nil(t, err)
// still returns user (being group-constrained has no effect) // still returns user (being group-constrained has no effect)