Add tests for pushrule conditions and fix bugs found when making tests

1 files changed, 134 insertions, 0 deletions

diff --git a/matrix/pushrules/condition_test.go b/matrix/pushrules/condition_test.go
new file mode 100644
index 0000000..7fd06ee
--- /dev/null
+++ b/matrix/pushrules/condition_test.go
@@ -0,0 +1,134 @@
+// gomuks - A terminal Matrix client written in Go.
+// Copyright (C) 2018 Tulir Asokan
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+package pushrules_test
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"maunium.net/go/gomatrix"
+	"maunium.net/go/gomuks/matrix/pushrules"
+	"maunium.net/go/gomuks/matrix/rooms"
+)
+
+var (
+	blankTestRoom       *rooms.Room
+	displaynameTestRoom pushrules.Room
+
+	countConditionTestEvent *gomatrix.Event
+
+	displaynamePushCondition *pushrules.PushCondition
+)
+
+func init() {
+	blankTestRoom = rooms.NewRoom("!fakeroom:maunium.net", "@tulir:maunium.net")
+
+	countConditionTestEvent = &gomatrix.Event{
+		Sender:    "@tulir:maunium.net",
+		Type:      "m.room.message",
+		Timestamp: 1523791120,
+		ID:        "$123:maunium.net",
+		RoomID:    "!fakeroom:maunium.net",
+		Content: map[string]interface{}{
+			"msgtype": "m.text",
+			"body":    "test",
+		},
+	}
+
+	displaynameTestRoom = newFakeRoom(4)
+	displaynamePushCondition = &pushrules.PushCondition{
+		Kind: pushrules.KindContainsDisplayName,
+	}
+}
+
+func newFakeEvent(evtType string, content map[string]interface{}) *gomatrix.Event {
+	return &gomatrix.Event{
+		Sender:    "@tulir:maunium.net",
+		Type:      evtType,
+		Timestamp: 1523791120,
+		ID:        "$123:maunium.net",
+		RoomID:    "!fakeroom:maunium.net",
+		Content:   content,
+	}
+}
+
+func newCountPushCondition(condition string) *pushrules.PushCondition {
+	return &pushrules.PushCondition{
+		Kind:                 pushrules.KindRoomMemberCount,
+		MemberCountCondition: condition,
+	}
+}
+
+func newMatchPushCondition(key, pattern string) *pushrules.PushCondition {
+	return &pushrules.PushCondition{
+		Kind:    pushrules.KindEventMatch,
+		Key:     key,
+		Pattern: pattern,
+	}
+}
+
+func TestPushCondition_Match_InvalidKind(t *testing.T) {
+	condition := &pushrules.PushCondition{
+		Kind: pushrules.PushCondKind("invalid"),
+	}
+	event := newFakeEvent("m.room.foobar", map[string]interface{}{})
+	assert.False(t, condition.Match(blankTestRoom, event))
+}
+
+type FakeRoom struct {
+	members map[string]*rooms.Member
+	owner   string
+}
+
+func newFakeRoom(memberCount int) *FakeRoom {
+	room := &FakeRoom{
+		owner:   "@tulir:maunium.net",
+		members: make(map[string]*rooms.Member),
+	}
+
+	if memberCount >= 1 {
+		room.members["@tulir:maunium.net"] = &rooms.Member{
+			UserID:      "@tulir:maunium.net",
+			Membership:  rooms.MembershipJoin,
+			DisplayName: "tulir",
+		}
+	}
+
+	for i := 0; i < memberCount-1; i++ {
+		mxid := fmt.Sprintf("@extrauser_%d:matrix.org", i)
+		room.members[mxid] = &rooms.Member{
+			UserID:      mxid,
+			Membership:  rooms.MembershipJoin,
+			DisplayName: fmt.Sprintf("Extra User %d", i),
+		}
+	}
+
+	return room
+}
+
+func (fr *FakeRoom) GetMember(mxid string) *rooms.Member {
+	return fr.members[mxid]
+}
+
+func (fr *FakeRoom) GetSessionOwner() *rooms.Member {
+	return fr.members[fr.owner]
+}
+
+func (fr *FakeRoom) GetMembers() map[string]*rooms.Member {
+	return fr.members
+}