mirror of
https://github.com/Luzifer/twitch-bot.git
synced 2024-09-20 02:22:55 +00:00
64 lines
1.6 KiB
Go
64 lines
1.6 KiB
Go
package counter
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/Luzifer/twitch-bot/v3/pkg/database"
|
|
)
|
|
|
|
func TestCounterStoreLoop(t *testing.T) {
|
|
dbc := database.GetTestDatabase(t)
|
|
dbc.DB().AutoMigrate(&Counter{})
|
|
|
|
counterName := "mytestcounter"
|
|
|
|
v, err := GetCounterValue(dbc, counterName)
|
|
assert.NoError(t, err, "reading non-existent counter")
|
|
assert.Equal(t, int64(0), v, "expecting 0 counter value on non-existent counter")
|
|
|
|
err = UpdateCounter(dbc, counterName, 5, true)
|
|
assert.NoError(t, err, "inserting counter")
|
|
|
|
err = UpdateCounter(dbc, counterName, 1, false)
|
|
assert.NoError(t, err, "updating counter")
|
|
|
|
v, err = GetCounterValue(dbc, counterName)
|
|
assert.NoError(t, err, "reading existent counter")
|
|
assert.Equal(t, int64(6), v, "expecting counter value on existing counter")
|
|
}
|
|
|
|
func TestCounterTopListAndRank(t *testing.T) {
|
|
dbc := database.GetTestDatabase(t)
|
|
dbc.DB().AutoMigrate(&Counter{})
|
|
|
|
counterTemplate := `#example:test:%v`
|
|
for i := 0; i < 6; i++ {
|
|
require.NoError(
|
|
t,
|
|
UpdateCounter(dbc, fmt.Sprintf(counterTemplate, i), int64(i), true),
|
|
"inserting counter %d", i,
|
|
)
|
|
}
|
|
|
|
cc, err := getCounterTopList(dbc, fmt.Sprintf(counterTemplate, ""), 3)
|
|
require.NoError(t, err)
|
|
assert.Len(t, cc, 3)
|
|
|
|
assert.Equal(t, []Counter{
|
|
{Name: "#example:test:5", Value: 5},
|
|
{Name: "#example:test:4", Value: 4},
|
|
{Name: "#example:test:3", Value: 3},
|
|
}, cc)
|
|
|
|
rank, count, err := getCounterRank(dbc,
|
|
fmt.Sprintf(counterTemplate, ""),
|
|
fmt.Sprintf(counterTemplate, 4))
|
|
require.NoError(t, err)
|
|
assert.Equal(t, int64(6), count)
|
|
assert.Equal(t, int64(2), rank)
|
|
}
|