func (s *keyupdaterSuite) SetUpTest(c *gc.C) {
s.JujuConnSuite.SetUpTest(c)
var stateAPI *api.State
stateAPI, s.rawMachine = s.OpenAPIAsNewMachine(c)
c.Assert(stateAPI, gc.NotNil)
s.keyupdater = stateAPI.KeyUpdater()
c.Assert(s.keyupdater, gc.NotNil)
}
func (s *workerSuite) SetUpTest(c *gc.C) {
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, gc.IsNil)
var apiRoot *api.State
apiRoot, s.machine = s.OpenAPIAsNewMachine(c)
c.Assert(apiRoot, gc.NotNil)
s.keyupdaterApi = apiRoot.KeyUpdater()
c.Assert(s.keyupdaterApi, gc.NotNil)
}
