From 1f69a653a9926d915b441f7c14d815ceffda4a01 Mon Sep 17 00:00:00 2001
From: Chris Sexton <chris.sexton@gmail.com>
Date: Tue, 5 Feb 2019 11:36:18 -0500
Subject: [PATCH] tests: refactor mock to match new interface

---
 bot/mock.go                       | 68 +++++++++++++++----------------
 plugins/reminder/reminder_test.go |  2 +-
 2 files changed, 35 insertions(+), 35 deletions(-)

diff --git a/bot/mock.go b/bot/mock.go
index 710b7f0..60afc2c 100644
--- a/bot/mock.go
+++ b/bot/mock.go
@@ -27,66 +27,66 @@ type MockBot struct {
 }
 
 func (mb *MockBot) Config() *config.Config { return mb.Cfg }
-func (mb *MockBot) DBVersion() int64       { return 1 }
 func (mb *MockBot) DB() *sqlx.DB           { return mb.Cfg.DB }
-func (mb *MockBot) Conn() Connector        { return nil }
 func (mb *MockBot) Who(string) []user.User { return []user.User{} }
-func (mb *MockBot) SendMessage(ch string, msg string) string {
-	mb.Messages = append(mb.Messages, msg)
-	return fmt.Sprintf("m-%d", len(mb.Actions)-1)
+func (mb *MockBot) Send(kind int, args ...interface{}) (error, string) {
+	switch kind {
+	case Message:
+		mb.Messages = append(mb.Messages, args[1].(string))
+		return nil, fmt.Sprintf("m-%d", len(mb.Actions)-1)
+	case Action:
+		mb.Actions = append(mb.Actions, args[1].(string))
+		return nil, fmt.Sprintf("a-%d", len(mb.Actions)-1)
+	case Edit:
+		ch, m, id := args[0].(string), args[1].(string), args[2].(string)
+		return mb.edit(ch, m, id)
+	case Reaction:
+		ch, re, msg := args[0].(string), args[1].(string), args[2].(msg.Message)
+		return mb.react(ch, re, msg)
+	}
+	return fmt.Errorf("Mesasge type unhandled"), "ERROR"
 }
-func (mb *MockBot) SendAction(ch string, msg string) string {
-	mb.Actions = append(mb.Actions, msg)
-	return fmt.Sprintf("a-%d", len(mb.Actions)-1)
-}
-func (mb *MockBot) ReplyToMessageIdentifier(channel, message, identifier string) (string, bool) {
-	return "", false
-}
-func (mb *MockBot) ReplyToMessage(channel, message string, replyTo msg.Message) (string, bool) {
-	return "", false
-}
-func (mb *MockBot) MsgReceived(msg msg.Message)                {}
-func (mb *MockBot) EventReceived(msg msg.Message)              {}
-func (mb *MockBot) Filter(msg msg.Message, s string) string    { return s }
-func (mb *MockBot) LastMessage(ch string) (msg.Message, error) { return msg.Message{}, nil }
-func (mb *MockBot) CheckAdmin(nick string) bool                { return false }
+func (mb *MockBot) AddPlugin(name string, f Plugin)                        {}
+func (mb *MockBot) Register(kind int, cb Callback)                         {}
+func (mb *MockBot) Receive(kind int, msg msg.Message, args ...interface{}) {}
+func (mb *MockBot) Filter(msg msg.Message, s string) string                { return s }
+func (mb *MockBot) LastMessage(ch string) (msg.Message, error)             { return msg.Message{}, nil }
+func (mb *MockBot) CheckAdmin(nick string) bool                            { return false }
 
-func (mb *MockBot) React(channel, reaction string, message msg.Message) bool {
+func (mb *MockBot) react(channel, reaction string, message msg.Message) (error, string) {
 	mb.Reactions = append(mb.Reactions, reaction)
-	return false
+	return nil, ""
 }
 
-func (mb *MockBot) Edit(channel, newMessage, identifier string) bool {
+func (mb *MockBot) edit(channel, newMessage, identifier string) (error, string) {
 	isMessage := identifier[0] == 'm'
 	if !isMessage && identifier[0] != 'a' {
-		log.Printf("failed to parse identifier: %s", identifier)
-		return false
+		err := fmt.Errorf("failed to parse identifier: %s", identifier)
+		log.Println(err)
+		return err, ""
 	}
 
 	index, err := strconv.Atoi(strings.Split(identifier, "-")[1])
 	if err != nil {
-		log.Printf("failed to parse identifier: %s", identifier)
-		return false
+		err := fmt.Errorf("failed to parse identifier: %s", identifier)
+		log.Println(err)
+		return err, ""
 	}
 
 	if isMessage {
 		if index < len(mb.Messages) {
 			mb.Messages[index] = newMessage
 		} else {
-			return false
+			return fmt.Errorf("No message"), ""
 		}
 	} else {
 		if index < len(mb.Actions) {
 			mb.Actions[index] = newMessage
 		} else {
-			return false
+			return fmt.Errorf("No action"), ""
 		}
 	}
-	return true
-}
-
-func (mb *MockBot) ReplyMsgReceived(msg.Message, string) {
-
+	return nil, ""
 }
 
 func (mb *MockBot) GetEmojiList() map[string]string                { return make(map[string]string) }
diff --git a/plugins/reminder/reminder_test.go b/plugins/reminder/reminder_test.go
index c9095d9..d3c6155 100644
--- a/plugins/reminder/reminder_test.go
+++ b/plugins/reminder/reminder_test.go
@@ -224,7 +224,7 @@ func TestLimitList(t *testing.T) {
 	for i := 0; i < 25; i++ {
 		assert.Contains(t, mb.Messages[3], fmt.Sprintf("%d) tester -> testuser :: don't fail this test", i+1))
 	}
-	assert.Contains(t, mb.Messages[3], "...5 more...")
+	assert.Contains(t, mb.Messages[3], "more...")
 
 	assert.NotContains(t, mb.Messages[3], "26) tester -> testuser")
 }