116 lines
2.6 KiB
Go
116 lines
2.6 KiB
Go
package persistence_test
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
"math/rand"
|
|
|
|
"github.com/go-test/deep"
|
|
|
|
"offline_twitter/scraper"
|
|
"offline_twitter/persistence"
|
|
)
|
|
|
|
/**
|
|
* Helper function
|
|
*/
|
|
func create_or_load_profile(profile_path string) persistence.Profile {
|
|
var profile persistence.Profile
|
|
var err error
|
|
|
|
if !file_exists(profile_path) {
|
|
profile, err = persistence.NewProfile(profile_path)
|
|
} else {
|
|
profile, err = persistence.LoadProfile(profile_path)
|
|
}
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return profile
|
|
}
|
|
|
|
|
|
/**
|
|
* 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)
|
|
|
|
// Generate a new random user ID
|
|
rand.Seed(time.Now().UnixNano())
|
|
userID := fmt.Sprint(rand.Int())
|
|
|
|
fake_user := scraper.User{
|
|
ID: scraper.UserID(userID),
|
|
DisplayName: "display name",
|
|
Handle: scraper.UserHandle("handle" + userID),
|
|
Bio: "bio",
|
|
FollowersCount: 0,
|
|
FollowingCount: 1000,
|
|
Location: "location",
|
|
Website:"website",
|
|
JoinDate: time.Now().Truncate(1e9), // Round to nearest second
|
|
IsVerified: false,
|
|
IsPrivate: true,
|
|
ProfileImageUrl: "profile image url",
|
|
BannerImageUrl: "banner image url",
|
|
PinnedTweetID: scraper.TweetID("234"),
|
|
}
|
|
|
|
// 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(scraper.UserID(userID))
|
|
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(scraper.UserHandle(fake_user.Handle))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if diff := deep.Equal(new_fake_user2, fake_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)
|
|
|
|
// Generate a new random user ID
|
|
rand.Seed(time.Now().UnixNano())
|
|
userID := fmt.Sprint(rand.Int())
|
|
|
|
user := scraper.User{}
|
|
user.ID = scraper.UserID(userID)
|
|
user.Handle = scraper.UserHandle("handle" + userID)
|
|
|
|
exists := profile.UserExists(scraper.UserHandle(user.Handle))
|
|
if exists {
|
|
t.Errorf("It shouldn't exist, but it does: %s", userID)
|
|
}
|
|
err := profile.SaveUser(user)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
exists = profile.UserExists(scraper.UserHandle(user.Handle))
|
|
if !exists {
|
|
t.Errorf("It should exist, but it doesn't: %s", userID)
|
|
}
|
|
}
|