aboutsummaryrefslogtreecommitdiff
path: root/matrix/pushrules/condition_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'matrix/pushrules/condition_test.go')
-rw-r--r--matrix/pushrules/condition_test.go134
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
+}