go-mastodon/cmd/mstdn/cmd_stream_test.go

74 lines
1.4 KiB
Go

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)
}
}