add test
parent
0eccbb0514
commit
aaa63c0e52
|
@ -0,0 +1,55 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
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)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue