mirror of
https://github.com/mattermost/mattermost.git
synced 2025-02-25 18:55:24 -06:00
9ef8c1e8b1
* api4: break out license and config from system * app: move some config functions from admin.go to config.go * add ExperimentalSettings.RestrictSystemAdmin * forbid various actions to restricted system admin * update default.json * fix function names in errors
106 lines
2.4 KiB
Go
106 lines
2.4 KiB
Go
package api4
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/mattermost/mattermost-server/model"
|
|
)
|
|
|
|
func TestGetOldClientLicense(t *testing.T) {
|
|
th := Setup().InitBasic()
|
|
defer th.TearDown()
|
|
Client := th.Client
|
|
|
|
license, resp := Client.GetOldClientLicense("")
|
|
CheckNoError(t, resp)
|
|
|
|
if len(license["IsLicensed"]) == 0 {
|
|
t.Fatal("license not returned correctly")
|
|
}
|
|
|
|
Client.Logout()
|
|
|
|
_, resp = Client.GetOldClientLicense("")
|
|
CheckNoError(t, resp)
|
|
|
|
if _, err := Client.DoApiGet("/license/client", ""); err == nil || err.StatusCode != http.StatusNotImplemented {
|
|
t.Fatal("should have errored with 501")
|
|
}
|
|
|
|
if _, err := Client.DoApiGet("/license/client?format=junk", ""); err == nil || err.StatusCode != http.StatusBadRequest {
|
|
t.Fatal("should have errored with 400")
|
|
}
|
|
|
|
license, resp = th.SystemAdminClient.GetOldClientLicense("")
|
|
CheckNoError(t, resp)
|
|
|
|
if len(license["IsLicensed"]) == 0 {
|
|
t.Fatal("license not returned correctly")
|
|
}
|
|
}
|
|
|
|
func TestUploadLicenseFile(t *testing.T) {
|
|
th := Setup().InitBasic()
|
|
defer th.TearDown()
|
|
Client := th.Client
|
|
|
|
t.Run("as system user", func(t *testing.T) {
|
|
ok, resp := Client.UploadLicenseFile([]byte{})
|
|
CheckForbiddenStatus(t, resp)
|
|
if ok {
|
|
t.Fatal("should fail")
|
|
}
|
|
})
|
|
|
|
t.Run("as system admin user", func(t *testing.T) {
|
|
ok, resp := th.SystemAdminClient.UploadLicenseFile([]byte{})
|
|
CheckBadRequestStatus(t, resp)
|
|
if ok {
|
|
t.Fatal("should fail")
|
|
}
|
|
})
|
|
|
|
t.Run("as restricted system admin user", func(t *testing.T) {
|
|
th.App.UpdateConfig(func(cfg *model.Config) { *cfg.ExperimentalSettings.RestrictSystemAdmin = true })
|
|
|
|
ok, resp := th.SystemAdminClient.UploadLicenseFile([]byte{})
|
|
CheckForbiddenStatus(t, resp)
|
|
if ok {
|
|
t.Fatal("should fail")
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestRemoveLicenseFile(t *testing.T) {
|
|
th := Setup().InitBasic()
|
|
defer th.TearDown()
|
|
Client := th.Client
|
|
|
|
t.Run("as system user", func(t *testing.T) {
|
|
ok, resp := Client.RemoveLicenseFile()
|
|
CheckForbiddenStatus(t, resp)
|
|
if ok {
|
|
t.Fatal("should fail")
|
|
}
|
|
})
|
|
|
|
t.Run("as system admin user", func(t *testing.T) {
|
|
ok, resp := th.SystemAdminClient.RemoveLicenseFile()
|
|
CheckNoError(t, resp)
|
|
if !ok {
|
|
t.Fatal("should pass")
|
|
}
|
|
})
|
|
|
|
t.Run("as restricted system admin user", func(t *testing.T) {
|
|
th.App.UpdateConfig(func(cfg *model.Config) { *cfg.ExperimentalSettings.RestrictSystemAdmin = true })
|
|
|
|
ok, resp := th.SystemAdminClient.RemoveLicenseFile()
|
|
CheckForbiddenStatus(t, resp)
|
|
if ok {
|
|
t.Fatal("should fail")
|
|
}
|
|
})
|
|
}
|