Merge pull request #65 from dispatchrun/more_tests

Add main command tests and `dispatch version` tests

Author: chicoxyzzy

cli/main.go

@@ -10,15 +10,14 @@ var (
 	DispatchCmdLong = `Welcome to Dispatch!
 
 To get started, use the login command to authenticate with Dispatch or create an account.
-	
+
 Documentation: https://docs.dispatch.run
 Discord: https://dispatch.run/discord
 Support: [email protected]
 `
 )
 
-// Main is the entry point of the command line.
-func Main() error {
+func createMainCommand() *cobra.Command {
 	cmd := &cobra.Command{
 		Version: version(),
 		Use:     "dispatch",
@@ -50,5 +49,10 @@ func Main() error {
 	cmd.AddCommand(runCommand())
 	cmd.AddCommand(versionCommand())
 
-	return cmd.ExecuteContext(context.Background())
+	return cmd
+}
+
+// Main is the entry point of the command line.
+func Main() error {
+	return createMainCommand().ExecuteContext(context.Background())
 }

cli/main_test.go

@@ -0,0 +1,37 @@
+package cli
+
+import (
+	"sort"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+var expectedCommands = []string{"login", "switch [organization]", "verification", "run", "version"}
+
+func TestMainCommand(t *testing.T) {
+	t.Run("Main command", func(t *testing.T) {
+		t.Parallel()
+
+		cmd := createMainCommand()
+		assert.NotNil(t, cmd, "Expected main command to be created")
+
+		groups := cmd.Groups()
+		assert.Len(t, groups, 2, "Expected 2 groups")
+		assert.Equal(t, "management", groups[0].ID, "Expected first group to be 'management'")
+		assert.Equal(t, "dispatch", groups[1].ID, "Expected second group to be 'dispatch'")
+
+		commands := cmd.Commands()
+		assert.Len(t, commands, 5, "Expected 5 commands")
+
+		// Extract the command IDs
+		commandIDs := make([]string, 0, len(commands))
+		for _, command := range commands {
+			commandIDs = append(commandIDs, command.Use)
+		}
+
+		// Sort slices alphabetically
+		sort.Strings(expectedCommands)
+		assert.Equal(t, expectedCommands, commandIDs, "All commands should be present in the main command")
+	})
+}

cli/version.go

@@ -1,7 +1,6 @@
 package cli
 
 import (
-	"fmt"
 	"runtime/debug"
 
 	"github.com/spf13/cobra"
@@ -13,7 +12,7 @@ func versionCommand() *cobra.Command {
 		Short: "Print the version",
 		RunE: func(cmd *cobra.Command, args []string) error {
 			// Match dispatch -v,--version output:
-			fmt.Println("dispatch version " + version())
+			cmd.Println("dispatch version " + version())
 			return nil
 		},
 	}

cli/version_test.go

@@ -0,0 +1,52 @@
+package cli
+
+import (
+	"bytes"
+	"os/exec"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+var versionText = "dispatch version devel"
+
+func TestVersionCommand(t *testing.T) {
+	t.Run("Print the version (test runtime)", func(t *testing.T) {
+		t.Parallel()
+
+		cmd := versionCommand()
+		stdout := &bytes.Buffer{}
+		cmd.SetOut(stdout)
+
+		if err := cmd.Execute(); err != nil {
+			t.Fatalf("Received unexpected error: %v", err)
+		}
+
+		assert.Equal(t, versionText+"\n", stdout.String())
+	})
+
+	t.Run("Print the version (binary)", func(t *testing.T) {
+		t.Parallel()
+
+		cmd := exec.Command(dispatchBinary, "version")
+		stderr := &bytes.Buffer{}
+		cmd.Stderr = stderr
+
+		if err := cmd.Run(); err != nil {
+			t.Fatalf("Received unexpected error: %v", err)
+		}
+
+		// get git commit hash
+		cmdGitHash := exec.Command("git", "rev-parse", "HEAD")
+		stdout := &bytes.Buffer{}
+		cmdGitHash.Stdout = stdout
+
+		if err := cmdGitHash.Run(); err != nil {
+			t.Fatalf("Received unexpected error: %v", err)
+		}
+
+		version := stdout.String()
+
+		assert.Equal(t, versionText+" "+version, stderr.String())
+	})
+}