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
|
// -*- 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 snapstate
import (
. "gopkg.in/check.v1"
"github.com/snapcore/snapd/overlord/state"
)
type progressAdapterTestSuite struct{}
var _ = Suite(&progressAdapterTestSuite{})
func (s *progressAdapterTestSuite) TestProgressAdapterWrite(c *C) {
st := state.New(nil)
st.Lock()
m := NewTaskProgressAdapterUnlocked(st.NewTask("op", "msg"))
st.Unlock()
p := m.(*taskProgressAdapter)
m.Start("msg", 161803)
c.Check(p.total, Equals, float64(161803))
m.Write([]byte("some-bytes"))
c.Check(p.current, Equals, float64(len("some-bytes")))
}
func (s *progressAdapterTestSuite) TestProgressAdapterWriteLocked(c *C) {
st := state.New(nil)
st.Lock()
defer st.Unlock()
m := NewTaskProgressAdapterLocked(st.NewTask("op", "msg"))
p := m.(*taskProgressAdapter)
m.Start("msg", 161803)
c.Check(p.total, Equals, float64(161803))
m.Write([]byte("some-bytes"))
c.Check(p.current, Equals, float64(len("some-bytes")))
}
func (s *progressAdapterTestSuite) TestProgressAdapterSetTaskProgress(c *C) {
st := state.New(nil)
st.Lock()
t := st.NewTask("op", "msg")
m := NewTaskProgressAdapterUnlocked(t)
st.Unlock()
// we expect 1000 bytes
m.Start("msg", 1000)
// write a single byte (0.1% of the total)
m.Write([]byte("1"))
// check that the progress is not updated yet
st.Lock()
msg, done, total := t.Progress()
st.Unlock()
c.Check(msg, Equals, "msg")
c.Check(done, Equals, 0)
c.Check(total, Equals, 1000)
// write another byte (0.2% now)
m.Write([]byte("2"))
// now the progress in the task gets updated (we update every 0.2%)
st.Lock()
_, done, total = t.Progress()
st.Unlock()
c.Check(done, Equals, 2)
c.Check(total, Equals, 1000)
}
|
