mirror of
https://github.com/mattermost/mattermost.git
synced 2025-02-25 18:55:24 -06:00
258 lines
6.7 KiB
Go
258 lines
6.7 KiB
Go
// Copyright (c) 2016-present Mattermost, Inc. All Rights Reserved.
|
|
// See License.txt for license information.
|
|
|
|
package storetest
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/mattermost/mattermost-server/model"
|
|
"github.com/mattermost/mattermost-server/store"
|
|
)
|
|
|
|
func TestCommandStore(t *testing.T, ss store.Store) {
|
|
t.Run("Save", func(t *testing.T) { testCommandStoreSave(t, ss) })
|
|
t.Run("Get", func(t *testing.T) { testCommandStoreGet(t, ss) })
|
|
t.Run("GetByTeam", func(t *testing.T) { testCommandStoreGetByTeam(t, ss) })
|
|
t.Run("GetByTrigger", func(t *testing.T) { testCommandStoreGetByTrigger(t, ss) })
|
|
t.Run("Delete", func(t *testing.T) { testCommandStoreDelete(t, ss) })
|
|
t.Run("DeleteByTeam", func(t *testing.T) { testCommandStoreDeleteByTeam(t, ss) })
|
|
t.Run("DeleteByUser", func(t *testing.T) { testCommandStoreDeleteByUser(t, ss) })
|
|
t.Run("Update", func(t *testing.T) { testCommandStoreUpdate(t, ss) })
|
|
t.Run("CommandCount", func(t *testing.T) { testCommandCount(t, ss) })
|
|
}
|
|
|
|
func testCommandStoreSave(t *testing.T, ss store.Store) {
|
|
o1 := model.Command{}
|
|
o1.CreatorId = model.NewId()
|
|
o1.Method = model.COMMAND_METHOD_POST
|
|
o1.TeamId = model.NewId()
|
|
o1.URL = "http://nowhere.com/"
|
|
o1.Trigger = "trigger"
|
|
|
|
if err := (<-ss.Command().Save(&o1)).Err; err != nil {
|
|
t.Fatal("couldn't save item", err)
|
|
}
|
|
|
|
if err := (<-ss.Command().Save(&o1)).Err; err == nil {
|
|
t.Fatal("shouldn't be able to update from save")
|
|
}
|
|
}
|
|
|
|
func testCommandStoreGet(t *testing.T, ss store.Store) {
|
|
o1 := &model.Command{}
|
|
o1.CreatorId = model.NewId()
|
|
o1.Method = model.COMMAND_METHOD_POST
|
|
o1.TeamId = model.NewId()
|
|
o1.URL = "http://nowhere.com/"
|
|
o1.Trigger = "trigger"
|
|
|
|
o1 = (<-ss.Command().Save(o1)).Data.(*model.Command)
|
|
|
|
if r1 := <-ss.Command().Get(o1.Id); r1.Err != nil {
|
|
t.Fatal(r1.Err)
|
|
} else {
|
|
if r1.Data.(*model.Command).CreateAt != o1.CreateAt {
|
|
t.Fatal("invalid returned command")
|
|
}
|
|
}
|
|
|
|
if err := (<-ss.Command().Get("123")).Err; err == nil {
|
|
t.Fatal("Missing id should have failed")
|
|
}
|
|
}
|
|
|
|
func testCommandStoreGetByTeam(t *testing.T, ss store.Store) {
|
|
o1 := &model.Command{}
|
|
o1.CreatorId = model.NewId()
|
|
o1.Method = model.COMMAND_METHOD_POST
|
|
o1.TeamId = model.NewId()
|
|
o1.URL = "http://nowhere.com/"
|
|
o1.Trigger = "trigger"
|
|
|
|
o1 = (<-ss.Command().Save(o1)).Data.(*model.Command)
|
|
|
|
if r1 := <-ss.Command().GetByTeam(o1.TeamId); r1.Err != nil {
|
|
t.Fatal(r1.Err)
|
|
} else {
|
|
if r1.Data.([]*model.Command)[0].CreateAt != o1.CreateAt {
|
|
t.Fatal("invalid returned command")
|
|
}
|
|
}
|
|
|
|
if result := <-ss.Command().GetByTeam("123"); result.Err != nil {
|
|
t.Fatal(result.Err)
|
|
} else {
|
|
if len(result.Data.([]*model.Command)) != 0 {
|
|
t.Fatal("no commands should have returned")
|
|
}
|
|
}
|
|
}
|
|
|
|
func testCommandStoreGetByTrigger(t *testing.T, ss store.Store) {
|
|
o1 := &model.Command{}
|
|
o1.CreatorId = model.NewId()
|
|
o1.Method = model.COMMAND_METHOD_POST
|
|
o1.TeamId = model.NewId()
|
|
o1.URL = "http://nowhere.com/"
|
|
o1.Trigger = "trigger1"
|
|
|
|
o2 := &model.Command{}
|
|
o2.CreatorId = model.NewId()
|
|
o2.Method = model.COMMAND_METHOD_POST
|
|
o2.TeamId = model.NewId()
|
|
o2.URL = "http://nowhere.com/"
|
|
o2.Trigger = "trigger1"
|
|
|
|
o1 = (<-ss.Command().Save(o1)).Data.(*model.Command)
|
|
o2 = (<-ss.Command().Save(o2)).Data.(*model.Command)
|
|
|
|
if r1 := <-ss.Command().GetByTrigger(o1.TeamId, o1.Trigger); r1.Err != nil {
|
|
t.Fatal(r1.Err)
|
|
} else {
|
|
if r1.Data.(*model.Command).Id != o1.Id {
|
|
t.Fatal("invalid returned command")
|
|
}
|
|
}
|
|
|
|
store.Must(ss.Command().Delete(o1.Id, model.GetMillis()))
|
|
|
|
if result := <-ss.Command().GetByTrigger(o1.TeamId, o1.Trigger); result.Err == nil {
|
|
t.Fatal("no commands should have returned")
|
|
}
|
|
}
|
|
|
|
func testCommandStoreDelete(t *testing.T, ss store.Store) {
|
|
o1 := &model.Command{}
|
|
o1.CreatorId = model.NewId()
|
|
o1.Method = model.COMMAND_METHOD_POST
|
|
o1.TeamId = model.NewId()
|
|
o1.URL = "http://nowhere.com/"
|
|
o1.Trigger = "trigger"
|
|
|
|
o1 = (<-ss.Command().Save(o1)).Data.(*model.Command)
|
|
|
|
if r1 := <-ss.Command().Get(o1.Id); r1.Err != nil {
|
|
t.Fatal(r1.Err)
|
|
} else {
|
|
if r1.Data.(*model.Command).CreateAt != o1.CreateAt {
|
|
t.Fatal("invalid returned command")
|
|
}
|
|
}
|
|
|
|
if r2 := <-ss.Command().Delete(o1.Id, model.GetMillis()); r2.Err != nil {
|
|
t.Fatal(r2.Err)
|
|
}
|
|
|
|
if r3 := (<-ss.Command().Get(o1.Id)); r3.Err == nil {
|
|
t.Log(r3.Data)
|
|
t.Fatal("Missing id should have failed")
|
|
}
|
|
}
|
|
|
|
func testCommandStoreDeleteByTeam(t *testing.T, ss store.Store) {
|
|
o1 := &model.Command{}
|
|
o1.CreatorId = model.NewId()
|
|
o1.Method = model.COMMAND_METHOD_POST
|
|
o1.TeamId = model.NewId()
|
|
o1.URL = "http://nowhere.com/"
|
|
o1.Trigger = "trigger"
|
|
|
|
o1 = (<-ss.Command().Save(o1)).Data.(*model.Command)
|
|
|
|
if r1 := <-ss.Command().Get(o1.Id); r1.Err != nil {
|
|
t.Fatal(r1.Err)
|
|
} else {
|
|
if r1.Data.(*model.Command).CreateAt != o1.CreateAt {
|
|
t.Fatal("invalid returned command")
|
|
}
|
|
}
|
|
|
|
if r2 := <-ss.Command().PermanentDeleteByTeam(o1.TeamId); r2.Err != nil {
|
|
t.Fatal(r2.Err)
|
|
}
|
|
|
|
if r3 := (<-ss.Command().Get(o1.Id)); r3.Err == nil {
|
|
t.Log(r3.Data)
|
|
t.Fatal("Missing id should have failed")
|
|
}
|
|
}
|
|
|
|
func testCommandStoreDeleteByUser(t *testing.T, ss store.Store) {
|
|
o1 := &model.Command{}
|
|
o1.CreatorId = model.NewId()
|
|
o1.Method = model.COMMAND_METHOD_POST
|
|
o1.TeamId = model.NewId()
|
|
o1.URL = "http://nowhere.com/"
|
|
o1.Trigger = "trigger"
|
|
|
|
o1 = (<-ss.Command().Save(o1)).Data.(*model.Command)
|
|
|
|
if r1 := <-ss.Command().Get(o1.Id); r1.Err != nil {
|
|
t.Fatal(r1.Err)
|
|
} else {
|
|
if r1.Data.(*model.Command).CreateAt != o1.CreateAt {
|
|
t.Fatal("invalid returned command")
|
|
}
|
|
}
|
|
|
|
if r2 := <-ss.Command().PermanentDeleteByUser(o1.CreatorId); r2.Err != nil {
|
|
t.Fatal(r2.Err)
|
|
}
|
|
|
|
if r3 := (<-ss.Command().Get(o1.Id)); r3.Err == nil {
|
|
t.Log(r3.Data)
|
|
t.Fatal("Missing id should have failed")
|
|
}
|
|
}
|
|
|
|
func testCommandStoreUpdate(t *testing.T, ss store.Store) {
|
|
o1 := &model.Command{}
|
|
o1.CreatorId = model.NewId()
|
|
o1.Method = model.COMMAND_METHOD_POST
|
|
o1.TeamId = model.NewId()
|
|
o1.URL = "http://nowhere.com/"
|
|
o1.Trigger = "trigger"
|
|
|
|
o1 = (<-ss.Command().Save(o1)).Data.(*model.Command)
|
|
|
|
o1.Token = model.NewId()
|
|
|
|
if r2 := <-ss.Command().Update(o1); r2.Err != nil {
|
|
t.Fatal(r2.Err)
|
|
}
|
|
|
|
o1.URL = "junk"
|
|
|
|
if r2 := <-ss.Command().Update(o1); r2.Err == nil {
|
|
t.Fatal("should have failed - bad URL")
|
|
}
|
|
}
|
|
|
|
func testCommandCount(t *testing.T, ss store.Store) {
|
|
o1 := &model.Command{}
|
|
o1.CreatorId = model.NewId()
|
|
o1.Method = model.COMMAND_METHOD_POST
|
|
o1.TeamId = model.NewId()
|
|
o1.URL = "http://nowhere.com/"
|
|
o1.Trigger = "trigger"
|
|
|
|
o1 = (<-ss.Command().Save(o1)).Data.(*model.Command)
|
|
|
|
if r1 := <-ss.Command().AnalyticsCommandCount(""); r1.Err != nil {
|
|
t.Fatal(r1.Err)
|
|
} else {
|
|
if r1.Data.(int64) == 0 {
|
|
t.Fatal("should be at least 1 command")
|
|
}
|
|
}
|
|
|
|
if r2 := <-ss.Command().AnalyticsCommandCount(o1.TeamId); r2.Err != nil {
|
|
t.Fatal(r2.Err)
|
|
} else {
|
|
if r2.Data.(int64) != 1 {
|
|
t.Fatal("should be 1 command")
|
|
}
|
|
}
|
|
}
|