/
main_test.go
47 lines (40 loc) · 761 Bytes
/
main_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
package main
import (
"github.com/KevinZonda/GoX/pkg/iox"
"os"
"testing"
)
const README = `
BEGIN
<!-- BEGIN LGBT-CN SIGNATURE -->
User1
User2
User5
<!-- END LGBT-CN SIGNATURE -->
<!-- BEGIN LGBT-CN COUNT -->
已有 -1 人签署!
<!-- END LGBT-CN COUNT -->
END
`
const EXPECTED = `BEGIN
<!-- BEGIN LGBT-CN SIGNATURE -->
User1
User2
User5
<!-- END LGBT-CN SIGNATURE -->
<!-- BEGIN LGBT-CN COUNT -->
已有 3 人签署!
<!-- END LGBT-CN COUNT -->
END`
const TEST_FILE = "test.md"
func TestMainFunc(t *testing.T) {
defer os.Remove(TEST_FILE)
iox.WriteAllText(TEST_FILE, README)
os.Args = []string{"main", TEST_FILE}
main()
txt, _ := iox.ReadAllText(TEST_FILE)
if txt != EXPECTED {
t.Errorf("Expected %s, got %s", EXPECTED, txt)
return
}
}