mirror of
https://github.com/mattermost/mattermost.git
synced 2025-02-25 18:55:24 -06:00
148 lines
3.2 KiB
Go
148 lines
3.2 KiB
Go
// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved.
|
|
// See License.txt for license information.
|
|
|
|
package store
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/mattermost/platform/model"
|
|
"github.com/mattermost/platform/utils"
|
|
)
|
|
|
|
var store Store
|
|
|
|
func Setup() {
|
|
if store == nil {
|
|
utils.LoadConfig("config.json")
|
|
utils.InitTranslations()
|
|
store = NewSqlStore()
|
|
|
|
store.MarkSystemRanUnitTests()
|
|
}
|
|
}
|
|
|
|
func TestSqlStore1(t *testing.T) {
|
|
utils.LoadConfig("config.json")
|
|
utils.Cfg.SqlSettings.Trace = true
|
|
|
|
store := NewSqlStore()
|
|
store.Close()
|
|
|
|
utils.LoadConfig("config.json")
|
|
}
|
|
|
|
func TestSqlStore2(t *testing.T) {
|
|
defer func() {
|
|
if r := recover(); r == nil {
|
|
t.Fatal("should have been fatal")
|
|
}
|
|
}()
|
|
|
|
utils.LoadConfig("config.json")
|
|
utils.Cfg.SqlSettings.DriverName = "missing"
|
|
store = NewSqlStore()
|
|
|
|
utils.LoadConfig("config.json")
|
|
}
|
|
|
|
func TestSqlStore3(t *testing.T) {
|
|
defer func() {
|
|
if r := recover(); r == nil {
|
|
t.Fatal("should have been fatal")
|
|
}
|
|
}()
|
|
|
|
utils.LoadConfig("config.json")
|
|
utils.Cfg.SqlSettings.DataSource = "missing"
|
|
store = NewSqlStore()
|
|
|
|
utils.LoadConfig("config.json")
|
|
}
|
|
|
|
func TestEncrypt(t *testing.T) {
|
|
m := make(map[string]string)
|
|
|
|
key := []byte("IPc17oYK9NAj6WfJeCqm5AxIBF6WBNuN") // AES-256
|
|
|
|
originalText1 := model.MapToJson(m)
|
|
cryptoText1, _ := encrypt(key, originalText1)
|
|
text1, _ := decrypt(key, cryptoText1)
|
|
rm1 := model.MapFromJson(strings.NewReader(text1))
|
|
|
|
if len(rm1) != 0 {
|
|
t.Fatal("error in encrypt")
|
|
}
|
|
|
|
m["key"] = "value"
|
|
originalText2 := model.MapToJson(m)
|
|
cryptoText2, _ := encrypt(key, originalText2)
|
|
text2, _ := decrypt(key, cryptoText2)
|
|
rm2 := model.MapFromJson(strings.NewReader(text2))
|
|
|
|
if rm2["key"] != "value" {
|
|
t.Fatal("error in encrypt")
|
|
}
|
|
}
|
|
|
|
func TestAlertDbCmds(t *testing.T) {
|
|
Setup()
|
|
|
|
sqlStore := store.(*SqlStore)
|
|
|
|
if !sqlStore.DoesTableExist("Systems") {
|
|
t.Fatal("Failed table exists")
|
|
}
|
|
|
|
if sqlStore.DoesColumnExist("Systems", "Test") {
|
|
t.Fatal("Column should not exist")
|
|
}
|
|
|
|
if !sqlStore.CreateColumnIfNotExists("Systems", "Test", "VARCHAR(50)", "VARCHAR(50)", "") {
|
|
t.Fatal("Failed to create column")
|
|
}
|
|
|
|
maxLen := sqlStore.GetMaxLengthOfColumnIfExists("Systems", "Test")
|
|
|
|
if maxLen != "50" {
|
|
t.Fatal("Failed to get max length found " + maxLen)
|
|
}
|
|
|
|
if !sqlStore.AlterColumnTypeIfExists("Systems", "Test", "VARCHAR(25)", "VARCHAR(25)") {
|
|
t.Fatal("failed to alter column size")
|
|
}
|
|
|
|
maxLen2 := sqlStore.GetMaxLengthOfColumnIfExists("Systems", "Test")
|
|
|
|
if maxLen2 != "25" {
|
|
t.Fatal("Failed to get max length")
|
|
}
|
|
|
|
if !sqlStore.RenameColumnIfExists("Systems", "Test", "Test1", "VARCHAR(25)") {
|
|
t.Fatal("Failed to rename column")
|
|
}
|
|
|
|
if sqlStore.DoesColumnExist("Systems", "Test") {
|
|
t.Fatal("Column should not exist")
|
|
}
|
|
|
|
if !sqlStore.DoesColumnExist("Systems", "Test1") {
|
|
t.Fatal("Column should exist")
|
|
}
|
|
|
|
sqlStore.CreateIndexIfNotExists("idx_systems_test1", "Systems", "Test1")
|
|
sqlStore.RemoveIndexIfExists("idx_systems_test1", "Systems")
|
|
|
|
sqlStore.CreateFullTextIndexIfNotExists("idx_systems_test1", "Systems", "Test1")
|
|
sqlStore.RemoveIndexIfExists("idx_systems_test1", "Systems")
|
|
|
|
if !sqlStore.RemoveColumnIfExists("Systems", "Test1") {
|
|
t.Fatal("Failed to remove columns")
|
|
}
|
|
|
|
if sqlStore.DoesColumnExist("Systems", "Test1") {
|
|
t.Fatal("Column should not exist")
|
|
}
|
|
}
|