catbase/plugins/tldr/tldr_test.go

package tldr

import (
	"github.com/velour/catbase/plugins/cli"
	"os"
	"strconv"
	"strings"
	"testing"
	"time"

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

func init() {
	log.Logger = log.Logger.Output(zerolog.ConsoleWriter{Out: os.Stderr})
}

var ch = "test"

func makeMessageBy(payload, by string) bot.Request {
	isCmd := strings.HasPrefix(payload, "!")
	if isCmd {
		payload = payload[1:]
	}

	return bot.Request{
		Conn: &cli.CliPlugin{},
		Kind: bot.Message,
		Msg: msg.Message{
			User:    &user.User{Name: by},
			Channel: ch,
			Body:    payload,
			Command: isCmd,
		},
	}
}

func makeMessage(payload string) bot.Request {
	return makeMessageBy(payload, "tester")
}

func setup(t *testing.T) (*TLDRPlugin, *bot.MockBot) {
	mb := bot.NewMockBot()
	r := New(mb)
	return r, mb
}

func TestAddHistoryLimitsDays(t *testing.T) {
	c, _ := setup(t)
	hrs := 24
	expected := 24
	c.b.Config().Set("TLDR.HistorySize", "100")
	c.b.Config().Set("TLDR.KeepHours", strconv.Itoa(hrs))
	t0 := time.Now().Add(-time.Duration(hrs*2) * time.Hour)
	for i := 0; i < 48; i++ {
		hist := history{
			body:      "test",
			user:      "tester",
			timestamp: t0.Add(time.Duration(i) * time.Hour),
		}
		c.addHistory(ch, hist)
	}
	assert.Len(t, c.history[ch], expected, "%d != %d", len(c.history), expected)
}