"README.md" did not exist on "adec1fc215cfd92d7dcfeb458658bca0e09092fd"
Unverified Commit f74bd99b authored by Drew Bailey's avatar Drew Bailey
Browse files

monitor command takes no args

rm extra new line

fix lint errors

return after close

fix, simplify test
Showing with 20 additions and 18 deletions
+20 -18
......@@ -244,7 +244,6 @@ type MonitorFrame struct {
// Monitor returns a channel which will receive streaming logs from the agent
// Providing a non-nil stopCh can be used to close the connection and stop log streaming
func (a *Agent) Monitor(stopCh <-chan struct{}, q *QueryOptions) (<-chan *StreamFrame, <-chan error) {
errCh := make(chan error, 1)
r, err := a.client.newRequest("GET", "/v1/agent/monitor")
if err != nil {
......
package api
import (
"bytes"
"fmt"
"reflect"
"sort"
......@@ -320,7 +319,6 @@ func TestAgent_MonitorWithNode(t *testing.T) {
require.NoError(t, err)
// Wait for a log message
var result bytes.Buffer
OUTER:
for {
select {
......
......@@ -29,10 +29,6 @@ func NewAgentEndpoint(c *Client) *Agent {
return m
}
type monitorFrame struct {
Data []byte `json:",omitempty"`
}
func (m *Agent) monitor(conn io.ReadWriteCloser) {
defer metrics.MeasureSince([]string{"client", "monitor", "monitor"}, time.Now())
defer conn.Close()
......
......@@ -22,20 +22,23 @@ func TestMonitor_Start(t *testing.T) {
})
closeCh := make(chan struct{})
defer close(closeCh)
logCh := m.Start(closeCh)
go func() {
for {
select {
case log := <-logCh:
require.Contains(t, string(log), "[DEBUG] test log")
case <-time.After(1 * time.Second):
t.Fatal("Expected to receive from log channel")
}
}
logger.Debug("test log")
time.Sleep(10 * time.Millisecond)
}()
logger.Debug("test log")
for {
select {
case log := <-logCh:
require.Contains(t, string(log), "[DEBUG] test log")
return
case <-time.After(3 * time.Second):
t.Fatal("Expected to receive from log channel")
}
}
}
// Ensure number of dropped messages are logged
......@@ -61,7 +64,6 @@ func TestMonitor_DroppedMessages(t *testing.T) {
}
received := ""
passed := make(chan struct{})
go func() {
for {
......
......@@ -74,6 +74,13 @@ func (c *MonitorCommand) Run(args []string) int {
return 1
}
args = flags.Args()
if l := len(args); l != 0 {
c.Ui.Error("This command takes no arguments")
c.Ui.Error(commandErrorText(c))
return 1
}
client, err := c.Meta.Client()
if err != nil {
c.Ui.Error(fmt.Sprintf("Error initializing client: %s", err))
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment