github.com/makyo/juju@v0.0.0-20160425123129-2608902037e9/api/export_test.go (about)

     1  // Copyright 2012, 2013 Canonical Ltd.
     2  // Licensed under the AGPLv3, see LICENCE file for details.
     3  
     4  package api
     5  
     6  import (
     7  	"github.com/juju/errors"
     8  	"github.com/juju/juju/api/base"
     9  	"github.com/juju/juju/network"
    10  )
    11  
    12  var (
    13  	CertDir               = &certDir
    14  	NewWebsocketDialer    = newWebsocketDialer
    15  	NewWebsocketDialerPtr = &newWebsocketDialer
    16  	WebsocketDialConfig   = &websocketDialConfig
    17  	SlideAddressToFront   = slideAddressToFront
    18  	BestVersion           = bestVersion
    19  	FacadeVersions        = &facadeVersions
    20  	ConnectWebsocket      = connectWebsocket
    21  )
    22  
    23  // SetServerAddress allows changing the URL to the internal API server
    24  // that AddLocalCharm uses in order to test NotImplementedError.
    25  func SetServerAddress(c *Client, scheme, addr string) {
    26  	c.st.serverScheme = scheme
    27  	c.st.addr = addr
    28  }
    29  
    30  // ServerRoot is exported so that we can test the built URL.
    31  func ServerRoot(c *Client) string {
    32  	return c.st.serverRoot()
    33  }
    34  
    35  // TestingStateParams is the parameters for NewTestingState, so that you can
    36  // only set the bits that you acutally want to test.
    37  type TestingStateParams struct {
    38  	Address        string
    39  	ModelTag       string
    40  	APIHostPorts   [][]network.HostPort
    41  	FacadeVersions map[string][]int
    42  	ServerScheme   string
    43  	ServerRoot     string
    44  }
    45  
    46  // NewTestingState creates an api.State object that can be used for testing. It
    47  // isn't backed onto an actual API server, so actual RPC methods can't be
    48  // called on it. But it can be used for testing general behavior.
    49  func NewTestingState(params TestingStateParams) Connection {
    50  	st := &state{
    51  		addr:              params.Address,
    52  		modelTag:          params.ModelTag,
    53  		hostPorts:         params.APIHostPorts,
    54  		facadeVersions:    params.FacadeVersions,
    55  		serverScheme:      params.ServerScheme,
    56  		serverRootAddress: params.ServerRoot,
    57  	}
    58  	return st
    59  }
    60  
    61  // PatchClientFacadeCall changes the internal FacadeCaller to one that lets
    62  // you mock out the FacadeCall method. The function returned by
    63  // PatchClientFacadeCall is a cleanup function that returns the client to its
    64  // original state.
    65  func PatchClientFacadeCall(c *Client, mockCall func(request string, params interface{}, response interface{}) error) func() {
    66  	orig := c.facade
    67  	c.facade = &resultCaller{mockCall}
    68  	return func() {
    69  		c.facade = orig
    70  	}
    71  }
    72  
    73  type resultCaller struct {
    74  	mockCall func(request string, params interface{}, response interface{}) error
    75  }
    76  
    77  func (f *resultCaller) FacadeCall(request string, params, response interface{}) error {
    78  	return f.mockCall(request, params, response)
    79  }
    80  
    81  func (f *resultCaller) Name() string {
    82  	return ""
    83  }
    84  
    85  func (f *resultCaller) BestAPIVersion() int {
    86  	return 0
    87  }
    88  
    89  func (f *resultCaller) RawAPICaller() base.APICaller {
    90  	return nil
    91  }
    92  
    93  // IsMinVersionError returns true if the given error was caused by the charm
    94  // having a minjujuversion higher than the juju environment's version.
    95  func IsMinVersionError(err error) bool {
    96  	_, ok := errors.Cause(err).(minJujuVersionErr)
    97  	return ok
    98  }