package main import ( "encoding/json" "flag" "io/ioutil" "log" "os" "testing" "github.com/urfave/cli" ) 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: "

foo

", want: "foo"}, {input: "

foo\nbar\nbaz

", want: "foobarbaz"}, {input: "

foo\nbar
baz

", 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 { log.Fatal(err) } err = ioutil.WriteFile(file, b, 0700) if err != nil { log.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) } }