// © 2016 the CatBase Authors under the WTFPL license. See AUTHORS for the list of authors. package bot import ( "log" "github.com/jmoiron/sqlx" "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 db *sqlx.DB Cfg config.Config Messages []string Actions []string } func (mb *MockBot) Config() *config.Config { return &mb.Cfg } func (mb *MockBot) DBVersion() int64 { return 1 } func (mb *MockBot) DB() *sqlx.DB { return mb.db } func (mb *MockBot) Who(string) []user.User { return []user.User{} } func (mb *MockBot) AddHandler(name string, f Handler) {} func (mb *MockBot) SendMessage(ch string, msg string) { mb.Messages = append(mb.Messages, msg) } func (mb *MockBot) SendAction(ch string, msg string) { mb.Actions = append(mb.Actions, msg) } 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 "" } 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(string, string, msg.Message) {} func NewMockBot() *MockBot { db, err := sqlx.Open("sqlite3_custom", ":memory:") if err != nil { log.Fatal("Failed to open database:", err) } b := MockBot{ db: db, Messages: make([]string, 0), Actions: make([]string, 0), } return &b }