1
0
Fork 0
mirror of https://github.com/diamondburned/arikawa.git synced 2024-12-10 23:45:42 +00:00
arikawa/utils/bot/ctx_plumb_test.go
2021-09-18 00:39:34 -07:00

156 lines
3.2 KiB
Go

package bot
import (
"testing"
"github.com/diamondburned/arikawa/v3/discord"
"github.com/diamondburned/arikawa/v3/gateway"
"github.com/diamondburned/arikawa/v3/state"
"github.com/diamondburned/arikawa/v3/state/store"
)
type hasPlumb struct {
Ctx *Context
Plumbed bool
PlumbedArgs string
NotPlumbed bool
NotPlumbedArgs string
}
func (h *hasPlumb) Setup(sub *Subcommand) {
sub.SetPlumb(h.Plumber)
}
func (h *hasPlumb) Plumber(_ *gateway.MessageCreateEvent, c RawArguments) error {
h.NotPlumbed = false
h.Plumbed = true
h.PlumbedArgs = string(c)
return nil
}
func (h *hasPlumb) Normal(_ *gateway.MessageCreateEvent, c RawArguments) error {
h.Plumbed = false
h.NotPlumbed = true
h.NotPlumbedArgs = string(c)
return nil
}
func TestSubcommandPlumb(t *testing.T) {
var s = &state.State{
Cabinet: store.NoopCabinet,
}
c, err := New(s, &testc{})
if err != nil {
t.Fatal("Failed to create new context:", err)
}
c.HasPrefix = NewPrefix("")
p := &hasPlumb{}
_, err = c.RegisterSubcommand(p)
if err != nil {
t.Fatal("Failed to register hasPlumb:", err)
}
sendFn := func(content string) {
m := &gateway.MessageCreateEvent{
Message: discord.Message{Content: content},
}
if err := c.callCmd(m); err != nil {
t.Fatal("Failed to call message:", err)
}
}
// Try call exactly what's in the Plumb example:
sendFn("hasPlumb")
if p.NotPlumbed || !p.Plumbed {
t.Error("Normal method called for hasPlumb")
}
sendFn("hasPlumb arg1")
if p.NotPlumbed || !p.Plumbed {
t.Error("Normal method called for hasPlumb with arguments")
}
if p.PlumbedArgs != "arg1" {
t.Errorf("Incorrect plumbed argument %q", p.PlumbedArgs)
}
sendFn("hasPlumb plumber arg1")
if p.NotPlumbed || !p.Plumbed {
t.Error("Normal method called for plumber command with arguments")
}
if p.PlumbedArgs != "arg1" {
t.Errorf("Incorrect normal plumbed argument %q", p.PlumbedArgs)
}
sendFn("hasPlumb normal")
if p.Plumbed || !p.NotPlumbed {
t.Error("Plumbed method called for normal command")
}
sendFn("hasPlumb normal args")
if p.Plumbed || !p.NotPlumbed {
t.Error("Plumbed method called for normal command with arguments")
}
if p.NotPlumbedArgs != "args" {
t.Errorf("Incorrect normal argument %q", p.NotPlumbedArgs)
}
}
type onlyPlumb struct {
Ctx *Context
Plumbed string
}
func (h *onlyPlumb) Setup(sub *Subcommand) {
sub.SetPlumb("Plumber")
}
func (h *onlyPlumb) Plumber(_ *gateway.MessageCreateEvent, c RawArguments) error {
h.Plumbed = string(c)
return nil
}
func TestSubcommandOnlyPlumb(t *testing.T) {
var s = &state.State{
Cabinet: store.NoopCabinet,
}
c, err := New(s, &testc{})
if err != nil {
t.Fatal("Failed to create new context:", err)
}
c.HasPrefix = NewPrefix("")
p := &onlyPlumb{}
_, err = c.RegisterSubcommand(p)
if err != nil {
t.Fatal("Failed to register hasPlumb:", err)
}
// Try call exactly what's in the Plumb example:
m := &gateway.MessageCreateEvent{
Message: discord.Message{
Content: "onlyPlumb test command",
},
}
if err := c.callCmd(m); err != nil {
t.Fatal("Failed to call message:", err)
}
if p.Plumbed != "test command" {
t.Fatal("Unexpected custom argument for plumbed:", p.Plumbed)
}
}