Add failing TestHTMLEntities test

src/github.com/matrix-org/go-neb/services/rssbot/rssbot_test.go

@@ -0,0 +1,120 @@
+package services
+
+import (
+	"bytes"
+	"encoding/json"
+	"errors"
+	"github.com/matrix-org/go-neb/database"
+	"github.com/matrix-org/go-neb/matrix"
+	"github.com/matrix-org/go-neb/types"
+	_ "github.com/mattn/go-sqlite3"
+	"io/ioutil"
+	"net/http"
+	"net/url"
+	"strings"
+	"sync"
+	"testing"
+	"time"
+)
+
+type MockTransport struct {
+	roundTrip func(*http.Request) (*http.Response, error)
+}
+
+func (t MockTransport) RoundTrip(req *http.Request) (*http.Response, error) {
+	return t.roundTrip(req)
+}
+
+func TestHTMLEntities(t *testing.T) {
+	// FIXME: Make ServiceDB an interface so we don't need to do this and import sqlite3!
+	//        We are NOT interested in db operations, but need them because OnPoll will
+	//        call StoreService.
+	db, err := database.Open("sqlite3", ":memory:")
+	if err != nil {
+		t.Fatal("Failed to create in-memory db: ", err)
+		return
+	}
+	database.SetServiceDB(db)
+
+	feedURL := "https://thehappymaskshop.hyrule"
+	// Replace the cachingClient with a mock so we can intercept RSS requests
+	rssTrans := struct{ MockTransport }{}
+	rssTrans.roundTrip = func(req *http.Request) (*http.Response, error) {
+		if req.URL.String() != feedURL {
+			return nil, errors.New("Unknown test URL")
+		}
+		return &http.Response{
+			StatusCode: 200,
+			Body: ioutil.NopCloser(bytes.NewBufferString(`
+			<?xml version="1.0" encoding="UTF-8"?>
+			<rss version="2.0"
+				xmlns:content="http://purl.org/rss/1.0/modules/content/"
+				xmlns:wfw="http://wellformedweb.org/CommentAPI/"
+				xmlns:dc="http://purl.org/dc/elements/1.1/"
+				xmlns:atom="http://www.w3.org/2005/Atom"
+				xmlns:sy="http://purl.org/rss/1.0/modules/syndication/"
+				xmlns:slash="http://purl.org/rss/1.0/modules/slash/"
+				>
+
+			<channel>
+				<title>Mask Shop</title>
+				<item>
+					<title>New Item: Majora&#8217;s Mask</title>
+					<link>http://go.neb/rss/majoras-mask</link>
+				</item>
+			</channel>
+			</rss>`)),
+		}, nil
+	}
+	cachingClient = &http.Client{Transport: rssTrans}
+
+	// Create the RSS service
+	srv, err := types.CreateService("id", "rssbot", "@happy_mask_salesman:hyrule", []byte(
+		`{"feeds": {"`+feedURL+`":{}}}`, // no config yet
+	))
+	if err != nil {
+		t.Fatal("Failed to create RSS bot: ", err)
+	}
+	rssbot := srv.(*rssBotService)
+	f := rssbot.Feeds[feedURL]
+	f.Rooms = []string{"!linksroom:hyrule"}
+	f.FeedUpdatedTimestampSecs = 12345
+	f.NextPollTimestampSecs = time.Now().Unix()
+	rssbot.Feeds[feedURL] = f
+
+	// Create the Matrix client which will send the notification
+	wg := sync.WaitGroup{}
+	wg.Add(1)
+	matrixTrans := struct{ MockTransport }{}
+	matrixTrans.roundTrip = func(req *http.Request) (*http.Response, error) {
+		if strings.HasPrefix(req.URL.Path, "/_matrix/client/r0/rooms/!linksroom:hyrule/send/m.room.message") {
+			// Check content body to make sure it is decoded
+			var msg matrix.HTMLMessage
+			if err := json.NewDecoder(req.Body).Decode(&msg); err != nil {
+				t.Fatal("Failed to decode request JSON: ", err)
+				return nil, errors.New("Error handling matrix client test request")
+			}
+			want := "New Item: Majora's Mask"
+			if !strings.Contains(msg.Body, want) {
+				t.Errorf("TestHTMLEntities: want '%s' in body, got '%s'", want, msg.Body)
+			}
+
+			wg.Done()
+			return &http.Response{
+				StatusCode: 200,
+				Body: ioutil.NopCloser(bytes.NewBufferString(`
+					{"event_id":"$123456:hyrule"}	
+				`)),
+			}, nil
+		}
+		return nil, errors.New("Unhandled matrix client test request")
+	}
+	u, _ := url.Parse("https://hyrule")
+	matrixClient := matrix.NewClient(&http.Client{Transport: matrixTrans}, u, "its_a_secret", "@happy_mask_salesman:hyrule")
+
+	// Invoke OnPoll to trigger the RSS feed update
+	_ = rssbot.OnPoll(matrixClient)
+
+	// Check that the Matrix client sent a message
+	wg.Wait()
+}