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
127
128
129
|
// -*- 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 main
import (
"bytes"
"encoding/json"
"fmt"
"net/http"
"net/http/httptest"
"os"
"path/filepath"
"testing"
. "gopkg.in/check.v1"
"github.com/snapcore/snapd/client"
)
func TestT(t *testing.T) { TestingT(t) }
type snapctlSuite struct {
server *httptest.Server
oldArgs []string
expectedContextID string
expectedArgs []string
expectedStdin []byte
}
var _ = Suite(&snapctlSuite{})
func (s *snapctlSuite) SetUpTest(c *C) {
os.Setenv("SNAP_COOKIE", "snap-context-test")
// don't use SNAP_CONTEXT, in case other tests accidentally leak this
os.Unsetenv("SNAP_CONTEXT")
n := 0
s.server = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
switch n {
case 0:
c.Assert(r.Method, Equals, "POST")
c.Assert(r.URL.Path, Equals, "/v2/snapctl")
c.Assert(r.Header.Get("Authorization"), Equals, "")
var snapctlPostData client.SnapCtlPostData
decoder := json.NewDecoder(r.Body)
c.Assert(decoder.Decode(&snapctlPostData), IsNil)
c.Assert(snapctlPostData.ContextID, Equals, s.expectedContextID)
c.Assert(snapctlPostData.Args, DeepEquals, s.expectedArgs)
c.Assert(snapctlPostData.Stdin, DeepEquals, s.expectedStdin)
fmt.Fprintln(w, `{"type": "sync", "result": {"stdout": "test stdout", "stderr": "test stderr"}}`)
default:
c.Fatalf("expected to get 1 request, now on %d", n+1)
}
n++
}))
clientConfig.BaseURL = s.server.URL
s.oldArgs = os.Args
os.Args = []string{"snapctl"}
s.expectedContextID = "snap-context-test"
s.expectedArgs = []string{}
fakeAuthPath := filepath.Join(c.MkDir(), "auth.json")
os.Setenv("SNAPD_AUTH_DATA_FILENAME", fakeAuthPath)
err := os.WriteFile(fakeAuthPath, []byte(`{"macaroon":"user-macaroon"}`), 0644)
c.Assert(err, IsNil)
}
func (s *snapctlSuite) TearDownTest(c *C) {
os.Unsetenv("SNAP_COOKIE")
os.Unsetenv("SNAPD_AUTH_DATA_FILENAME")
clientConfig.BaseURL = ""
s.server.Close()
os.Args = s.oldArgs
}
func (s *snapctlSuite) TestSnapctl(c *C) {
stdout, stderr, err := run(nil)
c.Check(err, IsNil)
c.Check(string(stdout), Equals, "test stdout")
c.Check(string(stderr), Equals, "test stderr")
}
func (s *snapctlSuite) TestSnapctlWithArgs(c *C) {
os.Args = []string{"snapctl", "foo", "--bar"}
s.expectedArgs = []string{"foo", "--bar"}
stdout, stderr, err := run(nil)
c.Check(err, IsNil)
c.Check(string(stdout), Equals, "test stdout")
c.Check(string(stderr), Equals, "test stderr")
}
func (s *snapctlSuite) TestSnapctlHelp(c *C) {
os.Unsetenv("SNAP_COOKIE")
s.expectedContextID = ""
os.Args = []string{"snapctl", "-h"}
s.expectedArgs = []string{"-h"}
_, _, err := run(nil)
c.Check(err, IsNil)
}
func (s *snapctlSuite) TestSnapctlWithStdin(c *C) {
s.expectedStdin = []byte("hello")
mockStdin := bytes.NewBuffer(s.expectedStdin)
_, _, err := run(mockStdin)
c.Check(err, IsNil)
}
|
