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) }