133 lines
3.2 KiB
Go
133 lines
3.2 KiB
Go
package persistence_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/go-test/deep"
|
|
)
|
|
|
|
|
|
/**
|
|
* Create a user, save it, reload it, and make sure it comes back the same
|
|
*/
|
|
func TestSaveAndLoadUser(t *testing.T) {
|
|
profile_path := "test_profiles/TestUserQueries"
|
|
profile := create_or_load_profile(profile_path)
|
|
|
|
fake_user := create_dummy_user()
|
|
|
|
// Save the user, then reload it and ensure it's the same
|
|
err := profile.SaveUser(fake_user)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
new_fake_user, err := profile.GetUserByID(fake_user.ID)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if diff := deep.Equal(new_fake_user, fake_user); diff != nil {
|
|
t.Error(diff)
|
|
}
|
|
|
|
// Same thing, but get by handle
|
|
new_fake_user2, err := profile.GetUserByHandle(fake_user.Handle)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if diff := deep.Equal(new_fake_user2, fake_user); diff != nil {
|
|
t.Error(diff)
|
|
}
|
|
}
|
|
|
|
|
|
func TestHandleIsCaseInsensitive(t *testing.T) {
|
|
profile_path := "test_profiles/TestUserQueries"
|
|
profile := create_or_load_profile(profile_path)
|
|
|
|
user := create_stable_user()
|
|
|
|
new_user, err := profile.GetUserByHandle("hANdle StaBlE")
|
|
if err != nil {
|
|
t.Fatalf("Couldn't find the user: %s", err.Error())
|
|
}
|
|
|
|
if diff := deep.Equal(user, new_user); diff != nil {
|
|
t.Error(diff)
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* Should correctly report whether the user exists in the database
|
|
*/
|
|
func TestUserExists(t *testing.T) {
|
|
profile_path := "test_profiles/TestUserQueries"
|
|
profile := create_or_load_profile(profile_path)
|
|
|
|
user := create_dummy_user()
|
|
|
|
exists := profile.UserExists(user.Handle)
|
|
if exists {
|
|
t.Errorf("It shouldn't exist, but it does: %d", user.ID)
|
|
}
|
|
err := profile.SaveUser(user)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
exists = profile.UserExists(user.Handle)
|
|
if !exists {
|
|
t.Errorf("It should exist, but it doesn't: %d", user.ID)
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Test scenarios relating to user content downloading
|
|
*/
|
|
func TestCheckUserContentDownloadNeeded(t *testing.T) {
|
|
profile_path := "test_profiles/TestUserQueries"
|
|
profile := create_or_load_profile(profile_path)
|
|
|
|
user := create_dummy_user()
|
|
|
|
// If user is not in database, should be "yes" automatically
|
|
if profile.CheckUserContentDownloadNeeded(user) != true {
|
|
t.Errorf("Non-saved user should always require content download")
|
|
}
|
|
|
|
// Save the user, but `is_content_downloaded` is still "false"
|
|
user.BannerImageUrl = "banner url1"
|
|
user.ProfileImageUrl = "profile url1"
|
|
user.IsContentDownloaded = false
|
|
err := profile.SaveUser(user)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
// If is_content_downloaded is false, then it needs download
|
|
if profile.CheckUserContentDownloadNeeded(user) != true {
|
|
t.Errorf("Non-downloaded user should require download")
|
|
}
|
|
|
|
// Mark `is_content_downloaded` as "true" again
|
|
user.IsContentDownloaded = true
|
|
err = profile.SaveUser(user)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
// If everything is up to date, no download should be required
|
|
if profile.CheckUserContentDownloadNeeded(user) != false {
|
|
t.Errorf("Up-to-date user shouldn't need a download")
|
|
}
|
|
|
|
// Change an URL, but don't save it-- needs to be different from what's in the DB
|
|
user.BannerImageUrl = "banner url2"
|
|
|
|
// Download needed for new banner image
|
|
if profile.CheckUserContentDownloadNeeded(user) != true {
|
|
t.Errorf("If banner image changed, user should require another download")
|
|
}
|
|
}
|