aboutsummaryrefslogtreecommitdiffstats
path: root/utils/merkletrie/noder/noder_test.go
blob: c1af99843bf28872f90a54ebd96e60b4e06d92ee (plain) (blame)
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
package noder

import (
	"testing"

	. "gopkg.in/check.v1"
)

func Test(t *testing.T) { TestingT(t) }

type NoderSuite struct{}

var _ = Suite(&NoderSuite{})

type noderMock struct {
	name     string
	hash     []byte
	isDir    bool
	children []Noder
}

func (n noderMock) String() string             { return n.Name() }
func (n noderMock) Hash() []byte               { return n.hash }
func (n noderMock) Name() string               { return n.name }
func (n noderMock) IsDir() bool                { return n.isDir }
func (n noderMock) Children() ([]Noder, error) { return n.children, nil }
func (n noderMock) NumChildren() (int, error)  { return len(n.children), nil }
func (n noderMock) Skip() bool                 { return false }

// Returns a sequence with the noders 3, 2, and 1 from the
// following diagram:
//
//   3
//   |
//   2
//   |
//   1
//  / \
// c1  c2
//
// This is also the path of "1".
func nodersFixture() []Noder {
	n1 := &noderMock{
		name:     "1",
		hash:     []byte{0x00, 0x01, 0x02},
		isDir:    true,
		children: childrenFixture(),
	}
	n2 := &noderMock{name: "2"}
	n3 := &noderMock{name: "3"}
	return []Noder{n3, n2, n1}
}

// Returns a collection of 2 noders: c1, c2.
func childrenFixture() []Noder {
	c1 := &noderMock{name: "c1"}
	c2 := &noderMock{name: "c2"}
	return []Noder{c1, c2}
}

// returns nodersFixture as the path of "1".
func pathFixture() Path {
	return Path(nodersFixture())
}

func (s *NoderSuite) TestString(c *C) {
	c.Assert(pathFixture().String(), Equals, "3/2/1")
}

func (s *NoderSuite) TestLast(c *C) {
	c.Assert(pathFixture().Last().Name(), Equals, "1")
}

func (s *NoderSuite) TestPathImplementsNoder(c *C) {
	p := pathFixture()
	c.Assert(p.Name(), Equals, "1")
	c.Assert(p.Hash(), DeepEquals, []byte{0x00, 0x01, 0x02})
	c.Assert(p.IsDir(), Equals, true)

	children, err := p.Children()
	c.Assert(err, IsNil)
	c.Assert(children, DeepEquals, childrenFixture())

	numChildren, err := p.NumChildren()
	c.Assert(err, IsNil)
	c.Assert(numChildren, Equals, 2)
}