Mock network call

cmd/main_test.go

@@ -4,10 +4,20 @@ import (
 	"testing"
 
 	"github.com/XiaoConstantine/mycli/pkg/commands/root"
+	"github.com/XiaoConstantine/mycli/pkg/commands/update"
+	"github.com/XiaoConstantine/mycli/pkg/iostreams"
 	"github.com/stretchr/testify/assert"
 )
 
 func TestRun(t *testing.T) {
+	// Save the original function and defer its restoration
+	originalCheckForUpdates := update.CheckForUpdatesFunc
+	defer func() { update.CheckForUpdatesFunc = originalCheckForUpdates }()
+
+	// Mock the CheckForUpdates function
+	update.CheckForUpdatesFunc = func(iostream *iostreams.IOStreams) (bool, string, error) {
+		return false, "v1.0.0", nil // No update available
+	}
 	// Define test cases
 	tests := []struct {
 		name     string

pkg/commands/root/root.go

@@ -92,7 +92,7 @@ func Run(args []string) ExitCode {
 	}
 
 	// Check for updates
-	hasUpdate, latestVersion, err := update.CheckForUpdates(iostream)
+	hasUpdate, latestVersion, err := update.CheckForUpdatesFunc(iostream)
 	if err != nil {
 		fmt.Fprintf(stderr, "Failed to check for updates: %s\n", err)
 	} else if hasUpdate {
@@ -157,10 +157,7 @@ func Run(args []string) ExitCode {
 		}
 		rootCmd.SetArgs([]string{selectedOption})
 	}
-	if err != nil {
-		fmt.Fprintf(stderr, "failed to create root command: %s\n", err)
-		return exitError
-	}
+
 	if _, err := rootCmd.ExecuteContextC(ctx); err != nil {
 		var pagerPipeError *iostreams.ErrClosedPagerPipe
 		var noResultsError utils.NoResultsError

pkg/commands/update/update.go

@@ -16,6 +16,8 @@ import (
 	"github.com/spf13/cobra"
 )
 
+var CheckForUpdatesFunc = CheckForUpdates
+
 func NewUpdateCmd(iostream *iostreams.IOStreams) *cobra.Command {
 	cmd := &cobra.Command{
 		Use:   "update",