~thecashewtrader/atomitter

3d1c872aed571848da611d27189823a5e97333a9 — thecashewtrader 1 year, 9 months ago 66fb310
Add test for getTweetTextWithFullURLs and getUntweetedArticles
1 files changed, 69 insertions(+), 0 deletions(-)

M main_test.go
M main_test.go => main_test.go +69 -0
@@ 5,6 5,7 @@ import (
	"testing"
	"text/template"

	"github.com/dghubble/go-twitter/twitter"
	"github.com/mmcdole/gofeed"
	"github.com/stretchr/testify/require"
)


@@ 55,3 56,71 @@ func TestIncorrectGetArticleText(t *testing.T) {

	require.NotEqual("Read my post \"Example Title\": https://example.com", articleText)
}

func TestGetTweetTextWithFullURLs(t *testing.T) {
	require := require.New(t)

	tweet := twitter.Tweet{
		Text: "Read my post \"Example Title\": https://t.co/mtXLLfYOYE",
		Entities: &twitter.Entities{
			Urls: []twitter.URLEntity{
				{
					URL:         "https://t.co/mtXLLfYOYE",
					ExpandedURL: "https://www.bbc.co.uk/news/blogs-trending-47975564",
				},
			},
		},
	}

	tweetText := getTweetTextWithFullURLs(tweet)

	require.Equal("Read my post \"Example Title\": https://www.bbc.co.uk/news/blogs-trending-47975564", tweetText)
}

func TestGetUntweetedArticles(t *testing.T) {
	require := require.New(t)

	format := "Read my post \"{{ .Title }}\": {{ .Link }}"
	tweetTemplate, err := template.New("main").Parse(format)
	require.NoError(err)

	articles := []*gofeed.Item{
		{
			Title: "Title 1",
			Link:  "https://example.com/1",
		},
		{
			Title: "Title 2",
			Link:  "https://example.com/2",
		},
		{
			Title: "Title 3",
			Link:  "https://example.com/3",
		},
	}

	tweets := []twitter.Tweet{
		{
			Text: "Read my post \"Title 1\": https://example.com/1",
			Entities: &twitter.Entities{
				Urls: []twitter.URLEntity{},
			},
		},
	}

	expectedUntweetedArticles := []*gofeed.Item{
		{
			Title: "Title 2",
			Link:  "https://example.com/2",
		},
		{
			Title: "Title 3",
			Link:  "https://example.com/3",
		},
	}

	untweetedArticles := getUntweetedArticles(articles, tweets, tweetTemplate)

	require.Equal(expectedUntweetedArticles, untweetedArticles)

}