fix bug after refine architect

ptt.go

@@ -179,8 +179,14 @@ func (p *PTT) GetPostStarByIndex(postIndex int) int {
 	return p.storedPost[postIndex].Likeint
 }
 
+//Set Ptt board psot number, fetch assigned number of posts.
+func (p *PTT) ParsePttByNumber(num int) {
+
+	return
+}
+
 //Set Ptt board page index, fetch all post and return article count back
-func (p *PTT) ParsePttPageByIndex(page int) int {
+func (p *PTT) ParsePttPageByIndex(page int, replace bool) int {
 	// Get https response with setting cookie over18=1
 	resp := getResponseWithCookie(p.entryAddress)
 	doc, err := goquery.NewDocumentFromResponse(resp)
@@ -234,6 +240,11 @@ func (p *PTT) ParsePttPageByIndex(page int) int {
 			posts = append(posts, newPost)
 		}
 	})
+	if replace {
+		p.storedPost = posts
+	} else {
+		p.storedPost = append(p.storedPost, posts...)
+	}
 
 	return len(p.storedPost)
 }

ptt_test.go

@@ -2,6 +2,14 @@ package photomgr
 
 import "testing"
 
+func TestGetPage(t *testing.T) {
+	ptt := NewPTT()
+	count := ptt.ParsePttPageByIndex(0, true)
+	if count == 0 {
+		t.Errorf("ParsePttPageByIndex: no content")
+	}
+}
+
 func TestURLPhoto(t *testing.T) {
 	ptt := NewPTT()
 	title := ptt.GetPostTitleByIndex(5)
@@ -36,7 +44,7 @@ func TestURLLike(t *testing.T) {
 
 func TestUAllGirls(t *testing.T) {
 	ptt := NewPTT()
-	count := ptt.ParsePttPageByIndex(0)
+	count := ptt.ParsePttPageByIndex(0, true)
 	for i := 0; i < count; i++ {
 		title := ptt.GetPostTitleByIndex(i)
 		if CheckTitleWithBeauty(title) {