offline-twitter/persistence/profile_test.go

package persistence_test

import (
	"testing"
	"os"
	"path"
	"errors"

	"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) {
	gibberish_path := "test_profiles/fjlwrefuvaaw23efwm"
	if file_exists(gibberish_path) {
		err := os.RemoveAll(gibberish_path)
		if err != nil {
			panic(err)
		}
	}
	err := os.Mkdir(gibberish_path, 0755)
	if err != nil {
		panic(err)
	}
	_, err = persistence.NewProfile(gibberish_path)
	if err == nil {
		t.Errorf("Should have failed to create a profile in an already existing directory!")
	}
	if _, is_right_type := err.(persistence.ErrTargetAlreadyExists); !is_right_type {
		t.Errorf("Expected 'ErrTargetAlreadyExists' error, got %T instead", err)
	}
}


/**
 * Should correctly create a new Profile
 */
func TestNewProfile(t *testing.T) {
	profile_path := "test_profiles/TestNewProfile"
	if file_exists(profile_path) {
		err := os.RemoveAll(profile_path)
		if err != nil {
			panic(err)
		}
	}

	profile, err := persistence.NewProfile(profile_path)
	if err != nil {
		t.Fatalf(err.Error())
	}

	if profile.ProfileDir != profile_path {
		t.Errorf("ProfileDir should be %s, but it is %s", 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)
	if err != nil {
		panic(err)
	}
	if len(contents) != 8 {
		t.Fatalf("Expected 8 contents, got %d instead", len(contents))
	}

	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 {
		if contents[i].Name() != v.filename || contents[i].IsDir() != v.isDir {
			t.Fatalf("Expected `%s` to be a %s, but got %s [%s]", v.filename, isdir_map(v.isDir), contents[i].Name(), isdir_map(contents[i].IsDir()))
		}
	}

	// Check database version is initialized
	version, err := profile.GetDatabaseVersion()
	if err != nil {
		panic(err)
	}
	if version != persistence.ENGINE_DATABASE_VERSION {
		t.Errorf("Expected database version %d, but got %d", persistence.ENGINE_DATABASE_VERSION, version)
	}
}


/**
 * Should correctly load the Profile
 */
func TestLoadProfile(t *testing.T) {
	profile_path := "test_profiles/TestLoadProfile"
	if file_exists(profile_path) {
		err := os.RemoveAll(profile_path)
		if err != nil {
			panic(err)
		}
	}

	_, err := persistence.NewProfile(profile_path)
	if err != nil {
		t.Fatalf(err.Error())
	}

	// Create some users
	err = os.WriteFile(path.Join(profile_path, "users.yaml"), []byte("- user: user1\n- user: user2\n"), 0644)
	if err != nil {
		t.Fatalf(err.Error())
	}

	profile, err := persistence.LoadProfile(profile_path)
	if err != nil {
		t.Fatalf(err.Error())
	}

	if profile.ProfileDir != profile_path {
		t.Errorf("Expected profile path to be %q, but got %q", profile_path, profile.ProfileDir)
	}

	if len(profile.UsersList) != 2 {
		t.Errorf("Expected 2 users, got %v", profile.UsersList)
	}
	if profile.UsersList[0].Handle != "user1" {
		t.Errorf("Expected first user to be %s, got %s", "user1", profile.UsersList[0].Handle)
	}
}