Adding enterprise commands support (#8327)

1 files changed, 26 insertions, 0 deletions

diff --git a/cmd/commands/sampledata_test.go b/cmd/commands/sampledata_test.go
new file mode 100644
index 000000000..d71ac0575
--- /dev/null
+++ b/cmd/commands/sampledata_test.go
@@ -0,0 +1,26 @@
+// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
+// See License.txt for license information.
+
+package commands
+
+import (
+	"testing"
+
+	"github.com/mattermost/mattermost-server/api"
+	"github.com/mattermost/mattermost-server/cmd"
+	"github.com/stretchr/testify/require"
+)
+
+func TestSampledataBadParameters(t *testing.T) {
+	th := api.Setup().InitBasic()
+	defer th.TearDown()
+
+	// should fail because you need at least 1 worker
+	require.Error(t, cmd.RunCommand(t, "sampledata", "--workers", "0"))
+
+	// should fail because you have more team memberships than teams
+	require.Error(t, cmd.RunCommand(t, "sampledata", "--teams", "10", "--teams-memberships", "11"))
+
+	// should fail because you have more channel memberships than channels per team
+	require.Error(t, cmd.RunCommand(t, "sampledata", "--channels-per-team", "10", "--channel-memberships", "11"))
+}