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
|
// Copyright 2015 Canonical Ltd.
// Licensed under the AGPLv3, see LICENCE file for details.
package undertaker_test
import (
"github.com/juju/testing"
jc "github.com/juju/testing/checkers"
gc "gopkg.in/check.v1"
"github.com/juju/juju/apiserver/params"
"github.com/juju/juju/environs"
"github.com/juju/juju/status"
"github.com/juju/juju/watcher"
"github.com/juju/juju/worker"
"github.com/juju/juju/worker/undertaker"
"github.com/juju/juju/worker/workertest"
)
type mockFacade struct {
stub *testing.Stub
info params.UndertakerModelInfoResult
}
func (mock *mockFacade) ModelInfo() (params.UndertakerModelInfoResult, error) {
mock.stub.AddCall("ModelInfo")
if err := mock.stub.NextErr(); err != nil {
return params.UndertakerModelInfoResult{}, err
}
return mock.info, nil
}
func (mock *mockFacade) WatchModelResources() (watcher.NotifyWatcher, error) {
mock.stub.AddCall("WatchModelResources")
if err := mock.stub.NextErr(); err != nil {
return nil, err
}
const count = 5
changes := make(chan struct{}, count)
for i := 0; i < count; i++ {
changes <- struct{}{}
}
return &mockWatcher{
Worker: workertest.NewErrorWorker(nil),
changes: changes,
}, nil
}
func (mock *mockFacade) ProcessDyingModel() error {
mock.stub.AddCall("ProcessDyingModel")
return mock.stub.NextErr()
}
func (mock *mockFacade) SetStatus(status status.Status, info string, data map[string]interface{}) error {
mock.stub.MethodCall(mock, "SetStatus", status, info, data)
return mock.stub.NextErr()
}
func (mock *mockFacade) RemoveModel() error {
mock.stub.AddCall("RemoveModel")
return mock.stub.NextErr()
}
type mockEnviron struct {
environs.Environ
stub *testing.Stub
}
func (mock *mockEnviron) Destroy() error {
mock.stub.AddCall("Destroy")
return mock.stub.NextErr()
}
type mockWatcher struct {
worker.Worker
changes chan struct{}
}
func (mock *mockWatcher) Changes() watcher.NotifyChannel {
return mock.changes
}
type fixture struct {
info params.UndertakerModelInfoResult
errors []error
dirty bool
}
func (fix fixture) cleanup(c *gc.C, w worker.Worker) {
if fix.dirty {
workertest.DirtyKill(c, w)
} else {
workertest.CleanKill(c, w)
}
}
func (fix fixture) run(c *gc.C, test func(worker.Worker)) *testing.Stub {
stub := &testing.Stub{}
environ := &mockEnviron{
stub: stub,
}
facade := &mockFacade{
stub: stub,
info: fix.info,
}
stub.SetErrors(fix.errors...)
w, err := undertaker.NewUndertaker(undertaker.Config{
Facade: facade,
Environ: environ,
})
c.Assert(err, jc.ErrorIsNil)
defer fix.cleanup(c, w)
test(w)
return stub
}
|