forked from snapcore/snapd
/
gadget.go
106 lines (87 loc) · 2.79 KB
/
gadget.go
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
// -*- Mode: Go; indent-tabs-mode: t -*-
/*
* Copyright (C) 2016-2019 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 devicestatetest
import (
"encoding/json"
"fmt"
. "gopkg.in/check.v1"
"gopkg.in/tomb.v2"
"gopkg.in/yaml.v2"
"github.com/snapcore/snapd/overlord/hookstate"
"github.com/snapcore/snapd/overlord/hookstate/ctlcmd"
"github.com/snapcore/snapd/overlord/snapstate"
"github.com/snapcore/snapd/overlord/state"
"github.com/snapcore/snapd/snap"
"github.com/snapcore/snapd/snap/snaptest"
)
type PrepareDeviceBehavior struct {
DeviceSvcURL string
Headers map[string]string
RegBody map[string]string
ProposedSerial string
}
func MockGadget(c *C, st *state.State, name string, revision snap.Revision, pDBhv *PrepareDeviceBehavior) (restore func()) {
sideInfoGadget := &snap.SideInfo{
RealName: name,
Revision: revision,
}
snapYaml := fmt.Sprintf(`name: %q
type: gadget
version: gadget
`, name)
if pDBhv != nil {
snapYaml += `hooks:
prepare-device:
`
}
snaptest.MockSnap(c, snapYaml, sideInfoGadget)
snapstate.Set(st, name, &snapstate.SnapState{
SnapType: "gadget",
Active: true,
Sequence: []*snap.SideInfo{sideInfoGadget},
Current: revision,
})
if pDBhv == nil {
// nothing to restore
return func() {}
}
// mock the prepare-device hook
return hookstate.MockRunHook(func(ctx *hookstate.Context, _ *tomb.Tomb) ([]byte, error) {
c.Assert(ctx.HookName(), Equals, "prepare-device")
// snapctl set the registration params
_, _, err := ctlcmd.Run(ctx, []string{"set", fmt.Sprintf("device-service.url=%q", pDBhv.DeviceSvcURL)}, 0)
c.Assert(err, IsNil)
if len(pDBhv.Headers) != 0 {
h, err := json.Marshal(pDBhv.Headers)
c.Assert(err, IsNil)
_, _, err = ctlcmd.Run(ctx, []string{"set", fmt.Sprintf("device-service.headers=%s", string(h))}, 0)
c.Assert(err, IsNil)
}
if pDBhv.ProposedSerial != "" {
_, _, err = ctlcmd.Run(ctx, []string{"set", fmt.Sprintf("registration.proposed-serial=%q", pDBhv.ProposedSerial)}, 0)
c.Assert(err, IsNil)
}
if len(pDBhv.RegBody) != 0 {
d, err := yaml.Marshal(pDBhv.RegBody)
c.Assert(err, IsNil)
_, _, err = ctlcmd.Run(ctx, []string{"set", fmt.Sprintf("registration.body=%q", d)}, 0)
c.Assert(err, IsNil)
}
return nil, nil
})
}