Cloud Freemium - Integrations (#20185)

Co-authored-by: Mattermod <mattermod@users.noreply.github.com>

app/plugin_test.go

@@ -18,9 +18,11 @@ import (
 
 	"github.com/gorilla/mux"
 	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/mock"
 	"github.com/stretchr/testify/require"
 
 	"github.com/mattermost/mattermost-server/v6/app/request"
+	"github.com/mattermost/mattermost-server/v6/einterfaces/mocks"
 	"github.com/mattermost/mattermost-server/v6/model"
 	"github.com/mattermost/mattermost-server/v6/plugin"
 	"github.com/mattermost/mattermost-server/v6/shared/mlog"
@@ -966,6 +968,70 @@ func TestProcessPrepackagedPlugins(t *testing.T) {
 	})
 }
 
+func TestEnablePluginWithCloudLimits(t *testing.T) {
+	th := Setup(t)
+	defer th.TearDown()
+
+	os.Setenv("MM_FEATUREFLAGS_CLOUDFREE", "true")
+	defer os.Unsetenv("MM_FEATUREFLAGS_CLOUDFREE")
+	th.App.ReloadConfig()
+	th.App.Srv().SetLicense(model.NewTestLicense("cloud"))
+
+	th.App.UpdateConfig(func(cfg *model.Config) {
+		*cfg.PluginSettings.Enable = true
+		*cfg.PluginSettings.RequirePluginSignature = false
+		cfg.PluginSettings.PluginStates["testplugin"] = &model.PluginState{Enable: false}
+		cfg.PluginSettings.PluginStates["testplugin2"] = &model.PluginState{Enable: false}
+	})
+
+	cloud := &mocks.CloudInterface{}
+	cloud.Mock.On("GetCloudLimits", mock.Anything).Return(&model.ProductLimits{
+		Integrations: &model.IntegrationsLimits{
+			Enabled: model.NewInt(1),
+		},
+	}, nil)
+
+	cloudImpl := th.App.Srv().Cloud
+	defer func() {
+		th.App.Srv().Cloud = cloudImpl
+	}()
+	th.App.Srv().Cloud = cloud
+
+	env := th.App.GetPluginsEnvironment()
+	require.NotNil(t, env)
+
+	path, _ := fileutils.FindDir("tests")
+	fileReader, err := os.Open(filepath.Join(path, "testplugin.tar.gz"))
+	require.NoError(t, err)
+	defer fileReader.Close()
+
+	_, appErr := th.App.WriteFile(fileReader, getBundleStorePath("testplugin"))
+	checkNoError(t, appErr)
+
+	fileReader, err = os.Open(filepath.Join(path, "testplugin2.tar.gz"))
+	require.NoError(t, err)
+	defer fileReader.Close()
+
+	_, appErr = th.App.WriteFile(fileReader, getBundleStorePath("testplugin2"))
+	checkNoError(t, appErr)
+
+	appErr = th.App.SyncPlugins()
+	checkNoError(t, appErr)
+
+	appErr = th.App.EnablePlugin("testplugin")
+	checkNoError(t, appErr)
+
+	appErr = th.App.EnablePlugin("testplugin2")
+	checkError(t, appErr)
+	require.Equal(t, "app.install_integration.reached_max_limit.error", appErr.Id)
+
+	os.Unsetenv("MM_FEATUREFLAGS_CLOUDFREE")
+	th.App.ReloadConfig()
+
+	appErr = th.App.EnablePlugin("testplugin2")
+	checkNoError(t, appErr)
+}
+
 func TestGetPluginStateOverride(t *testing.T) {
 	th := Setup(t)
 	defer th.TearDown()