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 TestHelpListsTheImportTargetFlag(t *testing.T) {
  26         help := commandOutput(t, "help")
  27         require.Contains(t, help, "import [-n] [--agent NAME] ZIP...")
  28         require.Contains(t, help, "--agent NAME           import into an existing direct child (default: root)")
  29 }
  30
  31 func TestHelpListsTheRootNotificationCommand(t *testing.T) {
  32         require.Contains(t, commandOutput(t, "help"), "notify --source SOURCE --id ID --text TEXT")
  33         require.Contains(t, commandOutput(t, "notify", "--help"), "not processed")
  34 }
  35
  36 func commandOutput(t *testing.T, args ...string) string {
  37         t.Helper()
  38
  39         reader, writer, err := os.Pipe()
  40         require.NoError(t, err)
  41
  42         originalArgs, originalStdout := os.Args, os.Stdout
  43         os.Args = append([]string{"onechat"}, args...)
  44         os.Stdout = writer
  45         t.Cleanup(func() {
  46                 os.Args, os.Stdout = originalArgs, originalStdout
  47
  48                 require.NoError(t, reader.Close())
  49         })
  50
  51         err = run()
  52         require.NoError(t, err)
  53
  54         require.NoError(t, writer.Close())
  55
  56         output, err := io.ReadAll(reader)
  57         require.NoError(t, err)
  58
  59         return string(output)
  60 }