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
|
// -*- Mode: Go; indent-tabs-mode: t -*-
/*
* Copyright (C) 2020 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 daemon_test
import (
"bytes"
"net/http"
"sort"
"time"
. "gopkg.in/check.v1"
"github.com/snapcore/snapd/daemon"
"github.com/snapcore/snapd/overlord/configstate/config"
"github.com/snapcore/snapd/overlord/snapstate"
"github.com/snapcore/snapd/overlord/state"
)
var _ = Suite(&consoleConfSuite{})
type consoleConfSuite struct {
apiBaseSuite
}
func (s *consoleConfSuite) TestPostConsoleConfStartRoutine(c *C) {
t0 := time.Now()
d := s.daemonWithOverlordMock()
snapMgr, err := snapstate.Manager(d.Overlord().State(), d.Overlord().TaskRunner())
c.Assert(err, IsNil)
d.Overlord().AddManager(snapMgr)
st := d.Overlord().State()
body := bytes.NewBuffer(nil)
req, err := http.NewRequest("POST", "/v2/internal/console-conf-start", body)
c.Assert(err, IsNil)
// no changes in state, no changes in response
rsp := s.syncReq(c, req, nil, actionIsExpected)
c.Assert(rsp.Result, DeepEquals, &daemon.ConsoleConfStartRoutineResult{})
// we did set the refresh.hold time back 20 minutes though
st.Lock()
defer st.Unlock()
tr := config.NewTransaction(st)
var t1 time.Time
err = tr.Get("core", "refresh.hold", &t1)
c.Assert(err, IsNil)
c.Assert(t0.Add(20*time.Minute).After(t1), Equals, false)
// if we add some changes to state that are in progress, then they are
// returned
// now make some auto-refresh changes to make sure we get those figured out
chg0 := st.NewChange("auto-refresh", "auto-refresh-the-things")
chg0.AddTask(st.NewTask("nop", "do nothing"))
// make it in doing state
chg0.SetStatus(state.DoingStatus)
chg0.Set("snap-names", []string{"doing-snap"})
// this one will be picked up too
chg1 := st.NewChange("auto-refresh", "auto-refresh-the-things")
chg1.AddTask(st.NewTask("nop", "do nothing"))
chg1.SetStatus(state.DoStatus)
chg1.Set("snap-names", []string{"do-snap"})
// this one won't, it's Done
chg2 := st.NewChange("auto-refresh", "auto-refresh-the-things")
chg2.AddTask(st.NewTask("nop", "do nothing"))
chg2.SetStatus(state.DoneStatus)
chg2.Set("snap-names", []string{"done-snap"})
// nor this one, it's Undone
chg3 := st.NewChange("auto-refresh", "auto-refresh-the-things")
chg3.AddTask(st.NewTask("nop", "do nothing"))
chg3.SetStatus(state.UndoneStatus)
chg3.Set("snap-names", []string{"undone-snap"})
st.Unlock()
defer st.Lock()
req2, err := http.NewRequest("POST", "/v2/internal/console-conf-start", body)
c.Assert(err, IsNil)
rsp2 := s.syncReq(c, req2, nil, actionIsExpected)
c.Assert(rsp2.Result, FitsTypeOf, &daemon.ConsoleConfStartRoutineResult{})
res := rsp2.Result.(*daemon.ConsoleConfStartRoutineResult)
sort.Strings(res.ActiveAutoRefreshChanges)
sort.Strings(res.ActiveAutoRefreshSnaps)
expChgs := []string{chg0.ID(), chg1.ID()}
sort.Strings(expChgs)
c.Assert(res, DeepEquals, &daemon.ConsoleConfStartRoutineResult{
ActiveAutoRefreshChanges: expChgs,
ActiveAutoRefreshSnaps: []string{"do-snap", "doing-snap"},
})
}
|
