1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
// -*- Mode: Go; indent-tabs-mode: t -*-
/*
* Copyright (C) 2016 Canonical Ltd
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 3 as
* published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*/
package client_test
import (
"bytes"
"encoding/base64"
"encoding/json"
"gopkg.in/check.v1"
"github.com/snapcore/snapd/client"
)
func (cs *clientSuite) TestClientRunSnapctlCallsEndpoint(c *check.C) {
options := &client.SnapCtlOptions{
ContextID: "1234ABCD",
Args: []string{"foo", "bar"},
}
cs.cli.RunSnapctl(options, nil)
c.Check(cs.req.Method, check.Equals, "POST")
c.Check(cs.req.URL.Path, check.Equals, "/v2/snapctl")
}
func (cs *clientSuite) TestClientRunSnapctl(c *check.C) {
cs.rsp = `{
"type": "sync",
"status-code": 200,
"result": {
"stdout": "test stdout",
"stderr": "test stderr"
}
}`
mockStdin := bytes.NewBufferString("some-input")
options := &client.SnapCtlOptions{
ContextID: "1234ABCD",
Args: []string{"foo", "bar"},
}
stdout, stderr, err := cs.cli.RunSnapctl(options, mockStdin)
c.Assert(err, check.IsNil)
c.Check(string(stdout), check.Equals, "test stdout")
c.Check(string(stderr), check.Equals, "test stderr")
var body map[string]any
decoder := json.NewDecoder(cs.req.Body)
err = decoder.Decode(&body)
c.Check(err, check.IsNil)
c.Check(body, check.DeepEquals, map[string]any{
"context-id": "1234ABCD",
"args": []any{"foo", "bar"},
// json byte-stream is b64 encoded
"stdin": base64.StdEncoding.EncodeToString([]byte("some-input")),
})
}
func (cs *clientSuite) TestInternalSnapctlCmdNeedsStdin(c *check.C) {
res := client.InternalSnapctlCmdNeedsStdin("fde-setup-result")
c.Check(res, check.Equals, true)
for _, s := range []string{"help", "other"} {
res := client.InternalSnapctlCmdNeedsStdin(s)
c.Check(res, check.Equals, false)
}
}
func (cs *clientSuite) TestClientRunSnapctlReadLimitOneTooMuch(c *check.C) {
cs.rsp = `{
"type": "sync",
"status-code": 200,
"result": {
}
}`
restore := client.MockStdinReadLimit(10)
defer restore()
mockStdin := bytes.NewBufferString("12345678901")
options := &client.SnapCtlOptions{
ContextID: "1234ABCD",
Args: []string{"foo", "bar"},
}
_, _, err := cs.cli.RunSnapctl(options, mockStdin)
c.Check(err, check.ErrorMatches, "cannot read more than 10 bytes of data from stdin")
}
func (cs *clientSuite) TestClientRunSnapctlReadLimitExact(c *check.C) {
cs.rsp = `{
"type": "sync",
"status-code": 200,
"result": {
}
}`
restore := client.MockStdinReadLimit(10)
defer restore()
mockStdin := bytes.NewBufferString("1234567890")
options := &client.SnapCtlOptions{
ContextID: "1234ABCD",
Args: []string{"foo", "bar"},
}
_, _, err := cs.cli.RunSnapctl(options, mockStdin)
c.Check(err, check.IsNil)
}
|
