]> ruderich.org/simon Gitweb - safcm/safcm.git/blob - cmd/safcm-remote/sync/sync_test.go
config: disallow negative permissions
[safcm/safcm.git] / 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         stdout, stderr, err := r.check("combinedOutput", cmd)
58         if stderr != nil {
59                 // stdout also contains stderr
60                 r.t.Fatalf("CombinedOutput: stderr != nil, but %v", stderr)
61         }
62         return stdout, err
63 }
64 func (r *testRunner) check(method string, cmd *exec.Cmd) (
65         []byte, []byte, error) {
66
67         if len(r.expCmds) == 0 {
68                 r.t.Fatalf("%s: empty expCmds", method)
69         }
70         if len(r.resStdout) == 0 {
71                 r.t.Fatalf("%s: empty resStdout", method)
72         }
73         if len(r.resStderr) == 0 {
74                 r.t.Fatalf("%s: empty resStderr", method)
75         }
76         if len(r.resError) == 0 {
77                 r.t.Fatalf("%s: empty resError", method)
78         }
79
80         exp := r.expCmds[0]
81         r.expCmds = r.expCmds[1:]
82         if !reflect.DeepEqual(exp, cmd) {
83                 opts := cmpopts.IgnoreUnexported(exec.Cmd{}, bytes.Buffer{})
84                 r.t.Errorf("%s: %s", method, cmp.Diff(exp, cmd, opts))
85         }
86
87         var stdout, stderr []byte
88         var err error
89
90         stdout, r.resStdout = r.resStdout[0], r.resStdout[1:]
91         stderr, r.resStderr = r.resStderr[0], r.resStderr[1:]
92         err, r.resError = r.resError[0], r.resError[1:]
93
94         return stdout, stderr, err
95 }
96
97 type syncTestResult struct {
98         ch     chan string
99         wg     sync.WaitGroup
100         dbg    []string
101         runner *testRunner
102 }
103
104 func prepareSync(req safcm.MsgSyncReq, runner *testRunner) (
105         *Sync, *syncTestResult) {
106
107         res := &syncTestResult{
108                 ch:     make(chan string),
109                 runner: runner,
110         }
111         res.wg.Add(1)
112         go func() {
113                 for {
114                         x, ok := <-res.ch
115                         if !ok {
116                                 break
117                         }
118                         res.dbg = append(res.dbg, x)
119                 }
120                 res.wg.Done()
121         }()
122
123         logger := log.NewLogger(logPrefix,
124                 func(level safcm.LogLevel, format string, a ...interface{}) {
125                         res.ch <- fmt.Sprintf("%d: %s", level,
126                                 fmt.Sprintf(format, a...))
127                 })
128         return &Sync{
129                 req: req,
130                 cmd: run.NewCmd(runner, logger),
131                 log: logger,
132         }, res
133 }
134
135 func (s *syncTestResult) Wait() []string {
136         close(s.ch)
137         s.wg.Wait()
138
139         // All expected commands must have been executed
140         if len(s.runner.expCmds) != 0 {
141                 s.runner.t.Errorf("expCmds left: %v", s.runner.expCmds)
142         }
143         if len(s.runner.resStdout) != 0 {
144                 s.runner.t.Errorf("resStdout left: %v", s.runner.resStdout)
145         }
146         if len(s.runner.resStderr) != 0 {
147                 s.runner.t.Errorf("resStderr left: %v", s.runner.resStderr)
148         }
149         if len(s.runner.resError) != 0 {
150                 s.runner.t.Errorf("resError left: %v", s.runner.resError)
151         }
152
153         return s.dbg
154 }