cws
/
catbase
mirror of https://github.com/velour/catbase.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Proper handling of the default pick quantity

This commit is contained in:
Steve McCoy 2018-08-30 22:32:10 -04:00 committed by Chris Sexton
parent ce91370313
commit 96a887917a
2 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
plugins/picker/picker.go
View File

@ -63,7 +63,7 @@ func (p *PickerPlugin) Message(message msg.Message) bool {
return true
}
var pickerListPrologue = regexp.MustCompile(`^pick[ \t]+([0-9]*)[ \t]+\{[ \t]*`)
var pickerListPrologue = regexp.MustCompile(`^pick[ \t]+([0-9]*)[ \t]*\{[ \t]*`)
var pickerListItem = regexp.MustCompile(`^([^,]+),[ \t]*`)
var pickerListFinalItem = regexp.MustCompile(`^([^,}]+),?[ \t]*\}[ \t]*`)
@ -73,7 +73,7 @@ func (p *PickerPlugin) parse(body string) (int, []string, error) {
return 0, nil, errors.New("saddle up for a syntax error")
}
n := 0
n := 1
var err error
if subs[1] != "" {
n, err = strconv.Atoi(subs[1])

11
plugins/picker/picker_test.go
View File

@ -25,7 +25,7 @@ func makeMessage(payload string) msg.Message {
}
}
func TestReplacement(t *testing.T) {
func TestPick2(t *testing.T) {
mb := bot.NewMockBot()
c := New(mb)
assert.NotNil(t, c)
@ -35,3 +35,12 @@ func TestReplacement(t *testing.T) {
t.Fatalf("expected a successful choice, got %q", mb.Messages[0])
}
}
func TestPickDefault(t *testing.T) {
mb := bot.NewMockBot()
c := New(mb)
assert.NotNil(t, c)
_ = c.Message(makeMessage("!pick { a}"))
assert.Len(t, mb.Messages, 1)
assert.Equal(t, `I've chosen "a" for you.`, mb.Messages[0])
}