git.giacomolli.uk - onechat/blob - command_test.go

   1 package main
   2
   3 import (
   4         "io"
   5         "os"
   6         "testing"
   7
   8         "github.com/stretchr/testify/require"
   9 )
  10
  11 func TestVersionCommandPrintsBuildVersion(t *testing.T) {
  12         original := version
  13         version = "test-version"
  14         t.Cleanup(func() {
  15                 version = original
  16         })
  17
  18         require.Equal(t, "test-version\n", commandOutput(t, "version"))
  19 }
  20
  21 func TestHelpListsVersionCommand(t *testing.T) {
  22         require.Contains(t, commandOutput(t, "help"), "  version                  show the build version\n")
  23 }
  24
  25 func commandOutput(t *testing.T, args ...string) string {
  26         t.Helper()
  27
  28         reader, writer, err := os.Pipe()
  29         require.NoError(t, err)
  30
  31         originalArgs, originalStdout := os.Args, os.Stdout
  32         os.Args = append([]string{"onechat"}, args...)
  33         os.Stdout = writer
  34         t.Cleanup(func() {
  35                 os.Args, os.Stdout = originalArgs, originalStdout
  36
  37                 require.NoError(t, reader.Close())
  38         })
  39
  40         err = run()
  41         require.NoError(t, err)
  42
  43         require.NoError(t, writer.Close())
  44
  45         output, err := io.ReadAll(reader)
  46         require.NoError(t, err)
  47
  48         return string(output)
  49 }