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) expectNilErr(t, e) exf(topic.Poll == 0, "t.Poll should be %d not %d", 0, topic.Poll) /*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()) topic, e = c.Topics.BypassGet(tid) expectNilErr(t, e) exf(topic.Poll == pid, "t.Poll should be %d not %d", pid, topic.Poll) 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) var vslice []int expectNilErr(t, p.Resultsf(func(votes int) error { vslice = append(vslice, votes) return nil })) //fmt.Printf("vslice: %+v\n", vslice) exf(vslice[0] == 1, "vslice[0] should be %d not %d", 0, vslice[0]) exf(vslice[1] == 0, "vslice[1] should be %d not %d", 1, vslice[1]) exf(vslice[2] == 0, "vslice[2] should be %d not %d", 0, vslice[2]) expectNilErr(t, p.CastVote(2, 1, "")) expectNilErr(t, c.Polls.Reload(p.ID)) p, e = c.Polls.Get(1) expectNilErr(t, e) testPoll(p, 1, tid, "topics", 0, false, 2) vslice = nil expectNilErr(t, p.Resultsf(func(votes int) error { vslice = append(vslice, votes) return nil })) //fmt.Printf("vslice: %+v\n", vslice) exf(vslice[0] == 1, "vslice[0] should be %d not %d", 1, vslice[0]) exf(vslice[1] == 0, "vslice[1] should be %d not %d", 0, vslice[1]) exf(vslice[2] == 1, "vslice[2] should be %d not %d", 1, vslice[2]) 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()) topic, e = c.Topics.BypassGet(tid) expectNilErr(t, e) exf(topic.Poll == pid, "t.Poll should be %d not %d", pid, topic.Poll) expectNilErr(t, topic.SetPoll(999)) topic, e = c.Topics.BypassGet(tid) expectNilErr(t, e) exf(topic.Poll == pid, "t.Poll should be %d not %d", pid, topic.Poll) expectNilErr(t, topic.SetPoll(0)) topic, e = c.Topics.BypassGet(tid) expectNilErr(t, e) exf(topic.Poll == pid, "t.Poll should be %d not %d", pid, topic.Poll) expectNilErr(t, topic.RemovePoll()) topic, e = c.Topics.BypassGet(tid) expectNilErr(t, e) exf(topic.Poll == 0, "t.Poll should be %d not %d", 0, topic.Poll) } 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 }