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
|
// Copyright 2014 Canonical Ltd.
// Licensed under the LGPLv3, see LICENCE file for details.
package hash_test
import (
"bytes"
"github.com/juju/errors"
"github.com/juju/testing"
jc "github.com/juju/testing/checkers"
"github.com/juju/testing/filetesting"
gc "gopkg.in/check.v1"
"github.com/juju/utils/hash"
)
var _ = gc.Suite(&WriterSuite{})
type WriterSuite struct {
testing.IsolationSuite
stub *testing.Stub
wBuffer *bytes.Buffer
writer *filetesting.StubWriter
hBuffer *bytes.Buffer
hash *filetesting.StubHash
}
func (s *WriterSuite) SetUpTest(c *gc.C) {
s.IsolationSuite.SetUpTest(c)
s.stub = &testing.Stub{}
s.wBuffer = new(bytes.Buffer)
s.writer = &filetesting.StubWriter{
Stub: s.stub,
ReturnWrite: s.wBuffer,
}
s.hBuffer = new(bytes.Buffer)
s.hash = filetesting.NewStubHash(s.stub, s.hBuffer)
}
func (s *WriterSuite) TestHashingWriterWriteEmpty(c *gc.C) {
w := hash.NewHashingWriter(s.writer, s.hash)
n, err := w.Write(nil)
c.Assert(err, jc.ErrorIsNil)
s.stub.CheckCallNames(c, "Write", "Write")
c.Check(n, gc.Equals, 0)
c.Check(s.wBuffer.String(), gc.Equals, "")
c.Check(s.hBuffer.String(), gc.Equals, "")
}
func (s *WriterSuite) TestHashingWriterWriteSmall(c *gc.C) {
w := hash.NewHashingWriter(s.writer, s.hash)
n, err := w.Write([]byte("spam"))
c.Assert(err, jc.ErrorIsNil)
s.stub.CheckCallNames(c, "Write", "Write")
c.Check(n, gc.Equals, 4)
c.Check(s.wBuffer.String(), gc.Equals, "spam")
c.Check(s.hBuffer.String(), gc.Equals, "spam")
}
func (s *WriterSuite) TestHashingWriterWriteFileError(c *gc.C) {
w := hash.NewHashingWriter(s.writer, s.hash)
failure := errors.New("<failed>")
s.stub.SetErrors(failure)
_, err := w.Write([]byte("spam"))
s.stub.CheckCallNames(c, "Write")
c.Check(errors.Cause(err), gc.Equals, failure)
}
func (s *WriterSuite) TestHashingWriterBase64Sum(c *gc.C) {
s.hash.ReturnSum = []byte("spam")
w := hash.NewHashingWriter(s.writer, s.hash)
b64sum := w.Base64Sum()
s.stub.CheckCallNames(c, "Sum")
c.Check(b64sum, gc.Equals, "c3BhbQ==")
}
|
