catbase/bot/mock.go

// © 2016 the CatBase Authors under the WTFPL license. See AUTHORS for the list of authors.

package bot

import (
	"fmt"
	bh "github.com/timshannon/bolthold"
	"io/ioutil"
	"net/http"
	"os"
	"regexp"
	"strconv"
	"strings"

	"github.com/rs/zerolog/log"
	"github.com/stretchr/testify/mock"
	"github.com/velour/catbase/bot/msg"
	"github.com/velour/catbase/bot/user"
	"github.com/velour/catbase/config"
)

type MockBot struct {
	mock.Mock
	store *bh.Store

	Cfg *config.Config

	Messages  []string
	Actions   []string
	Reactions []string

	storeFile *os.File
}

func (mb *MockBot) Config() *config.Config      { return mb.Cfg }
func (mb *MockBot) Store() *bh.Store            { return mb.Cfg.Store() }
func (mb *MockBot) Who(string) []user.User      { return []user.User{} }
func (mb *MockBot) WhoAmI() string              { return "tester" }
func (mb *MockBot) DefaultConnector() Connector { return nil }
func (mb *MockBot) GetPassword() string         { return "12345" }
func (mb *MockBot) SetQuiet(bool)               {}
func (mb *MockBot) Send(c Connector, kind Kind, args ...interface{}) (string, error) {
	switch kind {
	case Message:
		mb.Messages = append(mb.Messages, args[1].(string))
		return fmt.Sprintf("m-%d", len(mb.Actions)-1), nil
	case Action:
		mb.Actions = append(mb.Actions, args[1].(string))
		return fmt.Sprintf("a-%d", len(mb.Actions)-1), nil
	case Edit:
		ch, m, id := args[0].(string), args[1].(string), args[2].(string)
		return mb.edit(c, ch, m, id)
	case Reaction:
		ch, re, msg := args[0].(string), args[1].(string), args[2].(msg.Message)
		return mb.react(c, ch, re, msg)
	}
	return "ERR", fmt.Errorf("Mesasge type unhandled")
}
func (mb *MockBot) AddPlugin(f Plugin)                                                        {}
func (mb *MockBot) Register(p Plugin, kind Kind, cb Callback)                                 {}
func (mb *MockBot) RegisterTable(p Plugin, hs HandlerTable)                                   {}
func (mb *MockBot) RegisterRegex(p Plugin, kind Kind, r *regexp.Regexp, h ResponseHandler)    {}
func (mb *MockBot) RegisterRegexCmd(p Plugin, kind Kind, r *regexp.Regexp, h ResponseHandler) {}
func (mb *MockBot) RegisterWebName(_ http.Handler, _, _ string)                               {}
func (mb *MockBot) RegisterWeb(_ http.Handler, _ string)                                      {}
func (mb *MockBot) GetWebNavigation() []EndPoint                                              { return nil }
func (mb *MockBot) Receive(c Connector, kind Kind, msg msg.Message, args ...interface{}) bool {
	return false
}
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 nick == "admin" }

func (mb *MockBot) react(c Connector, channel, reaction string, message msg.Message) (string, error) {
	mb.Reactions = append(mb.Reactions, reaction)
	return "", nil
}

func (mb *MockBot) edit(c Connector, channel, newMessage, identifier string) (string, error) {
	isMessage := identifier[0] == 'm'
	if !isMessage && identifier[0] != 'a' {
		err := fmt.Errorf("failed to parse identifier: %s", identifier)
		log.Error().Err(err)
		return "", err
	}

	index, err := strconv.Atoi(strings.Split(identifier, "-")[1])
	if err != nil {
		err := fmt.Errorf("failed to parse identifier: %s", identifier)
		log.Error().Err(err)
		return "", err
	}

	if isMessage {
		if index < len(mb.Messages) {
			mb.Messages[index] = newMessage
		} else {
			return "", fmt.Errorf("No message")
		}
	} else {
		if index < len(mb.Actions) {
			mb.Actions[index] = newMessage
		} else {
			return "", fmt.Errorf("No action")
		}
	}
	return "", nil
}

func (mb *MockBot) GetEmojiList() map[string]string                { return make(map[string]string) }
func (mb *MockBot) RegisterFilter(s string, f func(string) string) {}

func NewMockBot() *MockBot {
	storeFile, err := ioutil.TempFile(os.TempDir(), "prefix-")
	if err != nil {
		panic(err)
	}
	log.Debug().Msgf("MockBot temp store: %s", storeFile.Name())
	cfg := config.ReadConfig(storeFile.Name())
	b := MockBot{
		Cfg:       cfg,
		Messages:  make([]string, 0),
		Actions:   make([]string, 0),
		storeFile: storeFile,
		store:     cfg.Store(),
	}
	// If any plugin registered a route, we need to reset those before any new test
	http.DefaultServeMux = new(http.ServeMux)
	return &b
}

func (mb *MockBot) GetPluginNames() []string                      { return nil }
func (mb *MockBot) RefreshPluginBlacklist() error                 { return nil }
func (mb *MockBot) RefreshPluginWhitelist() error                 { return nil }
func (mb *MockBot) GetWhitelist() []string                        { return []string{} }
func (mb *MockBot) OnBlacklist(ch, p string) bool                 { return false }
func (mb *MockBot) URLFormat(title, url string) string            { return title + url }
func (mb *MockBot) CheckPassword(secret, password string) bool    { return true }
func (mb *MockBot) ListenAndServe()                               {}
func (mb *MockBot) PubToASub(subject string, payload interface{}) {}
func (mb *MockBot) TearDown() error {
	mb.store.Close()
	return os.Remove(mb.storeFile.Name())
}