Exemplo n.º 1
0
func TestMessage(t *testing.T) {
	mStore := new(mockStore)
	mTarget := new(mockTarget)
	mShort := new(mockShort)
	mockConf.Propagate.QtyPerPage = 1

	Propagate(mStore, mTarget, mockPrices, mShort, *mockConf)
	if lastMsg != "" || lastUrl != "" {
		t.Error("There should be no messages, but got:", lastMsg, " with this url:", lastUrl)
	}
	var result feeds.RssResult
	result.Item.Title = "boom"
	result.Link = "http://www.boom.org"
	messages = append(messages, result)

	Propagate(mStore, mTarget, mockPrices, mShort, *mockConf)
	if lastMsg != "boom" || lastUrl != result.Link {
		t.Error("There should be boom messages, but got:", lastMsg, " with this url:", lastUrl)
	}

	var result2 feeds.RssResult
	result2.Item.Title = "bangbang"
	result2.Link = "http://www.bangbang.org"
	messages = append(messages, result2)

	Propagate(mStore, mTarget, mockPrices, mShort, *mockConf)
	if lastMsg != "bangbang" || lastUrl != result2.Link {
		t.Error("There should be bangbang messages, but got:", lastMsg, " with this url:", lastUrl)
	}
}
Exemplo n.º 2
0
func TestSend(t *testing.T) {
	mStore := new(mockStore)
	mTarget := new(mockTarget)
	mShort := new(mockShort)
	mockConf.Propagate.QtyPerPage = 1

	var checkLastAreSent = func() {
		for i := range lastSentItems {
			if last[i] != lastSentItems[i] {
				t.Error("The ", i, " item in store was not sent")
			}
		}
	}

	var result feeds.RssResult
	result.Item.Title = "boom"
	result.Link = "http://www.boom.org"
	last = append(last, result)
	var result2 feeds.RssResult
	result2.Item.Title = "boomboom"
	result2.Link = "http://www.boomboom.org"
	last = append(last, result2)
	Propagate(mStore, mTarget, mockPrices, mShort, *mockConf)
	checkLastAreSent()

	mockConf.Propagate.QtyPerPage = 1000
	Propagate(mStore, mTarget, mockPrices, mShort, *mockConf)
	checkLastAreSent()
}
Exemplo n.º 3
0
func (fs FileStore) Save(rssResult feeds.RssResult, link string, filename string) {
	if len(rssResult.Item.Title) == 0 {
		log.Printf("Skipping saving as this result has no title")
		return
	}
	if len(GotAlready) == 0 {
		GotAlready = fs.loadItems(filename)
	} else {
		log.Printf("Adding to the %v items done with %v \n", len(GotAlready), link)
	}
	GotAlready[link] = rssResult
	if rssResult.Failed {
		log.Printf("No message as the rss failed")
	} else {
		rssResult.Link = link
		toMessage = append(toMessage, rssResult)
	}
	jsave, _ := json.Marshal(GotAlready)
	var data []byte = jsave
	ioutil.WriteFile(filename, data, Perm)
}
Exemplo n.º 4
0
func TestPutInDateOrder(t *testing.T) {
	defer tearDown()

	var store = new(FileStore)
	store.ClearMessages()
	var mess = store.GetToMessage()
	if len(mess) != 0 {
		fmt.Println(mess)
		t.Error("Should have no new result to message but has", len(mess))
	}

	var data = new(feeds.RssResult)
	data.Item.Title = "Golang content two hours ahead"
	data.Date = time.Now().Add(time.Hour)
	var link = "http://one"
	store.Save(*data, link, filename)
	mess = store.GetToMessage()
	if len(mess) != 1 {
		t.Error("Should have one new result to message but has", len(mess))
	}

	var data2 = new(feeds.RssResult)
	data2.Item.Title = "Golang content one hour ahead"
	data2.Date = time.Now().Add(time.Hour).Add(time.Hour)
	var link2 = "http://two"
	store.Save(*data2, link2, filename)
	mess = store.GetToMessage()
	if len(mess) != 2 {
		t.Error("Should have two new result to message but has", len(mess))
	}
	var gotit = store.AlreadyHave(link2, filename)
	if !gotit {
		t.Error(link2, " has been stored but isn't")
	}

	var data3 = new(feeds.RssResult)
	data3.Item.Title = "Golang content now"
	data3.Date = time.Now()
	var link3 = "http://three"
	store.Save(*data3, link3, filename)
	mess = store.GetToMessage()
	if len(mess) != 3 {
		t.Error("Should have three new result to message but has", len(mess))
	}

	var data4 = new(feeds.RssResult)
	data4.Item.Title = "Golang content one hour behind"
	data4.Date = time.Now().Add(-time.Hour)
	data4.Failed = true
	var link4 = "http://four"
	store.Save(*data4, link4, filename)
	mess = store.GetToMessage()
	if len(mess) != 3 {
		t.Error("Should still have three new result to message but has", len(mess))
	}

	var data5 = new(feeds.RssResult)
	data5.Item.Title = "Golang content two hours behind"
	data5.Date = time.Now().Add(-time.Hour).Add(-time.Hour)
	data5.Failed = false
	var link5 = "http://five.rss"
	store.Save(*data5, link5, filename)
	mess = store.GetToMessage()
	if len(mess) != 4 {
		t.Error("Should have four new result to message but has", len(mess))
	}

	var res = store.GetLast(10, filename)
	if len(res) != 5 {
		t.Error("Should have five results with a title")
	}
	var check = func(no int, title string) {
		if res[no].Item.Title != title {
			fmt.Println(res[no])
			t.Error("Expected '" + res[no].Item.Title + "' to equal '" + title + "'")
		}
	}
	check(0, data2.Item.Title)
	check(1, data.Item.Title)
	check(2, data3.Item.Title)
	check(3, data4.Item.Title)
	check(4, data5.Item.Title)
	gotit = store.AlreadyHave(link, filename)
	if !gotit {
		t.Error(link, " has been stored but isn't")
	}
	gotit = store.AlreadyHave(link3, filename)
	if !gotit {
		t.Error(link3, " has been stored but isn't")
	}
	gotit = store.AlreadyHave(link4, filename)
	if !gotit {
		t.Error(link4, " has been stored but isn't")
	}
	gotit = store.AlreadyHave(link5, filename)
	if !gotit {
		t.Error(link5, " has been stored but isn't")
	}
	gotit = store.AlreadyHave("no not yet rss feed link result", filename)
	if gotit {
		t.Error("This has not been stored but is")
	}

	res = store.GetLast(4, filename)
	if len(res) != 4 {
		t.Error("Should be restricted to four results with a title")
	}

}