134 lines
3.0 KiB
Go
134 lines
3.0 KiB
Go
package persistence_test
|
|
|
|
import (
|
|
"testing"
|
|
"os"
|
|
"path"
|
|
"errors"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"offline_twitter/persistence"
|
|
)
|
|
|
|
// DUPE 1
|
|
func file_exists(path string) bool {
|
|
_, err := os.Stat(path)
|
|
if err == nil {
|
|
return true
|
|
} else if errors.Is(err, os.ErrNotExist) {
|
|
return false
|
|
} else {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func isdir_map(is_dir bool) string {
|
|
if is_dir {
|
|
return "directory"
|
|
}
|
|
return "file"
|
|
}
|
|
|
|
|
|
/**
|
|
* Should refuse to create a Profile if the target already exists (i.e., is a file or directory).
|
|
*/
|
|
func TestNewProfileInvalidPath(t *testing.T) {
|
|
require := require.New(t)
|
|
gibberish_path := "test_profiles/fjlwrefuvaaw23efwm"
|
|
if file_exists(gibberish_path) {
|
|
err := os.RemoveAll(gibberish_path)
|
|
require.NoError(err)
|
|
}
|
|
err := os.Mkdir(gibberish_path, 0755)
|
|
require.NoError(err)
|
|
|
|
_, err = persistence.NewProfile(gibberish_path)
|
|
require.Error(err, "Should have failed to create a profile in an already existing directory!")
|
|
|
|
_, is_right_type := err.(persistence.ErrTargetAlreadyExists)
|
|
assert.True(t, is_right_type, "Expected 'ErrTargetAlreadyExists' error, got %T instead", err)
|
|
}
|
|
|
|
|
|
/**
|
|
* Should correctly create a new Profile
|
|
*/
|
|
func TestNewProfile(t *testing.T) {
|
|
assert := assert.New(t)
|
|
require := require.New(t)
|
|
|
|
profile_path := "test_profiles/TestNewProfile"
|
|
if file_exists(profile_path) {
|
|
err := os.RemoveAll(profile_path)
|
|
require.NoError(err)
|
|
}
|
|
|
|
profile, err := persistence.NewProfile(profile_path)
|
|
require.NoError(err)
|
|
|
|
assert.Equal(profile_path,profile.ProfileDir)
|
|
if len(profile.UsersList) != 0 {
|
|
t.Errorf("Expected empty users list, got %v instead", profile.UsersList)
|
|
}
|
|
|
|
// Check files were created
|
|
contents, err := os.ReadDir(profile_path)
|
|
require.NoError(err)
|
|
assert.Len(contents, 8)
|
|
|
|
expected_files := []struct {
|
|
filename string
|
|
isDir bool
|
|
} {
|
|
{"images", true},
|
|
{"link_preview_images", true},
|
|
{"profile_images", true},
|
|
{"settings.yaml", false},
|
|
{"twitter.db", false},
|
|
{"users.yaml", false},
|
|
{"video_thumbnails", true},
|
|
{"videos", true},
|
|
}
|
|
|
|
for i, v := range expected_files {
|
|
assert.Equal(v.filename, contents[i].Name())
|
|
assert.Equal(v.isDir, contents[i].IsDir())
|
|
}
|
|
|
|
// Check database version is initialized
|
|
version, err := profile.GetDatabaseVersion()
|
|
require.NoError(err)
|
|
assert.Equal(persistence.ENGINE_DATABASE_VERSION, version)
|
|
}
|
|
|
|
|
|
/**
|
|
* Should correctly load the Profile
|
|
*/
|
|
func TestLoadProfile(t *testing.T) {
|
|
require := require.New(t)
|
|
|
|
profile_path := "test_profiles/TestLoadProfile"
|
|
if file_exists(profile_path) {
|
|
err := os.RemoveAll(profile_path)
|
|
require.NoError(err)
|
|
}
|
|
|
|
_, err := persistence.NewProfile(profile_path)
|
|
require.NoError(err)
|
|
|
|
// Create some users
|
|
err = os.WriteFile(path.Join(profile_path, "users.yaml"), []byte("- user: user1\n- user: user2\n"), 0644)
|
|
require.NoError(err)
|
|
|
|
profile, err := persistence.LoadProfile(profile_path)
|
|
require.NoError(err)
|
|
|
|
assert.Equal(t, profile_path, profile.ProfileDir)
|
|
assert.Len(t, profile.UsersList, 2)
|
|
assert.Equal(t, "user1", string(profile.UsersList[0].Handle))
|
|
}
|