mirror of
https://github.com/Luzifer/twitch-bot.git
synced 2024-11-14 02:52:42 +00:00
55 lines
1.8 KiB
Go
55 lines
1.8 KiB
Go
package punish
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/Luzifer/twitch-bot/v3/pkg/database"
|
|
)
|
|
|
|
func TestPunishmentRoundtrip(t *testing.T) {
|
|
dbc := database.GetTestDatabase(t)
|
|
require.NoError(t, dbc.DB().AutoMigrate(&punishLevel{}))
|
|
|
|
var (
|
|
channel = "#test"
|
|
user = "test"
|
|
uuid = "1befb33d-be89-4724-8ae1-0d465eb58007"
|
|
)
|
|
|
|
pl, err := getPunishment(dbc, channel, user, uuid)
|
|
assert.NoError(t, err, "query non-existent punishment")
|
|
assert.Equal(t, -1, pl.LastLevel, "check default level")
|
|
assert.Zero(t, pl.Executed, "check default time")
|
|
assert.Zero(t, pl.Cooldown, "check default cooldown")
|
|
|
|
err = setPunishment(dbc, channel, user, uuid, &levelConfig{
|
|
Cooldown: 500 * time.Millisecond,
|
|
Executed: time.Now().UTC(),
|
|
LastLevel: 1,
|
|
})
|
|
assert.NoError(t, err, "setting punishment")
|
|
|
|
pl, err = getPunishment(dbc, channel, user, uuid)
|
|
assert.NoError(t, err, "query existent punishment")
|
|
assert.Equal(t, 1, pl.LastLevel, "check level without cooldown")
|
|
|
|
time.Sleep(500 * time.Millisecond) // Wait for one cooldown to happen
|
|
|
|
pl, err = getPunishment(dbc, channel, user, uuid)
|
|
assert.NoError(t, err, "query existent punishment")
|
|
assert.Equal(t, 0, pl.LastLevel, "check level after one cooldown")
|
|
assert.NotZero(t, pl.Executed, "check non-zero-time after one cooldown")
|
|
assert.Equal(t, 500*time.Millisecond, pl.Cooldown, "check non-zero-cooldown after one cooldown")
|
|
|
|
time.Sleep(500 * time.Millisecond) // Wait for one cooldown to happen
|
|
|
|
pl, err = getPunishment(dbc, channel, user, uuid)
|
|
assert.NoError(t, err, "query existent punishment")
|
|
assert.Equal(t, -1, pl.LastLevel, "check level after two cooldown")
|
|
assert.Zero(t, pl.Executed, "check zero-time after two cooldown")
|
|
assert.Zero(t, pl.Cooldown, "check zero-cooldown after two cooldown")
|
|
}
|