cmd/safcm-remote/sync/sync_test.go

   1 // Copyright (C) 2021  Simon Ruderich
   2 //
   3 // This program is free software: you can redistribute it and/or modify
   4 // it under the terms of the GNU General Public License as published by
   5 // the Free Software Foundation, either version 3 of the License, or
   6 // (at your option) any later version.
   7 //
   8 // This program is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11 // GNU General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU General Public License
  14 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
  15
  16 package sync
  17
  18 import (
  19         "bytes"
  20         "fmt"
  21         "os/exec"
  22         "reflect"
  23         "sync"
  24         "testing"
  25
  26         "github.com/google/go-cmp/cmp"
  27         "github.com/google/go-cmp/cmp/cmpopts"
  28
  29         "ruderich.org/simon/safcm"
  30         "ruderich.org/simon/safcm/cmd/safcm-remote/log"
  31         "ruderich.org/simon/safcm/cmd/safcm-remote/run"
  32 )
  33
  34 // testRunner implements run.Runner to test commands without actually running
  35 // them.
  36 type testRunner struct {
  37         t         *testing.T
  38         expCmds   []*exec.Cmd
  39         resStdout [][]byte
  40         resStderr [][]byte
  41         resError  []error
  42 }
  43
  44 func (r *testRunner) Run(cmd *exec.Cmd) error {
  45         stdout, stderr, resErr := r.check("run", cmd)
  46         _, err := cmd.Stdout.Write(stdout)
  47         if err != nil {
  48                 panic(err)
  49         }
  50         _, err = cmd.Stderr.Write(stderr)
  51         if err != nil {
  52                 panic(err)
  53         }
  54         return resErr
  55 }
  56 func (r *testRunner) CombinedOutput(cmd *exec.Cmd) ([]byte, error) {
  57         r.t.Helper()
  58
  59         stdout, stderr, err := r.check("combinedOutput", cmd)
  60         if stderr != nil {
  61                 // stdout also contains stderr
  62                 r.t.Fatalf("CombinedOutput: stderr != nil, but %v", stderr)
  63         }
  64         return stdout, err
  65 }
  66 func (r *testRunner) check(method string, cmd *exec.Cmd) (
  67         []byte, []byte, error) {
  68         r.t.Helper()
  69
  70         if len(r.expCmds) == 0 {
  71                 r.t.Fatalf("%s: empty expCmds", method)
  72         }
  73         if len(r.resStdout) == 0 {
  74                 r.t.Fatalf("%s: empty resStdout", method)
  75         }
  76         if len(r.resStderr) == 0 {
  77                 r.t.Fatalf("%s: empty resStderr", method)
  78         }
  79         if len(r.resError) == 0 {
  80                 r.t.Fatalf("%s: empty resError", method)
  81         }
  82
  83         exp := r.expCmds[0]
  84         r.expCmds = r.expCmds[1:]
  85         if !reflect.DeepEqual(exp, cmd) {
  86                 opts := cmpopts.IgnoreUnexported(exec.Cmd{}, bytes.Buffer{})
  87                 r.t.Errorf("%s: %s", method, cmp.Diff(exp, cmd, opts))
  88         }
  89
  90         var stdout, stderr []byte
  91         var err error
  92
  93         stdout, r.resStdout = r.resStdout[0], r.resStdout[1:]
  94         stderr, r.resStderr = r.resStderr[0], r.resStderr[1:]
  95         err, r.resError = r.resError[0], r.resError[1:]
  96
  97         return stdout, stderr, err
  98 }
  99
 100 type syncTestResult struct {
 101         ch     chan string
 102         wg     sync.WaitGroup
 103         dbg    []string
 104         runner *testRunner
 105 }
 106
 107 func prepareSync(req safcm.MsgSyncReq, runner *testRunner) (
 108         *Sync, *syncTestResult) {
 109
 110         res := &syncTestResult{
 111                 ch:     make(chan string),
 112                 runner: runner,
 113         }
 114         res.wg.Add(1)
 115         go func() {
 116                 for {
 117                         x, ok := <-res.ch
 118                         if !ok {
 119                                 break
 120                         }
 121                         res.dbg = append(res.dbg, x)
 122                 }
 123                 res.wg.Done()
 124         }()
 125
 126         logger := log.NewLogger(
 127                 func(level safcm.LogLevel, msg string) {
 128                         res.ch <- fmt.Sprintf("%d: %s", level, msg)
 129                 })
 130         return &Sync{
 131                 req: req,
 132                 cmd: run.NewCmd(runner, logger),
 133                 log: logger,
 134         }, res
 135 }
 136
 137 func (s *syncTestResult) Wait() []string {
 138         s.runner.t.Helper()
 139
 140         close(s.ch)
 141         s.wg.Wait()
 142
 143         // All expected commands must have been executed
 144         if len(s.runner.expCmds) != 0 {
 145                 s.runner.t.Errorf("expCmds left: %v", s.runner.expCmds)
 146         }
 147         if len(s.runner.resStdout) != 0 {
 148                 s.runner.t.Errorf("resStdout left: %v", s.runner.resStdout)
 149         }
 150         if len(s.runner.resStderr) != 0 {
 151                 s.runner.t.Errorf("resStderr left: %v", s.runner.resStderr)
 152         }
 153         if len(s.runner.resError) != 0 {
 154                 s.runner.t.Errorf("resError left: %v", s.runner.resError)
 155         }
 156
 157         return s.dbg
 158 }