/
export_test.go
116 lines (101 loc) · 3.34 KB
/
export_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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
// Copyright 2012, 2013 Canonical Ltd.
// Licensed under the AGPLv3, see LICENCE file for details.
package api
import (
"github.com/juju/errors"
"github.com/juju/juju/api/base"
"github.com/juju/juju/network"
"github.com/juju/utils/clock"
"gopkg.in/juju/names.v2"
)
var (
CertDir = &certDir
NewWebsocketDialer = newWebsocketDialer
WebsocketDialConfig = &websocketDialConfig
SlideAddressToFront = slideAddressToFront
BestVersion = bestVersion
FacadeVersions = &facadeVersions
DialAPI = dialAPI
)
// RPCConnection defines the methods that are called on the rpc.Conn instance.
type RPCConnection rpcConnection
// SetServerAddress allows changing the URL to the internal API server
// that AddLocalCharm uses in order to test NotImplementedError.
func SetServerAddress(c *Client, scheme, addr string) {
c.st.serverScheme = scheme
c.st.addr = addr
}
// ServerRoot is exported so that we can test the built URL.
func ServerRoot(c *Client) string {
return c.st.serverRoot()
}
// TestingStateParams is the parameters for NewTestingState, so that you can
// only set the bits that you acutally want to test.
type TestingStateParams struct {
Address string
ModelTag string
APIHostPorts [][]network.HostPort
FacadeVersions map[string][]int
ServerScheme string
ServerRoot string
RPCConnection RPCConnection
Clock clock.Clock
Broken chan struct{}
}
// NewTestingState creates an api.State object that can be used for testing. It
// isn't backed onto an actual API server, so actual RPC methods can't be
// called on it. But it can be used for testing general behaviour.
func NewTestingState(params TestingStateParams) Connection {
var modelTag names.ModelTag
if params.ModelTag != "" {
t, err := names.ParseModelTag(params.ModelTag)
if err != nil {
panic("invalid model tag")
}
modelTag = t
}
st := &state{
client: params.RPCConnection,
clock: params.Clock,
addr: params.Address,
modelTag: modelTag,
hostPorts: params.APIHostPorts,
facadeVersions: params.FacadeVersions,
serverScheme: params.ServerScheme,
serverRootAddress: params.ServerRoot,
broken: params.Broken,
}
return st
}
// PatchClientFacadeCall changes the internal FacadeCaller to one that lets
// you mock out the FacadeCall method. The function returned by
// PatchClientFacadeCall is a cleanup function that returns the client to its
// original state.
func PatchClientFacadeCall(c *Client, mockCall func(request string, params interface{}, response interface{}) error) func() {
orig := c.facade
c.facade = &resultCaller{mockCall}
return func() {
c.facade = orig
}
}
type resultCaller struct {
mockCall func(request string, params interface{}, response interface{}) error
}
func (f *resultCaller) FacadeCall(request string, params, response interface{}) error {
return f.mockCall(request, params, response)
}
func (f *resultCaller) Name() string {
return ""
}
func (f *resultCaller) BestAPIVersion() int {
return 0
}
func (f *resultCaller) RawAPICaller() base.APICaller {
return nil
}
// IsMinVersionError returns true if the given error was caused by the charm
// having a minjujuversion higher than the juju environment's version.
func IsMinVersionError(err error) bool {
_, ok := errors.Cause(err).(minJujuVersionErr)
return ok
}