-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathplan_test.go
More file actions
103 lines (87 loc) · 1.69 KB
/
plan_test.go
File metadata and controls
103 lines (87 loc) · 1.69 KB
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
package vcrypt
import (
"reflect"
"testing"
"github.com/vcrypt/vcrypt/internal/test"
)
var (
twoManPlan = buildPlan(twoManGraph, "two-man rule plan")
twoPartyPlan = buildPlan(twoPartyGraph, "two-party lock plan")
diamondPlan = buildPlan(diamondGraph, "diamond plan")
dnsSecPlan = buildPlan(dnsSecGraph, "DNSSEC root key plan")
acmeBankPlan = buildPlan(acmeBankGraph, "Acme Bank Master Key Recovery plan")
)
func TestPlan(t *testing.T) {
tests := []struct {
*Plan
*Graph
}{
{
Plan: twoManPlan,
Graph: twoManGraph,
},
{
Plan: twoPartyPlan,
Graph: twoPartyGraph,
},
{
Plan: diamondPlan,
Graph: diamondGraph,
},
{
Plan: dnsSecPlan,
Graph: dnsSecGraph,
},
{
Plan: acmeBankPlan,
Graph: acmeBankGraph,
},
}
for _, tst := range tests {
plan, graph := tst.Plan, tst.Graph
ufp, err := plan.Digest()
if err != nil {
t.Fatal(err)
}
seal, err := plan.AddSeal(test.Sealer)
if err != nil {
t.Fatal(err)
}
data, err := plan.sealData()
if err != nil {
t.Fatal(err)
}
if err := seal.Check(data); err != nil {
t.Fatal(err)
}
sfp, err := plan.Digest()
if err != nil {
t.Fatal(err)
}
if reflect.DeepEqual(ufp, sfp) {
t.Errorf("plan digest unchanged by seal")
}
nodes, err := graph.Nodes()
if err != nil {
t.Fatal(err)
}
wfp, err := nodes[0].Digest()
if err != nil {
t.Fatal(err)
}
gfp, err := plan.Nodes[0].Digest()
if err != nil {
t.Fatal(err)
}
if !reflect.DeepEqual(wfp, gfp) {
t.Errorf("root node digests did not match %#v", nodes[0])
}
}
}
func buildPlan(g *Graph, desc string) *Plan {
plan, err := NewPlan(g, desc)
if err != nil {
panic(err)
}
return plan
}