arikawa/bot/extras/arguments/mention_test.go

package arguments

import (
	"testing"
)

func TestChannelMention(t *testing.T) {
	test := new(ChannelMention)
	str := "<#123123>"
	id := 123123

	if err := test.Parse(str); err != nil {
		t.Fatal("Expected", id, "error:", err)
	}

	if id := test.ID(); id != id {
		t.Fatal("Expected", id, "got", id)
	}

	if mention := test.Mention(); mention != str {
		t.Fatal("Expected", str, "got", mention)
	}
}

func TestUserMention(t *testing.T) {
	test := new(UserMention)
	str := "<@123123>"
	id := 123123

	if err := test.Parse(str); err != nil {
		t.Fatal("Expected", id, "error:", err)
	}

	if id := test.ID(); id != id {
		t.Fatal("Expected", id, "got", id)
	}

	if mention := test.Mention(); mention != str {
		t.Fatal("Expected", str, "got", mention)
	}
}

func TestRoleMention(t *testing.T) {
	test := new(RoleMention)
	str := "<@&123123>"
	id := 123123

	if err := test.Parse(str); err != nil {
		t.Fatal("Expected", id, "error:", err)
	}

	if id := test.ID(); id != id {
		t.Fatal("Expected", id, "got", id)
	}

	if mention := test.Mention(); mention != str {
		t.Fatal("Expected", str, "got", mention)
	}
}