log: fix logging

Author: postables

log.go

@@ -21,6 +21,11 @@ func (l Logger) Next() (map[string]interface{}, error) {
 	return out, nil
 }
 
+// Close is used to close our reader
+func (l Logger) Close() error {
+	return l.r.Close()
+}
+
 // GetLogs is used to retrieve a parsable logger object
 func (s *Shell) GetLogs() (Logger, error) {
 	logURL := fmt.Sprintf("http://%s/api/v0/log/tail", s.url)

log_test.go

@@ -8,6 +8,11 @@ func Test_Logger(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
+	defer func() {
+		if err := logger.Close(); err != nil {
+			t.Fatal(err)
+		}
+	}()
 	_, err = logger.Next()
 	if err != nil {
 		t.Fatal(err)