// writeStateAgentConfig creates and writes a state agent config.
func writeStateAgentConfig(c *gc.C, stateInfo *state.Info, dataDir, tag, password string, vers version.Binary) agent.ConfigSetterWriter {
port := gitjujutesting.FindTCPPort()
apiAddr := []string{fmt.Sprintf("localhost:%d", port)}
conf, err := agent.NewStateMachineConfig(
agent.AgentConfigParams{
DataDir: dataDir,
Tag: tag,
UpgradedToVersion: vers.Number,
Password: password,
Nonce: state.BootstrapNonce,
StateAddresses: stateInfo.Addrs,
APIAddresses: apiAddr,
CACert: stateInfo.CACert,
},
params.StateServingInfo{
Cert: coretesting.ServerCert,
PrivateKey: coretesting.ServerKey,
StatePort: gitjujutesting.MgoServer.Port(),
APIPort: port,
})
c.Assert(err, gc.IsNil)
conf.SetPassword(password)
c.Assert(conf.Write(), gc.IsNil)
return conf
}
// WriteStateAgentConfig creates and writes a state agent config.
func (s *AgentSuite) WriteStateAgentConfig(
c *gc.C,
tag names.Tag,
password string,
vers version.Binary,
modelTag names.ModelTag,
) agent.ConfigSetterWriter {
stateInfo := s.State.MongoConnectionInfo()
apiPort := gitjujutesting.FindTCPPort()
apiAddr := []string{fmt.Sprintf("localhost:%d", apiPort)}
conf, err := agent.NewStateMachineConfig(
agent.AgentConfigParams{
Paths: agent.NewPathsWithDefaults(agent.Paths{DataDir: s.DataDir()}),
Tag: tag,
UpgradedToVersion: vers.Number,
Password: password,
Nonce: agent.BootstrapNonce,
StateAddresses: stateInfo.Addrs,
APIAddresses: apiAddr,
CACert: stateInfo.CACert,
Model: modelTag,
},
params.StateServingInfo{
Cert: coretesting.ServerCert,
PrivateKey: coretesting.ServerKey,
CAPrivateKey: coretesting.CAKey,
StatePort: gitjujutesting.MgoServer.Port(),
APIPort: apiPort,
})
c.Assert(err, jc.ErrorIsNil)
conf.SetPassword(password)
c.Assert(conf.Write(), gc.IsNil)
return conf
}
// writeStateAgentConfig creates and writes a state agent config.
func writeStateAgentConfig(
c *gc.C, stateInfo *mongo.MongoInfo, dataDir string, tag names.Tag,
password string, vers version.Binary, envTag names.EnvironTag) agent.ConfigSetterWriter {
port := gitjujutesting.FindTCPPort()
apiAddr := []string{fmt.Sprintf("localhost:%d", port)}
conf, err := agent.NewStateMachineConfig(
agent.AgentConfigParams{
Paths: agent.NewPathsWithDefaults(agent.Paths{DataDir: dataDir}),
Tag: tag,
UpgradedToVersion: vers.Number,
Password: password,
Nonce: agent.BootstrapNonce,
StateAddresses: stateInfo.Addrs,
APIAddresses: apiAddr,
CACert: stateInfo.CACert,
Environment: envTag,
},
params.StateServingInfo{
Cert: coretesting.ServerCert,
PrivateKey: coretesting.ServerKey,
CAPrivateKey: coretesting.CAKey,
StatePort: gitjujutesting.MgoServer.Port(),
APIPort: port,
})
c.Assert(err, jc.ErrorIsNil)
conf.SetPassword(password)
c.Assert(conf.Write(), gc.IsNil)
return conf
}