3277 lines
110 KiB
Go
3277 lines
110 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"database/sql"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"runtime/debug"
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
c "github.com/Azareal/Gosora/common"
|
|
"github.com/Azareal/Gosora/common/gauth"
|
|
"github.com/Azareal/Gosora/common/phrases"
|
|
"github.com/Azareal/Gosora/routes"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
func miscinit(t *testing.T) {
|
|
if err := gloinit(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func recordMustExist(t *testing.T, err error, errmsg string, args ...interface{}) {
|
|
if err == ErrNoRows {
|
|
debug.PrintStack()
|
|
t.Errorf(errmsg, args...)
|
|
} else if err != nil {
|
|
debug.PrintStack()
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func recordMustNotExist(t *testing.T, err error, errmsg string, args ...interface{}) {
|
|
if err == nil {
|
|
debug.PrintStack()
|
|
t.Errorf(errmsg, args...)
|
|
} else if err != ErrNoRows {
|
|
debug.PrintStack()
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestUserStore(t *testing.T) {
|
|
miscinit(t)
|
|
if !c.PluginsInited {
|
|
c.InitPlugins()
|
|
}
|
|
|
|
var err error
|
|
uc := c.NewMemoryUserCache(c.Config.UserCacheCapacity)
|
|
c.Users, err = c.NewDefaultUserStore(uc)
|
|
expectNilErr(t, err)
|
|
uc.Flush()
|
|
userStoreTest(t, 2)
|
|
c.Users, err = c.NewDefaultUserStore(nil)
|
|
expectNilErr(t, err)
|
|
userStoreTest(t, 5)
|
|
}
|
|
func userStoreTest(t *testing.T, newUserID int) {
|
|
uc := c.Users.GetCache()
|
|
// Go doesn't have short-circuiting, so this'll allow us to do one liner tests
|
|
cacheLength := func(uc c.UserCache) int {
|
|
if uc == nil {
|
|
return 0
|
|
}
|
|
return uc.Length()
|
|
}
|
|
isCacheLengthZero := func(uc c.UserCache) bool {
|
|
return cacheLength(uc) == 0
|
|
}
|
|
ex, exf := exp(t), expf(t)
|
|
exf(isCacheLengthZero(uc), "The initial ucache length should be zero, not %d", cacheLength(uc))
|
|
|
|
_, err := c.Users.Get(-1)
|
|
recordMustNotExist(t, err, "UID #-1 shouldn't exist")
|
|
exf(isCacheLengthZero(uc), "We found %d items in the user cache and it's supposed to be empty", cacheLength(uc))
|
|
_, err = c.Users.Get(0)
|
|
recordMustNotExist(t, err, "UID #0 shouldn't exist")
|
|
exf(isCacheLengthZero(uc), "We found %d items in the user cache and it's supposed to be empty", cacheLength(uc))
|
|
|
|
user, err := c.Users.Get(1)
|
|
recordMustExist(t, err, "Couldn't find UID #1")
|
|
|
|
expectW := func(cond, expec bool, prefix, suffix string) {
|
|
midfix := "should not be"
|
|
if expec {
|
|
midfix = "should be"
|
|
}
|
|
ex(cond, prefix+" "+midfix+" "+suffix)
|
|
}
|
|
|
|
// TODO: Add email checks too? Do them separately?
|
|
expectUser := func(u *c.User, uid int, name string, group int, super, admin, mod, banned bool) {
|
|
exf(u.ID == uid, "u.ID should be %d. Got '%d' instead.", uid, u.ID)
|
|
exf(u.Name == name, "u.Name should be '%s', not '%s'", name, u.Name)
|
|
expectW(u.Group == group, true, u.Name, "in group"+strconv.Itoa(group))
|
|
expectW(u.IsSuperAdmin == super, super, u.Name, "a super admin")
|
|
expectW(u.IsAdmin == admin, admin, u.Name, "an admin")
|
|
expectW(u.IsSuperMod == mod, mod, u.Name, "a super mod")
|
|
expectW(u.IsMod == mod, mod, u.Name, "a mod")
|
|
expectW(u.IsBanned == banned, banned, u.Name, "banned")
|
|
}
|
|
expectUser(user, 1, "Admin", 1, true, true, true, false)
|
|
|
|
user, err = c.Users.GetByName("Admin")
|
|
recordMustExist(t, err, "Couldn't find user 'Admin'")
|
|
expectUser(user, 1, "Admin", 1, true, true, true, false)
|
|
us, err := c.Users.BulkGetByName([]string{"Admin"})
|
|
recordMustExist(t, err, "Couldn't find user 'Admin'")
|
|
exf(len(us) == 1, "len(us) should be 1, not %d", len(us))
|
|
expectUser(us[0], 1, "Admin", 1, true, true, true, false)
|
|
|
|
_, err = c.Users.Get(newUserID)
|
|
recordMustNotExist(t, err, fmt.Sprintf("UID #%d shouldn't exist", newUserID))
|
|
|
|
// TODO: GetByName tests for newUserID
|
|
|
|
if uc != nil {
|
|
expectIntToBeX(t, uc.Length(), 1, "User cache length should be 1, not %d")
|
|
_, err = uc.Get(-1)
|
|
recordMustNotExist(t, err, "UID #-1 shouldn't exist, even in the cache")
|
|
_, err = uc.Get(0)
|
|
recordMustNotExist(t, err, "UID #0 shouldn't exist, even in the cache")
|
|
user, err = uc.Get(1)
|
|
recordMustExist(t, err, "Couldn't find UID #1 in the cache")
|
|
|
|
exf(user.ID == 1, "user.ID does not match the requested UID. Got '%d' instead.", user.ID)
|
|
exf(user.Name == "Admin", "user.Name should be 'Admin', not '%s'", user.Name)
|
|
|
|
_, err = uc.Get(newUserID)
|
|
recordMustNotExist(t, err, "UID #%d shouldn't exist, even in the cache", newUserID)
|
|
uc.Flush()
|
|
expectIntToBeX(t, uc.Length(), 0, "User cache length should be 0, not %d")
|
|
}
|
|
|
|
// TODO: Lock onto the specific error type. Is this even possible without sacrificing the detailed information in the error message?
|
|
bulkGetMapEmpty := func(id int) {
|
|
userList, _ := c.Users.BulkGetMap([]int{id})
|
|
exf(len(userList) == 0, "The userList length should be 0, not %d", len(userList))
|
|
exf(isCacheLengthZero(uc), "User cache length should be 0, not %d", cacheLength(uc))
|
|
}
|
|
bulkGetMapEmpty(-1)
|
|
bulkGetMapEmpty(0)
|
|
|
|
userList, _ := c.Users.BulkGetMap([]int{1})
|
|
exf(len(userList) == 1, "Returned map should have one result (UID #1), not %d", len(userList))
|
|
user, ok := userList[1]
|
|
if !ok {
|
|
t.Error("We couldn't find UID #1 in the returned map")
|
|
t.Error("userList", userList)
|
|
return
|
|
}
|
|
exf(user.ID == 1, "user.ID does not match the requested UID. Got '%d' instead.", user.ID)
|
|
|
|
if uc != nil {
|
|
expectIntToBeX(t, uc.Length(), 1, "User cache length should be 1, not %d")
|
|
user, err = uc.Get(1)
|
|
recordMustExist(t, err, "Couldn't find UID #1 in the cache")
|
|
|
|
exf(user.ID == 1, "user.ID does not match the requested UID. Got '%d' instead.", user.ID)
|
|
uc.Flush()
|
|
}
|
|
|
|
ex(!c.Users.Exists(-1), "UID #-1 shouldn't exist")
|
|
ex(!c.Users.Exists(0), "UID #0 shouldn't exist")
|
|
ex(c.Users.Exists(1), "UID #1 should exist")
|
|
exf(!c.Users.Exists(newUserID), "UID #%d shouldn't exist", newUserID)
|
|
|
|
exf(isCacheLengthZero(uc), "User cache length should be 0, not %d", cacheLength(uc))
|
|
expectIntToBeX(t, c.Users.Count(), 1, "The number of users should be 1, not %d")
|
|
searchUser := func(name, email string, gid, count int) {
|
|
f := func(name, email string, gid, count int, m string) {
|
|
expectIntToBeX(t, c.Users.CountSearch(name, email, gid), count, "The number of users for "+m+", not %d")
|
|
}
|
|
f(name, email, 0, count, fmt.Sprintf("name '%s' and email '%s' should be %d", name, email, count))
|
|
f(name, "", 0, count, fmt.Sprintf("name '%s' should be %d", name, count))
|
|
f("", email, 0, count, fmt.Sprintf("email '%s' should be %d", email, count))
|
|
|
|
f2 := func(name, email string, gid, offset int, m string, args ...interface{}) {
|
|
ulist, err := c.Users.SearchOffset(name, email, gid, offset, 15)
|
|
expectNilErr(t, err)
|
|
expectIntToBeX(t, len(ulist), count, "The number of users for "+fmt.Sprintf(m, args...)+", not %d")
|
|
}
|
|
f2(name, email, 0, 0, "name '%s' and email '%s' should be %d", name, email, count)
|
|
f2(name, "", 0, 0, "name '%s' should be %d", name, count)
|
|
f2("", email, 0, 0, "email '%s' should be %d", email, count)
|
|
|
|
count = 0
|
|
f2(name, email, 0, 10, "name '%s' and email '%s' should be %d", name, email, count)
|
|
f2(name, "", 0, 10, "name '%s' should be %d", name, count)
|
|
f2("", email, 0, 10, "email '%s' should be %d", email, count)
|
|
|
|
f2(name, email, 999, 0, "name '%s' and email '%s' should be %d", name, email, 0)
|
|
f2(name, "", 999, 0, "name '%s' should be %d", name, 0)
|
|
f2("", email, 999, 0, "email '%s' should be %d", email, 0)
|
|
|
|
f2(name, email, 999, 10, "name '%s' and email '%s' should be %d", name, email, 0)
|
|
f2(name, "", 999, 10, "name '%s' should be %d", name, 0)
|
|
f2("", email, 999, 10, "email '%s' should be %d", email, 0)
|
|
}
|
|
searchUser("Sam", "sam@localhost.loc", 0, 0)
|
|
// TODO: CountSearch gid test
|
|
|
|
awaitingActivation := 5
|
|
// TODO: Write tests for the registration validators
|
|
uid, err := c.Users.Create("Sam", "ReallyBadPassword", "sam@localhost.loc", awaitingActivation, false)
|
|
expectNilErr(t, err)
|
|
exf(uid == newUserID, "The UID of the new user should be %d not %d", newUserID, uid)
|
|
exf(c.Users.Exists(newUserID), "UID #%d should exist", newUserID)
|
|
expectIntToBeX(t, c.Users.Count(), 2, "The number of users should be 2, not %d")
|
|
searchUser("Sam", "sam@localhost.loc", 0, 1)
|
|
// TODO: CountSearch gid test
|
|
|
|
user, err = c.Users.Get(newUserID)
|
|
recordMustExist(t, err, "Couldn't find UID #%d", newUserID)
|
|
expectUser(user, newUserID, "Sam", 5, false, false, false, false)
|
|
|
|
if uc != nil {
|
|
expectIntToBeX(t, uc.Length(), 1, "User cache length should be 1, not %d")
|
|
user, err = uc.Get(newUserID)
|
|
recordMustExist(t, err, "Couldn't find UID #%d in the cache", newUserID)
|
|
exf(user.ID == newUserID, "user.ID does not match the requested UID. Got '%d' instead.", user.ID)
|
|
}
|
|
|
|
userList, _ = c.Users.BulkGetMap([]int{1, uid})
|
|
exf(len(userList) == 2, "Returned map should have 2 results, not %d", len(userList))
|
|
// TODO: More tests on userList
|
|
|
|
{
|
|
userList, _ := c.Users.BulkGetByName([]string{"Admin", "Sam"})
|
|
exf(len(userList) == 2, "Returned list should have 2 results, not %d", len(userList))
|
|
}
|
|
|
|
if uc != nil {
|
|
expectIntToBeX(t, uc.Length(), 2, "User cache length should be 2, not %d")
|
|
user, err = uc.Get(1)
|
|
recordMustExist(t, err, "Couldn't find UID #%d in the cache", 1)
|
|
exf(user.ID == 1, "user.ID does not match the requested UID. Got '%d' instead.", user.ID)
|
|
user, err = uc.Get(newUserID)
|
|
recordMustExist(t, err, "Couldn't find UID #%d in the cache", newUserID)
|
|
exf(user.ID == newUserID, "user.ID does not match the requested UID. Got '%d' instead.", user.ID)
|
|
uc.Flush()
|
|
}
|
|
|
|
user, err = c.Users.Get(newUserID)
|
|
recordMustExist(t, err, "Couldn't find UID #%d", newUserID)
|
|
expectUser(user, newUserID, "Sam", 5, false, false, false, false)
|
|
|
|
if uc != nil {
|
|
expectIntToBeX(t, uc.Length(), 1, "User cache length should be 1, not %d")
|
|
user, err = uc.Get(newUserID)
|
|
recordMustExist(t, err, "Couldn't find UID #%d in the cache", newUserID)
|
|
exf(user.ID == newUserID, "user.ID does not match the requested UID. Got '%d' instead.", user.ID)
|
|
}
|
|
|
|
expectNilErr(t, user.Activate())
|
|
expectIntToBeX(t, user.Group, 5, "Sam should still be in group 5 in this copy")
|
|
|
|
// ? - What if we change the caching mechanism so it isn't hard purged and reloaded? We'll deal with that when we come to it, but for now, this is a sign of a cache bug
|
|
afterUserFlush := func(uid int) {
|
|
if uc != nil {
|
|
expectIntToBeX(t, uc.Length(), 0, "User cache length should be 0, not %d")
|
|
_, err = uc.Get(uid)
|
|
recordMustNotExist(t, err, "UID #%d shouldn't be in the cache", uid)
|
|
}
|
|
}
|
|
afterUserFlush(newUserID)
|
|
|
|
user, err = c.Users.Get(newUserID)
|
|
recordMustExist(t, err, "Couldn't find UID #%d", newUserID)
|
|
expectUser(user, newUserID, "Sam", c.Config.DefaultGroup, false, false, false, false)
|
|
|
|
// Permanent ban
|
|
duration, _ := time.ParseDuration("0")
|
|
|
|
// TODO: Attempt a double ban, double activation, and double unban
|
|
expectNilErr(t, user.Ban(duration, 1))
|
|
exf(user.Group == c.Config.DefaultGroup, "Sam should be in group %d, not %d", c.Config.DefaultGroup, user.Group)
|
|
afterUserFlush(newUserID)
|
|
|
|
user, err = c.Users.Get(newUserID)
|
|
recordMustExist(t, err, "Couldn't find UID #%d", newUserID)
|
|
expectUser(user, newUserID, "Sam", c.BanGroup, false, false, false, true)
|
|
|
|
// TODO: Do tests against the scheduled updates table and the task system to make sure the ban exists there and gets revoked when it should
|
|
|
|
expectNilErr(t, user.Unban())
|
|
expectIntToBeX(t, user.Group, c.BanGroup, "Sam should still be in the ban group in this copy")
|
|
afterUserFlush(newUserID)
|
|
|
|
user, err = c.Users.Get(newUserID)
|
|
recordMustExist(t, err, "Couldn't find UID #%d", newUserID)
|
|
expectUser(user, newUserID, "Sam", c.Config.DefaultGroup, false, false, false, false)
|
|
|
|
reportsForumID := 1 // TODO: Use the constant in common?
|
|
generalForumID := 2
|
|
dummyResponseRecorder := httptest.NewRecorder()
|
|
bytesBuffer := bytes.NewBuffer([]byte(""))
|
|
dummyRequest1 := httptest.NewRequest("", "/forum/"+strconv.Itoa(reportsForumID), bytesBuffer)
|
|
dummyRequest2 := httptest.NewRequest("", "/forum/"+strconv.Itoa(generalForumID), bytesBuffer)
|
|
var user2 *c.User
|
|
|
|
changeGroupTest := func(oldGroup, newGroup int) {
|
|
expectNilErr(t, user.ChangeGroup(newGroup))
|
|
// ! I don't think ChangeGroup should be changing the value of user... Investigate this.
|
|
ex(oldGroup == user.Group, "Someone's mutated this pointer elsewhere")
|
|
|
|
user, err = c.Users.Get(newUserID)
|
|
recordMustExist(t, err, "Couldn't find UID #%d", newUserID)
|
|
user2 = c.BlankUser()
|
|
*user2 = *user
|
|
}
|
|
|
|
changeGroupTest2 := func(rank string, firstShouldBe, secondShouldBe bool) {
|
|
head, e := c.UserCheck(dummyResponseRecorder, dummyRequest1, user)
|
|
if e != nil {
|
|
t.Fatal(e)
|
|
}
|
|
head2, e := c.UserCheck(dummyResponseRecorder, dummyRequest2, user2)
|
|
if e != nil {
|
|
t.Fatal(e)
|
|
}
|
|
ferr := c.ForumUserCheck(head, dummyResponseRecorder, dummyRequest1, user, reportsForumID)
|
|
ex(ferr == nil, "There shouldn't be any errors in forumUserCheck")
|
|
ex(user.Perms.ViewTopic == firstShouldBe, rank+" should be able to access the reports forum")
|
|
ferr = c.ForumUserCheck(head2, dummyResponseRecorder, dummyRequest2, user2, generalForumID)
|
|
ex(ferr == nil, "There shouldn't be any errors in forumUserCheck")
|
|
ex(user2.Perms.ViewTopic == secondShouldBe, "Sam should be able to access the general forum")
|
|
}
|
|
|
|
changeGroupTest(c.Config.DefaultGroup, 1)
|
|
expectUser(user, newUserID, "Sam", 1, false, true, true, false)
|
|
changeGroupTest2("Admins", true, true)
|
|
|
|
changeGroupTest(1, 2)
|
|
expectUser(user, newUserID, "Sam", 2, false, false, true, false)
|
|
changeGroupTest2("Mods", true, true)
|
|
|
|
changeGroupTest(2, 3)
|
|
expectUser(user, newUserID, "Sam", 3, false, false, false, false)
|
|
changeGroupTest2("Members", false, true)
|
|
ex(user.Perms.ViewTopic != user2.Perms.ViewTopic, "user.Perms.ViewTopic and user2.Perms.ViewTopic should never match")
|
|
|
|
changeGroupTest(3, 4)
|
|
expectUser(user, newUserID, "Sam", 4, false, false, false, true)
|
|
changeGroupTest2("Members", false, true)
|
|
|
|
changeGroupTest(4, 5)
|
|
expectUser(user, newUserID, "Sam", 5, false, false, false, false)
|
|
changeGroupTest2("Members", false, true)
|
|
|
|
changeGroupTest(5, 6)
|
|
expectUser(user, newUserID, "Sam", 6, false, false, false, false)
|
|
changeGroupTest2("Members", false, true)
|
|
|
|
err = user.ChangeGroup(c.Config.DefaultGroup)
|
|
expectNilErr(t, err)
|
|
ex(user.Group == 6, "Someone's mutated this pointer elsewhere")
|
|
|
|
exf(user.LastIP == "", "user.LastIP should be blank not %s", user.LastIP)
|
|
expectNilErr(t, user.UpdateIP("::1"))
|
|
user, err = c.Users.Get(newUserID)
|
|
recordMustExist(t, err, "Couldn't find UID #%d", newUserID)
|
|
exf(user.LastIP == "::1", "user.LastIP should be %s not %s", "::1", user.LastIP)
|
|
expectNilErr(t, c.Users.ClearLastIPs())
|
|
expectNilErr(t, c.Users.Reload(newUserID))
|
|
user, err = c.Users.Get(newUserID)
|
|
recordMustExist(t, err, "Couldn't find UID #%d", newUserID)
|
|
exf(user.LastIP == "", "user.LastIP should be blank not %s", user.LastIP)
|
|
|
|
expectNilErr(t, user.Delete())
|
|
exf(!c.Users.Exists(newUserID), "UID #%d should no longer exist", newUserID)
|
|
afterUserFlush(newUserID)
|
|
expectIntToBeX(t, c.Users.Count(), 1, "The number of users should be 1, not %d")
|
|
searchUser("Sam", "sam@localhost.loc", 0, 0)
|
|
// TODO: CountSearch gid test
|
|
|
|
_, err = c.Users.Get(newUserID)
|
|
recordMustNotExist(t, err, "UID #%d shouldn't exist", newUserID)
|
|
|
|
// And a unicode test, even though I doubt it'll fail
|
|
uid, err = c.Users.Create("サム", "😀😀😀", "sam@localhost.loc", awaitingActivation, false)
|
|
expectNilErr(t, err)
|
|
exf(uid == newUserID+1, "The UID of the new user should be %d", newUserID+1)
|
|
exf(c.Users.Exists(newUserID+1), "UID #%d should exist", newUserID+1)
|
|
|
|
user, err = c.Users.Get(newUserID + 1)
|
|
recordMustExist(t, err, "Couldn't find UID #%d", newUserID+1)
|
|
expectUser(user, newUserID+1, "サム", 5, false, false, false, false)
|
|
|
|
expectNilErr(t, user.Delete())
|
|
exf(!c.Users.Exists(newUserID+1), "UID #%d should no longer exist", newUserID+1)
|
|
|
|
// MySQL utf8mb4 username test
|
|
uid, err = c.Users.Create("😀😀😀", "😀😀😀", "sam@localhost.loc", awaitingActivation, false)
|
|
expectNilErr(t, err)
|
|
exf(uid == newUserID+2, "The UID of the new user should be %d", newUserID+2)
|
|
exf(c.Users.Exists(newUserID+2), "UID #%d should exist", newUserID+2)
|
|
|
|
user, err = c.Users.Get(newUserID + 2)
|
|
recordMustExist(t, err, "Couldn't find UID #%d", newUserID+1)
|
|
expectUser(user, newUserID+2, "😀😀😀", 5, false, false, false, false)
|
|
|
|
expectNilErr(t, user.Delete())
|
|
exf(!c.Users.Exists(newUserID+2), "UID #%d should no longer exist", newUserID+2)
|
|
|
|
// TODO: Add unicode login tests somewhere? Probably with the rest of the auth tests
|
|
// TODO: Add tests for the Cache* methods
|
|
}
|
|
|
|
// TODO: Add an error message to this?
|
|
func expectNilErr(t *testing.T, item error) {
|
|
if item != nil {
|
|
debug.PrintStack()
|
|
t.Fatal(item)
|
|
}
|
|
}
|
|
|
|
func expectIntToBeX(t *testing.T, item, expect int, errmsg string) {
|
|
if item != expect {
|
|
debug.PrintStack()
|
|
t.Fatalf(errmsg, item)
|
|
}
|
|
}
|
|
|
|
func expect(t *testing.T, item bool, errmsg string) {
|
|
if !item {
|
|
debug.PrintStack()
|
|
t.Fatal(errmsg)
|
|
}
|
|
}
|
|
|
|
func expectf(t *testing.T, item bool, errmsg string, args ...interface{}) {
|
|
if !item {
|
|
debug.PrintStack()
|
|
t.Fatalf(errmsg, args...)
|
|
}
|
|
}
|
|
|
|
func exp(t *testing.T) func(bool, string) {
|
|
return func(val bool, errmsg string) {
|
|
if !val {
|
|
debug.PrintStack()
|
|
t.Fatal(errmsg)
|
|
}
|
|
}
|
|
}
|
|
|
|
func expf(t *testing.T) func(bool, string, ...interface{}) {
|
|
return func(val bool, errmsg string, params ...interface{}) {
|
|
if !val {
|
|
debug.PrintStack()
|
|
t.Fatalf(errmsg, params...)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestPermsMiddleware(t *testing.T) {
|
|
miscinit(t)
|
|
if !c.PluginsInited {
|
|
c.InitPlugins()
|
|
}
|
|
|
|
dummyResponseRecorder := httptest.NewRecorder()
|
|
bytesBuffer := bytes.NewBuffer([]byte(""))
|
|
dummyRequest := httptest.NewRequest("", "/forum/1", bytesBuffer)
|
|
user := c.BlankUser()
|
|
ex := exp(t)
|
|
|
|
f := func(ff func(w http.ResponseWriter, r *http.Request, u *c.User) c.RouteError) bool {
|
|
ferr := ff(dummyResponseRecorder, dummyRequest, user)
|
|
return ferr == nil
|
|
}
|
|
|
|
ex(!f(c.SuperModOnly), "Blank users shouldn't be supermods")
|
|
user.IsSuperMod = false
|
|
ex(!f(c.SuperModOnly), "Non-supermods shouldn't be allowed through supermod gates")
|
|
user.IsSuperMod = true
|
|
ex(f(c.SuperModOnly), "Supermods should be allowed through supermod gates")
|
|
|
|
// TODO: Loop over the Control Panel routes and make sure only supermods can get in
|
|
|
|
user = c.BlankUser()
|
|
|
|
ex(!f(c.MemberOnly), "Blank users shouldn't be considered loggedin")
|
|
user.Loggedin = false
|
|
ex(!f(c.MemberOnly), "Guests shouldn't be able to access member areas")
|
|
user.Loggedin = true
|
|
ex(f(c.MemberOnly), "Logged in users should be able to access member areas")
|
|
|
|
// TODO: Loop over the /user/ routes and make sure only members can access the ones other than /user/username
|
|
|
|
user = c.BlankUser()
|
|
|
|
ex(!f(c.AdminOnly), "Blank users shouldn't be considered admins")
|
|
user.IsAdmin = false
|
|
ex(!f(c.AdminOnly), "Non-admins shouldn't be able to access admin areas")
|
|
user.IsAdmin = true
|
|
ex(f(c.AdminOnly), "Admins should be able to access admin areas")
|
|
|
|
user = c.BlankUser()
|
|
|
|
ex(!f(c.SuperAdminOnly), "Blank users shouldn't be considered super admins")
|
|
user.IsSuperAdmin = false
|
|
ex(!f(c.SuperAdminOnly), "Non-super admins shouldn't be allowed through the super admin gate")
|
|
user.IsSuperAdmin = true
|
|
ex(f(c.SuperAdminOnly), "Super admins should be allowed through super admin gates")
|
|
|
|
// TODO: Make sure only super admins can access the backups route
|
|
|
|
//dummyResponseRecorder = httptest.NewRecorder()
|
|
//bytesBuffer = bytes.NewBuffer([]byte(""))
|
|
//dummyRequest = httptest.NewRequest("", "/panel/backups/", bytesBuffer)
|
|
}
|
|
|
|
func TestTopicStore(t *testing.T) {
|
|
miscinit(t)
|
|
if !c.PluginsInited {
|
|
c.InitPlugins()
|
|
}
|
|
|
|
var err error
|
|
tcache := c.NewMemoryTopicCache(c.Config.TopicCacheCapacity)
|
|
c.Topics, err = c.NewDefaultTopicStore(tcache)
|
|
expectNilErr(t, err)
|
|
c.Config.DisablePostIP = false
|
|
topicStoreTest(t, 2, "::1")
|
|
c.Config.DisablePostIP = true
|
|
topicStoreTest(t, 3, "")
|
|
|
|
c.Topics, err = c.NewDefaultTopicStore(nil)
|
|
expectNilErr(t, err)
|
|
c.Config.DisablePostIP = false
|
|
topicStoreTest(t, 4, "::1")
|
|
c.Config.DisablePostIP = true
|
|
topicStoreTest(t, 5, "")
|
|
}
|
|
func topicStoreTest(t *testing.T, newID int, ip string) {
|
|
var topic *c.Topic
|
|
var err error
|
|
ex, exf := exp(t), expf(t)
|
|
|
|
_, err = c.Topics.Get(-1)
|
|
recordMustNotExist(t, err, "TID #-1 shouldn't exist")
|
|
_, err = c.Topics.Get(0)
|
|
recordMustNotExist(t, err, "TID #0 shouldn't exist")
|
|
|
|
topic, err = c.Topics.Get(1)
|
|
recordMustExist(t, err, "Couldn't find TID #1")
|
|
exf(topic.ID == 1, "topic.ID does not match the requested TID. Got '%d' instead.", topic.ID)
|
|
|
|
// TODO: Add BulkGetMap() to the TopicStore
|
|
|
|
ex(!c.Topics.Exists(-1), "TID #-1 shouldn't exist")
|
|
ex(!c.Topics.Exists(0), "TID #0 shouldn't exist")
|
|
ex(c.Topics.Exists(1), "TID #1 should exist")
|
|
|
|
count := c.Topics.Count()
|
|
exf(count == 1, "Global count for topics should be 1, not %d", count)
|
|
|
|
//Create(fid int, topicName string, content string, uid int, ip string) (tid int, err error)
|
|
tid, err := c.Topics.Create(2, "Test Topic", "Topic Content", 1, ip)
|
|
expectNilErr(t, err)
|
|
exf(tid == newID, "TID for the new topic should be %d, not %d", newID, tid)
|
|
exf(c.Topics.Exists(newID), "TID #%d should exist", newID)
|
|
|
|
count = c.Topics.Count()
|
|
exf(count == 2, "Global count for topics should be 2, not %d", count)
|
|
|
|
iFrag := func(cond bool) string {
|
|
if !cond {
|
|
return "n't"
|
|
}
|
|
return ""
|
|
}
|
|
|
|
testTopic := func(tid int, title, content string, createdBy int, ip string, parentID int, isClosed, sticky bool) {
|
|
topic, err = c.Topics.Get(tid)
|
|
recordMustExist(t, err, fmt.Sprintf("Couldn't find TID #%d", tid))
|
|
exf(topic.ID == tid, "topic.ID does not match the requested TID. Got '%d' instead.", topic.ID)
|
|
exf(topic.GetID() == tid, "topic.ID does not match the requested TID. Got '%d' instead.", topic.GetID())
|
|
exf(topic.Title == title, "The topic's name should be '%s', not %s", title, topic.Title)
|
|
exf(topic.Content == content, "The topic's body should be '%s', not %s", content, topic.Content)
|
|
exf(topic.CreatedBy == createdBy, "The topic's creator should be %d, not %d", createdBy, topic.CreatedBy)
|
|
exf(topic.IP == ip, "The topic's IP should be '%s', not %s", ip, topic.IP)
|
|
exf(topic.ParentID == parentID, "The topic's parent forum should be %d, not %d", parentID, topic.ParentID)
|
|
exf(topic.IsClosed == isClosed, "This topic should%s be locked", iFrag(topic.IsClosed))
|
|
exf(topic.Sticky == sticky, "This topic should%s be sticky", iFrag(topic.Sticky))
|
|
exf(topic.GetTable() == "topics", "The topic's table should be 'topics', not %s", topic.GetTable())
|
|
}
|
|
|
|
tc := c.Topics.GetCache()
|
|
shouldNotBeIn := func(tid int) {
|
|
if tc != nil {
|
|
_, err = tc.Get(tid)
|
|
recordMustNotExist(t, err, "Topic cache should be empty")
|
|
}
|
|
}
|
|
if tc != nil {
|
|
_, err = tc.Get(newID)
|
|
expectNilErr(t, err)
|
|
}
|
|
|
|
testTopic(newID, "Test Topic", "Topic Content", 1, ip, 2, false, false)
|
|
|
|
expectNilErr(t, topic.Lock())
|
|
shouldNotBeIn(newID)
|
|
testTopic(newID, "Test Topic", "Topic Content", 1, ip, 2, true, false)
|
|
|
|
expectNilErr(t, topic.Unlock())
|
|
shouldNotBeIn(newID)
|
|
testTopic(newID, "Test Topic", "Topic Content", 1, ip, 2, false, false)
|
|
|
|
expectNilErr(t, topic.Stick())
|
|
shouldNotBeIn(newID)
|
|
testTopic(newID, "Test Topic", "Topic Content", 1, ip, 2, false, true)
|
|
|
|
expectNilErr(t, topic.Unstick())
|
|
shouldNotBeIn(newID)
|
|
testTopic(newID, "Test Topic", "Topic Content", 1, ip, 2, false, false)
|
|
|
|
expectNilErr(t, topic.MoveTo(1))
|
|
shouldNotBeIn(newID)
|
|
testTopic(newID, "Test Topic", "Topic Content", 1, ip, 1, false, false)
|
|
// TODO: Add more tests for more *Topic methods
|
|
|
|
expectNilErr(t, c.Topics.ClearIPs())
|
|
expectNilErr(t, c.Topics.Reload(topic.ID))
|
|
testTopic(newID, "Test Topic", "Topic Content", 1, "", 1, false, false)
|
|
// TODO: Add more tests for more *Topic methods
|
|
|
|
expectNilErr(t, topic.Delete())
|
|
shouldNotBeIn(newID)
|
|
|
|
_, err = c.Topics.Get(newID)
|
|
recordMustNotExist(t, err, fmt.Sprintf("TID #%d shouldn't exist", newID))
|
|
exf(!c.Topics.Exists(newID), "TID #%d shouldn't exist", newID)
|
|
|
|
// TODO: Test topic creation and retrieving that created topic plus reload and inspecting the cache
|
|
}
|
|
|
|
func TestForumStore(t *testing.T) {
|
|
miscinit(t)
|
|
if !c.PluginsInited {
|
|
c.InitPlugins()
|
|
}
|
|
ex, exf := exp(t), expf(t)
|
|
// TODO: Test ForumStore.Reload
|
|
|
|
fcache, ok := c.Forums.(c.ForumCache)
|
|
ex(ok, "Unable to cast ForumStore to ForumCache")
|
|
ex(c.Forums.Count() == 2, "The forumstore global count should be 2")
|
|
ex(fcache.Length() == 2, "The forum cache length should be 2")
|
|
|
|
_, err := c.Forums.Get(-1)
|
|
recordMustNotExist(t, err, "FID #-1 shouldn't exist")
|
|
_, err = c.Forums.Get(0)
|
|
recordMustNotExist(t, err, "FID #0 shouldn't exist")
|
|
|
|
testForum := func(f *c.Forum, fid int, name string, active bool, desc string) {
|
|
exf(f.ID == fid, "forum.ID should be %d, not %d.", fid, f.ID)
|
|
// TODO: Check the preset and forum permissions
|
|
exf(f.Name == name, "forum.Name should be %s, not %s", name, f.Name)
|
|
str := ""
|
|
if !active {
|
|
str = "n't"
|
|
}
|
|
exf(f.Active == active, "The reports forum should%s be active", str)
|
|
exf(f.Desc == desc, "forum.Desc should be '%s' not '%s'", desc, f.Desc)
|
|
}
|
|
|
|
forum, err := c.Forums.Get(1)
|
|
recordMustExist(t, err, "Couldn't find FID #1")
|
|
expectDesc := "All the reports go here"
|
|
testForum(forum, 1, "Reports", false, expectDesc)
|
|
forum, err = c.Forums.BypassGet(1)
|
|
recordMustExist(t, err, "Couldn't find FID #1")
|
|
|
|
forum, err = c.Forums.Get(2)
|
|
recordMustExist(t, err, "Couldn't find FID #2")
|
|
forum, err = c.Forums.BypassGet(2)
|
|
recordMustExist(t, err, "Couldn't find FID #2")
|
|
expectDesc = "A place for general discussions which don't fit elsewhere"
|
|
testForum(forum, 2, "General", true, expectDesc)
|
|
|
|
// Forum reload test, kind of hacky but gets the job done
|
|
/*
|
|
CacheGet(id int) (*Forum, error)
|
|
CacheSet(forum *Forum) error
|
|
*/
|
|
ex(ok, "ForumCache should be available")
|
|
forum.Name = "nanana"
|
|
fcache.CacheSet(forum)
|
|
forum, err = c.Forums.Get(2)
|
|
recordMustExist(t, err, "Couldn't find FID #2")
|
|
exf(forum.Name == "nanana", "The faux name should be nanana not %s", forum.Name)
|
|
expectNilErr(t, c.Forums.Reload(2))
|
|
forum, err = c.Forums.Get(2)
|
|
recordMustExist(t, err, "Couldn't find FID #2")
|
|
exf(forum.Name == "General", "The proper name should be 2 not %s", forum.Name)
|
|
|
|
ex(!c.Forums.Exists(-1), "FID #-1 shouldn't exist")
|
|
ex(!c.Forums.Exists(0), "FID #0 shouldn't exist")
|
|
ex(c.Forums.Exists(1), "FID #1 should exist")
|
|
ex(c.Forums.Exists(2), "FID #2 should exist")
|
|
ex(!c.Forums.Exists(3), "FID #3 shouldn't exist")
|
|
|
|
_, err = c.Forums.Create("", "", true, "all")
|
|
ex(err != nil, "A forum shouldn't be successfully created, if it has a blank name")
|
|
|
|
fid, err := c.Forums.Create("Test Forum", "", true, "all")
|
|
expectNilErr(t, err)
|
|
ex(fid == 3, "The first forum we create should have an ID of 3")
|
|
ex(c.Forums.Exists(3), "FID #2 should exist")
|
|
|
|
ex(c.Forums.Count() == 3, "The forumstore global count should be 3")
|
|
ex(fcache.Length() == 3, "The forum cache length should be 3")
|
|
|
|
forum, err = c.Forums.Get(3)
|
|
recordMustExist(t, err, "Couldn't find FID #3")
|
|
forum, err = c.Forums.BypassGet(3)
|
|
recordMustExist(t, err, "Couldn't find FID #3")
|
|
testForum(forum, 3, "Test Forum", true, "")
|
|
|
|
// TODO: More forum creation tests
|
|
|
|
expectNilErr(t, c.Forums.Delete(3))
|
|
ex(forum.ID == 3, "forum pointer shenanigans")
|
|
ex(c.Forums.Count() == 2, "The forumstore global count should be 2")
|
|
ex(fcache.Length() == 2, "The forum cache length should be 2")
|
|
ex(!c.Forums.Exists(3), "FID #3 shouldn't exist after being deleted")
|
|
_, err = c.Forums.Get(3)
|
|
recordMustNotExist(t, err, "FID #3 shouldn't exist after being deleted")
|
|
_, err = c.Forums.BypassGet(3)
|
|
recordMustNotExist(t, err, "FID #3 shouldn't exist after being deleted")
|
|
|
|
ex(c.Forums.Delete(c.ReportForumID) != nil, "The reports forum shouldn't be deletable")
|
|
exf(c.Forums.Exists(c.ReportForumID), "FID #%d should still exist", c.ReportForumID)
|
|
_, err = c.Forums.Get(c.ReportForumID)
|
|
exf(err == nil, "FID #%d should still exist", c.ReportForumID)
|
|
_, err = c.Forums.BypassGet(c.ReportForumID)
|
|
exf(err == nil, "FID #%d should still exist", c.ReportForumID)
|
|
|
|
eforums := map[int]bool{1: true, 2: true}
|
|
{
|
|
forums, err := c.Forums.GetAll()
|
|
expectNilErr(t, err)
|
|
found := make(map[int]*c.Forum)
|
|
for _, forum := range forums {
|
|
_, ok := eforums[forum.ID]
|
|
exf(ok, "unknown forum #%d in forums", forum.ID)
|
|
found[forum.ID] = forum
|
|
}
|
|
for fid, _ := range eforums {
|
|
_, ok := found[fid]
|
|
exf(ok, "unable to find expected forum #%d in forums", fid)
|
|
}
|
|
}
|
|
|
|
{
|
|
fids, err := c.Forums.GetAllIDs()
|
|
expectNilErr(t, err)
|
|
found := make(map[int]bool)
|
|
for _, fid := range fids {
|
|
_, ok := eforums[fid]
|
|
exf(ok, "unknown fid #%d in fids", fid)
|
|
found[fid] = true
|
|
}
|
|
for fid, _ := range eforums {
|
|
_, ok := found[fid]
|
|
exf(ok, "unable to find expected fid #%d in fids", fid)
|
|
}
|
|
}
|
|
|
|
vforums := map[int]bool{2: true}
|
|
{
|
|
forums, err := c.Forums.GetAllVisible()
|
|
expectNilErr(t, err)
|
|
found := make(map[int]*c.Forum)
|
|
for _, forum := range forums {
|
|
_, ok := vforums[forum.ID]
|
|
exf(ok, "unknown forum #%d in forums", forum.ID)
|
|
found[forum.ID] = forum
|
|
}
|
|
for fid, _ := range vforums {
|
|
_, ok := found[fid]
|
|
exf(ok, "unable to find expected forum #%d in forums", fid)
|
|
}
|
|
}
|
|
|
|
{
|
|
fids, err := c.Forums.GetAllVisibleIDs()
|
|
expectNilErr(t, err)
|
|
found := make(map[int]bool)
|
|
for _, fid := range fids {
|
|
_, ok := vforums[fid]
|
|
exf(ok, "unknown fid #%d in fids", fid)
|
|
found[fid] = true
|
|
}
|
|
for fid, _ := range vforums {
|
|
_, ok := found[fid]
|
|
exf(ok, "unable to find expected fid #%d in fids", fid)
|
|
}
|
|
}
|
|
|
|
forum, err = c.Forums.Get(2)
|
|
expectNilErr(t, err)
|
|
prevTopicCount := forum.TopicCount
|
|
tid, err := c.Topics.Create(forum.ID, "Forum Meta Test", "Forum Meta Test", 1, "")
|
|
expectNilErr(t, err)
|
|
forum, err = c.Forums.Get(2)
|
|
expectNilErr(t, err)
|
|
exf(forum.TopicCount == (prevTopicCount+1), "forum.TopicCount should be %d not %d", prevTopicCount+1, forum.TopicCount)
|
|
exf(forum.LastTopicID == tid, "forum.LastTopicID should be %d not %d", tid, forum.LastTopicID)
|
|
exf(forum.LastPage == 1, "forum.LastPage should be %d not %d", 1, forum.LastPage)
|
|
|
|
// TODO: Test topic creation and forum topic metadata
|
|
|
|
// TODO: Test forum update
|
|
// TODO: Other forumstore stuff and forumcache?
|
|
}
|
|
|
|
// TODO: Implement this
|
|
func TestForumPermsStore(t *testing.T) {
|
|
miscinit(t)
|
|
if !c.PluginsInited {
|
|
c.InitPlugins()
|
|
}
|
|
ex := exp(t)
|
|
|
|
f := func(fid, gid int, msg string, inv ...bool) {
|
|
fp, err := c.FPStore.Get(fid, gid)
|
|
if err == ErrNoRows {
|
|
fp = c.BlankForumPerms()
|
|
} else {
|
|
expectNilErr(t, err)
|
|
}
|
|
vt := fp.ViewTopic
|
|
if len(inv) > 0 && inv[0] == true {
|
|
vt = !vt
|
|
}
|
|
ex(vt, msg)
|
|
}
|
|
|
|
// TODO: Test reporting
|
|
initialState := func() {
|
|
f(1, 1, "admins should be able to see reports")
|
|
f(1, 2, "mods should be able to see reports")
|
|
f(1, 3, "members should not be able to see reports", true)
|
|
f(1, 4, "banned users should not be able to see reports", true)
|
|
f(2, 1, "admins should be able to see general")
|
|
f(2, 3, "members should be able to see general")
|
|
f(2, 6, "guests should be able to see general")
|
|
}
|
|
initialState()
|
|
|
|
expectNilErr(t, c.FPStore.Reload(1))
|
|
initialState()
|
|
expectNilErr(t, c.FPStore.Reload(2))
|
|
initialState()
|
|
|
|
gid, err := c.Groups.Create("FP Test", "FP Test", false, false, false)
|
|
expectNilErr(t, err)
|
|
fid, err := c.Forums.Create("FP Test", "FP Test", true, "")
|
|
expectNilErr(t, err)
|
|
|
|
u := c.GuestUser.Copy()
|
|
rt := func(gid, fid int, shouldSucceed bool) {
|
|
w := httptest.NewRecorder()
|
|
bytesBuffer := bytes.NewBuffer([]byte(""))
|
|
sfid := strconv.Itoa(fid)
|
|
req := httptest.NewRequest("", "/forum/"+sfid, bytesBuffer)
|
|
u.Group = gid
|
|
h, err := c.UserCheck(w, req, &u)
|
|
expectNilErr(t, err)
|
|
rerr := routes.ViewForum(w, req, &u, h, sfid)
|
|
if shouldSucceed {
|
|
ex(rerr == nil, "ViewForum should succeed")
|
|
} else {
|
|
ex(rerr != nil, "ViewForum should not succeed")
|
|
}
|
|
}
|
|
rt(1, fid, false)
|
|
rt(2, fid, false)
|
|
rt(3, fid, false)
|
|
rt(4, fid, false)
|
|
rt(gid, fid, false)
|
|
|
|
fp, err := c.FPStore.GetCopy(fid, gid)
|
|
if err == sql.ErrNoRows {
|
|
fp = *c.BlankForumPerms()
|
|
} else if err != nil {
|
|
expectNilErr(t, err)
|
|
}
|
|
fmt.Printf("fp: %+v\n", fp)
|
|
|
|
f(fid, 1, "admins should not be able to see fp test", true)
|
|
f(fid, 2, "mods should not be able to see fp test", true)
|
|
f(fid, 3, "members should not be able to see fp test", true)
|
|
f(fid, 4, "banned users should not be able to see fp test", true)
|
|
f(fid, gid, "fp test should not be able to see fp test", true)
|
|
|
|
fp.ViewTopic = true
|
|
|
|
forum, err := c.Forums.Get(fid)
|
|
expectNilErr(t, err)
|
|
expectNilErr(t, forum.SetPerms(&fp, "custom", gid))
|
|
|
|
rt(1, fid, false)
|
|
rt(2, fid, false)
|
|
rt(3, fid, false)
|
|
rt(4, fid, false)
|
|
rt(gid, fid, true)
|
|
|
|
fp, err = c.FPStore.GetCopy(fid, gid)
|
|
if err == sql.ErrNoRows {
|
|
fp = *c.BlankForumPerms()
|
|
} else if err != nil {
|
|
expectNilErr(t, err)
|
|
}
|
|
|
|
f(fid, 1, "admins should not be able to see fp test", true)
|
|
f(fid, 2, "mods should not be able to see fp test", true)
|
|
f(fid, 3, "members should not be able to see fp test", true)
|
|
f(fid, 4, "banned users should not be able to see fp test", true)
|
|
f(fid, gid, "fp test should be able to see fp test")
|
|
|
|
expectNilErr(t, c.Forums.Delete(fid))
|
|
rt(1, fid, false)
|
|
rt(2, fid, false)
|
|
rt(3, fid, false)
|
|
rt(4, fid, false)
|
|
rt(gid, fid, false)
|
|
|
|
// TODO: Test changing forum permissions
|
|
}
|
|
|
|
// TODO: Test the group permissions
|
|
// TODO: Test group.CanSee for forum presets + group perms
|
|
func TestGroupStore(t *testing.T) {
|
|
miscinit(t)
|
|
if !c.PluginsInited {
|
|
c.InitPlugins()
|
|
}
|
|
ex, exf := exp(t), expf(t)
|
|
|
|
_, err := c.Groups.Get(-1)
|
|
recordMustNotExist(t, err, "GID #-1 shouldn't exist")
|
|
|
|
// TODO: Refactor the group store to remove GID #0
|
|
g, err := c.Groups.Get(0)
|
|
recordMustExist(t, err, "Couldn't find GID #0")
|
|
exf(g.ID == 0, "g.ID doesn't not match the requested GID. Got '%d' instead.", g.ID)
|
|
exf(g.Name == "Unknown", "GID #0 is named '%s' and not 'Unknown'", g.Name)
|
|
|
|
g, err = c.Groups.Get(1)
|
|
recordMustExist(t, err, "Couldn't find GID #1")
|
|
exf(g.ID == 1, "g.ID doesn't not match the requested GID. Got '%d' instead.'", g.ID)
|
|
ex(len(g.CanSee) > 0, "g.CanSee should not be zero")
|
|
|
|
ex(!c.Groups.Exists(-1), "GID #-1 shouldn't exist")
|
|
// 0 aka Unknown, for system posts and other oddities
|
|
ex(c.Groups.Exists(0), "GID #0 should exist")
|
|
ex(c.Groups.Exists(1), "GID #1 should exist")
|
|
|
|
isAdmin, isMod, isBanned := true, true, false
|
|
gid, err := c.Groups.Create("Testing", "Test", isAdmin, isMod, isBanned)
|
|
expectNilErr(t, err)
|
|
ex(c.Groups.Exists(gid), "The group we just made doesn't exist")
|
|
|
|
ff := func(i bool) string {
|
|
if !i {
|
|
return "n't"
|
|
}
|
|
return ""
|
|
}
|
|
f := func(gid int, isBanned, isMod, isAdmin bool) {
|
|
ex(g.ID == gid, "The group ID should match the requested ID")
|
|
exf(g.IsAdmin == isAdmin, "This should%s be an admin group", ff(isAdmin))
|
|
exf(g.IsMod == isMod, "This should%s be a mod group", ff(isMod))
|
|
exf(g.IsBanned == isBanned, "This should%s be a ban group", ff(isBanned))
|
|
}
|
|
|
|
g, err = c.Groups.Get(gid)
|
|
expectNilErr(t, err)
|
|
f(gid, false, true, true)
|
|
ex(len(g.CanSee) == 0, "g.CanSee should be empty")
|
|
|
|
isAdmin, isMod, isBanned = false, true, true
|
|
gid, err = c.Groups.Create("Testing 2", "Test", isAdmin, isMod, isBanned)
|
|
expectNilErr(t, err)
|
|
ex(c.Groups.Exists(gid), "The group we just made doesn't exist")
|
|
|
|
g, err = c.Groups.Get(gid)
|
|
expectNilErr(t, err)
|
|
f(gid, false, true, false)
|
|
|
|
// TODO: Make sure this pointer doesn't change once we refactor the group store to stop updating the pointer
|
|
expectNilErr(t, g.ChangeRank(false, false, true))
|
|
|
|
g, err = c.Groups.Get(gid)
|
|
expectNilErr(t, err)
|
|
f(gid, true, false, false)
|
|
|
|
expectNilErr(t, g.ChangeRank(true, true, true))
|
|
|
|
g, err = c.Groups.Get(gid)
|
|
expectNilErr(t, err)
|
|
f(gid, false, true, true)
|
|
ex(len(g.CanSee) == 0, "len(g.CanSee) should be 0")
|
|
|
|
expectNilErr(t, g.ChangeRank(false, true, true))
|
|
|
|
forum, err := c.Forums.Get(2)
|
|
expectNilErr(t, err)
|
|
forumPerms, err := c.FPStore.GetCopy(2, gid)
|
|
if err == sql.ErrNoRows {
|
|
forumPerms = *c.BlankForumPerms()
|
|
} else if err != nil {
|
|
expectNilErr(t, err)
|
|
}
|
|
forumPerms.ViewTopic = true
|
|
|
|
err = forum.SetPerms(&forumPerms, "custom", gid)
|
|
expectNilErr(t, err)
|
|
|
|
g, err = c.Groups.Get(gid)
|
|
expectNilErr(t, err)
|
|
f(gid, false, true, false)
|
|
ex(g.CanSee != nil, "g.CanSee must not be nil")
|
|
ex(len(g.CanSee) == 1, "len(g.CanSee) should not be one")
|
|
ex(g.CanSee[0] == 2, "g.CanSee[0] should be 2")
|
|
canSee := g.CanSee
|
|
|
|
// Make sure the data is static
|
|
expectNilErr(t, c.Groups.Reload(gid))
|
|
|
|
g, err = c.Groups.Get(gid)
|
|
expectNilErr(t, err)
|
|
f(gid, false, true, false)
|
|
|
|
// TODO: Don't enforce a specific order here
|
|
canSeeTest := func(a, b []int) bool {
|
|
if (a == nil) != (b == nil) {
|
|
return false
|
|
}
|
|
if len(a) != len(b) {
|
|
return false
|
|
}
|
|
for i := range a {
|
|
if a[i] != b[i] {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
ex(canSeeTest(g.CanSee, canSee), "g.CanSee is not being reused")
|
|
|
|
// TODO: Test group deletion
|
|
// TODO: Test group reload
|
|
// TODO: Test group cache set
|
|
}
|
|
|
|
func TestGroupPromotions(t *testing.T) {
|
|
miscinit(t)
|
|
if !c.PluginsInited {
|
|
c.InitPlugins()
|
|
}
|
|
ex, exf := exp(t), expf(t)
|
|
|
|
_, err := c.GroupPromotions.Get(-1)
|
|
recordMustNotExist(t, err, "GP #-1 shouldn't exist")
|
|
_, err = c.GroupPromotions.Get(0)
|
|
recordMustNotExist(t, err, "GP #0 shouldn't exist")
|
|
_, err = c.GroupPromotions.Get(1)
|
|
recordMustNotExist(t, err, "GP #1 shouldn't exist")
|
|
expectNilErr(t, c.GroupPromotions.Delete(1))
|
|
|
|
//GetByGroup(gid int) (gps []*GroupPromotion, err error)
|
|
|
|
testPromo := func(exid, from, to, level, posts, registeredFor int, shouldFail bool) {
|
|
gpid, err := c.GroupPromotions.Create(from, to, false, level, posts, registeredFor)
|
|
exf(gpid == exid, "gpid should be %d not %d", exid, gpid)
|
|
//fmt.Println("gpid:", gpid)
|
|
gp, err := c.GroupPromotions.Get(gpid)
|
|
expectNilErr(t, err)
|
|
exf(gp.ID == gpid, "gp.ID should be %d not %d", gpid, gp.ID)
|
|
exf(gp.From == from, "gp.From should be %d not %d", from, gp.From)
|
|
exf(gp.To == to, "gp.To should be %d not %d", to, gp.To)
|
|
ex(!gp.TwoWay, "gp.TwoWay should be false not true")
|
|
exf(gp.Level == level, "gp.Level should be %d not %d", level, gp.Level)
|
|
exf(gp.Posts == posts, "gp.Posts should be %d not %d", posts, gp.Posts)
|
|
exf(gp.MinTime == 0, "gp.MinTime should be %d not %d", 0, gp.MinTime)
|
|
exf(gp.RegisteredFor == registeredFor, "gp.RegisteredFor should be %d not %d", registeredFor, gp.RegisteredFor)
|
|
|
|
uid, err := c.Users.Create("Lord_"+strconv.Itoa(gpid), "I_Rule", "", from, false)
|
|
expectNilErr(t, err)
|
|
u, err := c.Users.Get(uid)
|
|
expectNilErr(t, err)
|
|
exf(u.ID == uid, "u.ID should be %d not %d", uid, u.ID)
|
|
exf(u.Group == from, "u.Group should be %d not %d", from, u.Group)
|
|
err = c.GroupPromotions.PromoteIfEligible(u, u.Level, u.Posts, u.CreatedAt)
|
|
expectNilErr(t, err)
|
|
u.CacheRemove()
|
|
u, err = c.Users.Get(uid)
|
|
expectNilErr(t, err)
|
|
exf(u.ID == uid, "u.ID should be %d not %d", uid, u.ID)
|
|
if shouldFail {
|
|
exf(u.Group == from, "u.Group should be (from-group) %d not %d", from, u.Group)
|
|
} else {
|
|
exf(u.Group == to, "u.Group should be (to-group)%d not %d", to, u.Group)
|
|
}
|
|
|
|
expectNilErr(t, c.GroupPromotions.Delete(gpid))
|
|
_, err = c.GroupPromotions.Get(gpid)
|
|
recordMustNotExist(t, err, fmt.Sprintf("GP #%d should no longer exist", gpid))
|
|
}
|
|
testPromo(1, 1, 2, 0, 0, 0, false)
|
|
testPromo(2, 1, 2, 5, 5, 0, true)
|
|
testPromo(3, 1, 2, 0, 0, 1, true)
|
|
}
|
|
|
|
func TestReplyStore(t *testing.T) {
|
|
miscinit(t)
|
|
if !c.PluginsInited {
|
|
c.InitPlugins()
|
|
}
|
|
_, e := c.Rstore.Get(-1)
|
|
recordMustNotExist(t, e, "RID #-1 shouldn't exist")
|
|
_, e = c.Rstore.Get(0)
|
|
recordMustNotExist(t, e, "RID #0 shouldn't exist")
|
|
|
|
c.Config.DisablePostIP = false
|
|
testReplyStore(t, 2, "::1")
|
|
c.Config.DisablePostIP = true
|
|
testReplyStore(t, 5, "")
|
|
}
|
|
|
|
func testReplyStore(t *testing.T, newID int, ip string) {
|
|
ex, exf := exp(t), expf(t)
|
|
replyTest2 := func(r *c.Reply, e error, rid, parentID, createdBy int, content, ip string) {
|
|
expectNilErr(t, e)
|
|
exf(r.ID == rid, "RID #%d has the wrong ID. It should be %d not %d", rid, rid, r.ID)
|
|
exf(r.ParentID == parentID, "The parent topic of RID #%d should be %d not %d", rid, parentID, r.ParentID)
|
|
exf(r.CreatedBy == createdBy, "The creator of RID #%d should be %d not %d", rid, createdBy, r.CreatedBy)
|
|
exf(r.Content == content, "The contents of RID #%d should be '%s' not %s", rid, content, r.Content)
|
|
exf(r.IP == ip, "The IP of RID#%d should be '%s' not %s", rid, ip, r.IP)
|
|
}
|
|
|
|
replyTest := func(rid, parentID, createdBy int, content, ip string) {
|
|
r, e := c.Rstore.Get(rid)
|
|
replyTest2(r, e, rid, parentID, createdBy, content, ip)
|
|
r, e = c.Rstore.GetCache().Get(rid)
|
|
replyTest2(r, e, rid, parentID, createdBy, content, ip)
|
|
}
|
|
replyTest(1, 1, 1, "A reply!", "")
|
|
|
|
// ! This is hard to do deterministically as the system may pre-load certain items but let's give it a try:
|
|
//_, err = c.Rstore.GetCache().Get(1)
|
|
//recordMustNotExist(t, err, "RID #1 shouldn't be in the cache")
|
|
|
|
_, err := c.Rstore.Get(newID)
|
|
recordMustNotExist(t, err, "RID #2 shouldn't exist")
|
|
|
|
newPostCount := 1
|
|
tid, err := c.Topics.Create(2, "Reply Test Topic", "Reply Test Topic", 1, "")
|
|
expectNilErr(t, err)
|
|
|
|
topic, err := c.Topics.Get(tid)
|
|
expectNilErr(t, err)
|
|
exf(topic.PostCount == newPostCount, "topic.PostCount should be %d, not %d", newPostCount, topic.PostCount)
|
|
exf(topic.LastReplyID == 0, "topic.LastReplyID should be %d not %d", 0, topic.LastReplyID)
|
|
ex(topic.CreatedAt == topic.LastReplyAt, "topic.LastReplyAt should equal it's topic.CreatedAt")
|
|
exf(topic.LastReplyBy == 1, "topic.LastReplyBy should be %d not %d", 1, topic.LastReplyBy)
|
|
|
|
_, err = c.Rstore.GetCache().Get(newID)
|
|
recordMustNotExist(t, err, "RID #%d shouldn't be in the cache", newID)
|
|
|
|
time.Sleep(2 * time.Second)
|
|
|
|
uid, err := c.Users.Create("Reply Topic Test User"+strconv.Itoa(newID), "testpassword", "", 2, true)
|
|
expectNilErr(t, err)
|
|
rid, err := c.Rstore.Create(topic, "Fofofo", ip, uid)
|
|
expectNilErr(t, err)
|
|
exf(rid == newID, "The next reply ID should be %d not %d", newID, rid)
|
|
exf(topic.PostCount == newPostCount, "The old topic in memory's post count should be %d, not %d", newPostCount+1, topic.PostCount)
|
|
// TODO: Test the reply count on the topic
|
|
exf(topic.LastReplyID == 0, "topic.LastReplyID should be %d not %d", 0, topic.LastReplyID)
|
|
ex(topic.CreatedAt == topic.LastReplyAt, "topic.LastReplyAt should equal it's topic.CreatedAt")
|
|
|
|
replyTest(newID, tid, uid, "Fofofo", ip)
|
|
|
|
topic, err = c.Topics.Get(tid)
|
|
expectNilErr(t, err)
|
|
exf(topic.PostCount == newPostCount+1, "topic.PostCount should be %d, not %d", newPostCount+1, topic.PostCount)
|
|
exf(topic.LastReplyID == rid, "topic.LastReplyID should be %d not %d", rid, topic.LastReplyID)
|
|
ex(topic.CreatedAt != topic.LastReplyAt, "topic.LastReplyAt should not equal it's topic.CreatedAt")
|
|
exf(topic.LastReplyBy == uid, "topic.LastReplyBy should be %d not %d", uid, topic.LastReplyBy)
|
|
|
|
expectNilErr(t, topic.CreateActionReply("destroy", ip, 1))
|
|
exf(topic.PostCount == newPostCount+1, "The old topic in memory's post count should be %d, not %d", newPostCount+1, topic.PostCount)
|
|
replyTest(newID+1, tid, 1, "", ip)
|
|
// TODO: Check the actionType field of the reply, this might not be loaded by TopicStore, maybe we should add it there?
|
|
|
|
topic, err = c.Topics.Get(tid)
|
|
expectNilErr(t, err)
|
|
exf(topic.PostCount == newPostCount+2, "topic.PostCount should be %d, not %d", newPostCount+2, topic.PostCount)
|
|
exf(topic.LastReplyID != rid, "topic.LastReplyID should not be %d", rid)
|
|
arid := topic.LastReplyID
|
|
|
|
// TODO: Expand upon this
|
|
rid, err = c.Rstore.Create(topic, "hiii", ip, 1)
|
|
expectNilErr(t, err)
|
|
replyTest(rid, topic.ID, 1, "hiii", ip)
|
|
|
|
reply, err := c.Rstore.Get(rid)
|
|
expectNilErr(t, err)
|
|
expectNilErr(t, reply.SetPost("huuu"))
|
|
exf(reply.Content == "hiii", "topic.Content should be hiii, not %s", reply.Content)
|
|
|
|
reply, err = c.Rstore.Get(rid)
|
|
replyTest2(reply, err, rid, topic.ID, 1, "huuu", ip)
|
|
expectNilErr(t, c.Rstore.ClearIPs())
|
|
_ = c.Rstore.GetCache().Remove(rid)
|
|
replyTest(rid, topic.ID, 1, "huuu", "")
|
|
|
|
expectNilErr(t, reply.Delete())
|
|
// No pointer shenanigans x.x
|
|
// TODO: Log reply.ID and rid in cases of pointer shenanigans?
|
|
ex(reply.ID == rid, "pointer shenanigans")
|
|
|
|
_, err = c.Rstore.GetCache().Get(rid)
|
|
recordMustNotExist(t, err, fmt.Sprintf("RID #%d shouldn't be in the cache", rid))
|
|
_, err = c.Rstore.Get(rid)
|
|
recordMustNotExist(t, err, fmt.Sprintf("RID #%d shouldn't exist", rid))
|
|
|
|
topic, err = c.Topics.Get(tid)
|
|
expectNilErr(t, err)
|
|
exf(topic.LastReplyID == arid, "topic.LastReplyID should be %d not %d", arid, topic.LastReplyID)
|
|
|
|
// TODO: Write a test for this
|
|
//(topic *TopicUser) Replies(offset int, pFrag int, user *User) (rlist []*ReplyUser, ogdesc string, err error)
|
|
|
|
// TODO: Add tests for *Reply
|
|
// TODO: Add tests for ReplyCache
|
|
}
|
|
|
|
func TestLikes(t *testing.T) {
|
|
miscinit(t)
|
|
if !c.PluginsInited {
|
|
c.InitPlugins()
|
|
}
|
|
_, exf := exp(t), expf(t)
|
|
bulkExists := func(iids []int, sentBy int, targetType string, expCount int) {
|
|
ids, e := c.Likes.BulkExists(iids, sentBy, "replies")
|
|
//recordMustNotExist(t, e, "no likes should be found")
|
|
expectNilErr(t, e)
|
|
exf(len(ids) == expCount, "len ids should be %d", expCount)
|
|
|
|
idMap := make(map[int]struct{})
|
|
for _, id := range ids {
|
|
idMap[id] = struct{}{}
|
|
}
|
|
for _, iid := range iids {
|
|
_, ok := idMap[iid]
|
|
exf(ok, "missing iid %d in idMap", iid)
|
|
}
|
|
|
|
idCount := 0
|
|
expectNilErr(t, c.Likes.BulkExistsFunc(iids, sentBy, targetType, func(_ int) error {
|
|
idCount++
|
|
return nil
|
|
}))
|
|
exf(idCount == expCount, "idCount should be %d not %d", expCount, idCount)
|
|
}
|
|
|
|
uid := 1
|
|
bulkExists([]int{}, uid, "replies", 0)
|
|
|
|
topic, e := c.Topics.Get(1)
|
|
expectNilErr(t, e)
|
|
rid, e := c.Rstore.Create(topic, "hiii", "", uid)
|
|
expectNilErr(t, e)
|
|
r, e := c.Rstore.Get(rid)
|
|
expectNilErr(t, e)
|
|
expectNilErr(t, r.Like(uid))
|
|
bulkExists([]int{rid}, uid, "replies", 1)
|
|
|
|
rid2, e := c.Rstore.Create(topic, "hi 2 u 2", "", uid)
|
|
expectNilErr(t, e)
|
|
r2, e := c.Rstore.Get(rid2)
|
|
expectNilErr(t, e)
|
|
expectNilErr(t, r2.Like(uid))
|
|
bulkExists([]int{rid, rid2}, uid, "replies", 2)
|
|
|
|
expectNilErr(t, r.Unlike(uid))
|
|
bulkExists([]int{rid2}, uid, "replies", 1)
|
|
expectNilErr(t, r2.Unlike(uid))
|
|
bulkExists([]int{}, uid, "replies", 0)
|
|
|
|
//BulkExists(ids []int, sentBy int, targetType string) (eids []int, err error)
|
|
|
|
expectNilErr(t, topic.Like(1, uid))
|
|
expectNilErr(t, topic.Unlike(uid))
|
|
}
|
|
|
|
func TestAttachments(t *testing.T) {
|
|
miscinit(t)
|
|
if !c.PluginsInited {
|
|
c.InitPlugins()
|
|
}
|
|
ex, exf := exp(t), expf(t)
|
|
|
|
filename := "n0-48.png"
|
|
srcFile := "./test_data/" + filename
|
|
destFile := "./attachs/" + filename
|
|
|
|
ft := func(e error) {
|
|
if e != nil && e != sql.ErrNoRows {
|
|
t.Error(e)
|
|
}
|
|
}
|
|
|
|
ex(c.Attachments.Count() == 0, "the number of attachments should be 0")
|
|
ex(c.Attachments.CountIn("topics", 1) == 0, "the number of attachments in topic 1 should be 0")
|
|
exf(c.Attachments.CountInPath(filename) == 0, "the number of attachments with path '%s' should be 0", filename)
|
|
_, e := c.Attachments.FGet(1)
|
|
ft(e)
|
|
ex(e == sql.ErrNoRows, ".FGet should have no results")
|
|
_, e = c.Attachments.Get(1)
|
|
ft(e)
|
|
ex(e == sql.ErrNoRows, ".Get should have no results")
|
|
_, e = c.Attachments.MiniGetList("topics", 1)
|
|
ft(e)
|
|
ex(e == sql.ErrNoRows, ".MiniGetList should have no results")
|
|
_, e = c.Attachments.BulkMiniGetList("topics", []int{1})
|
|
ft(e)
|
|
ex(e == sql.ErrNoRows, ".BulkMiniGetList should have no results")
|
|
|
|
simUpload := func() {
|
|
// Sim an upload, try a proper upload through the proper pathway later on
|
|
_, e = os.Stat(destFile)
|
|
if e != nil && !os.IsNotExist(e) {
|
|
expectNilErr(t, e)
|
|
} else if e == nil {
|
|
expectNilErr(t, os.Remove(destFile))
|
|
}
|
|
|
|
input, e := ioutil.ReadFile(srcFile)
|
|
expectNilErr(t, e)
|
|
expectNilErr(t, ioutil.WriteFile(destFile, input, 0644))
|
|
}
|
|
simUpload()
|
|
|
|
tid, e := c.Topics.Create(2, "Attach Test", "Filler Body", 1, "")
|
|
expectNilErr(t, e)
|
|
aid, e := c.Attachments.Add(2, "forums", tid, "topics", 1, filename, "")
|
|
expectNilErr(t, e)
|
|
exf(aid == 1, "aid should be 1 not %d", aid)
|
|
expectNilErr(t, c.Attachments.AddLinked("topics", tid))
|
|
ex(c.Attachments.Count() == 1, "the number of attachments should be 1")
|
|
exf(c.Attachments.CountIn("topics", tid) == 1, "the number of attachments in topic %d should be 1", tid)
|
|
exf(c.Attachments.CountInPath(filename) == 1, "the number of attachments with path '%s' should be 1", filename)
|
|
|
|
et := func(a *c.MiniAttachment, aid, sid, oid, uploadedBy int, path, extra, ext string) {
|
|
exf(a.ID == aid, "ID should be %d not %d", aid, a.ID)
|
|
exf(a.SectionID == sid, "SectionID should be %d not %d", sid, a.SectionID)
|
|
exf(a.OriginID == oid, "OriginID should be %d not %d", oid, a.OriginID)
|
|
exf(a.UploadedBy == uploadedBy, "UploadedBy should be %d not %d", uploadedBy, a.UploadedBy)
|
|
exf(a.Path == path, "Path should be %s not %s", path, a.Path)
|
|
exf(a.Extra == extra, "Extra should be %s not %s", extra, a.Extra)
|
|
ex(a.Image, "Image should be true")
|
|
exf(a.Ext == ext, "Ext should be %s not %s", ext, a.Ext)
|
|
}
|
|
et2 := func(a *c.Attachment, aid, sid, oid, uploadedBy int, path, extra, ext string) {
|
|
exf(a.ID == aid, "ID should be %d not %d", aid, a.ID)
|
|
exf(a.SectionID == sid, "SectionID should be %d not %d", sid, a.SectionID)
|
|
exf(a.OriginID == oid, "OriginID should be %d not %d", oid, a.OriginID)
|
|
exf(a.UploadedBy == uploadedBy, "UploadedBy should be %d not %d", uploadedBy, a.UploadedBy)
|
|
exf(a.Path == path, "Path should be %s not %s", path, a.Path)
|
|
exf(a.Extra == extra, "Extra should be %s not %s", extra, a.Extra)
|
|
ex(a.Image, "Image should be true")
|
|
exf(a.Ext == ext, "Ext should be %s not %s", ext, a.Ext)
|
|
}
|
|
|
|
f2 := func(aid, sid, oid int, extra string, topic bool) {
|
|
var tbl string
|
|
if topic {
|
|
tbl = "topics"
|
|
} else {
|
|
tbl = "replies"
|
|
}
|
|
fa, e := c.Attachments.FGet(aid)
|
|
expectNilErr(t, e)
|
|
et2(fa, aid, sid, oid, 1, filename, extra, "png")
|
|
|
|
a, e := c.Attachments.Get(aid)
|
|
expectNilErr(t, e)
|
|
et(a, aid, sid, oid, 1, filename, extra, "png")
|
|
|
|
alist, e := c.Attachments.MiniGetList(tbl, oid)
|
|
expectNilErr(t, e)
|
|
exf(len(alist) == 1, "len(alist) should be 1 not %d", len(alist))
|
|
a = alist[0]
|
|
et(a, aid, sid, oid, 1, filename, extra, "png")
|
|
|
|
amap, e := c.Attachments.BulkMiniGetList(tbl, []int{oid})
|
|
expectNilErr(t, e)
|
|
exf(len(amap) == 1, "len(amap) should be 1 not %d", len(amap))
|
|
alist, ok := amap[oid]
|
|
if !ok {
|
|
t.Logf("key %d not found in amap", oid)
|
|
}
|
|
exf(len(alist) == 1, "len(alist) should be 1 not %d", len(alist))
|
|
a = alist[0]
|
|
et(a, aid, sid, oid, 1, filename, extra, "png")
|
|
}
|
|
|
|
topic, e := c.Topics.Get(tid)
|
|
expectNilErr(t, e)
|
|
exf(topic.AttachCount == 1, "topic.AttachCount should be 1 not %d", topic.AttachCount)
|
|
f2(aid, 2, tid, "", true)
|
|
expectNilErr(t, topic.MoveTo(1))
|
|
f2(aid, 1, tid, "", true)
|
|
expectNilErr(t, c.Attachments.MoveTo(2, tid, "topics"))
|
|
f2(aid, 2, tid, "", true)
|
|
|
|
// TODO: ShowAttachment test
|
|
|
|
deleteTest := func(aid, oid int, topic bool) {
|
|
var tbl string
|
|
if topic {
|
|
tbl = "topics"
|
|
} else {
|
|
tbl = "replies"
|
|
}
|
|
//expectNilErr(t, c.Attachments.Delete(aid))
|
|
expectNilErr(t, c.DeleteAttachment(aid))
|
|
ex(c.Attachments.Count() == 0, "the number of attachments should be 0")
|
|
exf(c.Attachments.CountIn(tbl, oid) == 0, "the number of attachments in topic %d should be 0", tid)
|
|
exf(c.Attachments.CountInPath(filename) == 0, "the number of attachments with path '%s' should be 0", filename)
|
|
_, e = c.Attachments.FGet(aid)
|
|
ft(e)
|
|
ex(e == sql.ErrNoRows, ".FGet should have no results")
|
|
_, e = c.Attachments.Get(aid)
|
|
ft(e)
|
|
ex(e == sql.ErrNoRows, ".Get should have no results")
|
|
_, e = c.Attachments.MiniGetList(tbl, oid)
|
|
ft(e)
|
|
ex(e == sql.ErrNoRows, ".MiniGetList should have no results")
|
|
_, e = c.Attachments.BulkMiniGetList(tbl, []int{oid})
|
|
ft(e)
|
|
ex(e == sql.ErrNoRows, ".BulkMiniGetList should have no results")
|
|
}
|
|
deleteTest(aid, tid, true)
|
|
topic, e = c.Topics.Get(tid)
|
|
expectNilErr(t, e)
|
|
exf(topic.AttachCount == 0, "topic.AttachCount should be 0 not %d", topic.AttachCount)
|
|
|
|
simUpload()
|
|
rid, e := c.Rstore.Create(topic, "Reply Filler", "", 1)
|
|
expectNilErr(t, e)
|
|
aid, e = c.Attachments.Add(2, "forums", rid, "replies", 1, filename, strconv.Itoa(topic.ID))
|
|
expectNilErr(t, e)
|
|
exf(aid == 2, "aid should be 2 not %d", aid)
|
|
expectNilErr(t, c.Attachments.AddLinked("replies", rid))
|
|
r, e := c.Rstore.Get(rid)
|
|
expectNilErr(t, e)
|
|
exf(r.AttachCount == 1, "r.AttachCount should be 1 not %d", r.AttachCount)
|
|
f2(aid, 2, rid, strconv.Itoa(topic.ID), false)
|
|
expectNilErr(t, c.Attachments.MoveTo(1, rid, "replies"))
|
|
f2(aid, 1, rid, strconv.Itoa(topic.ID), false)
|
|
deleteTest(aid, rid, false)
|
|
r, e = c.Rstore.Get(rid)
|
|
expectNilErr(t, e)
|
|
exf(r.AttachCount == 0, "r.AttachCount should be 0 not %d", r.AttachCount)
|
|
|
|
// TODO: Path overlap tests
|
|
}
|
|
|
|
func TestPolls(t *testing.T) {
|
|
miscinit(t)
|
|
if !c.PluginsInited {
|
|
c.InitPlugins()
|
|
}
|
|
ex, exf := exp(t), expf(t)
|
|
|
|
shouldNotExist := func(id int) {
|
|
exf(!c.Polls.Exists(id), "poll %d should not exist", id)
|
|
_, e := c.Polls.Get(id)
|
|
recordMustNotExist(t, e, fmt.Sprintf("poll %d shouldn't exist", id))
|
|
}
|
|
shouldNotExist(-1)
|
|
shouldNotExist(0)
|
|
shouldNotExist(1)
|
|
exf(c.Polls.Count() == 0, "count should be %d not %d", 0, c.Polls.Count())
|
|
|
|
tid, e := c.Topics.Create(2, "Poll Test", "Filler Body", 1, "")
|
|
expectNilErr(t, e)
|
|
topic, e := c.Topics.Get(tid)
|
|
/*Options map[int]string
|
|
Results map[int]int // map[optionIndex]points
|
|
QuickOptions []PollOption // TODO: Fix up the template transpiler so we don't need to use this hack anymore
|
|
}*/
|
|
pollType := 0 // Basic single choice
|
|
pid, e := c.Polls.Create(topic, pollType, map[int]string{0: "item 1", 1: "item 2", 2: "item 3"})
|
|
expectNilErr(t, e)
|
|
exf(pid == 1, "poll id should be 1 not %d", pid)
|
|
ex(c.Polls.Exists(1), "poll 1 should exist")
|
|
exf(c.Polls.Count() == 1, "count should be %d not %d", 1, c.Polls.Count())
|
|
|
|
testPoll := func(p *c.Poll, id, parentID int, parentTable string, ptype int, antiCheat bool, voteCount int) {
|
|
ef := exf
|
|
ef(p.ID == id, "p.ID should be %d not %d", id, p.ID)
|
|
ef(p.ParentID == parentID, "p.ParentID should be %d not %d", parentID, p.ParentID)
|
|
ef(p.ParentTable == parentTable, "p.ParentID should be %s not %s", parentTable, p.ParentTable)
|
|
ef(p.Type == ptype, "p.ParentID should be %d not %d", ptype, p.Type)
|
|
s := "false"
|
|
if p.AntiCheat {
|
|
s = "true"
|
|
}
|
|
ef(p.AntiCheat == antiCheat, "p.AntiCheat should be ", s)
|
|
// TODO: More fields
|
|
ef(p.VoteCount == voteCount, "p.VoteCount should be %d not %d", voteCount, p.VoteCount)
|
|
}
|
|
|
|
p, e := c.Polls.Get(1)
|
|
expectNilErr(t, e)
|
|
testPoll(p, 1, tid, "topics", 0, false, 0)
|
|
|
|
expectNilErr(t, p.CastVote(0, 1, ""))
|
|
expectNilErr(t, c.Polls.Reload(p.ID))
|
|
p, e = c.Polls.Get(1)
|
|
expectNilErr(t, e)
|
|
testPoll(p, 1, tid, "topics", 0, false, 1)
|
|
|
|
expectNilErr(t, c.Polls.ClearIPs())
|
|
// TODO: Test to see if it worked
|
|
|
|
expectNilErr(t, p.Delete())
|
|
ex(!c.Polls.Exists(1), "poll 1 should no longer exist")
|
|
_, e = c.Polls.Get(1)
|
|
recordMustNotExist(t, e, "poll 1 should no longer exist")
|
|
exf(c.Polls.Count() == 0, "count should be %d not %d", 0, c.Polls.Count())
|
|
}
|
|
|
|
func TestSearch(t *testing.T) {
|
|
miscinit(t)
|
|
if !c.PluginsInited {
|
|
c.InitPlugins()
|
|
}
|
|
exf := expf(t)
|
|
|
|
title := "search"
|
|
body := "bab bab bab bab"
|
|
q := "search"
|
|
tid, e := c.Topics.Create(2, title, body, 1, "")
|
|
expectNilErr(t, e)
|
|
|
|
tids, e := c.RepliesSearch.Query(q, []int{2})
|
|
//fmt.Printf("tids: %+v\n", tids)
|
|
expectNilErr(t, e)
|
|
exf(len(tids) == 1, "len(tids) should be 1 not %d", len(tids))
|
|
|
|
topic, e := c.Topics.Get(tids[0])
|
|
expectNilErr(t, e)
|
|
exf(topic.ID == tid, "topic.ID should be %d not %d", tid, topic.ID)
|
|
exf(topic.Title == title, "topic.Title should be %s not %s", title, topic.Title)
|
|
|
|
tids, e = c.RepliesSearch.Query(q, []int{1, 2})
|
|
//fmt.Printf("tids: %+v\n", tids)
|
|
expectNilErr(t, e)
|
|
exf(len(tids) == 1, "len(tids) should be 1 not %d", len(tids))
|
|
|
|
q = "bab"
|
|
tids, e = c.RepliesSearch.Query(q, []int{1, 2})
|
|
//fmt.Printf("tids: %+v\n", tids)
|
|
expectNilErr(t, e)
|
|
exf(len(tids) == 1, "len(tids) should be 1 not %d", len(tids))
|
|
}
|
|
|
|
func TestProfileReplyStore(t *testing.T) {
|
|
miscinit(t)
|
|
if !c.PluginsInited {
|
|
c.InitPlugins()
|
|
}
|
|
|
|
_, e := c.Prstore.Get(-1)
|
|
recordMustNotExist(t, e, "PRID #-1 shouldn't exist")
|
|
_, e = c.Prstore.Get(0)
|
|
recordMustNotExist(t, e, "PRID #0 shouldn't exist")
|
|
_, e = c.Prstore.Get(1)
|
|
recordMustNotExist(t, e, "PRID #1 shouldn't exist")
|
|
|
|
c.Config.DisablePostIP = false
|
|
testProfileReplyStore(t, 1, "::1")
|
|
c.Config.DisablePostIP = true
|
|
testProfileReplyStore(t, 2, "")
|
|
}
|
|
func testProfileReplyStore(t *testing.T, newID int, ip string) {
|
|
exf := expf(t)
|
|
// ? - Commented this one out as strong constraints like this put an unreasonable load on the database, we only want errors if a delete which should succeed fails
|
|
//profileReply := c.BlankProfileReply(1)
|
|
//e = profileReply.Delete()
|
|
//expect(t,e != nil,"You shouldn't be able to delete profile replies which don't exist")
|
|
|
|
profileID := 1
|
|
prid, e := c.Prstore.Create(profileID, "Haha", 1, ip)
|
|
expectNilErr(t, e)
|
|
exf(prid == newID, "The first profile reply should have an ID of %d", newID)
|
|
|
|
pr, e := c.Prstore.Get(newID)
|
|
expectNilErr(t, e)
|
|
exf(pr.ID == newID, "The profile reply should have an ID of %d not %d", newID, pr.ID)
|
|
exf(pr.ParentID == 1, "The parent ID of the profile reply should be 1 not %d", pr.ParentID)
|
|
exf(pr.Content == "Haha", "The profile reply's contents should be 'Haha' not '%s'", pr.Content)
|
|
exf(pr.CreatedBy == 1, "The profile reply's creator should be 1 not %d", pr.CreatedBy)
|
|
exf(pr.IP == ip, "The profile reply's IP should be '%s' not '%s'", ip, pr.IP)
|
|
|
|
expectNilErr(t, c.Prstore.ClearIPs())
|
|
|
|
pr, e = c.Prstore.Get(newID)
|
|
expectNilErr(t, e)
|
|
exf(pr.ID == newID, "The profile reply should have an ID of %d not %d", newID, pr.ID)
|
|
exf(pr.ParentID == 1, "The parent ID of the profile reply should be 1 not %d", pr.ParentID)
|
|
exf(pr.Content == "Haha", "The profile reply's contents should be 'Haha' not '%s'", pr.Content)
|
|
exf(pr.CreatedBy == 1, "The profile reply's creator should be 1 not %d", pr.CreatedBy)
|
|
ip = ""
|
|
exf(pr.IP == ip, "The profile reply's IP should be '%s' not '%s'", ip, pr.IP)
|
|
|
|
expectNilErr(t, pr.Delete())
|
|
_, e = c.Prstore.Get(newID)
|
|
exf(e != nil, "PRID #%d shouldn't exist after being deleted", newID)
|
|
|
|
// TODO: Test pr.SetBody() and pr.Creator()
|
|
}
|
|
|
|
func TestConvos(t *testing.T) {
|
|
miscinit(t)
|
|
if !c.PluginsInited {
|
|
c.InitPlugins()
|
|
}
|
|
ex, exf := exp(t), expf(t)
|
|
|
|
sf := func(i interface{}, e error) error {
|
|
return e
|
|
}
|
|
mf := func(e error, msg string, exists bool) {
|
|
if !exists {
|
|
recordMustNotExist(t, e, msg)
|
|
} else {
|
|
recordMustExist(t, e, msg)
|
|
}
|
|
}
|
|
gu := func(uid, offset int, exists bool) {
|
|
s := ""
|
|
if !exists {
|
|
s = " not"
|
|
}
|
|
mf(sf(c.Convos.GetUser(uid, offset)), fmt.Sprintf("convo getuser %d %d should%s exist", uid, offset, s), exists)
|
|
}
|
|
gue := func(uid, offset int, exists bool) {
|
|
s := ""
|
|
if !exists {
|
|
s = " not"
|
|
}
|
|
mf(sf(c.Convos.GetUserExtra(uid, offset)), fmt.Sprintf("convo getuserextra %d %d should%s exist", uid, offset, s), exists)
|
|
}
|
|
|
|
ex(c.Convos.GetUserCount(-1) == 0, "getusercount should be 0")
|
|
ex(c.Convos.GetUserCount(0) == 0, "getusercount should be 0")
|
|
mf(sf(c.Convos.Get(-1)), "convo -1 should not exist", false)
|
|
mf(sf(c.Convos.Get(0)), "convo 0 should not exist", false)
|
|
gu(-1, -1, false)
|
|
gu(-1, 0, false)
|
|
gu(0, 0, false)
|
|
gue(-1, -1, false)
|
|
gue(-1, 0, false)
|
|
gue(0, 0, false)
|
|
|
|
nf := func(cid, count int) {
|
|
ex := count > 0
|
|
s := ""
|
|
if !ex {
|
|
s = " not"
|
|
}
|
|
mf(sf(c.Convos.Get(cid)), fmt.Sprintf("convo %d should%s exist", cid, s), ex)
|
|
gu(1, 0, ex)
|
|
gu(1, 5, false) // invariant may change in future tests
|
|
|
|
exf(c.Convos.GetUserCount(1) == count, "getusercount should be %d", count)
|
|
gue(1, 0, ex)
|
|
gue(1, 5, false) // invariant may change in future tests
|
|
exf(c.Convos.Count() == count, "convos count should be %d", count)
|
|
}
|
|
nf(1, 0)
|
|
|
|
awaitingActivation := 5
|
|
uid, err := c.Users.Create("Saturn", "ReallyBadPassword", "", awaitingActivation, false)
|
|
expectNilErr(t, err)
|
|
|
|
cid, err := c.Convos.Create("hehe", 1, []int{uid})
|
|
expectNilErr(t, err)
|
|
ex(cid == 1, "cid should be 1")
|
|
ex(c.Convos.Count() == 1, "convos count should be 1")
|
|
|
|
co, err := c.Convos.Get(cid)
|
|
expectNilErr(t, err)
|
|
ex(co.ID == 1, "co.ID should be 1")
|
|
ex(co.CreatedBy == 1, "co.CreatedBy should be 1")
|
|
// TODO: CreatedAt test
|
|
ex(co.LastReplyBy == 1, "co.LastReplyBy should be 1")
|
|
// TODO: LastReplyAt test
|
|
expectIntToBeX(t, co.PostsCount(), 1, "postscount should be 1, not %d")
|
|
ex(co.Has(uid), "saturn should be in the conversation")
|
|
ex(!co.Has(9999), "uid 9999 should not be in the conversation")
|
|
uids, err := co.Uids()
|
|
expectNilErr(t, err)
|
|
expectIntToBeX(t, len(uids), 2, "uids length should be 2, not %d")
|
|
exf(uids[0] == uid, "uids[0] should be %d, not %d", uid, uids[0])
|
|
exf(uids[1] == 1, "uids[1] should be %d, not %d", 1, uids[1])
|
|
nf(cid, 1)
|
|
|
|
expectNilErr(t, c.Convos.Delete(cid))
|
|
expectIntToBeX(t, co.PostsCount(), 0, "postscount should be 0, not %d")
|
|
ex(!co.Has(uid), "saturn should not be in a deleted conversation")
|
|
uids, err = co.Uids()
|
|
expectNilErr(t, err)
|
|
expectIntToBeX(t, len(uids), 0, "uids length should be 0, not %d")
|
|
nf(cid, 0)
|
|
|
|
// TODO: More tests
|
|
|
|
// Block tests
|
|
|
|
ok, err := c.UserBlocks.IsBlockedBy(1, 1)
|
|
expectNilErr(t, err)
|
|
ex(!ok, "there shouldn't be any blocks")
|
|
ok, err = c.UserBlocks.BulkIsBlockedBy([]int{1}, 1)
|
|
expectNilErr(t, err)
|
|
ex(!ok, "there shouldn't be any blocks")
|
|
bf := func(blocker, offset, perPage, expectLen, blockee int) {
|
|
l, err := c.UserBlocks.BlockedByOffset(blocker, offset, perPage)
|
|
expectNilErr(t, err)
|
|
exf(len(l) == expectLen, "there should be %d users blocked by %d not %d", expectLen, blocker, len(l))
|
|
if len(l) > 0 {
|
|
exf(l[0] == blockee, "blocked uid should be %d not %d", blockee, l[0])
|
|
}
|
|
}
|
|
nbf := func(blocker, blockee int) {
|
|
ok, err := c.UserBlocks.IsBlockedBy(1, 2)
|
|
expectNilErr(t, err)
|
|
ex(!ok, "there shouldn't be any blocks")
|
|
ok, err = c.UserBlocks.BulkIsBlockedBy([]int{1}, 2)
|
|
expectNilErr(t, err)
|
|
ex(!ok, "there shouldn't be any blocks")
|
|
expectIntToBeX(t, c.UserBlocks.BlockedByCount(1), 0, "blockedbycount for 1 should be 1, not %d")
|
|
bf(1, 0, 1, 0, 0)
|
|
bf(1, 0, 15, 0, 0)
|
|
bf(1, 1, 15, 0, 0)
|
|
bf(1, 5, 15, 0, 0)
|
|
}
|
|
nbf(1, 2)
|
|
|
|
expectNilErr(t, c.UserBlocks.Add(1, 2))
|
|
ok, err = c.UserBlocks.IsBlockedBy(1, 2)
|
|
expectNilErr(t, err)
|
|
ex(ok, "2 should be blocked by 1")
|
|
expectIntToBeX(t, c.UserBlocks.BlockedByCount(1), 1, "blockedbycount for 1 should be 1, not %d")
|
|
bf(1, 0, 1, 1, 2)
|
|
bf(1, 0, 15, 1, 2)
|
|
bf(1, 1, 15, 0, 0)
|
|
bf(1, 5, 15, 0, 0)
|
|
|
|
// Double add test
|
|
expectNilErr(t, c.UserBlocks.Add(1, 2))
|
|
ok, err = c.UserBlocks.IsBlockedBy(1, 2)
|
|
expectNilErr(t, err)
|
|
ex(ok, "2 should be blocked by 1")
|
|
//expectIntToBeX(t, c.UserBlocks.BlockedByCount(1), 1, "blockedbycount for 1 should be 1, not %d") // todo: fix this
|
|
//bf(1, 0, 1, 1, 2) // todo: fix this
|
|
//bf(1, 0, 15, 1, 2) // todo: fix this
|
|
//bf(1, 1, 15, 0, 0) // todo: fix this
|
|
bf(1, 5, 15, 0, 0)
|
|
|
|
expectNilErr(t, c.UserBlocks.Remove(1, 2))
|
|
nbf(1, 2)
|
|
// Double remove test
|
|
expectNilErr(t, c.UserBlocks.Remove(1, 2))
|
|
nbf(1, 2)
|
|
|
|
// TODO: Self-block test
|
|
|
|
// TODO: More Block tests
|
|
}
|
|
|
|
func TestActivityStream(t *testing.T) {
|
|
miscinit(t)
|
|
ex, exf := exp(t), expf(t)
|
|
|
|
ex(c.Activity.Count() == 0, "activity stream count should be 0")
|
|
gNone := func(id int) {
|
|
_, e := c.Activity.Get(id)
|
|
recordMustNotExist(t, e, "activity item "+strconv.Itoa(id)+" shouldn't exist")
|
|
}
|
|
gNone(-1)
|
|
gNone(0)
|
|
gNone(1)
|
|
countAsid := func(asid, count int) {
|
|
exf(c.ActivityMatches.CountAsid(asid) == count, "activity stream matches count for asid %d should be %d not %d", asid, count, c.ActivityMatches.CountAsid(asid))
|
|
}
|
|
countAsid(-1, 0)
|
|
countAsid(0, 0)
|
|
countAsid(1, 0)
|
|
|
|
a := c.Alert{ActorID: 1, TargetUserID: 1, Event: "like", ElementType: "topic", ElementID: 1}
|
|
id, e := c.Activity.Add(a)
|
|
expectNilErr(t, e)
|
|
ex(id == 1, "new activity item id should be 1")
|
|
|
|
ex(c.Activity.Count() == 1, "activity stream count should be 1")
|
|
al, e := c.Activity.Get(1)
|
|
expectNilErr(t, e)
|
|
exf(al.ASID == id, "alert asid should be %d not %d", id, al.ASID)
|
|
ex(al.ActorID == 1, "alert actorid should be 1")
|
|
ex(al.TargetUserID == 1, "alert targetuserid should be 1")
|
|
ex(al.Event == "like", "alert event type should be like")
|
|
ex(al.ElementType == "topic", "alert element type should be topic")
|
|
ex(al.ElementID == 1, "alert element id should be 1")
|
|
|
|
countAsid(id, 0)
|
|
|
|
tuid, e := c.Users.Create("Activity Target", "Activity Target", "", 1, true)
|
|
expectNilErr(t, e)
|
|
expectNilErr(t, c.ActivityMatches.Add(tuid, 1))
|
|
countAsid(id, 1)
|
|
expectNilErr(t, c.ActivityMatches.Delete(tuid, id))
|
|
countAsid(id, 0)
|
|
|
|
expectNilErr(t, c.ActivityMatches.Add(tuid, 1))
|
|
countAsid(id, 1)
|
|
changed, e := c.ActivityMatches.DeleteAndCountChanged(tuid, id)
|
|
expectNilErr(t, e)
|
|
exf(changed == 1, "changed should be %d not %d", 1, changed)
|
|
countAsid(id, 0)
|
|
|
|
expectNilErr(t, c.ActivityMatches.Add(tuid, 1))
|
|
countAsid(id, 1)
|
|
|
|
// TODO: Add more tests
|
|
|
|
expectNilErr(t, c.Activity.Delete(id))
|
|
ex(c.Activity.Count() == 0, "activity stream count should be 0")
|
|
gNone(id)
|
|
countAsid(id, 0)
|
|
|
|
// TODO: More tests
|
|
}
|
|
|
|
func TestLogs(t *testing.T) {
|
|
ex, exf := exp(t), expf(t)
|
|
miscinit(t)
|
|
gTests := func(s c.LogStore, phrase string) {
|
|
ex(s.Count() == 0, "There shouldn't be any "+phrase)
|
|
logs, err := s.GetOffset(0, 25)
|
|
expectNilErr(t, err)
|
|
ex(len(logs) == 0, "The log slice should be empty")
|
|
}
|
|
gTests(c.ModLogs, "modlogs")
|
|
gTests(c.AdminLogs, "adminlogs")
|
|
|
|
gTests2 := func(s c.LogStore, phrase string) {
|
|
err := s.Create("something", 0, "bumblefly", "::1", 1)
|
|
expectNilErr(t, err)
|
|
count := s.Count()
|
|
exf(count == 1, "store.Count should return one, not %d", count)
|
|
logs, err := s.GetOffset(0, 25)
|
|
recordMustExist(t, err, "We should have at-least one "+phrase)
|
|
ex(len(logs) == 1, "The length of the log slice should be one")
|
|
|
|
l := logs[0]
|
|
ex(l.Action == "something", "l.Action is not something")
|
|
ex(l.ElementID == 0, "l.ElementID is not 0")
|
|
ex(l.ElementType == "bumblefly", "l.ElementType is not bumblefly")
|
|
ex(l.IP == "::1", "l.IP is not ::1")
|
|
ex(l.ActorID == 1, "l.ActorID is not 1")
|
|
// TODO: Add a test for log.DoneAt? Maybe throw in some dates and times which are clearly impossible but which may occur due to timezone bugs?
|
|
}
|
|
gTests2(c.ModLogs, "modlog")
|
|
gTests2(c.AdminLogs, "adminlog")
|
|
}
|
|
|
|
func TestRegLogs(t *testing.T) {
|
|
miscinit(t)
|
|
if !c.PluginsInited {
|
|
c.InitPlugins()
|
|
}
|
|
exf := expf(t)
|
|
|
|
mustNone := func() {
|
|
exf(c.RegLogs.Count() == 0, "count should be %d not %d", 0, c.RegLogs.Count())
|
|
items, e := c.RegLogs.GetOffset(0, 10)
|
|
expectNilErr(t, e)
|
|
exf(len(items) == 0, "len(items) should be %d not %d", 0, len(items))
|
|
expectNilErr(t, c.RegLogs.Purge())
|
|
exf(c.RegLogs.Count() == 0, "count should be %d not %d", 0, c.RegLogs.Count())
|
|
items, e = c.RegLogs.GetOffset(0, 10)
|
|
expectNilErr(t, e)
|
|
exf(len(items) == 0, "len(items) should be %d not %d", 0, len(items))
|
|
}
|
|
mustNone()
|
|
|
|
regLog := &c.RegLogItem{Username: "Aa", Email: "aa@example.com", FailureReason: "fake", Success: false, IP: ""}
|
|
id, e := regLog.Create()
|
|
exf(id == 1, "id should be %d not %d", 1, id)
|
|
expectNilErr(t, e)
|
|
|
|
exf(c.RegLogs.Count() == 1, "count should be %d not %d", 1, c.RegLogs.Count())
|
|
items, e := c.RegLogs.GetOffset(0, 10)
|
|
expectNilErr(t, e)
|
|
exf(len(items) == 1, "len(items) should be %d not %d", 1, len(items))
|
|
// TODO: Add more tests
|
|
|
|
expectNilErr(t, c.RegLogs.DeleteOlderThanDays(2))
|
|
|
|
exf(c.RegLogs.Count() == 1, "count should be %d not %d", 1, c.RegLogs.Count())
|
|
items, e = c.RegLogs.GetOffset(0, 10)
|
|
expectNilErr(t, e)
|
|
exf(len(items) == 1, "len(items) should be %d not %d", 1, len(items))
|
|
// TODO: Add more tests
|
|
|
|
// TODO: Commit() test?
|
|
dayAgo := time.Now().AddDate(0, 0, -5)
|
|
items[0].DoneAt = dayAgo.Format("2006-01-02 15:04:05")
|
|
expectNilErr(t, items[0].Commit())
|
|
|
|
exf(c.RegLogs.Count() == 1, "count should be %d not %d", 1, c.RegLogs.Count())
|
|
items, e = c.RegLogs.GetOffset(0, 10)
|
|
expectNilErr(t, e)
|
|
exf(len(items) == 1, "len(items) should be %d not %d", 1, len(items))
|
|
// TODO: Add more tests
|
|
|
|
expectNilErr(t, c.RegLogs.DeleteOlderThanDays(2))
|
|
mustNone()
|
|
|
|
regLog = &c.RegLogItem{Username: "Aa", Email: "aa@example.com", FailureReason: "fake", Success: false, IP: ""}
|
|
id, e = regLog.Create()
|
|
exf(id == 2, "id should be %d not %d", 2, id)
|
|
expectNilErr(t, e)
|
|
|
|
exf(c.RegLogs.Count() == 1, "count should be %d not %d", 1, c.RegLogs.Count())
|
|
items, e = c.RegLogs.GetOffset(0, 10)
|
|
expectNilErr(t, e)
|
|
exf(len(items) == 1, "len(items) should be %d not %d", 1, len(items))
|
|
// TODO: Add more tests
|
|
|
|
expectNilErr(t, c.RegLogs.Purge())
|
|
mustNone()
|
|
|
|
// TODO: Add more tests
|
|
}
|
|
|
|
func TestLoginLogs(t *testing.T) {
|
|
miscinit(t)
|
|
if !c.PluginsInited {
|
|
c.InitPlugins()
|
|
}
|
|
ex, exf := exp(t), expf(t)
|
|
uid, e := c.Users.Create("Log Test", "Log Test", "", 1, true)
|
|
expectNilErr(t, e)
|
|
|
|
exf(c.LoginLogs.CountUser(-1) == 0, "countuser(-1) should be %d not %d", 0, c.LoginLogs.CountUser(-1))
|
|
exf(c.LoginLogs.CountUser(0) == 0, "countuser(0) should be %d not %d", 0, c.LoginLogs.CountUser(0))
|
|
exf(c.LoginLogs.CountUser(1) == 0, "countuser(1) should be %d not %d", 0, c.LoginLogs.CountUser(1))
|
|
goNone := func(uid, offset, perPage int) {
|
|
items, e := c.LoginLogs.GetOffset(uid, offset, perPage)
|
|
expectNilErr(t, e)
|
|
exf(len(items) == 0, "len(items) should be %d not %d", 0, len(items))
|
|
}
|
|
goNone(-1, 0, 10)
|
|
goNone(0, 0, 10)
|
|
goNone(1, 0, 10)
|
|
goNone(1, 1, 10)
|
|
goNone(1, 0, 0)
|
|
|
|
mustNone := func() {
|
|
exf(c.LoginLogs.Count() == 0, "count should be %d not %d", 0, c.LoginLogs.Count())
|
|
exf(c.LoginLogs.CountUser(uid) == 0, "countuser(%d) should be %d not %d", uid, 0, c.LoginLogs.CountUser(uid))
|
|
goNone(uid, 0, 10)
|
|
goNone(uid, 1, 10)
|
|
goNone(uid, 0, 0)
|
|
}
|
|
mustNone()
|
|
|
|
logItem := &c.LoginLogItem{UID: uid, Success: true, IP: ""}
|
|
_, e = logItem.Create()
|
|
expectNilErr(t, e)
|
|
|
|
exf(c.LoginLogs.Count() == 1, "count should be %d not %d", 1, c.LoginLogs.Count())
|
|
exf(c.LoginLogs.CountUser(uid) == 1, "countuser(%d) should be %d not %d", uid, 1, c.LoginLogs.CountUser(uid))
|
|
items, e := c.LoginLogs.GetOffset(uid, 0, 10)
|
|
expectNilErr(t, e)
|
|
exf(len(items) == 1, "len(items) should be %d not %d", 1, len(items))
|
|
// TODO: More tests
|
|
exf(items[0].UID == uid, "UID should be %d not %d", uid, items[0].UID)
|
|
ex(items[0].Success, "Success should be true")
|
|
ex(items[0].IP == "", "IP should be blank")
|
|
goNone(uid, 1, 10)
|
|
goNone(uid, 0, 0)
|
|
|
|
dayAgo := time.Now().AddDate(0, 0, -5)
|
|
items[0].DoneAt = dayAgo.Format("2006-01-02 15:04:05")
|
|
prevDoneAt := items[0].DoneAt
|
|
expectNilErr(t, items[0].Commit())
|
|
|
|
items, e = c.LoginLogs.GetOffset(uid, 0, 10)
|
|
expectNilErr(t, e)
|
|
exf(len(items) == 1, "len(items) should be %d not %d", 1, len(items))
|
|
// TODO: More tests
|
|
exf(items[0].UID == uid, "UID should be %d not %d", uid, items[0].UID)
|
|
ex(items[0].Success, "Success should be true")
|
|
ex(items[0].IP == "", "IP should be blank")
|
|
exf(items[0].DoneAt == prevDoneAt, "DoneAt should be %s not %s", prevDoneAt, items[0].DoneAt)
|
|
goNone(uid, 1, 10)
|
|
goNone(uid, 0, 0)
|
|
|
|
expectNilErr(t, c.LoginLogs.DeleteOlderThanDays(2))
|
|
mustNone()
|
|
|
|
logItem = &c.LoginLogItem{UID: uid, Success: true, IP: ""}
|
|
_, e = logItem.Create()
|
|
expectNilErr(t, e)
|
|
|
|
exf(c.LoginLogs.Count() == 1, "count should be %d not %d", 1, c.LoginLogs.Count())
|
|
exf(c.LoginLogs.CountUser(uid) == 1, "countuser(%d) should be %d not %d", uid, 1, c.LoginLogs.CountUser(uid))
|
|
items, e = c.LoginLogs.GetOffset(uid, 0, 10)
|
|
expectNilErr(t, e)
|
|
exf(len(items) == 1, "len(items) should be %d not %d", 1, len(items))
|
|
// TODO: More tests
|
|
exf(items[0].UID == uid, "UID should be %d not %d", uid, items[0].UID)
|
|
ex(items[0].Success, "Success should be true")
|
|
ex(items[0].IP == "", "IP should be blank")
|
|
goNone(uid, 1, 10)
|
|
goNone(uid, 0, 0)
|
|
|
|
expectNilErr(t, c.LoginLogs.Purge())
|
|
mustNone()
|
|
}
|
|
|
|
func TestPluginManager(t *testing.T) {
|
|
miscinit(t)
|
|
if !c.PluginsInited {
|
|
c.InitPlugins()
|
|
}
|
|
ex := exp(t)
|
|
|
|
_, ok := c.Plugins["fairy-dust"]
|
|
ex(!ok, "Plugin fairy-dust shouldn't exist")
|
|
pl, ok := c.Plugins["bbcode"]
|
|
ex(ok, "Plugin bbcode should exist")
|
|
ex(!pl.Installable, "Plugin bbcode shouldn't be installable")
|
|
ex(!pl.Installed, "Plugin bbcode shouldn't be 'installed'")
|
|
ex(!pl.Active, "Plugin bbcode shouldn't be active")
|
|
active, e := pl.BypassActive()
|
|
expectNilErr(t, e)
|
|
ex(!active, "Plugin bbcode shouldn't be active in the database either")
|
|
hasPlugin, e := pl.InDatabase()
|
|
expectNilErr(t, e)
|
|
ex(!hasPlugin, "Plugin bbcode shouldn't exist in the database")
|
|
// TODO: Add some test cases for SetActive and SetInstalled before calling AddToDatabase
|
|
|
|
expectNilErr(t, pl.AddToDatabase(true, false))
|
|
ex(!pl.Installable, "Plugin bbcode shouldn't be installable")
|
|
ex(!pl.Installed, "Plugin bbcode shouldn't be 'installed'")
|
|
ex(pl.Active, "Plugin bbcode should be active")
|
|
active, e = pl.BypassActive()
|
|
expectNilErr(t, e)
|
|
ex(active, "Plugin bbcode should be active in the database too")
|
|
hasPlugin, e = pl.InDatabase()
|
|
expectNilErr(t, e)
|
|
ex(hasPlugin, "Plugin bbcode should exist in the database")
|
|
ex(pl.Init != nil, "Plugin bbcode should have an init function")
|
|
expectNilErr(t, pl.Init(pl))
|
|
|
|
expectNilErr(t, pl.SetActive(true))
|
|
ex(!pl.Installable, "Plugin bbcode shouldn't be installable")
|
|
ex(!pl.Installed, "Plugin bbcode shouldn't be 'installed'")
|
|
ex(pl.Active, "Plugin bbcode should still be active")
|
|
active, e = pl.BypassActive()
|
|
expectNilErr(t, e)
|
|
ex(active, "Plugin bbcode should still be active in the database too")
|
|
hasPlugin, e = pl.InDatabase()
|
|
expectNilErr(t, e)
|
|
ex(hasPlugin, "Plugin bbcode should still exist in the database")
|
|
|
|
expectNilErr(t, pl.SetActive(false))
|
|
ex(!pl.Installable, "Plugin bbcode shouldn't be installable")
|
|
ex(!pl.Installed, "Plugin bbcode shouldn't be 'installed'")
|
|
ex(!pl.Active, "Plugin bbcode shouldn't be active")
|
|
active, e = pl.BypassActive()
|
|
expectNilErr(t, e)
|
|
ex(!active, "Plugin bbcode shouldn't be active in the database")
|
|
hasPlugin, e = pl.InDatabase()
|
|
expectNilErr(t, e)
|
|
ex(hasPlugin, "Plugin bbcode should still exist in the database")
|
|
ex(pl.Deactivate != nil, "Plugin bbcode should have an init function")
|
|
pl.Deactivate(pl) // Returns nothing
|
|
|
|
// Not installable, should not be mutated
|
|
ex(pl.SetInstalled(true) == c.ErrPluginNotInstallable, "Plugin was set as installed despite not being installable")
|
|
ex(!pl.Installable, "Plugin bbcode shouldn't be installable")
|
|
ex(!pl.Installed, "Plugin bbcode shouldn't be 'installed'")
|
|
ex(!pl.Active, "Plugin bbcode shouldn't be active")
|
|
active, e = pl.BypassActive()
|
|
expectNilErr(t, e)
|
|
ex(!active, "Plugin bbcode shouldn't be active in the database either")
|
|
hasPlugin, e = pl.InDatabase()
|
|
expectNilErr(t, e)
|
|
ex(hasPlugin, "Plugin bbcode should still exist in the database")
|
|
|
|
ex(pl.SetInstalled(false) == c.ErrPluginNotInstallable, "Plugin was set as not installed despite not being installable")
|
|
ex(!pl.Installable, "Plugin bbcode shouldn't be installable")
|
|
ex(!pl.Installed, "Plugin bbcode shouldn't be 'installed'")
|
|
ex(!pl.Active, "Plugin bbcode shouldn't be active")
|
|
active, e = pl.BypassActive()
|
|
expectNilErr(t, e)
|
|
ex(!active, "Plugin bbcode shouldn't be active in the database either")
|
|
hasPlugin, e = pl.InDatabase()
|
|
expectNilErr(t, e)
|
|
ex(hasPlugin, "Plugin bbcode should still exist in the database")
|
|
|
|
// This isn't really installable, but we want to get a few tests done before getting plugins which are stateful
|
|
pl.Installable = true
|
|
expectNilErr(t, pl.SetInstalled(true))
|
|
ex(pl.Installable, "Plugin bbcode should be installable")
|
|
ex(pl.Installed, "Plugin bbcode should be 'installed'")
|
|
ex(!pl.Active, "Plugin bbcode shouldn't be active")
|
|
active, e = pl.BypassActive()
|
|
expectNilErr(t, e)
|
|
ex(!active, "Plugin bbcode shouldn't be active in the database either")
|
|
hasPlugin, e = pl.InDatabase()
|
|
expectNilErr(t, e)
|
|
ex(hasPlugin, "Plugin bbcode should still exist in the database")
|
|
|
|
expectNilErr(t, pl.SetInstalled(false))
|
|
ex(pl.Installable, "Plugin bbcode should be installable")
|
|
ex(!pl.Installed, "Plugin bbcode shouldn't be 'installed'")
|
|
ex(!pl.Active, "Plugin bbcode shouldn't be active")
|
|
active, e = pl.BypassActive()
|
|
expectNilErr(t, e)
|
|
ex(!active, "Plugin bbcode shouldn't be active in the database either")
|
|
hasPlugin, e = pl.InDatabase()
|
|
expectNilErr(t, e)
|
|
ex(hasPlugin, "Plugin bbcode should still exist in the database")
|
|
|
|
// Bugs sometimes arise when we try to delete a hook when there are multiple, so test for that
|
|
// TODO: Do a finer grained test for that case...? A bigger test might catch more odd cases with multiple plugins
|
|
pl2, ok := c.Plugins["markdown"]
|
|
ex(ok, "Plugin markdown should exist")
|
|
ex(!pl2.Installable, "Plugin markdown shouldn't be installable")
|
|
ex(!pl2.Installed, "Plugin markdown shouldn't be 'installed'")
|
|
ex(!pl2.Active, "Plugin markdown shouldn't be active")
|
|
active, e = pl2.BypassActive()
|
|
expectNilErr(t, e)
|
|
ex(!active, "Plugin markdown shouldn't be active in the database either")
|
|
hasPlugin, e = pl2.InDatabase()
|
|
expectNilErr(t, e)
|
|
ex(!hasPlugin, "Plugin markdown shouldn't exist in the database")
|
|
|
|
expectNilErr(t, pl2.AddToDatabase(true, false))
|
|
expectNilErr(t, pl2.Init(pl2))
|
|
expectNilErr(t, pl.SetActive(true))
|
|
expectNilErr(t, pl.Init(pl))
|
|
pl2.Deactivate(pl2)
|
|
expectNilErr(t, pl2.SetActive(false))
|
|
pl.Deactivate(pl)
|
|
expectNilErr(t, pl.SetActive(false))
|
|
|
|
// Hook tests
|
|
ht := func() *c.HookTable {
|
|
return c.GetHookTable()
|
|
}
|
|
ex(ht().Sshook("haha", "ho") == "ho", "Sshook shouldn't have anything bound to it yet")
|
|
handle := func(in string) (out string) {
|
|
return in + "hi"
|
|
}
|
|
pl.AddHook("haha", handle)
|
|
ex(ht().Sshook("haha", "ho") == "hohi", "Sshook didn't give hohi")
|
|
pl.RemoveHook("haha", handle)
|
|
ex(ht().Sshook("haha", "ho") == "ho", "Sshook shouldn't have anything bound to it anymore")
|
|
|
|
/*ex(ht().Hook("haha", "ho") == "ho", "Hook shouldn't have anything bound to it yet")
|
|
handle2 := func(inI interface{}) (out interface{}) {
|
|
return inI.(string) + "hi"
|
|
}
|
|
pl.AddHook("hehe", handle2)
|
|
ex(ht().Hook("hehe", "ho").(string) == "hohi", "Hook didn't give hohi")
|
|
pl.RemoveHook("hehe", handle2)
|
|
ex(ht().Hook("hehe", "ho").(string) == "ho", "Hook shouldn't have anything bound to it anymore")*/
|
|
|
|
// TODO: Add tests for more hook types
|
|
}
|
|
|
|
func TestPhrases(t *testing.T) {
|
|
getPhrase := phrases.GetPermPhrase
|
|
tp := func(name, expects string) {
|
|
res := getPhrase(name)
|
|
expect(t, res == expects, "Not the expected phrase, got '"+res+"' instead")
|
|
}
|
|
tp("BanUsers", "Can ban users")
|
|
tp("NoSuchPerm", "{lang.perms[NoSuchPerm]}")
|
|
tp("ViewTopic", "Can view topics")
|
|
tp("NoSuchPerm", "{lang.perms[NoSuchPerm]}")
|
|
|
|
// TODO: Cover the other phrase types, also try switching between languages to see if anything strange happens
|
|
}
|
|
|
|
func TestMetaStore(t *testing.T) {
|
|
ex, exf := exp(t), expf(t)
|
|
m, e := c.Meta.Get("magic")
|
|
ex(m == "", "meta var magic should be empty")
|
|
recordMustNotExist(t, e, "meta var magic should not exist")
|
|
|
|
expectVal := func(name, expect string) {
|
|
m, e = c.Meta.Get(name)
|
|
expectNilErr(t, e)
|
|
exf(m == expect, "meta var %s should be %s", name, expect)
|
|
}
|
|
|
|
expectNilErr(t, c.Meta.Set("magic", "lol"))
|
|
expectVal("magic", "lol")
|
|
expectNilErr(t, c.Meta.Set("magic", "wha"))
|
|
expectVal("magic", "wha")
|
|
|
|
m, e = c.Meta.Get("giggle")
|
|
ex(m == "", "meta var giggle should be empty")
|
|
recordMustNotExist(t, e, "meta var giggle should not exist")
|
|
|
|
expectNilErr(t, c.Meta.SetInt("magic", 1))
|
|
expectVal("magic", "1")
|
|
expectNilErr(t, c.Meta.SetInt64("magic", 5))
|
|
expectVal("magic", "5")
|
|
}
|
|
|
|
func TestPages(t *testing.T) {
|
|
ex := exp(t)
|
|
ex(c.Pages.Count() == 0, "Page count should be 0")
|
|
_, e := c.Pages.Get(1)
|
|
recordMustNotExist(t, e, "Page 1 should not exist yet")
|
|
expectNilErr(t, c.Pages.Delete(-1))
|
|
expectNilErr(t, c.Pages.Delete(0))
|
|
expectNilErr(t, c.Pages.Delete(1))
|
|
_, e = c.Pages.Get(1)
|
|
recordMustNotExist(t, e, "Page 1 should not exist yet")
|
|
//e = c.Pages.Reload(1)
|
|
//recordMustNotExist(t,e,"Page 1 should not exist yet")
|
|
|
|
ipage := c.BlankCustomPage()
|
|
ipage.Name = "test"
|
|
ipage.Title = "Test"
|
|
ipage.Body = "A test page"
|
|
pid, e := ipage.Create()
|
|
expectNilErr(t, e)
|
|
ex(pid == 1, "The first page should have an ID of 1")
|
|
ex(c.Pages.Count() == 1, "Page count should be 1")
|
|
|
|
test := func(pid int, ep *c.CustomPage) {
|
|
p, e := c.Pages.Get(pid)
|
|
expectNilErr(t, e)
|
|
ex(p.Name == ep.Name, "The page name should be "+ep.Name)
|
|
ex(p.Title == ep.Title, "The page title should be "+ep.Title)
|
|
ex(p.Body == ep.Body, "The page body should be "+ep.Body)
|
|
}
|
|
test(1, ipage)
|
|
|
|
opage, err := c.Pages.Get(1)
|
|
expectNilErr(t, err)
|
|
opage.Name = "t"
|
|
opage.Title = "T"
|
|
opage.Body = "testing"
|
|
expectNilErr(t, opage.Commit())
|
|
|
|
test(1, opage)
|
|
|
|
expectNilErr(t, c.Pages.Delete(1))
|
|
ex(c.Pages.Count() == 0, "Page count should be 0")
|
|
_, e = c.Pages.Get(1)
|
|
recordMustNotExist(t, e, "Page 1 should not exist")
|
|
//e = c.Pages.Reload(1)
|
|
//recordMustNotExist(t,e,"Page 1 should not exist")
|
|
|
|
// TODO: More tests
|
|
}
|
|
|
|
func TestWordFilters(t *testing.T) {
|
|
ex, exf := exp(t), expf(t)
|
|
// TODO: Test the word filters and their store
|
|
ex(c.WordFilters.Length() == 0, "Word filter list should be empty")
|
|
ex(c.WordFilters.EstCount() == 0, "Word filter list should be empty")
|
|
ex(c.WordFilters.Count() == 0, "Word filter list should be empty")
|
|
filters, err := c.WordFilters.GetAll()
|
|
expectNilErr(t, err) // TODO: Slightly confusing that we don't get ErrNoRow here
|
|
ex(len(filters) == 0, "Word filter map should be empty")
|
|
// TODO: Add a test for ParseMessage relating to word filters
|
|
_, err = c.WordFilters.Get(1)
|
|
recordMustNotExist(t, err, "filter 1 should not exist")
|
|
|
|
wfid, err := c.WordFilters.Create("imbecile", "lovely")
|
|
expectNilErr(t, err)
|
|
ex(wfid == 1, "The first word filter should have an ID of 1")
|
|
ex(c.WordFilters.Length() == 1, "Word filter list should not be empty")
|
|
ex(c.WordFilters.EstCount() == 1, "Word filter list should not be empty")
|
|
ex(c.WordFilters.Count() == 1, "Word filter list should not be empty")
|
|
|
|
ftest := func(f *c.WordFilter, id int, find, replace string) {
|
|
exf(f.ID == id, "Word filter ID should be %d, not %d", id, f.ID)
|
|
exf(f.Find == find, "Word filter needle should be '%s', not '%s'", find, f.Find)
|
|
exf(f.Replace == replace, "Word filter replacement should be '%s', not '%s'", replace, f.Replace)
|
|
}
|
|
|
|
filters, err = c.WordFilters.GetAll()
|
|
expectNilErr(t, err)
|
|
ex(len(filters) == 1, "Word filter map should not be empty")
|
|
ftest(filters[1], 1, "imbecile", "lovely")
|
|
|
|
filter, err := c.WordFilters.Get(1)
|
|
expectNilErr(t, err)
|
|
ftest(filter, 1, "imbecile", "lovely")
|
|
|
|
// Update
|
|
expectNilErr(t, c.WordFilters.Update(1, "b", "a"))
|
|
|
|
ex(c.WordFilters.Length() == 1, "Word filter list should not be empty")
|
|
ex(c.WordFilters.EstCount() == 1, "Word filter list should not be empty")
|
|
ex(c.WordFilters.Count() == 1, "Word filter list should not be empty")
|
|
|
|
filters, err = c.WordFilters.GetAll()
|
|
expectNilErr(t, err)
|
|
ex(len(filters) == 1, "Word filter map should not be empty")
|
|
ftest(filters[1], 1, "b", "a")
|
|
|
|
filter, err = c.WordFilters.Get(1)
|
|
expectNilErr(t, err)
|
|
ftest(filter, 1, "b", "a")
|
|
|
|
// TODO: Add a test for ParseMessage relating to word filters
|
|
|
|
expectNilErr(t, c.WordFilters.Delete(1))
|
|
|
|
ex(c.WordFilters.Length() == 0, "Word filter list should be empty")
|
|
ex(c.WordFilters.EstCount() == 0, "Word filter list should be empty")
|
|
ex(c.WordFilters.Count() == 0, "Word filter list should be empty")
|
|
filters, err = c.WordFilters.GetAll()
|
|
expectNilErr(t, err) // TODO: Slightly confusing that we don't get ErrNoRow here
|
|
ex(len(filters) == 0, "Word filter map should be empty")
|
|
_, err = c.WordFilters.Get(1)
|
|
recordMustNotExist(t, err, "filter 1 should not exist")
|
|
|
|
// TODO: Any more tests we could do?
|
|
}
|
|
|
|
func TestMFAStore(t *testing.T) {
|
|
exf := expf(t)
|
|
|
|
mustNone := func() {
|
|
_, e := c.MFAstore.Get(-1)
|
|
recordMustNotExist(t, e, "mfa uid -1 should not exist")
|
|
_, e = c.MFAstore.Get(0)
|
|
recordMustNotExist(t, e, "mfa uid 0 should not exist")
|
|
_, e = c.MFAstore.Get(1)
|
|
recordMustNotExist(t, e, "mfa uid 1 should not exist")
|
|
}
|
|
mustNone()
|
|
|
|
secret, e := c.GenerateGAuthSecret()
|
|
expectNilErr(t, e)
|
|
expectNilErr(t, c.MFAstore.Create(secret, 1))
|
|
_, e = c.MFAstore.Get(0)
|
|
recordMustNotExist(t, e, "mfa uid 0 should not exist")
|
|
var scratches []string
|
|
it, e := c.MFAstore.Get(1)
|
|
test := func(j int) {
|
|
expectNilErr(t, e)
|
|
exf(it.UID == 1, "UID should be 1 not %d", it.UID)
|
|
exf(it.Secret == secret, "Secret should be '%s' not %s", secret, it.Secret)
|
|
exf(len(it.Scratch) == 8, "Scratch should be 8 not %d", len(it.Scratch))
|
|
for i, scratch := range it.Scratch {
|
|
exf(scratch != "", "scratch %d should not be empty", i)
|
|
if scratches != nil {
|
|
if j == i {
|
|
exf(scratches[i] != scratch, "scratches[%d] should not be %s", i, scratches[i])
|
|
} else {
|
|
exf(scratches[i] == scratch, "scratches[%d] should be %s not %s", i, scratches[i], scratch)
|
|
}
|
|
}
|
|
}
|
|
scratches = make([]string, 8)
|
|
copy(scratches, it.Scratch)
|
|
}
|
|
test(0)
|
|
for i := 0; i < len(scratches); i++ {
|
|
expectNilErr(t, it.BurnScratch(i))
|
|
it, e = c.MFAstore.Get(1)
|
|
test(i)
|
|
}
|
|
token, e := gauth.GetTOTPToken(secret)
|
|
expectNilErr(t, e)
|
|
expectNilErr(t, c.Auth.ValidateMFAToken(token, 1))
|
|
expectNilErr(t, it.Delete())
|
|
mustNone()
|
|
}
|
|
|
|
// TODO: Expand upon the valid characters which can go in URLs?
|
|
func TestSlugs(t *testing.T) {
|
|
l := &MEPairList{nil}
|
|
c.Config.BuildSlugs = true // Flip this switch, otherwise all the tests will fail
|
|
|
|
l.Add("Unknown", "unknown")
|
|
l.Add("Unknown2", "unknown2")
|
|
l.Add("Unknown ", "unknown")
|
|
l.Add("Unknown 2", "unknown-2")
|
|
l.Add("Unknown 2", "unknown-2")
|
|
l.Add("Admin Alice", "admin-alice")
|
|
l.Add("Admin_Alice", "adminalice")
|
|
l.Add("Admin_Alice-", "adminalice")
|
|
l.Add("-Admin_Alice-", "adminalice")
|
|
l.Add("-Admin@Alice-", "adminalice")
|
|
l.Add("-Admin😀Alice-", "adminalice")
|
|
l.Add("u", "u")
|
|
l.Add("", "untitled")
|
|
l.Add(" ", "untitled")
|
|
l.Add("-", "untitled")
|
|
l.Add("--", "untitled")
|
|
l.Add("é", "é")
|
|
l.Add("-é-", "é")
|
|
l.Add("-你好-", "untitled")
|
|
l.Add("-こにちは-", "untitled")
|
|
|
|
for _, item := range l.Items {
|
|
t.Log("Testing string '" + item.Msg + "'")
|
|
res := c.NameToSlug(item.Msg)
|
|
if res != item.Expects {
|
|
t.Error("Bad output:", "'"+res+"'")
|
|
t.Error("Expected:", item.Expects)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestWidgets(t *testing.T) {
|
|
ex, exf := exp(t), expf(t)
|
|
_, e := c.Widgets.Get(1)
|
|
recordMustNotExist(t, e, "There shouldn't be any widgets by default")
|
|
widgets := c.Docks.RightSidebar.Items
|
|
exf(len(widgets) == 0, "RightSidebar should have 0 items, not %d", len(widgets))
|
|
|
|
widget := &c.Widget{Position: 0, Side: "rightSidebar", Type: "simple", Enabled: true, Location: "global"}
|
|
ewidget := &c.WidgetEdit{widget, map[string]string{"Name": "Test", "Text": "Testing"}}
|
|
wid, e := ewidget.Create()
|
|
expectNilErr(t, e)
|
|
ex(wid == 1, "wid should be 1")
|
|
|
|
wtest := func(w, w2 *c.Widget) {
|
|
ex(w.Position == w2.Position, "wrong position")
|
|
ex(w.Side == w2.Side, "wrong side")
|
|
ex(w.Type == w2.Type, "wrong type")
|
|
ex(w.Enabled == w2.Enabled, "wrong enabled")
|
|
ex(w.Location == w2.Location, "wrong location")
|
|
}
|
|
|
|
// TODO: Do a test for the widget body
|
|
widget2, e := c.Widgets.Get(1)
|
|
expectNilErr(t, e)
|
|
wtest(widget, widget2)
|
|
|
|
widgets = c.Docks.RightSidebar.Items
|
|
exf(len(widgets) == 1, "RightSidebar should have 1 item, not %d", len(widgets))
|
|
wtest(widget, widgets[0])
|
|
|
|
widget2.Enabled = false
|
|
ewidget = &c.WidgetEdit{widget2, map[string]string{"Name": "Test", "Text": "Testing"}}
|
|
expectNilErr(t, ewidget.Commit())
|
|
|
|
widget2, e = c.Widgets.Get(1)
|
|
expectNilErr(t, e)
|
|
widget.Enabled = false
|
|
wtest(widget, widget2)
|
|
|
|
widgets = c.Docks.RightSidebar.Items
|
|
exf(len(widgets) == 1, "RightSidebar should have 1 item, not %d", len(widgets))
|
|
widget.Enabled = false
|
|
wtest(widget, widgets[0])
|
|
|
|
expectNilErr(t, widget2.Delete())
|
|
|
|
_, e = c.Widgets.Get(1)
|
|
recordMustNotExist(t, e, "There shouldn't be any widgets anymore")
|
|
widgets = c.Docks.RightSidebar.Items
|
|
exf(len(widgets) == 0, "RightSidebar should have 0 items, not %d", len(widgets))
|
|
}
|
|
|
|
/*type ForumActionStoreInt interface {
|
|
Get(faid int) (*ForumAction, error)
|
|
GetInForum(fid int) ([]*ForumAction, error)
|
|
GetAll() ([]*ForumAction, error)
|
|
GetNewTopicActions(fid int) ([]*ForumAction, error)
|
|
|
|
Add(fa *ForumAction) (int, error)
|
|
Delete(faid int) error
|
|
Exists(faid int) bool
|
|
Count() int
|
|
CountInForum(fid int) int
|
|
|
|
DailyTick() error
|
|
}*/
|
|
|
|
func TestForumActions(t *testing.T) {
|
|
ex, exf, s := exp(t), expf(t), c.ForumActionStore
|
|
|
|
count := s.CountInForum(-1)
|
|
exf(count == 0, "count should be %d not %d", 0, count)
|
|
count = s.CountInForum(0)
|
|
exf(count == 0, "count in 0 should be %d not %d", 0, count)
|
|
ex(!s.Exists(-1), "faid -1 should not exist")
|
|
ex(!s.Exists(0), "faid 0 should not exist")
|
|
_, e := s.Get(-1)
|
|
recordMustNotExist(t, e, "faid -1 should not exist")
|
|
_, e = s.Get(0)
|
|
recordMustNotExist(t, e, "faid 0 should not exist")
|
|
|
|
noActions := func(fid, faid int) {
|
|
/*sfid, */ sfaid := /*strconv.Itoa(fid), */ strconv.Itoa(faid)
|
|
count := s.Count()
|
|
exf(count == 0, "count should be %d not %d", 0, count)
|
|
count = s.CountInForum(fid)
|
|
exf(count == 0, "count in %d should be %d not %d", fid, 0, count)
|
|
exf(!s.Exists(faid), "faid %d should not exist", faid)
|
|
_, e := s.Get(faid)
|
|
recordMustNotExist(t, e, "faid "+sfaid+" should not exist")
|
|
//exf(fa == nil, "fa should be nil not %+v", fa)
|
|
fas, e := s.GetInForum(fid)
|
|
//recordMustNotExist(t, e, "fid "+sfid+" should not have any actions")
|
|
expectNilErr(t, e) // TODO: Why does this not return ErrNoRows?
|
|
exf(len(fas) == 0, "len(fas) should be %d not %d", 0, len(fas))
|
|
fas, e = s.GetAll()
|
|
//recordMustNotExist(t, e, "there should not be any actions")
|
|
expectNilErr(t, e) // TODO: Why does this not return ErrNoRows?
|
|
exf(len(fas) == 0, "len(fas) should be %d not %d", 0, len(fas))
|
|
fas, e = s.GetNewTopicActions(fid)
|
|
//recordMustNotExist(t, e, "fid "+sfid+" should not have any new topic actions")
|
|
expectNilErr(t, e) // TODO: Why does this not return ErrNoRows?
|
|
exf(len(fas) == 0, "len(fas) should be %d not %d", 0, len(fas))
|
|
}
|
|
noActions(1, 1)
|
|
|
|
fid, e := c.Forums.Create("Forum Action Test", "Forum Action Test", true, "")
|
|
expectNilErr(t, e)
|
|
noActions(fid, 1)
|
|
|
|
faid, e := c.ForumActionStore.Add(&c.ForumAction{
|
|
Forum: fid,
|
|
RunOnTopicCreation: false,
|
|
RunDaysAfterTopicCreation: 1,
|
|
RunDaysAfterTopicLastReply: 0,
|
|
Action: c.ForumActionLock,
|
|
Extra: "",
|
|
})
|
|
expectNilErr(t, e)
|
|
exf(faid == 1, "faid should be %d not %d", 1, faid)
|
|
count = s.Count()
|
|
exf(count == 1, "count should be %d not %d", 1, count)
|
|
count = s.CountInForum(fid)
|
|
exf(count == 1, "count in %d should be %d not %d", fid, 1, count)
|
|
exf(s.Exists(faid), "faid %d should exist", faid)
|
|
|
|
fa, e := s.Get(faid)
|
|
expectNilErr(t, e)
|
|
exf(fa.ID == faid, "fa.ID should be %d not %d", faid, fa.ID)
|
|
exf(fa.Forum == fid, "fa.Forum should be %d not %d", fid, fa.Forum)
|
|
exf(fa.RunOnTopicCreation == false, "fa.RunOnTopicCreation should be false")
|
|
exf(fa.RunDaysAfterTopicCreation == 1, "fa.RunDaysAfterTopicCreation should be %d not %d", 1, fa.RunDaysAfterTopicCreation)
|
|
exf(fa.RunDaysAfterTopicLastReply == 0, "fa.RunDaysAfterTopicLastReply should be %d not %d", 0, fa.RunDaysAfterTopicLastReply)
|
|
exf(fa.Action == c.ForumActionLock, "fa.Action should be %d not %d", c.ForumActionLock, fa.Action)
|
|
exf(fa.Extra == "", "fa.Extra should be '%s' not '%s'", "", fa.Extra)
|
|
|
|
tid, e := c.Topics.Create(fid, "Forum Action Topic", "Forum Action Topic", 1, "")
|
|
expectNilErr(t, e)
|
|
topic, e := c.Topics.Get(tid)
|
|
expectNilErr(t, e)
|
|
ex(!topic.IsClosed, "topic.IsClosed should be false")
|
|
dayAgo := time.Now().AddDate(0, 0, -5)
|
|
expectNilErr(t, topic.TestSetCreatedAt(dayAgo))
|
|
expectNilErr(t, fa.Run())
|
|
topic, e = c.Topics.Get(tid)
|
|
expectNilErr(t, e)
|
|
ex(topic.IsClosed, "topic.IsClosed should be true")
|
|
/*_, e = c.Rstore.Create(topic, "Forum Action Reply", "", 1)
|
|
expectNilErr(t, e)*/
|
|
|
|
tid, e = c.Topics.Create(fid, "Forum Action Topic 2", "Forum Action Topic 2", 1, "")
|
|
expectNilErr(t, e)
|
|
topic, e = c.Topics.Get(tid)
|
|
expectNilErr(t, e)
|
|
ex(!topic.IsClosed, "topic.IsClosed should be false")
|
|
expectNilErr(t, fa.Run())
|
|
topic, e = c.Topics.Get(tid)
|
|
expectNilErr(t, e)
|
|
ex(!topic.IsClosed, "topic.IsClosed should be false")
|
|
|
|
_ = tid
|
|
|
|
expectNilErr(t, s.Delete(faid))
|
|
noActions(fid, faid)
|
|
|
|
// TODO: Bulk lock tests
|
|
faid, e = c.ForumActionStore.Add(&c.ForumAction{
|
|
Forum: fid,
|
|
RunOnTopicCreation: false,
|
|
RunDaysAfterTopicCreation: 2,
|
|
RunDaysAfterTopicLastReply: 0,
|
|
Action: c.ForumActionLock,
|
|
Extra: "",
|
|
})
|
|
expectNilErr(t, e)
|
|
|
|
var l []int
|
|
addTopic := func() {
|
|
tid, e = c.Topics.Create(fid, "Forum Action Topic 2", "Forum Action Topic 2", 1, "")
|
|
expectNilErr(t, e)
|
|
topic, e := c.Topics.Get(tid)
|
|
expectNilErr(t, e)
|
|
ex(!topic.IsClosed, "topic.IsClosed should be false")
|
|
dayAgo := time.Now().AddDate(0, 0, -5)
|
|
expectNilErr(t, topic.TestSetCreatedAt(dayAgo))
|
|
l = append(l, tid)
|
|
}
|
|
lTest := func() {
|
|
for _, ll := range l {
|
|
to, e := c.Topics.Get(ll)
|
|
expectNilErr(t, e)
|
|
ex(to.IsClosed, "to.IsClosed should be true")
|
|
}
|
|
l = nil
|
|
}
|
|
|
|
addTopic()
|
|
addTopic()
|
|
addTopic()
|
|
addTopic()
|
|
addTopic()
|
|
addTopic()
|
|
addTopic()
|
|
addTopic()
|
|
addTopic()
|
|
addTopic()
|
|
expectNilErr(t, fa.Run())
|
|
lTest()
|
|
// TODO: Create a method on the *ForumAction to get the count of topics which it could be run on and add a test to verify the count is as expected.
|
|
|
|
addTopic()
|
|
addTopic()
|
|
addTopic()
|
|
addTopic()
|
|
addTopic()
|
|
addTopic()
|
|
addTopic()
|
|
addTopic()
|
|
addTopic()
|
|
addTopic()
|
|
addTopic()
|
|
expectNilErr(t, fa.Run())
|
|
lTest()
|
|
// TODO: Create a method on the *ForumAction to get the count of topics which it could be run on and add a test to verify the count is as expected.
|
|
}
|
|
|
|
func TestTopicList(t *testing.T) {
|
|
ex, exf := exp(t), expf(t)
|
|
fid, err := c.Forums.Create("Test Forum", "Desc for test forum", true, "")
|
|
expectNilErr(t, err)
|
|
tint := c.TopicList.(c.TopicListIntTest)
|
|
|
|
testPagi := func(p c.Paginator, pageList []int, page, lastPage int) {
|
|
exf(len(p.PageList) == len(pageList), "len(pagi.PageList) should be %d not %d", len(pageList), len(p.PageList))
|
|
for i, page := range pageList {
|
|
exf(p.PageList[i] == page, "pagi.PageList[%d] should be %d not %d", i, page, p.PageList[i])
|
|
}
|
|
exf(p.Page == page, "pagi.Page should be %d not %d", page, p.Page)
|
|
exf(p.LastPage == lastPage, "pagi.LastPage should be %d not %d", lastPage, p.LastPage)
|
|
}
|
|
test := func(topicList []*c.TopicsRow, pagi c.Paginator, listLen int, pagi2 c.Paginator, tid1 int) {
|
|
exf(len(topicList) == listLen, "len(topicList) should be %d not %d", listLen, len(topicList))
|
|
if len(topicList) > 0 {
|
|
topic := topicList[0]
|
|
exf(topic.ID == tid1, "topic.ID should be %d not %d", tid1, topic.ID)
|
|
}
|
|
testPagi(pagi, pagi2.PageList, pagi2.Page, pagi2.LastPage)
|
|
}
|
|
noTopics := func(topicList []*c.TopicsRow, pagi c.Paginator) {
|
|
exf(len(topicList) == 0, "len(topicList) should be 0 not %d", len(topicList))
|
|
testPagi(pagi, []int{}, 1, 1)
|
|
}
|
|
noTopicsOnPage2 := func(topicList []*c.TopicsRow, pagi c.Paginator) {
|
|
exf(len(topicList) == 0, "len(topicList) should be 0 not %d", len(topicList))
|
|
testPagi(pagi, []int{1}, 2, 1)
|
|
}
|
|
|
|
forum, err := c.Forums.Get(fid)
|
|
expectNilErr(t, err)
|
|
|
|
isAdmin, isMod, isBanned := false, false, false
|
|
gid, err := c.Groups.Create("Topic List Test", "Test", isAdmin, isMod, isBanned)
|
|
expectNilErr(t, err)
|
|
ex(c.Groups.Exists(gid), "The group we just made doesn't exist")
|
|
|
|
fp, err := c.FPStore.GetCopy(fid, gid)
|
|
if err == sql.ErrNoRows {
|
|
fp = *c.BlankForumPerms()
|
|
} else if err != nil {
|
|
expectNilErr(t, err)
|
|
}
|
|
fp.ViewTopic = true
|
|
|
|
forum, err = c.Forums.Get(fid)
|
|
expectNilErr(t, err)
|
|
expectNilErr(t, forum.SetPerms(&fp, "custom", gid))
|
|
|
|
g, err := c.Groups.Get(gid)
|
|
expectNilErr(t, err)
|
|
|
|
noTopicsTests := func() {
|
|
rr := func(page, orderby int) {
|
|
topicList, forumList, pagi, err := c.TopicList.GetListByGroup(g, page, orderby, []int{fid})
|
|
expectNilErr(t, err)
|
|
noTopics(topicList, pagi)
|
|
exf(len(forumList) == 1, "len(forumList) should be 1 not %d", len(forumList))
|
|
}
|
|
rr(1, 0)
|
|
rr(2, 0)
|
|
rr(1, 1)
|
|
rr(2, 1)
|
|
|
|
topicList, pagi, err := c.TopicList.GetListByForum(forum, 1, 0)
|
|
expectNilErr(t, err)
|
|
noTopics(topicList, pagi)
|
|
|
|
topicList, pagi, err = tint.RawGetListByForum(forum, 1, 0)
|
|
expectNilErr(t, err)
|
|
noTopics(topicList, pagi)
|
|
|
|
topicList, forumList, pagi, err := c.TopicList.GetList(1, 0, []int{fid})
|
|
expectNilErr(t, err)
|
|
noTopics(topicList, pagi)
|
|
exf(len(forumList) == 1, "len(forumList) should be 1 not %d", len(forumList))
|
|
|
|
topicList, forumList, pagi, err = c.TopicList.GetListByCanSee([]int{fid}, 1, 0, []int{fid})
|
|
expectNilErr(t, err)
|
|
noTopics(topicList, pagi)
|
|
exf(len(forumList) == 1, "len(forumList) should be 1 not %d", len(forumList))
|
|
|
|
topicList, forumList, pagi, err = c.TopicList.GetListByCanSee([]int{}, 1, 0, []int{fid})
|
|
expectNilErr(t, err)
|
|
noTopics(topicList, pagi)
|
|
// TODO: Why is there a discrepency between this and GetList()?
|
|
exf(len(forumList) == 0, "len(forumList) should be 0 not %d", len(forumList))
|
|
}
|
|
noTopicsTests()
|
|
|
|
tid, err := c.Topics.Create(fid, "New Topic", "New Topic Body", 1, "")
|
|
expectNilErr(t, err)
|
|
|
|
topicList, forumList, pagi, err := c.TopicList.GetListByGroup(g, 1, 0, []int{fid})
|
|
expectNilErr(t, err)
|
|
test(topicList, pagi, 1, c.Paginator{[]int{1}, 1, 1}, tid)
|
|
exf(len(forumList) == 1, "len(forumList) should be 1 not %d", len(forumList))
|
|
|
|
topicList, forumList, pagi, err = c.TopicList.GetListByGroup(g, 2, 0, []int{fid})
|
|
expectNilErr(t, err)
|
|
noTopics(topicList, pagi)
|
|
exf(len(forumList) == 1, "len(forumList) should be 1 not %d", len(forumList))
|
|
|
|
topicList, forumList, pagi, err = c.TopicList.GetListByGroup(g, 1, 1, []int{fid})
|
|
expectNilErr(t, err)
|
|
test(topicList, pagi, 1, c.Paginator{[]int{1}, 1, 1}, tid)
|
|
exf(len(forumList) == 1, "len(forumList) should be 1 not %d", len(forumList))
|
|
|
|
topicList, forumList, pagi, err = c.TopicList.GetListByGroup(g, 2, 1, []int{fid})
|
|
expectNilErr(t, err)
|
|
noTopicsOnPage2(topicList, pagi)
|
|
exf(len(forumList) == 1, "len(forumList) should be 1 not %d", len(forumList))
|
|
|
|
topicList, pagi, err = tint.RawGetListByForum(forum, 1, 0)
|
|
expectNilErr(t, err)
|
|
test(topicList, pagi, 1, c.Paginator{[]int{1}, 1, 1}, tid)
|
|
|
|
topicList, pagi, err = tint.RawGetListByForum(forum, 0, 0)
|
|
expectNilErr(t, err)
|
|
test(topicList, pagi, 1, c.Paginator{[]int{1}, 1, 1}, tid)
|
|
|
|
expectNilErr(t, tint.Tick())
|
|
forum, err = c.Forums.Get(fid)
|
|
expectNilErr(t, err)
|
|
topicList, pagi, err = c.TopicList.GetListByForum(forum, 1, 0)
|
|
expectNilErr(t, err)
|
|
test(topicList, pagi, 1, c.Paginator{[]int{1}, 1, 1}, tid)
|
|
|
|
topicList, forumList, pagi, err = c.TopicList.GetList(1, 0, []int{fid})
|
|
expectNilErr(t, err)
|
|
test(topicList, pagi, 1, c.Paginator{[]int{1}, 1, 1}, tid)
|
|
exf(len(forumList) == 1, "len(forumList) should be 1 not %d", len(forumList))
|
|
|
|
topicList, forumList, pagi, err = c.TopicList.GetListByCanSee([]int{fid}, 1, 0, []int{fid})
|
|
expectNilErr(t, err)
|
|
test(topicList, pagi, 1, c.Paginator{[]int{1}, 1, 1}, tid)
|
|
exf(len(forumList) == 1, "len(forumList) should be 1 not %d", len(forumList))
|
|
|
|
topicList, forumList, pagi, err = c.TopicList.GetListByCanSee([]int{}, 1, 0, []int{fid})
|
|
expectNilErr(t, err)
|
|
noTopics(topicList, pagi)
|
|
exf(len(forumList) == 0, "len(forumList) should be 0 not %d", len(forumList))
|
|
|
|
topic, err := c.Topics.Get(tid)
|
|
expectNilErr(t, err)
|
|
expectNilErr(t, topic.Delete())
|
|
|
|
forum, err = c.Forums.Get(fid)
|
|
expectNilErr(t, err)
|
|
noTopicsTests()
|
|
|
|
// TODO: More tests
|
|
|
|
_ = ex
|
|
}
|
|
|
|
func TestUtils(t *testing.T) {
|
|
ee := func(email, eemail string) {
|
|
cemail := c.CanonEmail(email)
|
|
expectf(t, cemail == eemail, "%s should be %s", cemail, eemail)
|
|
}
|
|
ee("test@example.com", "test@example.com")
|
|
ee("test.test@example.com", "test.test@example.com")
|
|
ee("", "")
|
|
ee("ddd", "ddd")
|
|
ee("test.test@gmail.com", "testtest@gmail.com")
|
|
ee("TEST.test@gmail.com", "testtest@gmail.com")
|
|
ee("test.TEST.test@gmail.com", "testtesttest@gmail.com")
|
|
ee("test..TEST.test@gmail.com", "testtesttest@gmail.com")
|
|
ee("TEST.test@example.com", "test.test@example.com")
|
|
ee("test.TEST.test@example.com", "test.test.test@example.com")
|
|
// TODO: Exotic unicode email types? Are there those?
|
|
|
|
// TODO: More utils.go tests
|
|
}
|
|
|
|
func TestWeakPassword(t *testing.T) {
|
|
ex := exp(t)
|
|
/*weakPass := func(password, name, email string) func(error,string,...interface{}) {
|
|
err := c.WeakPassword(password, name, email)
|
|
return func(expectErr error, m string, p ...interface{}) {
|
|
m = fmt.Sprintf("pass=%s, user=%s, email=%s ", password, name, email) + m
|
|
expect(t, err == expectErr, fmt.Sprintf(m,p...))
|
|
}
|
|
}*/
|
|
nilErrStr := func(e error) error {
|
|
if e == nil {
|
|
e = errors.New("nil")
|
|
}
|
|
return e
|
|
}
|
|
weakPass := func(password, name, email string) func(error) {
|
|
err := c.WeakPassword(password, name, email)
|
|
e := nilErrStr(err)
|
|
m := fmt.Sprintf("pass=%s, user=%s, email=%s ", password, name, email)
|
|
return func(expectErr error) {
|
|
ee := nilErrStr(expectErr)
|
|
ex(err == expectErr, m+fmt.Sprintf("err should be '%s' not '%s'", ee, e))
|
|
}
|
|
}
|
|
|
|
//weakPass("test", "test", "test@example.com")(c.ErrWeakPasswordContains,"err should be ErrWeakPasswordContains not '%s'")
|
|
weakPass("", "draw", "test@example.com")(c.ErrWeakPasswordNone)
|
|
weakPass("test", "draw", "test@example.com")(c.ErrWeakPasswordShort)
|
|
weakPass("testtest", "draw", "test@example.com")(c.ErrWeakPasswordContains)
|
|
weakPass("testdraw", "draw", "test@example.com")(c.ErrWeakPasswordNameInPass)
|
|
weakPass("test@example.com", "draw", "test@example.com")(c.ErrWeakPasswordEmailInPass)
|
|
weakPass("meet@example.com2", "draw", "")(c.ErrWeakPasswordNoUpper)
|
|
weakPass("Meet@example.com2", "draw", "")(nil)
|
|
weakPass("test2", "draw", "test@example.com")(c.ErrWeakPasswordShort)
|
|
weakPass("test22222222", "draw", "test@example.com")(c.ErrWeakPasswordContains)
|
|
weakPass("superman", "draw", "test@example.com")(c.ErrWeakPasswordCommon)
|
|
weakPass("Superman", "draw", "test@example.com")(c.ErrWeakPasswordCommon)
|
|
weakPass("Superma2", "draw", "test@example.com")(nil)
|
|
weakPass("superman2", "draw", "test@example.com")(c.ErrWeakPasswordCommon)
|
|
weakPass("Superman2", "draw", "test@example.com")(c.ErrWeakPasswordCommon)
|
|
weakPass("superman22", "draw", "test@example.com")(c.ErrWeakPasswordNoUpper)
|
|
weakPass("K\\@<^s}1", "draw", "test@example.com")(nil)
|
|
weakPass("K\\@<^s}r", "draw", "test@example.com")(c.ErrWeakPasswordNoNumbers)
|
|
weakPass("k\\@<^s}1", "draw", "test@example.com")(c.ErrWeakPasswordNoUpper)
|
|
weakPass("aaaaaaaa", "draw", "test@example.com")(c.ErrWeakPasswordNoUpper)
|
|
weakPass("aA1aA1aA1", "draw", "test@example.com")(c.ErrWeakPasswordUniqueChars)
|
|
weakPass("abababab", "draw", "test@example.com")(c.ErrWeakPasswordNoUpper)
|
|
weakPass("11111111111111111111", "draw", "test@example.com")(c.ErrWeakPasswordNoUpper)
|
|
weakPass("aaaaaaaaaaAAAAAAAAAA", "draw", "test@example.com")(c.ErrWeakPasswordUniqueChars)
|
|
weakPass("-:u/nMxb,A!n=B;H\\sjM", "draw", "test@example.com")(nil)
|
|
}
|
|
|
|
func TestAuth(t *testing.T) {
|
|
ex := exp(t)
|
|
// bcrypt likes doing stupid things, so this test will probably fail
|
|
realPassword := "Madame Cassandra's Mystic Orb"
|
|
t.Logf("Set realPassword to '%s'", realPassword)
|
|
t.Log("Hashing the real password with bcrypt")
|
|
hashedPassword, _, err := c.BcryptGeneratePassword(realPassword)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
passwordTest(t, realPassword, hashedPassword)
|
|
// TODO: Peek at the prefix to verify this is a bcrypt hash
|
|
|
|
t.Log("Hashing the real password")
|
|
hashedPassword2, _, err := c.GeneratePassword(realPassword)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
passwordTest(t, realPassword, hashedPassword2)
|
|
// TODO: Peek at the prefix to verify this is a bcrypt hash
|
|
|
|
_, err, _ = c.Auth.Authenticate("None", "password")
|
|
errmsg := "Name None shouldn't exist"
|
|
if err != nil {
|
|
errmsg += "\n" + err.Error()
|
|
}
|
|
ex(err == c.ErrNoUserByName, errmsg)
|
|
|
|
uid, err, _ := c.Auth.Authenticate("Admin", "password")
|
|
expectNilErr(t, err)
|
|
expectf(t, uid == 1, "Default admin uid should be 1 not %d", uid)
|
|
|
|
_, err, _ = c.Auth.Authenticate("Sam", "ReallyBadPassword")
|
|
errmsg = "Name Sam shouldn't exist"
|
|
if err != nil {
|
|
errmsg += "\n" + err.Error()
|
|
}
|
|
ex(err == c.ErrNoUserByName, errmsg)
|
|
|
|
admin, err := c.Users.Get(1)
|
|
expectNilErr(t, err)
|
|
// TODO: Move this into the user store tests to provide better coverage? E.g. To see if the installer and the user creator initialise the field differently
|
|
ex(admin.Session == "", "Admin session should be blank")
|
|
|
|
session, err := c.Auth.CreateSession(1)
|
|
expectNilErr(t, err)
|
|
ex(session != "", "Admin session shouldn't be blank")
|
|
// TODO: Test the actual length set in the setting in addition to this "too short" test
|
|
// TODO: We might be able to push up this minimum requirement
|
|
ex(len(session) > 10, "Admin session shouldn't be too short")
|
|
ex(admin.Session != session, "Old session should not match new one")
|
|
admin, err = c.Users.Get(1)
|
|
expectNilErr(t, err)
|
|
ex(admin.Session == session, "Sessions should match")
|
|
|
|
// TODO: Create a user with a unicode password and see if we can login as them
|
|
// TODO: Tests for SessionCheck, GetCookies, and ForceLogout
|
|
// TODO: Tests for MFA Verification
|
|
}
|
|
|
|
// TODO: Vary the salts? Keep in mind that some algorithms store the salt in the hash therefore the salt string may be blank
|
|
func passwordTest(t *testing.T, realPassword, hashedPassword string) {
|
|
if len(hashedPassword) < 10 {
|
|
t.Error("Hash too short")
|
|
}
|
|
salt := ""
|
|
password := realPassword
|
|
t.Logf("Testing password '%s'", password)
|
|
t.Logf("Testing salt '%s'", salt)
|
|
err := c.CheckPassword(hashedPassword, password, salt)
|
|
if err == c.ErrMismatchedHashAndPassword {
|
|
t.Error("The two don't match")
|
|
} else if err == c.ErrPasswordTooLong {
|
|
t.Error("CheckPassword thinks the password is too long")
|
|
} else if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
password = "hahaha"
|
|
t.Logf("Testing password '%s'", password)
|
|
t.Logf("Testing salt '%s'", salt)
|
|
err = c.CheckPassword(hashedPassword, password, salt)
|
|
if err == c.ErrPasswordTooLong {
|
|
t.Error("CheckPassword thinks the password is too long")
|
|
} else if err == nil {
|
|
t.Error("The two shouldn't match!")
|
|
}
|
|
|
|
password = "Madame Cassandra's Mystic"
|
|
t.Logf("Testing password '%s'", password)
|
|
t.Logf("Testing salt '%s'", salt)
|
|
err = c.CheckPassword(hashedPassword, password, salt)
|
|
expect(t, err != c.ErrPasswordTooLong, "CheckPassword thinks the password is too long")
|
|
expect(t, err != nil, "The two shouldn't match!")
|
|
}
|
|
|
|
func TestUserPrivacy(t *testing.T) {
|
|
pu, u := c.BlankUser(), &c.GuestUser
|
|
pu.ID = 1
|
|
ex, exf := exp(t), expf(t)
|
|
ex(!pu.Privacy.NoPresence, "pu.Privacy.NoPresence should be false")
|
|
ex(!u.Privacy.NoPresence, "u.Privacy.NoPresence should be false")
|
|
|
|
var msg string
|
|
test := func(expects bool, level int) {
|
|
pu.Privacy.ShowComments = level
|
|
val := c.PrivacyCommentsShow(pu, u)
|
|
var bit string
|
|
if !expects {
|
|
bit = " not"
|
|
val = !val
|
|
}
|
|
exf(val, "%s should%s be able to see comments on level %d", msg, bit, level)
|
|
}
|
|
// 0 = default, 1 = public, 2 = registered, 3 = friends, 4 = self, 5 = disabled
|
|
|
|
msg = "guest users"
|
|
test(true, 0)
|
|
test(true, 1)
|
|
test(false, 2)
|
|
test(false, 3)
|
|
test(false, 4)
|
|
test(false, 5)
|
|
|
|
u = c.BlankUser()
|
|
msg = "blank users"
|
|
test(true, 0)
|
|
test(true, 1)
|
|
test(false, 2)
|
|
//test(false,3)
|
|
test(false, 4)
|
|
test(false, 5)
|
|
|
|
u.Loggedin = true
|
|
msg = "registered users"
|
|
test(true, 0)
|
|
test(true, 1)
|
|
test(true, 2)
|
|
test(false, 3)
|
|
test(false, 4)
|
|
test(false, 5)
|
|
|
|
u.IsBanned = true
|
|
msg = "banned users"
|
|
test(true, 0)
|
|
test(true, 1)
|
|
test(true, 2)
|
|
test(false, 3)
|
|
test(false, 4)
|
|
test(false, 5)
|
|
u.IsBanned = false
|
|
|
|
u.IsMod = true
|
|
msg = "mods"
|
|
test(true, 0)
|
|
test(true, 1)
|
|
test(true, 2)
|
|
test(false, 3)
|
|
test(false, 4)
|
|
test(false, 5)
|
|
u.IsMod = false
|
|
|
|
u.IsSuperMod = true
|
|
msg = "super mods"
|
|
test(true, 0)
|
|
test(true, 1)
|
|
test(true, 2)
|
|
test(false, 3)
|
|
test(false, 4)
|
|
test(false, 5)
|
|
u.IsSuperMod = false
|
|
|
|
u.IsAdmin = true
|
|
msg = "admins"
|
|
test(true, 0)
|
|
test(true, 1)
|
|
test(true, 2)
|
|
test(false, 3)
|
|
test(false, 4)
|
|
test(false, 5)
|
|
u.IsAdmin = false
|
|
|
|
u.IsSuperAdmin = true
|
|
msg = "super admins"
|
|
test(true, 0)
|
|
test(true, 1)
|
|
test(true, 2)
|
|
test(false, 3)
|
|
test(false, 4)
|
|
test(false, 5)
|
|
u.IsSuperAdmin = false
|
|
|
|
u.ID = 1
|
|
test(true, 0)
|
|
test(true, 1)
|
|
test(true, 2)
|
|
test(true, 3)
|
|
test(true, 4)
|
|
test(false, 5)
|
|
}
|
|
|
|
type METri struct {
|
|
Name string // Optional, this is here for tests involving invisible characters so we know what's going in
|
|
Msg string
|
|
Expects string
|
|
}
|
|
|
|
type METriList struct {
|
|
Items []METri
|
|
}
|
|
|
|
func (l *METriList) Add(args ...string) {
|
|
if len(args) < 2 {
|
|
panic("need 2 or more args")
|
|
}
|
|
if len(args) > 2 {
|
|
l.Items = append(l.Items, METri{args[0], args[1], args[2]})
|
|
} else {
|
|
l.Items = append(l.Items, METri{"", args[0], args[1]})
|
|
}
|
|
}
|
|
|
|
type CountTest struct {
|
|
Name string
|
|
Msg string
|
|
Expects int
|
|
}
|
|
|
|
type CountTestList struct {
|
|
Items []CountTest
|
|
}
|
|
|
|
func (l *CountTestList) Add(name, msg string, expects int) {
|
|
l.Items = append(l.Items, CountTest{name, msg, expects})
|
|
}
|
|
|
|
func TestWordCount(t *testing.T) {
|
|
l := &CountTestList{nil}
|
|
l.Add("blank", "", 0)
|
|
l.Add("single-letter", "h", 1)
|
|
l.Add("single-kana", "お", 1)
|
|
l.Add("single-letter-words", "h h", 2)
|
|
l.Add("two-letter", "h", 1)
|
|
l.Add("two-kana", "おは", 1)
|
|
l.Add("two-letter-words", "hh hh", 2)
|
|
l.Add("", "h,h", 2)
|
|
l.Add("", "h,,h", 2)
|
|
l.Add("", "h, h", 2)
|
|
l.Add("", " h, h", 2)
|
|
l.Add("", "h, h ", 2)
|
|
l.Add("", " h, h ", 2)
|
|
l.Add("", "h, h", 2)
|
|
l.Add("", "h\nh", 2)
|
|
l.Add("", "h\"h", 2)
|
|
l.Add("", "h[r]h", 3)
|
|
l.Add("", "お,お", 2)
|
|
l.Add("", "お、お", 2)
|
|
l.Add("", "お\nお", 2)
|
|
l.Add("", "お”お", 2)
|
|
l.Add("", "お「あ」お", 3)
|
|
|
|
for _, item := range l.Items {
|
|
res := c.WordCount(item.Msg)
|
|
if res != item.Expects {
|
|
if item.Name != "" {
|
|
t.Error("Name: ", item.Name)
|
|
}
|
|
t.Error("Testing string '" + item.Msg + "'")
|
|
t.Error("Bad output:", res)
|
|
t.Error("Expected:", item.Expects)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestTick(t *testing.T) {
|
|
expectNilErr(t, c.StartupTasks())
|
|
expectNilErr(t, c.Dailies())
|
|
|
|
expectNilErr(t, c.Tasks.HalfSec.Run())
|
|
expectNilErr(t, c.Tasks.Sec.Run())
|
|
expectNilErr(t, c.Tasks.FifteenMin.Run())
|
|
expectNilErr(t, c.Tasks.Hour.Run())
|
|
expectNilErr(t, c.Tasks.Day.Run())
|
|
|
|
thumbChan := make(chan bool)
|
|
expectNilErr(t, tickLoop(thumbChan))
|
|
expectNilErr(t, c.CTickLoop.HalfSecf())
|
|
expectNilErr(t, c.CTickLoop.Secf())
|
|
expectNilErr(t, c.CTickLoop.FifteenMinf())
|
|
expectNilErr(t, c.CTickLoop.Hourf())
|
|
expectNilErr(t, c.CTickLoop.Dayf())
|
|
}
|
|
|
|
func TestWSHub(t *testing.T) {
|
|
ex, exf, h := exp(t), expf(t), &c.WsHub
|
|
exf(h.GuestCount() == 0, "GuestCount should be %d not %d", 0, h.GuestCount())
|
|
exf(h.UserCount() == 0, "UserCount should be %d not %d", 0, h.UserCount())
|
|
ex(!h.HasUser(-1), "HasUser(-1) should be false")
|
|
ex(!h.HasUser(0), "HasUser(0) should be false")
|
|
ex(!h.HasUser(1), "HasUser(1) should be false")
|
|
|
|
uid, e := c.Users.Create("WsHub Test", "WsHub Test", "", 1, true)
|
|
expectNilErr(t, e)
|
|
exf(!h.HasUser(uid), "HasUser(%d) should be false", uid)
|
|
exf(len(h.AllUsers()) == 0, "len(AllUsers()) should be %d not %d", 0, len(h.AllUsers()))
|
|
|
|
f := func(uid, guestCount, userCount, allUserListLen int, hasUser bool) {
|
|
exf(h.GuestCount() == guestCount, "GuestCount should be %d not %d", guestCount, h.GuestCount())
|
|
exf(h.UserCount() == userCount, "UserCount should be %d not %d", userCount, h.UserCount())
|
|
exf(len(h.AllUsers()) == allUserListLen, "len(AllUsers()) should be %d not %d", allUserListLen, len(h.AllUsers()))
|
|
if hasUser {
|
|
exf(h.HasUser(uid), "HasUser(%d) should be true", uid)
|
|
} else {
|
|
exf(!h.HasUser(uid), "HasUser(%d) should be false", uid)
|
|
}
|
|
}
|
|
|
|
u, e := c.Users.Get(uid)
|
|
expectNilErr(t, e)
|
|
wsUser, e := h.AddConn(u, nil)
|
|
expectNilErr(t, e)
|
|
f(uid, 0, 1, 1, true)
|
|
|
|
uid, e = c.Users.Create("WsHub Test 2", "WsHub Test 2", "", 1, true)
|
|
expectNilErr(t, e)
|
|
u2, e := c.Users.Get(uid)
|
|
expectNilErr(t, e)
|
|
wsUser2, e := h.AddConn(u2, nil)
|
|
expectNilErr(t, e)
|
|
f(uid, 0, 2, 2, true)
|
|
|
|
h.RemoveConn(wsUser2, nil)
|
|
f(uid, 0, 1, 1, false)
|
|
h.RemoveConn(wsUser2, nil)
|
|
f(uid, 0, 1, 1, false)
|
|
h.RemoveConn(wsUser, nil)
|
|
f(uid, 0, 0, 0, false)
|
|
|
|
countSockets := func(wsUser *c.WSUser, expect int) {
|
|
exf(wsUser.CountSockets() == expect, "CountSockets() should be %d not %d", expect, wsUser.CountSockets())
|
|
}
|
|
wsUser2, e = h.AddConn(u2, nil)
|
|
expectNilErr(t, e)
|
|
f(uid, 0, 1, 1, true)
|
|
countSockets(wsUser2, 1)
|
|
wsUser2.RemoveSocket(nil)
|
|
f(uid, 0, 1, 1, true)
|
|
countSockets(wsUser2, 0)
|
|
h.RemoveConn(wsUser2, nil)
|
|
f(uid, 0, 0, 0, false)
|
|
countSockets(wsUser2, 0)
|
|
|
|
wsUser2, e = h.AddConn(u2, nil)
|
|
expectNilErr(t, e)
|
|
f(uid, 0, 1, 1, true)
|
|
countSockets(wsUser2, 1)
|
|
expectNilErr(t, wsUser2.Ping())
|
|
f(uid, 0, 1, 1, true)
|
|
countSockets(wsUser2, 0)
|
|
h.RemoveConn(wsUser2, nil)
|
|
f(uid, 0, 0, 0, false)
|
|
countSockets(wsUser2, 0)
|
|
|
|
// TODO: Add more tests
|
|
}
|