cws
/
catbase
mirror of https://github.com/velour/catbase.git
1
0
Fork 0
Code Issues Releases Wiki Activity
catbase/plugins/stats/stats_test.go

298 lines
5.2 KiB
Go
Raw Normal View History

stats: WIP demo of stats * Added a plugin for stats * Using BoltDB for ease of not screwing with SQL ** Perhaps everything should be Bolt... * Should be relatively easy to add new stat counters: ** Create a function that makes the stat{} struct ** Add it to the list of active counters * Should be relatively easy to add a views page later
2017-06-05 22:37:46 +00:00
package stats
import (
"encoding/json"
"os"
"strings"
"testing"
stats: Add tests and clean a few things up * Make statFromDB return 0 if no stats are in the DB instead of error * Convert tests to use the assert library * Add tests for actual messages
2017-06-06 00:51:29 +00:00
"github.com/stretchr/testify/assert"
"github.com/velour/catbase/bot"
"github.com/velour/catbase/bot/msg"
"github.com/velour/catbase/bot/user"
stats: WIP demo of stats * Added a plugin for stats * Using BoltDB for ease of not screwing with SQL ** Perhaps everything should be Bolt... * Should be relatively easy to add new stat counters: ** Create a function that makes the stat{} struct ** Add it to the list of active counters * Should be relatively easy to add a views page later
2017-06-05 22:37:46 +00:00
)
stats: Add tests and clean a few things up * Make statFromDB return 0 if no stats are in the DB instead of error * Convert tests to use the assert library * Add tests for actual messages
2017-06-06 00:51:29 +00:00
var dbPath = "test.db"
stats: WIP demo of stats * Added a plugin for stats * Using BoltDB for ease of not screwing with SQL ** Perhaps everything should be Bolt... * Should be relatively easy to add new stat counters: ** Create a function that makes the stat{} struct ** Add it to the list of active counters * Should be relatively easy to add a views page later
2017-06-05 22:37:46 +00:00
func TestJSON(t *testing.T) {
expected := 5
b, err := json.Marshal(expected)
stats: Add tests and clean a few things up * Make statFromDB return 0 if no stats are in the DB instead of error * Convert tests to use the assert library * Add tests for actual messages
2017-06-06 00:51:29 +00:00
assert.Nil(t, err)
stats: WIP demo of stats * Added a plugin for stats * Using BoltDB for ease of not screwing with SQL ** Perhaps everything should be Bolt... * Should be relatively easy to add new stat counters: ** Create a function that makes the stat{} struct ** Add it to the list of active counters * Should be relatively easy to add a views page later
2017-06-05 22:37:46 +00:00
t.Logf("%+v", expected)
t.Log(string(b))
}
func TestValueConversion(t *testing.T) {
expected := value(5)
b, err := expected.Bytes()
stats: Add tests and clean a few things up * Make statFromDB return 0 if no stats are in the DB instead of error * Convert tests to use the assert library * Add tests for actual messages
2017-06-06 00:51:29 +00:00
assert.Nil(t, err)
stats: WIP demo of stats * Added a plugin for stats * Using BoltDB for ease of not screwing with SQL ** Perhaps everything should be Bolt... * Should be relatively easy to add new stat counters: ** Create a function that makes the stat{} struct ** Add it to the list of active counters * Should be relatively easy to add a views page later
2017-06-05 22:37:46 +00:00
t.Log(string(b))
actual, err := valueFromBytes(b)
stats: Add tests and clean a few things up * Make statFromDB return 0 if no stats are in the DB instead of error * Convert tests to use the assert library * Add tests for actual messages
2017-06-06 00:51:29 +00:00
assert.Nil(t, err)
stats: WIP demo of stats * Added a plugin for stats * Using BoltDB for ease of not screwing with SQL ** Perhaps everything should be Bolt... * Should be relatively easy to add new stat counters: ** Create a function that makes the stat{} struct ** Add it to the list of active counters * Should be relatively easy to add a views page later
2017-06-05 22:37:46 +00:00
stats: Add tests and clean a few things up * Make statFromDB return 0 if no stats are in the DB instead of error * Convert tests to use the assert library * Add tests for actual messages
2017-06-06 00:51:29 +00:00
assert.Equal(t, actual, expected)
stats: WIP demo of stats * Added a plugin for stats * Using BoltDB for ease of not screwing with SQL ** Perhaps everything should be Bolt... * Should be relatively easy to add new stat counters: ** Create a function that makes the stat{} struct ** Add it to the list of active counters * Should be relatively easy to add a views page later
2017-06-05 22:37:46 +00:00
}
stats: Add tests and clean a few things up * Make statFromDB return 0 if no stats are in the DB instead of error * Convert tests to use the assert library * Add tests for actual messages
2017-06-06 00:51:29 +00:00
func rmDB(t *testing.T) {
stats: WIP demo of stats * Added a plugin for stats * Using BoltDB for ease of not screwing with SQL ** Perhaps everything should be Bolt... * Should be relatively easy to add new stat counters: ** Create a function that makes the stat{} struct ** Add it to the list of active counters * Should be relatively easy to add a views page later
2017-06-05 22:37:46 +00:00
err := os.Remove(dbPath)
if err != nil && !strings.Contains(err.Error(), "no such file or directory") {
t.Fatal(err)
}
}
func TestWithDB(t *testing.T) {
stats: Add tests and clean a few things up * Make statFromDB return 0 if no stats are in the DB instead of error * Convert tests to use the assert library * Add tests for actual messages
2017-06-06 00:51:29 +00:00
rmDB(t)
stats: WIP demo of stats * Added a plugin for stats * Using BoltDB for ease of not screwing with SQL ** Perhaps everything should be Bolt... * Should be relatively easy to add new stat counters: ** Create a function that makes the stat{} struct ** Add it to the list of active counters * Should be relatively easy to add a views page later
2017-06-05 22:37:46 +00:00
t.Run("TestDBReadWrite", func(t *testing.T) {
bucket := "testBucket"
key := "testKey"
expected := stats{stat{
bucket,
key,
1,
}}
err := expected.toDB(dbPath)
stats: Add tests and clean a few things up * Make statFromDB return 0 if no stats are in the DB instead of error * Convert tests to use the assert library * Add tests for actual messages
2017-06-06 00:51:29 +00:00
assert.Nil(t, err)
stats: WIP demo of stats * Added a plugin for stats * Using BoltDB for ease of not screwing with SQL ** Perhaps everything should be Bolt... * Should be relatively easy to add new stat counters: ** Create a function that makes the stat{} struct ** Add it to the list of active counters * Should be relatively easy to add a views page later
2017-06-05 22:37:46 +00:00
actual, err := statFromDB(dbPath, bucket, key)
stats: Add tests and clean a few things up * Make statFromDB return 0 if no stats are in the DB instead of error * Convert tests to use the assert library * Add tests for actual messages
2017-06-06 00:51:29 +00:00
assert.Nil(t, err)
stats: WIP demo of stats * Added a plugin for stats * Using BoltDB for ease of not screwing with SQL ** Perhaps everything should be Bolt... * Should be relatively easy to add new stat counters: ** Create a function that makes the stat{} struct ** Add it to the list of active counters * Should be relatively easy to add a views page later
2017-06-05 22:37:46 +00:00
stats: Add tests and clean a few things up * Make statFromDB return 0 if no stats are in the DB instead of error * Convert tests to use the assert library * Add tests for actual messages
2017-06-06 00:51:29 +00:00
assert.Equal(t, actual, expected[0])
stats: WIP demo of stats * Added a plugin for stats * Using BoltDB for ease of not screwing with SQL ** Perhaps everything should be Bolt... * Should be relatively easy to add new stat counters: ** Create a function that makes the stat{} struct ** Add it to the list of active counters * Should be relatively easy to add a views page later
2017-06-05 22:37:46 +00:00
})
stats: Add tests and clean a few things up * Make statFromDB return 0 if no stats are in the DB instead of error * Convert tests to use the assert library * Add tests for actual messages
2017-06-06 00:51:29 +00:00
rmDB(t)
stats: WIP demo of stats * Added a plugin for stats * Using BoltDB for ease of not screwing with SQL ** Perhaps everything should be Bolt... * Should be relatively easy to add new stat counters: ** Create a function that makes the stat{} struct ** Add it to the list of active counters * Should be relatively easy to add a views page later
2017-06-05 22:37:46 +00:00
t.Run("TestDBAddStatInLoop", func(t *testing.T) {
bucket := "testBucket"
key := "testKey"
expected := value(25)
statPack := stats{stat{
bucket,
key,
5,
}}
for i := 0; i < 5; i++ {
err := statPack.toDB(dbPath)
stats: Add tests and clean a few things up * Make statFromDB return 0 if no stats are in the DB instead of error * Convert tests to use the assert library * Add tests for actual messages
2017-06-06 00:51:29 +00:00
assert.Nil(t, err)
stats: WIP demo of stats * Added a plugin for stats * Using BoltDB for ease of not screwing with SQL ** Perhaps everything should be Bolt... * Should be relatively easy to add new stat counters: ** Create a function that makes the stat{} struct ** Add it to the list of active counters * Should be relatively easy to add a views page later
2017-06-05 22:37:46 +00:00
}
actual, err := statFromDB(dbPath, bucket, key)
stats: Add tests and clean a few things up * Make statFromDB return 0 if no stats are in the DB instead of error * Convert tests to use the assert library * Add tests for actual messages
2017-06-06 00:51:29 +00:00
assert.Nil(t, err)
stats: WIP demo of stats * Added a plugin for stats * Using BoltDB for ease of not screwing with SQL ** Perhaps everything should be Bolt... * Should be relatively easy to add new stat counters: ** Create a function that makes the stat{} struct ** Add it to the list of active counters * Should be relatively easy to add a views page later
2017-06-05 22:37:46 +00:00
stats: Add tests and clean a few things up * Make statFromDB return 0 if no stats are in the DB instead of error * Convert tests to use the assert library * Add tests for actual messages
2017-06-06 00:51:29 +00:00
assert.Equal(t, actual.val, expected)
stats: WIP demo of stats * Added a plugin for stats * Using BoltDB for ease of not screwing with SQL ** Perhaps everything should be Bolt... * Should be relatively easy to add new stat counters: ** Create a function that makes the stat{} struct ** Add it to the list of active counters * Should be relatively easy to add a views page later
2017-06-05 22:37:46 +00:00
})
stats: Add tests and clean a few things up * Make statFromDB return 0 if no stats are in the DB instead of error * Convert tests to use the assert library * Add tests for actual messages
2017-06-06 00:51:29 +00:00
rmDB(t)
stats: WIP demo of stats * Added a plugin for stats * Using BoltDB for ease of not screwing with SQL ** Perhaps everything should be Bolt... * Should be relatively easy to add new stat counters: ** Create a function that makes the stat{} struct ** Add it to the list of active counters * Should be relatively easy to add a views page later
2017-06-05 22:37:46 +00:00
t.Run("TestDBAddStats", func(t *testing.T) {
bucket := "testBucket"
key := "testKey"
expected := value(5)
statPack := stats{
stat{
bucket,
key,
1,
},
stat{
bucket,
key,
1,
},
stat{
bucket,
key,
1,
},
stat{
bucket,
key,
1,
},
stat{
bucket,
key,
1,
},
}
err := statPack.toDB(dbPath)
stats: Add tests and clean a few things up * Make statFromDB return 0 if no stats are in the DB instead of error * Convert tests to use the assert library * Add tests for actual messages
2017-06-06 00:51:29 +00:00
assert.Nil(t, err)
stats: WIP demo of stats * Added a plugin for stats * Using BoltDB for ease of not screwing with SQL ** Perhaps everything should be Bolt... * Should be relatively easy to add new stat counters: ** Create a function that makes the stat{} struct ** Add it to the list of active counters * Should be relatively easy to add a views page later
2017-06-05 22:37:46 +00:00
actual, err := statFromDB(dbPath, bucket, key)
stats: Add tests and clean a few things up * Make statFromDB return 0 if no stats are in the DB instead of error * Convert tests to use the assert library * Add tests for actual messages
2017-06-06 00:51:29 +00:00
assert.Nil(t, err)
assert.Equal(t, actual.val, expected)
})
rmDB(t)
}
func makeMessage(payload string) msg.Message {
isCmd := strings.HasPrefix(payload, "!")
if isCmd {
payload = payload[1:]
}
return msg.Message{
User: &user.User{Name: "tester"},
Channel: "test",
Body: payload,
Command: isCmd,
}
}
func testUserCounter(t *testing.T, count int) {
expected := value(count)
mb := bot.NewMockBot()
mb.Cfg.Stats.DBPath = dbPath
s := New(mb)
assert.NotNil(t, s)
for i := 0; i < count; i++ {
s.Message(makeMessage("test"))
}
_, err := os.Stat(dbPath)
assert.Nil(t, err)
stat, err := statFromDB(mb.Config().Stats.DBPath, "user", "tester")
assert.Nil(t, err)
actual := stat.val
assert.Equal(t, actual, expected)
}
func TestMessages(t *testing.T) {
_, err := os.Stat(dbPath)
assert.NotNil(t, err)
t.Run("TestOneUserCounter", func(t *testing.T) {
count := 5
expected := value(count)
mb := bot.NewMockBot()
mb.Cfg.Stats.DBPath = dbPath
s := New(mb)
assert.NotNil(t, s)
for i := 0; i < count; i++ {
s.Message(makeMessage("test"))
stats: WIP demo of stats * Added a plugin for stats * Using BoltDB for ease of not screwing with SQL ** Perhaps everything should be Bolt... * Should be relatively easy to add new stat counters: ** Create a function that makes the stat{} struct ** Add it to the list of active counters * Should be relatively easy to add a views page later
2017-06-05 22:37:46 +00:00
}
stats: Add tests and clean a few things up * Make statFromDB return 0 if no stats are in the DB instead of error * Convert tests to use the assert library * Add tests for actual messages
2017-06-06 00:51:29 +00:00
_, err := os.Stat(dbPath)
assert.Nil(t, err)
stat, err := statFromDB(mb.Config().Stats.DBPath, "user", "tester")
assert.Nil(t, err)
actual := stat.val
assert.Equal(t, actual, expected)
})
rmDB(t)
t.Run("TestTenUserCounter", func(t *testing.T) {
count := 5
expected := value(count)
mb := bot.NewMockBot()
mb.Cfg.Stats.DBPath = dbPath
s := New(mb)
assert.NotNil(t, s)
for i := 0; i < count; i++ {
s.Message(makeMessage("test"))
}
_, err := os.Stat(dbPath)
assert.Nil(t, err)
stat, err := statFromDB(mb.Config().Stats.DBPath, "user", "tester")
assert.Nil(t, err)
actual := stat.val
assert.Equal(t, actual, expected)
})
rmDB(t)
t.Run("TestChannelCounter", func(t *testing.T) {
count := 5
expected := value(count)
mb := bot.NewMockBot()
mb.Cfg.Stats.DBPath = dbPath
s := New(mb)
assert.NotNil(t, s)
for i := 0; i < count; i++ {
s.Message(makeMessage("test"))
stats: WIP demo of stats * Added a plugin for stats * Using BoltDB for ease of not screwing with SQL ** Perhaps everything should be Bolt... * Should be relatively easy to add new stat counters: ** Create a function that makes the stat{} struct ** Add it to the list of active counters * Should be relatively easy to add a views page later
2017-06-05 22:37:46 +00:00
}
stats: Add tests and clean a few things up * Make statFromDB return 0 if no stats are in the DB instead of error * Convert tests to use the assert library * Add tests for actual messages
2017-06-06 00:51:29 +00:00
_, err := os.Stat(dbPath)
assert.Nil(t, err)
stat, err := statFromDB(mb.Config().Stats.DBPath, "channel", "test")
assert.Nil(t, err)
actual := stat.val
assert.Equal(t, actual, expected)
})
rmDB(t)
t.Run("TestSightingCounter", func(t *testing.T) {
count := 5
expected := value(count)
mb := bot.NewMockBot()
mb.Cfg.Stats.DBPath = dbPath
mb.Cfg.Stats.Sightings = []string{"user", "nobody"}
s := New(mb)
assert.NotNil(t, s)
for i := 0; i < count; i++ {
s.Message(makeMessage("user sighting"))
}
_, err := os.Stat(dbPath)
assert.Nil(t, err)
stat, err := statFromDB(mb.Config().Stats.DBPath, "sighting", "user")
assert.Nil(t, err)
actual := stat.val
assert.Equal(t, actual, expected)
})
rmDB(t)
t.Run("TestSightingCounterNoResults", func(t *testing.T) {
count := 5
expected := value(0)
mb := bot.NewMockBot()
mb.Cfg.Stats.DBPath = dbPath
mb.Cfg.Stats.Sightings = []string{}
s := New(mb)
assert.NotNil(t, s)
for i := 0; i < count; i++ {
s.Message(makeMessage("user sighting"))
}
_, err := os.Stat(dbPath)
assert.Nil(t, err)
stat, err := statFromDB(mb.Config().Stats.DBPath, "sighting", "user")
assert.Nil(t, err)
actual := stat.val
assert.Equal(t, actual, expected)
stats: WIP demo of stats * Added a plugin for stats * Using BoltDB for ease of not screwing with SQL ** Perhaps everything should be Bolt... * Should be relatively easy to add new stat counters: ** Create a function that makes the stat{} struct ** Add it to the list of active counters * Should be relatively easy to add a views page later
2017-06-05 22:37:46 +00:00
})
stats: Add tests and clean a few things up * Make statFromDB return 0 if no stats are in the DB instead of error * Convert tests to use the assert library * Add tests for actual messages
2017-06-06 00:51:29 +00:00
rmDB(t)
stats: WIP demo of stats * Added a plugin for stats * Using BoltDB for ease of not screwing with SQL ** Perhaps everything should be Bolt... * Should be relatively easy to add new stat counters: ** Create a function that makes the stat{} struct ** Add it to the list of active counters * Should be relatively easy to add a views page later
2017-06-05 22:37:46 +00:00
}