func (s *collectMetricsSuite) TestCollectMetricsFailsOnNonLocalCharm(c *gc.C) {
runClient := &testRunClient{}
serviceClient := &testServiceClient{}
serviceClient.charmURL = "cs:quantal/charm"
s.PatchValue(metricsdebug.NewAPIConn, noConn)
s.PatchValue(metricsdebug.NewRunClient, metricsdebug.NewRunClientFnc(runClient))
s.PatchValue(metricsdebug.NewServiceClient, metricsdebug.NewServiceClientFnc(serviceClient))
_, err := coretesting.RunCommand(c, metricsdebug.NewCollectMetricsCommand(), "foobar")
c.Assert(err, gc.ErrorMatches, `"foobar" is not a local charm`)
runClient.CheckCallNames(c, "Close")
}
func (s *collectMetricsSuite) TestCollectMetrics(c *gc.C) {
runClient := &testRunClient{}
serviceClient := &testServiceClient{}
serviceClient.charmURL = "local:quantal/charm"
s.PatchValue(metricsdebug.NewAPIConn, noConn)
s.PatchValue(metricsdebug.NewRunClient, metricsdebug.NewRunClientFnc(runClient))
s.PatchValue(metricsdebug.NewServiceClient, metricsdebug.NewServiceClientFnc(serviceClient))
actionTag1 := names.NewActionTag("01234567-89ab-cdef-0123-456789abcdef")
actionTag2 := names.NewActionTag("11234567-89ab-cdef-0123-456789abcdef")
tests := []struct {
about string
args []string
stdout string
results [][]params.ActionResult
actionMap map[string]params.ActionResult
err string
}{{
about: "missing args",
err: "you need to specify a unit or application.",
}, {
about: "invalid application name",
args: []string{"application_1-0"},
err: `"application_1-0" is not a valid unit or application`,
}, {
about: "all is well",
args: []string{"uptime"},
results: [][]params.ActionResult{
[]params.ActionResult{{
Action: ¶ms.Action{
Tag: actionTag1.String(),
},
}},
[]params.ActionResult{{
Action: ¶ms.Action{
Tag: actionTag2.String(),
},
}},
},
actionMap: map[string]params.ActionResult{
actionTag1.Id(): params.ActionResult{
Action: ¶ms.Action{
Tag: actionTag1.String(),
Receiver: "unit-uptime-0",
},
Output: map[string]interface{}{
"Stdout": "ok",
"Stderr": "",
},
},
actionTag2.Id(): params.ActionResult{
Action: ¶ms.Action{
Tag: actionTag2.String(),
},
Output: map[string]interface{}{
"Stdout": "ok",
"Stderr": "",
},
},
},
}, {
about: "invalid tag returned",
args: []string{"uptime"},
results: [][]params.ActionResult{
[]params.ActionResult{{
Action: ¶ms.Action{
Tag: "invalid",
},
}},
},
stdout: `failed to collect metrics: "invalid" is not a valid tag\n`,
}, {
about: "no action found",
args: []string{"uptime"},
results: [][]params.ActionResult{
[]params.ActionResult{{
Action: ¶ms.Action{
Tag: actionTag1.String(),
},
}},
},
stdout: "failed to collect metrics: plm\n",
}, {
about: "fail to parse result",
args: []string{"uptime"},
results: [][]params.ActionResult{
[]params.ActionResult{{
Action: ¶ms.Action{
Tag: actionTag1.String(),
},
}},
},
actionMap: map[string]params.ActionResult{
actionTag1.Id(): params.ActionResult{},
},
stdout: "failed to collect metrics: could not read stdout\n",
}, {
about: "no results on sendResults",
args: []string{"uptime"},
results: [][]params.ActionResult{
[]params.ActionResult{{
Action: ¶ms.Action{
Tag: actionTag1.String(),
},
}},
},
actionMap: map[string]params.ActionResult{
actionTag1.Id(): params.ActionResult{
Action: ¶ms.Action{
Tag: actionTag2.String(),
Receiver: "unit-uptime-0",
},
Output: map[string]interface{}{
"Stdout": "ok",
"Stderr": "",
},
},
},
stdout: "failed to send metrics for unit uptime/0: no results\n",
}, {
about: "too many sendResults",
args: []string{"uptime"},
results: [][]params.ActionResult{
[]params.ActionResult{{
Action: ¶ms.Action{
Tag: actionTag1.String(),
},
}},
[]params.ActionResult{{
Action: ¶ms.Action{
Tag: actionTag1.String(),
},
}, {
Action: ¶ms.Action{
Tag: actionTag2.String(),
},
}},
},
actionMap: map[string]params.ActionResult{
actionTag1.Id(): params.ActionResult{
Action: ¶ms.Action{
Tag: actionTag2.String(),
Receiver: "unit-uptime-0",
},
Output: map[string]interface{}{
"Stdout": "ok",
"Stderr": "",
},
},
},
stdout: "failed to send metrics for unit uptime/0\n",
}, {
about: "sendResults error",
args: []string{"uptime"},
results: [][]params.ActionResult{
[]params.ActionResult{{
Action: ¶ms.Action{
Tag: actionTag1.String(),
},
}},
[]params.ActionResult{{
Error: ¶ms.Error{
Message: "permission denied",
},
}},
},
actionMap: map[string]params.ActionResult{
actionTag1.Id(): params.ActionResult{
Action: ¶ms.Action{
Tag: actionTag2.String(),
Receiver: "unit-uptime-0",
},
Output: map[string]interface{}{
"Stdout": "ok",
"Stderr": "",
},
},
},
stdout: "failed to send metrics for unit uptime/0: permission denied\n",
}, {
about: "couldn't get sendResults action",
args: []string{"uptime"},
results: [][]params.ActionResult{
[]params.ActionResult{{
Action: ¶ms.Action{
Tag: actionTag1.String(),
},
}},
[]params.ActionResult{{
Action: ¶ms.Action{
Tag: actionTag2.String(),
},
}},
},
actionMap: map[string]params.ActionResult{
actionTag1.Id(): params.ActionResult{
Action: ¶ms.Action{
Tag: actionTag2.String(),
Receiver: "unit-uptime-0",
},
Output: map[string]interface{}{
"Stdout": "ok",
"Stderr": "",
},
},
},
stdout: "failed to send metrics for unit uptime/0: plm\n",
}, {
about: "couldn't parse sendResults action",
args: []string{"uptime"},
results: [][]params.ActionResult{
[]params.ActionResult{{
Action: ¶ms.Action{
Tag: actionTag1.String(),
},
}},
[]params.ActionResult{{
Action: ¶ms.Action{
Tag: actionTag2.String(),
},
}},
},
actionMap: map[string]params.ActionResult{
actionTag1.Id(): params.ActionResult{
Action: ¶ms.Action{
Tag: actionTag2.String(),
Receiver: "unit-uptime-0",
},
Output: map[string]interface{}{
"Stdout": "ok",
"Stderr": "",
},
},
actionTag2.Id(): params.ActionResult{},
},
stdout: "failed to send metrics for unit uptime/0: could not read stdout\n",
}, {
about: "sendResults action stderr",
args: []string{"uptime"},
results: [][]params.ActionResult{
[]params.ActionResult{{
Action: ¶ms.Action{
Tag: actionTag1.String(),
},
}},
[]params.ActionResult{{
Action: ¶ms.Action{
Tag: actionTag2.String(),
},
}},
},
actionMap: map[string]params.ActionResult{
actionTag1.Id(): params.ActionResult{
Action: ¶ms.Action{
Tag: actionTag2.String(),
Receiver: "unit-uptime-0",
},
Output: map[string]interface{}{
"Stdout": "ok",
"Stderr": "",
},
},
actionTag2.Id(): params.ActionResult{
Action: ¶ms.Action{
Tag: actionTag2.String(),
Receiver: "unit-uptime-0",
},
Output: map[string]interface{}{
"Stdout": "garbage",
"Stderr": "kek",
},
},
},
stdout: "failed to send metrics for unit uptime/0: kek\n",
}}
for i, test := range tests {
c.Logf("running test %d: %v", i, test.about)
runClient.reset()
if test.results != nil {
runClient.results = test.results
}
metricsdebug.PatchGetActionResult(s.PatchValue, test.actionMap)
ctx, err := coretesting.RunCommand(c, metricsdebug.NewCollectMetricsCommand(), test.args...)
if test.err != "" {
c.Assert(err, gc.ErrorMatches, test.err)
} else {
c.Assert(err, jc.ErrorIsNil)
c.Assert(coretesting.Stdout(ctx), gc.Matches, test.stdout)
}
}
}
func (s *collectMetricsSuite) TestCollectMetrics(c *gc.C) {
runClient := &testRunClient{}
cleanup := testing.PatchValue(metricsdebug.NewRunClient, metricsdebug.NewRunClientFnc(runClient))
defer cleanup()
tests := []struct {
about string
args []string
stdout string
results [][]params.RunResult
err string
}{{
about: "missing args",
err: "you need to specify a unit or service.",
}, {
about: "invalid service name",
args: []string{"service_1-0"},
err: `"service_1-0" is not a valid unit or service`,
}, {
about: "all is well",
args: []string{"uptime"},
results: [][]params.RunResult{
[]params.RunResult{{
UnitId: "uptime/0",
ExecResponse: exec.ExecResponse{
Stdout: []byte("ok"),
},
}},
[]params.RunResult{{
UnitId: "uptime/0",
ExecResponse: exec.ExecResponse{
Stdout: []byte("ok"),
},
}},
},
}, {
about: "fail to collect metrics",
args: []string{"wordpress"},
results: [][]params.RunResult{
[]params.RunResult{{
UnitId: "wordpress/0",
ExecResponse: exec.ExecResponse{
Stderr: []byte("nc: unix connect failed: No such file or directory"),
},
}},
},
stdout: "failed to collect metrics for unit wordpress/0: not a metered charm\n",
}, {
about: "fail to send metrics",
args: []string{"uptime"},
results: [][]params.RunResult{
[]params.RunResult{{
UnitId: "uptime/0",
ExecResponse: exec.ExecResponse{
Stdout: []byte("ok"),
},
}},
[]params.RunResult{{
UnitId: "uptime/0",
ExecResponse: exec.ExecResponse{
Stderr: []byte("an embarrassing error"),
},
}},
},
stdout: "failed to send metrics for unit uptime/0: an embarrassing error\n",
},
}
for i, test := range tests {
c.Logf("running test %d: %v", i, test.about)
runClient.reset()
if test.results != nil {
runClient.results = test.results
}
ctx, err := coretesting.RunCommand(c, metricsdebug.NewCollectMetricsCommand(), test.args...)
if test.err != "" {
c.Assert(err, gc.ErrorMatches, test.err)
} else {
c.Assert(err, jc.ErrorIsNil)
c.Assert(coretesting.Stdout(ctx), gc.Matches, test.stdout)
}
}
}
