upstream-mattermost/server/channels/app/platform/license_test.go

// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
// See LICENSE.txt for license information.

package platform

import (
	"encoding/json"
	"net/http"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"github.com/mattermost/mattermost/server/public/model"
	"github.com/mattermost/mattermost/server/public/plugin/plugintest/mock"
	"github.com/mattermost/mattermost/server/v8/channels/utils"
	mocks2 "github.com/mattermost/mattermost/server/v8/channels/utils/mocks"
	"github.com/mattermost/mattermost/server/v8/channels/utils/testutils"
)

func TestLoadLicense(t *testing.T) {
	mainHelper.Parallel(t)
	th := Setup(t)

	th.Service.LoadLicense()
	require.Nil(t, th.Service.License(), "shouldn't have a valid license")
}

func TestSaveLicense(t *testing.T) {
	mainHelper.Parallel(t)
	th := Setup(t)

	b1 := []byte("junk")

	_, err := th.Service.SaveLicense(b1)
	require.NotNil(t, err, "shouldn't have saved license")
}

func TestSaveLicenseWrongEnvironment(t *testing.T) {
	// t.Setenv prevents t.Parallel — the service environment has no config equivalent.
	th := Setup(t)

	t.Setenv("MM_SERVICEENVIRONMENT", model.ServiceEnvironmentProduction)

	// validTestLicense is signed with the test key, so uploading it to a server running
	// in the production service environment must report an environment mismatch rather
	// than the generic invalid-license error.
	_, appErr := th.Service.SaveLicense(validTestLicense)
	require.NotNil(t, appErr, "shouldn't have saved a license signed for another environment")
	require.Equal(t, model.WrongEnvironmentTestLicenseError, appErr.Id)
	require.Equal(t, http.StatusBadRequest, appErr.StatusCode)
}

func TestSaveEnterpriseAdvancedLicense(t *testing.T) {
	th := Setup(t)

	defer testutils.ResetLicenseValidator()
	mockLicenseValidator := mocks2.LicenseValidatorIface{}

	license := &model.License{
		Id: model.NewId(),
		Features: &model.Features{
			Users: new(100),
		},
		Customer: &model.Customer{
			Name:  "TestName",
			Email: "test@example.com",
		},
		SkuName:      "SKU NAME",
		SkuShortName: model.LicenseShortSkuEnterpriseAdvanced,
		StartsAt:     model.GetMillis() - 1000,
		ExpiresAt:    model.GetMillis() + 100000,
	}

	mockLicenseValidator.On("LicenseFromBytes", mock.Anything).Return(license, nil).Once()
	licenseBytes, err := json.Marshal(license)
	require.NoError(t, err)

	mockLicenseValidator.On("ValidateLicense", mock.Anything).Return(string(licenseBytes), nil)
	utils.LicenseValidator = &mockLicenseValidator

	_, appErr := th.Service.SaveLicense(licenseBytes)

	require.Nil(t, appErr, "should have saved license")
}

func TestRemoveLicense(t *testing.T) {
	mainHelper.Parallel(t)
	th := Setup(t)

	err := th.Service.RemoveLicense()
	require.Nil(t, err, "should have removed license")
}

func TestSetLicense(t *testing.T) {
	mainHelper.Parallel(t)
	th := Setup(t)

	l1 := &model.License{}
	l1.Features = &model.Features{}
	l1.Customer = &model.Customer{}
	l1.StartsAt = model.GetMillis() - 1000
	l1.ExpiresAt = model.GetMillis() + 100000
	ok := th.Service.SetLicense(l1)
	require.True(t, ok, "license should have worked")

	l3 := &model.License{}
	l3.Features = &model.Features{}
	l3.Customer = &model.Customer{}
	l3.StartsAt = model.GetMillis() + 10000
	l3.ExpiresAt = model.GetMillis() + 100000
	ok = th.Service.SetLicense(l3)
	require.True(t, ok, "license should have passed")
}

func TestGetSanitizedClientLicense(t *testing.T) {
	mainHelper.Parallel(t)
	th := Setup(t)

	setLicense(th, nil)

	m := th.Service.GetSanitizedClientLicense()

	_, ok := m["Name"]
	assert.False(t, ok)
	_, ok = m["SkuName"]
	assert.False(t, ok)
}

func setLicense(th *TestHelper, customer *model.Customer) {
	l1 := &model.License{}
	l1.Features = &model.Features{}
	if customer != nil {
		l1.Customer = customer
	} else {
		l1.Customer = &model.Customer{}
		l1.Customer.Name = "TestName"
		l1.Customer.Email = "test@example.com"
	}
	l1.SkuName = "SKU NAME"
	l1.SkuShortName = "SKU SHORT NAME"
	l1.StartsAt = model.GetMillis() - 1000
	l1.ExpiresAt = model.GetMillis() + 100000
	th.Service.SetLicense(l1)
}