go-mastodon/cmd/mstdn/main_test.go

package main

import (
	"encoding/json"
	"flag"
	"io/ioutil"
	"os"
	"testing"

	"github.com/urfave/cli/v2"
)

func TestReadFileFile(t *testing.T) {
	b, err := readFile("main.go")
	if err != nil {
		t.Fatal(err)
	}
	if len(b) == 0 {
		t.Fatalf("should read something: %v", err)
	}
}

func TestReadFileStdin(t *testing.T) {
	f, err := os.Open("main.go")
	if err != nil {
		t.Fatal(err)
	}
	defer f.Close()
	stdin := os.Stdin
	os.Stdin = f
	defer func() {
		os.Stdin = stdin
	}()

	b, err := readFile("-")
	if err != nil {
		t.Fatal(err)
	}
	if len(b) == 0 {
		t.Fatalf("should read something: %v", err)
	}
}

func TestTextContent(t *testing.T) {
	tests := []struct {
		input string
		want  string
	}{
		{input: "", want: ""},
		{input: "<p>foo</p>", want: "foo"},
		{input: "<p>foo<span>\nbar\n</span>baz</p>", want: "foobarbaz"},
		{input: "<p>foo<span>\nbar<br></span>baz</p>", want: "foobar\nbaz"},
	}
	for _, test := range tests {
		got := textContent(test.input)
		if got != test.want {
			t.Fatalf("want %q but %q", test.want, got)
		}
	}
}

func TestGetConfig(t *testing.T) {
	tmpdir, err := ioutil.TempDir("", "mstdn")
	if err != nil {
		t.Fatal(err)
	}
	home := os.Getenv("HOME")
	appdata := os.Getenv("APPDATA")
	os.Setenv("HOME", tmpdir)
	os.Setenv("APPDATA", tmpdir)
	defer func() {
		os.RemoveAll(tmpdir)
		os.Setenv("HOME", home)
		os.Setenv("APPDATA", appdata)
	}()

	app := makeApp()
	set := flag.NewFlagSet("test", 0)
	set.Parse([]string{"mstdn", "-profile", ""})
	c := cli.NewContext(app, set, nil)
	file, config, err := getConfig(c)
	if err != nil {
		t.Fatal(err)
	}
	if _, err := os.Stat(file); err == nil {
		t.Fatal("should not exists")
	}
	if config.AccessToken != "" {
		t.Fatalf("should be empty: %v", config.AccessToken)
	}
	if config.ClientID == "" {
		t.Fatalf("should not be empty")
	}
	if config.ClientSecret == "" {
		t.Fatalf("should not be empty")
	}
	config.AccessToken = "foo"
	b, err := json.MarshalIndent(config, "", "  ")
	if err != nil {
		t.Fatal(err)
	}
	err = ioutil.WriteFile(file, b, 0700)
	if err != nil {
		t.Fatal(err)
	}
	file, config, err = getConfig(c)
	if err != nil {
		t.Fatal(err)
	}
	if _, err := os.Stat(file); err != nil {
		t.Fatalf("should exists: %v", err)
	}
	if got := config.AccessToken; got != "foo" {
		t.Fatalf("want %q but %q", "foo", got)
	}
}

func TestPrompt(t *testing.T) {
	readUsername = func() (string, error) {
		return "foo", nil
	}
	readPassword = func() (string, error) {
		return "bar", nil
	}
	username, password, err := prompt()
	if err != nil {
		t.Fatal(err)
	}
	if username != "foo" {
		t.Fatalf("want %q but %q", "foo", username)
	}
	if password != "bar" {
		t.Fatalf("want %q but %q", "bar", password)
	}
}