func (s *workerSuite) SetUpTest(c *gc.C) {
//TODO(bogdanteleaga): Fix this on windows
if runtime.GOOS == "windows" {
c.Skip("bug 1403084: authentication worker not implemented yet on windows")
}
s.JujuConnSuite.SetUpTest(c)
// Default ssh user is currently "ubuntu".
c.Assert(authenticationworker.SSHUser, gc.Equals, "ubuntu")
// Set the ssh user to empty (the current user) as required by the test infrastructure.
s.PatchValue(&authenticationworker.SSHUser, "")
// Replace the default dummy key in the test environment with a valid one.
// This will be added to the ssh authorised keys when the agent starts.
s.setAuthorisedKeys(c, sshtesting.ValidKeyOne.Key+" firstuser@host")
// Record the existing key with its prefix for testing later.
s.existingEnvKey = sshtesting.ValidKeyOne.Key + " Juju:firstuser@host"
// Set up an existing key (which is not in the environment) in the ssh authorised_keys file.
s.existingKeys = []string{sshtesting.ValidKeyTwo.Key + " existinguser@host"}
err := ssh.AddKeys(authenticationworker.SSHUser, s.existingKeys...)
c.Assert(err, jc.ErrorIsNil)
var apiRoot api.Connection
apiRoot, s.machine = s.OpenAPIAsNewMachine(c)
c.Assert(apiRoot, gc.NotNil)
s.keyupdaterApi = apiRoot.KeyUpdater()
c.Assert(s.keyupdaterApi, gc.NotNil)
}
func (s *loginSuite) assertRemoteModel(c *gc.C, api api.Connection, expected names.ModelTag) {
// Look at what the api thinks it has.
tag, ok := api.ModelTag()
c.Assert(ok, jc.IsTrue)
c.Assert(tag, gc.Equals, expected)
// Look at what the api Client thinks it has.
client := api.Client()
// ModelUUID looks at the env tag on the api state connection.
uuid, ok := client.ModelUUID()
c.Assert(ok, jc.IsTrue)
c.Assert(uuid, gc.Equals, expected.Id())
// The code below is to verify that the API connection is operating on
// the expected model. We make a change in state on that model, and
// then check that it is picked up by a call to the API.
st, err := s.State.ForModel(tag)
c.Assert(err, jc.ErrorIsNil)
defer st.Close()
expectedCons := constraints.MustParse("mem=8G")
err = st.SetModelConstraints(expectedCons)
c.Assert(err, jc.ErrorIsNil)
cons, err := client.GetModelConstraints()
c.Assert(err, jc.ErrorIsNil)
c.Assert(cons, jc.DeepEquals, expectedCons)
}
func opClientWatchAll(c *gc.C, st api.Connection, mst *state.State) (func(), error) {
watcher, err := st.Client().WatchAll()
if err == nil {
watcher.Stop()
}
return func() {}, err
}
// updateAllMachines finds all machines and resets the stored state address
// in each of them. The address does not include the port.
func updateAllMachines(apiState api.Connection, stateAddr string) ([]restoreResult, error) {
client := apiState.Client()
status, err := client.Status(nil)
if err != nil {
return nil, errors.Annotate(err, "cannot get status")
}
pendingMachineCount := 0
done := make(chan restoreResult)
for _, machineStatus := range status.Machines {
// A newly resumed state server requires no updating, and more
// than one state server is not yet support by this plugin.
if machineStatus.HasVote || machineStatus.WantsVote || machineStatus.Life == "dead" {
continue
}
pendingMachineCount++
machine := machineStatus
go func() {
err := runMachineUpdate(client, machine.Id, setAgentAddressScript(stateAddr))
if err != nil {
logger.Errorf("failed to update machine %s: %v", machine.Id, err)
} else {
progress("updated machine %s", machine.Id)
}
r := restoreResult{machineName: machine.Id, err: err}
done <- r
}()
}
results := make([]restoreResult, pendingMachineCount)
for ; pendingMachineCount > 0; pendingMachineCount-- {
results[pendingMachineCount-1] = <-done
}
return results, nil
}
func opClientAddServiceUnits(c *gc.C, st api.Connection, mst *state.State) (func(), error) {
_, err := st.Client().AddServiceUnits("nosuch", 1, "")
if params.IsCodeNotFound(err) {
err = nil
}
return func() {}, err
}
func opClientServiceUnexpose(c *gc.C, st api.Connection, mst *state.State) (func(), error) {
err := st.Client().ServiceUnexpose("wordpress")
if err != nil {
return func() {}, err
}
return func() {}, nil
}
func opClientServiceDestroy(c *gc.C, st api.Connection, mst *state.State) (func(), error) {
err := st.Client().ServiceDestroy("non-existent")
if params.IsCodeNotFound(err) {
err = nil
}
return func() {}, err
}
func opClientServiceDeployWithNetworks(c *gc.C, st api.Connection, mst *state.State) (func(), error) {
err := st.Client().ServiceDeployWithNetworks("mad:bad/url-1", "x", 1, "", constraints.Value{}, "", nil)
if err.Error() == `charm or bundle URL has invalid schema: "mad:bad/url-1"` {
err = nil
}
return func() {}, err
}
func opClientDestroyRelation(c *gc.C, st api.Connection, mst *state.State) (func(), error) {
err := st.Client().DestroyRelation("nosuch1", "nosuch2")
if params.IsCodeNotFound(err) {
err = nil
}
return func() {}, err
}
func restoreBootstrapMachine(st api.Connection, backupFile string, agentConf agentConfig) (addr string, err error) {
client := st.Client()
addr, err = client.PublicAddress("0")
if err != nil {
return "", errors.Annotate(err, "cannot get public address of bootstrap machine")
}
paddr, err := client.PrivateAddress("0")
if err != nil {
return "", errors.Annotate(err, "cannot get private address of bootstrap machine")
}
status, err := client.Status(nil)
if err != nil {
return "", errors.Annotate(err, "cannot get environment status")
}
info, ok := status.Machines["0"]
if !ok {
return "", fmt.Errorf("cannot find bootstrap machine in status")
}
newInstId := instance.Id(info.InstanceId)
progress("copying backup file to bootstrap host")
if err := sendViaScp(backupFile, addr, "~/juju-backup.tgz"); err != nil {
return "", errors.Annotate(err, "cannot copy backup file to bootstrap instance")
}
progress("updating bootstrap machine")
if err := runViaSsh(addr, updateBootstrapMachineScript(newInstId, agentConf, addr, paddr)); err != nil {
return "", errors.Annotate(err, "update script failed")
}
return addr, nil
}
func opClientDestroyServiceUnits(c *gc.C, st api.Connection, mst *state.State) (func(), error) {
err := st.Client().DestroyServiceUnits("wordpress/99")
if err != nil && strings.HasPrefix(err.Error(), "no units were destroyed") {
err = nil
}
return func() {}, err
}
func opClientServiceSetCharm(c *gc.C, st api.Connection, mst *state.State) (func(), error) {
err := st.Client().ServiceSetCharm("nosuch", "local:quantal/wordpress", false)
if params.IsCodeNotFound(err) {
err = nil
}
return func() {}, err
}
func opClientServiceSetYAML(c *gc.C, st api.Connection, mst *state.State) (func(), error) {
err := st.Client().ServiceSetYAML("wordpress", `"wordpress": {"blog-title": "foo"}`)
if err != nil {
return func() {}, err
}
return resetBlogTitle(c, st), nil
}
func opClientEnvironmentGet(c *gc.C, st api.Connection, mst *state.State) (func(), error) {
_, err := st.Client().EnvironmentGet()
if err != nil {
return func() {}, err
}
return func() {}, nil
}
func opClientGetAnnotations(c *gc.C, st api.Connection, mst *state.State) (func(), error) {
ann, err := st.Client().GetAnnotations("service-wordpress")
if err != nil {
return func() {}, err
}
c.Assert(ann, gc.DeepEquals, make(map[string]string))
return func() {}, nil
}
func (s *loggerSuite) SetUpTest(c *gc.C) {
s.JujuConnSuite.SetUpTest(c)
var stateAPI api.Connection
stateAPI, s.rawMachine = s.OpenAPIAsNewMachine(c)
// Create the logger facade.
s.logger = stateAPI.Logger()
c.Assert(s.logger, gc.NotNil)
}
func (s *keyupdaterSuite) SetUpTest(c *gc.C) {
s.JujuConnSuite.SetUpTest(c)
var stateAPI api.Connection
stateAPI, s.rawMachine = s.OpenAPIAsNewMachine(c)
c.Assert(stateAPI, gc.NotNil)
s.keyupdater = stateAPI.KeyUpdater()
c.Assert(s.keyupdater, gc.NotNil)
}
func resetBlogTitle(c *gc.C, st api.Connection) func() {
return func() {
err := st.Client().ServiceSet("wordpress", map[string]string{
"blog-title": "",
})
c.Assert(err, jc.ErrorIsNil)
}
}
func opClientSetServiceConstraints(c *gc.C, st api.Connection, mst *state.State) (func(), error) {
nullConstraints := constraints.Value{}
err := st.Client().SetServiceConstraints("wordpress", nullConstraints)
if err != nil {
return func() {}, err
}
return func() {}, nil
}
func opClientSetEnvironmentConstraints(c *gc.C, st api.Connection, mst *state.State) (func(), error) {
nullConstraints := constraints.Value{}
err := st.Client().SetModelConstraints(nullConstraints)
if err != nil {
return func() {}, err
}
return func() {}, nil
}
func opClientServiceSet(c *gc.C, st api.Connection, mst *state.State) (func(), error) {
err := st.Client().ServiceSet("wordpress", map[string]string{
"blog-title": "foo",
})
if err != nil {
return func() {}, err
}
return resetBlogTitle(c, st), nil
}
// checkAvailable ensures the Juju GUI is available on the controller at the
// given URL.
func (c *guiCommand) checkAvailable(rawURL string, conn api.Connection) error {
client, err := conn.HTTPClient()
if err != nil {
return errors.Annotate(err, "cannot retrieve HTTP client")
}
if err = clientGet(client, rawURL); err != nil {
return errors.Annotate(err, "Juju GUI is not available")
}
return nil
}
func opClientStatus(c *gc.C, st api.Connection, mst *state.State) (func(), error) {
status, err := st.Client().Status(nil)
if err != nil {
c.Check(status, gc.IsNil)
return func() {}, err
}
clearSinceTimes(status)
c.Assert(status, jc.DeepEquals, scenarioStatus)
return func() {}, nil
}
func opClientServiceExpose(c *gc.C, st api.Connection, mst *state.State) (func(), error) {
err := st.Client().ServiceExpose("wordpress")
if err != nil {
return func() {}, err
}
return func() {
svc, err := mst.Service("wordpress")
c.Assert(err, jc.ErrorIsNil)
svc.ClearExposed()
}, nil
}
func newAPIClient(apiCaller api.Connection) (*client.Client, error) {
caller := base.NewFacadeCallerForVersion(apiCaller, resource.ComponentName, server.Version)
httpClient, err := apiCaller.HTTPClient()
if err != nil {
return nil, errors.Trace(err)
}
// The apiCaller takes care of prepending /environment/<envUUID>.
apiClient := client.NewClient(caller, httpClient, apiCaller)
return apiClient, nil
}
func opClientEnvironmentSet(c *gc.C, st api.Connection, mst *state.State) (func(), error) {
args := map[string]interface{}{"some-key": "some-value"}
err := st.Client().EnvironmentSet(args)
if err != nil {
return func() {}, err
}
return func() {
args["some-key"] = nil
st.Client().EnvironmentSet(args)
}, nil
}
func opClientSetAnnotations(c *gc.C, st api.Connection, mst *state.State) (func(), error) {
pairs := map[string]string{"key1": "value1", "key2": "value2"}
err := st.Client().SetAnnotations("service-wordpress", pairs)
if err != nil {
return func() {}, err
}
return func() {
pairs := map[string]string{"key1": "", "key2": ""}
st.Client().SetAnnotations("service-wordpress", pairs)
}, nil
}
// uninstallerManifold defines a simple start function which retrieves
// some dependencies, checks if the machine is dead and causes the
// agent to uninstall itself if it is. This doubles up on part of the
// machiner's functionality but the machiner doesn't run until
// upgrades are complete, and the upgrade related workers may not be
// able to make API requests if the machine is dead.
func uninstallerManifold(config uninstallerManifoldConfig) dependency.Manifold {
return dependency.Manifold{
Inputs: []string{
config.AgentName,
config.APICallerName,
},
Start: func(getResource dependency.GetResourceFunc) (worker.Worker, error) {
if config.WriteUninstallFile == nil {
return nil, errors.New("WriteUninstallFile not specified")
}
// Get the agent.
var agent agent.Agent
if err := getResource(config.AgentName, &agent); err != nil {
return nil, err
}
// Grab the tag and ensure that it's for a machine.
tag, ok := agent.CurrentConfig().Tag().(names.MachineTag)
if !ok {
return nil, errors.New("agent's tag is not a machine tag")
}
// Get API connection.
//
// TODO(mjs) - this should really be a base.APICaller to
// remove the possibility of the API connection being closed
// here.
var apiConn api.Connection
if err := getResource(config.APICallerName, &apiConn); err != nil {
return nil, err
}
// Check if the machine is dead and set the agent to
// uninstall if it is.
//
// TODO(mjs) - ideally this would be using its own facade.
machine, err := apiConn.Agent().Entity(tag)
if err != nil {
return nil, err
}
if machine.Life() == params.Dead {
if err := config.WriteUninstallFile(); err != nil {
return nil, errors.Annotate(err, "writing uninstall agent file")
}
return nil, worker.ErrTerminateAgent
}
// All is well - we're done (no actual worker is actually returned).
return nil, dependency.ErrUninstall
},
}
}
func (s *pingerSuite) checkConnectionDies(c *gc.C, conn api.Connection) {
attempt := utils.AttemptStrategy{
Total: coretesting.LongWait,
Delay: coretesting.ShortWait,
}
for a := attempt.Start(); a.Next(); {
err := conn.Ping()
if err != nil {
c.Assert(err, gc.ErrorMatches, "connection is shut down")
return
}
}
c.Fatal("connection didn't get shut down")
}
func opClientServiceUpdate(c *gc.C, st api.Connection, mst *state.State) (func(), error) {
args := params.ServiceUpdate{
ServiceName: "no-such-charm",
CharmUrl: "cs:quantal/wordpress-42",
ForceCharmUrl: true,
SettingsStrings: map[string]string{"blog-title": "foo"},
SettingsYAML: `"wordpress": {"blog-title": "foo"}`,
}
err := st.Client().ServiceUpdate(args)
if params.IsCodeNotFound(err) {
err = nil
}
return func() {}, err
}
