Function TestHandleVersionCommand

cli/cli_test.go:39–93

(t *testing.T)

}

func TestHandleVersionCommand(t *testing.T) {
	logger, _ := zap.NewDevelopment()
	cliInstance := &ChatCLI{logger: logger}
	handler := NewCommandHandler(cliInstance)

	originalCheckImpl := version.CheckLatestVersionImpl
	originalBuildImpl := version.GetBuildInfoImpl

	mockChecker := new(mockVersionChecker)

	version.CheckLatestVersionImpl = func(ctx context.Context) (string, bool, error) {
		return mockChecker.Check(ctx)
	}
	version.GetBuildInfoImpl = func() (string, string, string) {
		return "1.25.0", "abc1234", "2024-09-15"
	}
	defer func() {
		version.CheckLatestVersionImpl = originalCheckImpl
		version.GetBuildInfoImpl = originalBuildImpl
	}()

	testCases := []struct {
		name       string
		mockLatest string
		mockUpdate bool
		mockErr    error
		expectOut  string
	}{
		{"Update available", "1.26.0", true, nil, "Disponível! Atualize"},
		{"No update", "1.25.0", false, nil, "Você está na versão mais recente."},
		{"With error", "", false, errors.New("network error"), "Não foi possível verificar: network error"},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			mockChecker.On("Check", mock.Anything).Return(tc.mockLatest, tc.mockUpdate, tc.mockErr).Once()

			oldStdout := os.Stdout
			r, w, _ := os.Pipe()
			os.Stdout = w
			defer func() { os.Stdout = oldStdout }()

			handler.handleVersionCommand(context.Background())

			w.Close()
			out, _ := io.ReadAll(r)

			cleanOut := stripANSI(string(out))
			normalized := normalizeSpaces(cleanOut)

			assert.Contains(t, normalized, tc.expectOut)
			mockChecker.AssertExpectations(t)
		})
	}
}