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
116
117
118
119
120
121
122
|
// Copyright 2013 Canonical Ltd.
// Licensed under the AGPLv3, see LICENCE file for details.
package addressupdater_test
import (
"fmt"
"reflect"
"time"
gc "launchpad.net/gocheck"
"launchpad.net/juju-core/instance"
"launchpad.net/juju-core/juju/testing"
"launchpad.net/juju-core/provider/dummy"
"launchpad.net/juju-core/state"
coretesting "launchpad.net/juju-core/testing"
"launchpad.net/juju-core/testing/testbase"
"launchpad.net/juju-core/worker"
"launchpad.net/juju-core/worker/addressupdater"
)
var _ = gc.Suite(&workerSuite{})
type workerSuite struct {
testing.JujuConnSuite
}
func (*workerSuite) instId(i int) instance.Id {
return instance.Id(fmt.Sprint(i))
}
func (*workerSuite) addressesForIndex(i int) []instance.Address {
return []instance.Address{
instance.NewAddress(fmt.Sprintf("127.0.0.%d", i)),
}
}
func (s *workerSuite) TestWorker(c *gc.C) {
// Most functionality is already tested in detail - we
// just need to test that things are wired together
// correctly.
defer testbase.PatchValue(addressupdater.ShortPoll, 10*time.Millisecond).Restore()
defer testbase.PatchValue(addressupdater.LongPoll, 10*time.Millisecond).Restore()
machines, insts := s.setupScenario(c)
s.State.StartSync()
w := addressupdater.NewWorker(s.State)
defer func() {
c.Assert(worker.Stop(w), gc.IsNil)
}()
// Wait for the odd numbered machines in the
// first half of the machine slice to be given their
// addresses.
for a := coretesting.LongAttempt.Start(); a.Next(); {
if !a.HasNext() {
c.Fatalf("timed out waiting for machine addresses")
}
if machinesSatisfy(c, machines, func(i int, m *state.Machine) bool {
if i < len(machines)/2 && i%2 == 1 {
return reflect.DeepEqual(m.Addresses(), s.addressesForIndex(i))
}
return len(m.Addresses()) == 0
}) {
break
}
}
// Now provision the second half of the machines and
// watch them get addresses.
for i := len(insts) / 2; i < len(insts); i++ {
dummy.SetInstanceAddresses(insts[i], s.addressesForIndex(i))
}
for a := coretesting.LongAttempt.Start(); a.Next(); {
if !a.HasNext() {
c.Fatalf("timed out waiting for machine addresses")
}
if machinesSatisfy(c, machines, func(i int, m *state.Machine) bool {
return reflect.DeepEqual(m.Addresses(), s.addressesForIndex(i))
}) {
break
}
}
}
// TODO(rog)
// - check that the environment observer is actually hooked up.
// - check that the environment observer is stopped.
// - check that the errors propagate correctly.
func machinesSatisfy(c *gc.C, machines []*state.Machine, f func(i int, m *state.Machine) bool) bool {
for i, m := range machines {
err := m.Refresh()
c.Assert(err, gc.IsNil)
if !f(i, m) {
return false
}
}
return true
}
func (s *workerSuite) setupScenario(c *gc.C) ([]*state.Machine, []instance.Instance) {
var machines []*state.Machine
var insts []instance.Instance
for i := 0; i < 10; i++ {
m, err := s.State.AddMachine("series", state.JobHostUnits)
c.Assert(err, gc.IsNil)
machines = append(machines, m)
inst, _ := testing.AssertStartInstance(c, s.Conn.Environ, m.Id())
insts = append(insts, inst)
}
// Associate the odd-numbered machines with an instance.
for i := 1; i < len(machines); i += 2 {
m := machines[i]
err := m.SetProvisioned(insts[i].Id(), "nonce", nil)
c.Assert(err, gc.IsNil)
}
// Associate the first half of the instances with an address.
for i := 0; i < len(machines)/2; i++ {
dummy.SetInstanceAddresses(insts[i], s.addressesForIndex(i))
}
return machines, insts
}
|