forked from juju/juju
-
Notifications
You must be signed in to change notification settings - Fork 0
/
utils_test.go
99 lines (84 loc) · 2.3 KB
/
utils_test.go
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
// Copyright 2015 Canonical Ltd.
// Licensed under the AGPLv3, see LICENCE file for details.
package apiserver
import (
jc "github.com/juju/testing/checkers"
gc "gopkg.in/check.v1"
"github.com/juju/juju/state"
"github.com/juju/juju/state/testing"
)
type utilsSuite struct {
testing.StateSuite
pool *state.StatePool
}
var _ = gc.Suite(&utilsSuite{})
func (s *utilsSuite) SetUpTest(c *gc.C) {
s.StateSuite.SetUpTest(c)
s.pool = state.NewStatePool(s.State)
s.AddCleanup(func(*gc.C) { s.pool.Close() })
}
func (s *utilsSuite) TestValidateEmpty(c *gc.C) {
uuid, err := validateModelUUID(
validateArgs{
statePool: s.pool,
})
c.Assert(err, jc.ErrorIsNil)
c.Assert(uuid, gc.Equals, s.State.ModelUUID())
}
func (s *utilsSuite) TestValidateEmptyStrict(c *gc.C) {
_, err := validateModelUUID(
validateArgs{
statePool: s.pool,
strict: true,
})
c.Assert(err, gc.ErrorMatches, `unknown model: ""`)
}
func (s *utilsSuite) TestValidateController(c *gc.C) {
uuid, err := validateModelUUID(
validateArgs{
statePool: s.pool,
modelUUID: s.State.ModelUUID(),
})
c.Assert(err, jc.ErrorIsNil)
c.Assert(uuid, gc.Equals, s.State.ModelUUID())
}
func (s *utilsSuite) TestValidateControllerStrict(c *gc.C) {
uuid, err := validateModelUUID(
validateArgs{
statePool: s.pool,
modelUUID: s.State.ModelUUID(),
strict: true,
})
c.Assert(err, jc.ErrorIsNil)
c.Assert(uuid, gc.Equals, s.State.ModelUUID())
}
func (s *utilsSuite) TestValidateBadModelUUID(c *gc.C) {
_, err := validateModelUUID(
validateArgs{
statePool: s.pool,
modelUUID: "bad",
})
c.Assert(err, gc.ErrorMatches, `unknown model: "bad"`)
}
func (s *utilsSuite) TestValidateOtherModel(c *gc.C) {
envState := s.Factory.MakeModel(c, nil)
defer envState.Close()
uuid, err := validateModelUUID(
validateArgs{
statePool: s.pool,
modelUUID: envState.ModelUUID(),
})
c.Assert(err, jc.ErrorIsNil)
c.Assert(uuid, gc.Equals, envState.ModelUUID())
}
func (s *utilsSuite) TestValidateOtherModelControllerOnly(c *gc.C) {
envState := s.Factory.MakeModel(c, nil)
defer envState.Close()
_, err := validateModelUUID(
validateArgs{
statePool: s.pool,
modelUUID: envState.ModelUUID(),
controllerModelOnly: true,
})
c.Assert(err, gc.ErrorMatches, `requested model ".*" is not the controller model`)
}