From cc5f65b3bf221c93079ce2668553baff5617a3ae Mon Sep 17 00:00:00 2001
From: Yasuhiro Matsumoto <mattn.jp@gmail.com>
Date: Thu, 20 Apr 2017 10:39:20 +0900
Subject: [PATCH] add test for stream command

---
 cmd/mstdn/cmd_stream.go      |  2 +
 cmd/mstdn/cmd_stream_test.go | 73 ++++++++++++++++++++++++++++++++++++
 2 files changed, 75 insertions(+)
 create mode 100644 cmd/mstdn/cmd_stream_test.go

diff --git a/cmd/mstdn/cmd_stream.go b/cmd/mstdn/cmd_stream.go
index a40f998..b37fd86 100644
--- a/cmd/mstdn/cmd_stream.go
+++ b/cmd/mstdn/cmd_stream.go
@@ -56,6 +56,8 @@ func cmdStream(c *cli.Context) error {
 		close(q)
 	}()
 
+	c.App.Metadata["signal"] = sc
+
 	s := newScreen(config)
 	for e := range q {
 		if asJSON {
diff --git a/cmd/mstdn/cmd_stream_test.go b/cmd/mstdn/cmd_stream_test.go
new file mode 100644
index 0000000..fdb3427
--- /dev/null
+++ b/cmd/mstdn/cmd_stream_test.go
@@ -0,0 +1,73 @@
+package main
+
+import (
+	"bytes"
+	"fmt"
+	"net/http"
+	"net/http/httptest"
+	"os"
+	"strings"
+	"testing"
+	"time"
+
+	"github.com/mattn/go-mastodon"
+)
+
+func TestCmdStream(t *testing.T) {
+	var ts *httptest.Server
+	ts = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if r.URL.Path != "/api/v1/streaming/public/local" {
+			http.Error(w, http.StatusText(http.StatusNotFound), http.StatusNotFound)
+			return
+		}
+		f, _ := w.(http.Flusher)
+		fmt.Fprintln(w, `
+event: update
+data: {"content": "foo", "account":{"acct":"FOO"}}
+		`)
+		f.Flush()
+
+		fmt.Fprintln(w, `
+event: update
+data: {"content": "bar", "account":{"acct":"BAR"}}
+		`)
+		f.Flush()
+		return
+	}))
+	defer ts.Close()
+
+	config := &mastodon.Config{
+		Server:       ts.URL,
+		ClientID:     "foo",
+		ClientSecret: "bar",
+		AccessToken:  "zoo",
+	}
+	client := mastodon.NewClient(config)
+
+	var buf bytes.Buffer
+	app := makeApp()
+	app.Writer = &buf
+	app.Metadata = map[string]interface{}{
+		"client": client,
+		"config": config,
+	}
+
+	go func() {
+		time.Sleep(5 * time.Second)
+		if sig, ok := app.Metadata["signal"]; ok {
+			sig.(chan os.Signal) <- os.Interrupt
+			return
+		}
+		panic("timeout")
+	}()
+
+	app.Run([]string{"mstdn", "stream"})
+
+	out := buf.String()
+	if !strings.Contains(out, "FOO@") {
+		t.Fatalf("%q should be contained in output of command: %v", "FOO@", out)
+	}
+	if !strings.Contains(out, "foo") {
+		t.Fatalf("%q should be contained in output of command: %v", "foo", out)
+	}
+}