mirror of
https://github.com/mattermost/mattermost.git
synced 2025-02-25 18:55:24 -06:00
e89b26e8f3
* format using `goimports -local github.com/mattermost/mattermost-server/v5 -w` * added goimports lint check to .golangci.yml * format using `goimports -local github.com/mattermost/mattermost-server/v5 -w` for a corner case * make app-layers, *-mocks and store-layers for ci check Co-authored-by: Mahmudul Haque <mahmudulhaque@protonmail.com> Co-authored-by: Mattermod <mattermod@users.noreply.github.com>
103 lines
3.0 KiB
Go
103 lines
3.0 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package app
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/mock"
|
|
|
|
"github.com/mattermost/mattermost-server/v5/einterfaces"
|
|
"github.com/mattermost/mattermost-server/v5/einterfaces/mocks"
|
|
"github.com/mattermost/mattermost-server/v5/model"
|
|
storemocks "github.com/mattermost/mattermost-server/v5/store/storetest/mocks"
|
|
)
|
|
|
|
func TestSAMLSettings(t *testing.T) {
|
|
tt := []struct {
|
|
name string
|
|
setNewInterface bool
|
|
useNewSAMLLibrary bool
|
|
isNil bool
|
|
metadata string
|
|
}{
|
|
{
|
|
name: "No SAML Interfaces, default setting",
|
|
setNewInterface: false,
|
|
useNewSAMLLibrary: false,
|
|
isNil: true,
|
|
},
|
|
{
|
|
name: "No SAML Interfaces, set config true",
|
|
setNewInterface: false,
|
|
useNewSAMLLibrary: true,
|
|
isNil: true,
|
|
},
|
|
{
|
|
name: "Both SAML Interfaces, default setting",
|
|
setNewInterface: true,
|
|
useNewSAMLLibrary: false,
|
|
isNil: false,
|
|
metadata: "samlTwo",
|
|
},
|
|
{
|
|
name: "Both SAML Interfaces, config true",
|
|
setNewInterface: true,
|
|
useNewSAMLLibrary: true,
|
|
isNil: false,
|
|
metadata: "samlTwo",
|
|
},
|
|
}
|
|
|
|
for _, tc := range tt {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
saml2 := &mocks.SamlInterface{}
|
|
saml2.Mock.On("ConfigureSP").Return(nil)
|
|
saml2.Mock.On("GetMetadata").Return("samlTwo", nil)
|
|
if tc.setNewInterface {
|
|
RegisterNewSamlInterface(func(a *App) einterfaces.SamlInterface {
|
|
return saml2
|
|
})
|
|
} else {
|
|
RegisterNewSamlInterface(nil)
|
|
}
|
|
|
|
th := SetupEnterpriseWithStoreMock(t)
|
|
defer th.TearDown()
|
|
|
|
mockStore := th.App.Srv().Store.(*storemocks.Store)
|
|
mockUserStore := storemocks.UserStore{}
|
|
mockUserStore.On("Count", mock.Anything).Return(int64(10), nil)
|
|
mockPostStore := storemocks.PostStore{}
|
|
mockPostStore.On("GetMaxPostSize").Return(65535, nil)
|
|
mockSystemStore := storemocks.SystemStore{}
|
|
mockSystemStore.On("GetByName", "UpgradedFromTE").Return(&model.System{Name: "UpgradedFromTE", Value: "false"}, nil)
|
|
mockSystemStore.On("GetByName", "InstallationDate").Return(&model.System{Name: "InstallationDate", Value: "10"}, nil)
|
|
mockSystemStore.On("GetByName", "FirstServerRunTimestamp").Return(&model.System{Name: "FirstServerRunTimestamp", Value: "10"}, nil)
|
|
|
|
mockStore.On("User").Return(&mockUserStore)
|
|
mockStore.On("Post").Return(&mockPostStore)
|
|
mockStore.On("System").Return(&mockSystemStore)
|
|
|
|
if tc.useNewSAMLLibrary {
|
|
th.App.UpdateConfig(func(cfg *model.Config) {
|
|
*cfg.ExperimentalSettings.UseNewSAMLLibrary = tc.useNewSAMLLibrary
|
|
})
|
|
}
|
|
|
|
th.Server.initEnterprise()
|
|
th.App.initEnterprise()
|
|
if tc.isNil {
|
|
assert.Nil(t, th.App.Srv().Saml)
|
|
} else {
|
|
assert.NotNil(t, th.App.Srv().Saml)
|
|
metadata, err := th.App.Srv().Saml.GetMetadata()
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, tc.metadata, metadata)
|
|
}
|
|
})
|
|
}
|
|
}
|