func (s *RsyslogSuite) TestModeForwarding(c *gc.C) {
err := s.APIState.Client().EnvironmentSet(map[string]interface{}{"rsyslog-ca-cert": coretesting.CACert})
c.Assert(err, gc.IsNil)
st, m := s.OpenAPIAsNewMachine(c, state.JobHostUnits)
addrs := []string{"0.1.2.3", "0.2.4.6"}
worker, err := rsyslog.NewRsyslogConfigWorker(st.Rsyslog(), rsyslog.RsyslogModeForwarding, m.Tag().String(), "", addrs)
c.Assert(err, gc.IsNil)
defer func() { c.Assert(worker.Wait(), gc.IsNil) }()
defer worker.Kill()
// We should get a ca-cert.pem with the contents introduced into state config.
waitForFile(c, filepath.Join(*rsyslog.LogDir, "ca-cert.pem"))
caCertPEM, err := ioutil.ReadFile(filepath.Join(*rsyslog.LogDir, "ca-cert.pem"))
c.Assert(err, gc.IsNil)
c.Assert(string(caCertPEM), gc.DeepEquals, coretesting.CACert)
// Verify rsyslog configuration.
waitForFile(c, filepath.Join(*rsyslog.RsyslogConfDir, "25-juju.conf"))
rsyslogConf, err := ioutil.ReadFile(filepath.Join(*rsyslog.RsyslogConfDir, "25-juju.conf"))
c.Assert(err, gc.IsNil)
syslogPort := s.Conn.Environ.Config().SyslogPort()
syslogConfig := syslog.NewForwardConfig(m.Tag().String(), *rsyslog.LogDir, syslogPort, "", addrs)
syslogConfig.ConfigDir = *rsyslog.RsyslogConfDir
rendered, err := syslogConfig.Render()
c.Assert(err, gc.IsNil)
c.Assert(string(rsyslogConf), gc.DeepEquals, string(rendered))
}
func (s *syslogConfigSuite) TestForwardConfigRender(c *gc.C) {
cfg := cfg()
syslog.NewForwardConfig(cfg)
s.assertRsyslogConfigContents(
c,
cfg,
syslogtesting.ExpectedForwardSyslogConf(c, args()),
)
}
func (s *syslogConfigSuite) TestForwardConfigRenderWithNamespace(c *gc.C) {
syslogConfigRenderer := syslog.NewForwardConfig(
"some-machine", agent.DefaultLogDir, 999, "namespace", []string{"server"},
)
s.assertRsyslogConfigContents(
c, syslogConfigRenderer, syslogtesting.ExpectedForwardSyslogConf(
c, "some-machine", agent.DefaultLogDir, "namespace", "server", 999,
),
)
}
func (s *syslogConfigSuite) TestForwardConfigRenderWithNamespace(c *gc.C) {
cfg := cfg()
cfg.Namespace = "namespace"
args := args()
args.Namespace = "namespace"
syslog.NewForwardConfig(cfg)
s.assertRsyslogConfigContents(
c,
cfg,
syslogtesting.ExpectedForwardSyslogConf(c, args),
)
}
func newRsyslogConfigHandler(st *apirsyslog.State, mode RsyslogMode, tag names.Tag, namespace string, stateServerAddrs []string, jujuConfigDir string) (*RsyslogConfigHandler, error) {
if namespace != "" {
jujuConfigDir += "-" + namespace
}
jujuConfigDir = filepath.Join(jujuConfigDir, "rsyslog")
if err := os.MkdirAll(jujuConfigDir, 0755); err != nil {
return nil, errors.Trace(err)
}
syslogConfig := &syslog.SyslogConfig{
LogFileName: tag.String(),
LogDir: logDir,
JujuConfigDir: jujuConfigDir,
Port: 0,
Namespace: namespace,
StateServerAddresses: stateServerAddrs,
}
if mode == RsyslogModeAccumulate {
syslog.NewAccumulateConfig(syslogConfig)
} else {
syslog.NewForwardConfig(syslogConfig)
}
// Historically only machine-0 includes the namespace in the log
// dir/file; for backwards compatibility we continue the tradition.
if tag != names.NewMachineTag("0") {
namespace = ""
}
switch tag := tag.(type) {
case names.MachineTag:
if namespace == "" {
syslogConfig.ConfigFileName = "25-juju.conf"
} else {
syslogConfig.ConfigFileName = fmt.Sprintf("25-juju-%s.conf", namespace)
}
default:
syslogConfig.ConfigFileName = fmt.Sprintf("26-juju-%s.conf", tag)
}
syslogConfig.ConfigDir = rsyslogConfDir
syslogConfig.LogDir = logDir
if namespace != "" {
syslogConfig.LogDir += "-" + namespace
}
return &RsyslogConfigHandler{
st: st,
mode: mode,
syslogConfig: syslogConfig,
tag: tag,
}, nil
}
func (s *syslogConfigSuite) TestForwardConfigWrite(c *gc.C) {
syslogConfigRenderer := cfg()
syslogConfigRenderer.ConfigDir = s.configDir
syslogConfigRenderer.ConfigFileName = "rsyslog.conf"
syslog.NewForwardConfig(syslogConfigRenderer)
s.assertRsyslogConfigPath(c, syslogConfigRenderer)
err := syslogConfigRenderer.Write()
c.Assert(err, jc.ErrorIsNil)
syslogConfData, err := ioutil.ReadFile(syslogConfigRenderer.ConfigFilePath())
c.Assert(err, jc.ErrorIsNil)
c.Assert(
string(syslogConfData),
gc.Equals,
syslogtesting.ExpectedForwardSyslogConf(c, args()),
)
}
func newRsyslogConfigHandler(st *apirsyslog.State, mode RsyslogMode, tag, namespace string, stateServerAddrs []string) (*RsyslogConfigHandler, error) {
var syslogConfig *syslog.SyslogConfig
if mode == RsyslogModeAccumulate {
syslogConfig = syslog.NewAccumulateConfig(
tag, logDir, 0, namespace, stateServerAddrs,
)
} else {
syslogConfig = syslog.NewForwardConfig(
tag, logDir, 0, namespace, stateServerAddrs,
)
}
// Historically only machine-0 includes the namespace in the log
// dir/file; for backwards compatibility we continue the tradition.
if tag != "machine-0" {
namespace = ""
}
kind, err := names.TagKind(tag)
if err != nil {
return nil, err
}
if kind == names.MachineTagKind {
if namespace == "" {
syslogConfig.ConfigFileName = "25-juju.conf"
} else {
syslogConfig.ConfigFileName = fmt.Sprintf("25-juju-%s.conf", namespace)
}
} else {
syslogConfig.ConfigFileName = fmt.Sprintf("26-juju-%s.conf", tag)
}
syslogConfig.ConfigDir = rsyslogConfDir
syslogConfig.LogDir = logDir
if namespace != "" {
syslogConfig.LogDir += "-" + namespace
}
return &RsyslogConfigHandler{
st: st,
mode: mode,
syslogConfig: syslogConfig,
tag: tag,
}, nil
}
func (s *syslogConfigSuite) TestForwardConfigWrite(c *gc.C) {
syslogConfigRenderer := syslog.NewForwardConfig(
"some-machine", agent.DefaultLogDir, 999, "", []string{"server"},
)
syslogConfigRenderer.ConfigDir = s.configDir
syslogConfigRenderer.ConfigFileName = "rsyslog.conf"
s.assertRsyslogConfigPath(c, syslogConfigRenderer)
err := syslogConfigRenderer.Write()
c.Assert(err, gc.IsNil)
syslogConfData, err := ioutil.ReadFile(syslogConfigRenderer.ConfigFilePath())
c.Assert(err, gc.IsNil)
c.Assert(
string(syslogConfData),
gc.Equals,
syslogtesting.ExpectedForwardSyslogConf(
c, "some-machine", agent.DefaultLogDir, "", "server", 999,
),
)
}
func newRsyslogConfigHandler(st *apirsyslog.State, mode RsyslogMode, tag names.Tag, namespace string, stateServerAddrs []string) (*RsyslogConfigHandler, error) {
var syslogConfig *syslog.SyslogConfig
if mode == RsyslogModeAccumulate {
syslogConfig = syslog.NewAccumulateConfig(tag.String(), logDir, 0, namespace, stateServerAddrs)
} else {
syslogConfig = syslog.NewForwardConfig(tag.String(), logDir, 0, namespace, stateServerAddrs)
}
// Historically only machine-0 includes the namespace in the log
// dir/file; for backwards compatibility we continue the tradition.
if tag != names.NewMachineTag("0") {
namespace = ""
}
switch tag := tag.(type) {
case names.MachineTag:
if namespace == "" {
syslogConfig.ConfigFileName = "25-juju.conf"
} else {
syslogConfig.ConfigFileName = fmt.Sprintf("25-juju-%s.conf", namespace)
}
default:
syslogConfig.ConfigFileName = fmt.Sprintf("26-juju-%s.conf", tag)
}
syslogConfig.ConfigDir = rsyslogConfDir
syslogConfig.LogDir = logDir
if namespace != "" {
syslogConfig.LogDir += "-" + namespace
}
return &RsyslogConfigHandler{
st: st,
mode: mode,
syslogConfig: syslogConfig,
tag: tag,
}, nil
}