gomuks/matrix/pushrules/condition_test.go

// 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
}