Add QueryOptions to api package's monitor

hashicorp · Nov 28, 2016 · 8079c49 · 8079c49
1 parent 338e36c
commit 8079c49
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/api/agent.go b/api/agent.go
@@ -416,8 +416,9 @@ func (a *Agent) DisableNodeMaintenance() error {
 // 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 the
 // log stream
-func (a *Agent) Monitor(loglevel string, stopCh chan struct{}) (chan string, error) {
+func (a *Agent) Monitor(loglevel string, stopCh chan struct{}, q *QueryOptions) (chan string, error) {
 	r := a.c.newRequest("GET", "/v1/agent/monitor")
+	r.setQueryOptions(q)
 	if loglevel != "" {
 		r.params.Add("loglevel", loglevel)
 	}

diff --git a/api/agent_test.go b/api/agent_test.go
@@ -566,7 +566,7 @@ func TestAgent_Monitor(t *testing.T) {
 
 	agent := c.Agent()
 
-	logCh, err := agent.Monitor("info", nil)
+	logCh, err := agent.Monitor("info", nil, nil)
 	if err != nil {
 		t.Fatalf("err: %v", err)
 	}