mirror of
https://github.com/mattermost/mattermost.git
synced 2026-04-15 05:57:37 -04:00
8740152df7
* MM-66944 Change PlatformService.IsLeader to always be true when license doesn't support clustering * Address feedback * Add thorough null-checking for license
62 lines
1.9 KiB
Go
62 lines
1.9 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package platform
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
"github.com/mattermost/mattermost/server/v8/channels/testlib"
|
|
)
|
|
|
|
func TestIsLeader(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
|
|
t.Run("no license returns true", func(t *testing.T) {
|
|
th := Setup(t)
|
|
th.Service.SetLicense(nil)
|
|
|
|
assert.True(t, th.Service.IsLeader())
|
|
})
|
|
|
|
t.Run("license with cluster enabled and cluster interface returns cluster leader", func(t *testing.T) {
|
|
fakeCluster := &testlib.FakeClusterInterface{}
|
|
th := SetupWithCluster(t, fakeCluster)
|
|
th.Service.UpdateConfig(func(cfg *model.Config) {
|
|
*cfg.ClusterSettings.Enable = true
|
|
})
|
|
|
|
// This is the only case where IsLeader returns false because it's the only case where we actually call
|
|
// FakeClusterInterface.IsLeader() which returns false
|
|
assert.False(t, th.Service.IsLeader())
|
|
})
|
|
|
|
t.Run("license without cluster feature but cluster enabled returns true", func(t *testing.T) {
|
|
fakeCluster := &testlib.FakeClusterInterface{}
|
|
th := SetupWithCluster(t, fakeCluster)
|
|
|
|
// Set a license without the Cluster feature (like an Entry license)
|
|
license := model.NewTestLicenseWithFalseDefaults("cluster")
|
|
th.Service.SetLicense(license)
|
|
th.Service.UpdateConfig(func(cfg *model.Config) {
|
|
*cfg.ClusterSettings.Enable = true
|
|
})
|
|
|
|
// Even though ClusterSettings.Enable is true and clusterIFace is set, the license doesn't support clustering,
|
|
// so this node should be leader
|
|
assert.True(t, th.Service.IsLeader())
|
|
})
|
|
|
|
t.Run("cluster settings disabled returns true", func(t *testing.T) {
|
|
fakeCluster := &testlib.FakeClusterInterface{}
|
|
th := SetupWithCluster(t, fakeCluster)
|
|
th.Service.UpdateConfig(func(cfg *model.Config) {
|
|
*cfg.ClusterSettings.Enable = false
|
|
})
|
|
|
|
assert.True(t, th.Service.IsLeader())
|
|
})
|
|
}
|