70 lines
1.3 KiB
Go
70 lines
1.3 KiB
Go
|
// +build unit
|
||
|
|
|
||
|
|
package bot
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/diamondburned/arikawa/gateway"
|
||
|
|
"github.com/diamondburned/arikawa/state"
|
||
|
|
)
|
||
|
|
|
||
|
|
type hasPlumb struct {
|
||
|
|
Ctx *Context
|
||
|
|
|
||
|
|
Plumbed string
|
||
|
|
NotPlumbed bool
|
||
|
|
}
|
||
|
|
|
||
|
|
func (h *hasPlumb) Normal(_ *gateway.MessageCreateEvent) error {
|
||
|
|
h.NotPlumbed = true
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (h *hasPlumb) PーPlumber(
|
||
|
|
_ *gateway.MessageCreateEvent, c Content) error {
|
||
|
|
|
||
|
|
h.Plumbed = string(c)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSubcommandPlumb(t *testing.T) {
|
||
|
|
var state = &state.State{
|
||
|
|
Store: state.NewDefaultStore(nil),
|
||
|
|
}
|
||
|
|
|
||
|
|
c, err := New(state, &testCommands{})
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal("Failed to create new context:", err)
|
||
|
|
}
|
||
|
|
c.Prefix = ""
|
||
|
|
|
||
|
|
p := &hasPlumb{}
|
||
|
|
|
||
|
|
_, err = c.RegisterSubcommand(p)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal("Failed to register hasPlumb:", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if l := len(c.subcommands[0].Commands); l != 1 {
|
||
|
|
t.Fatal("Unexpected length for sub.Commands:", l)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Try call exactly what's in the Plumb example:
|
||
|
|
m := &gateway.MessageCreateEvent{
|
||
|
|
Content: "hasPlumb test command",
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := c.callCmd(m); err != nil {
|
||
|
|
t.Fatal("Failed to call message:", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if p.NotPlumbed {
|
||
|
|
t.Fatal("Normal method called for hasPlumb")
|
||
|
|
}
|
||
|
|
|
||
|
|
if p.Plumbed != "test command" {
|
||
|
|
t.Fatal("Unexpected custom argument for plumbed:", p.Plumbed)
|
||
|
|
}
|
||
|
|
}
|