mirror of
https://github.com/diamondburned/arikawa.git
synced 2024-12-10 23:45:42 +00:00
156 lines
3.2 KiB
Go
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)
|
|
}
|
|
}
|