2023-10-04 13:50:16 +03:00
|
|
|
use std::{collections::HashMap, fs, path};
|
2023-08-28 10:34:30 +03:00
|
|
|
|
|
|
|
use tempfile::tempdir;
|
|
|
|
|
2023-10-04 13:50:16 +03:00
|
|
|
use crate::{database, gb_repository, git, keys, project_repository, projects, storage, users};
|
|
|
|
|
|
|
|
pub struct Suite {
|
|
|
|
pub local_app_data: path::PathBuf,
|
|
|
|
pub user_storage: users::Storage,
|
|
|
|
pub projects_storage: projects::Storage,
|
|
|
|
pub keys_storage: keys::Storage,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Default for Suite {
|
|
|
|
fn default() -> Self {
|
|
|
|
let local_app_data = temp_dir();
|
|
|
|
let storage = storage::Storage::from(&local_app_data);
|
|
|
|
let user_storage = users::Storage::from(&storage);
|
|
|
|
let projects_storage = projects::Storage::from(&storage);
|
|
|
|
let keys_storage = keys::Storage::from(&storage);
|
|
|
|
Self {
|
|
|
|
local_app_data,
|
|
|
|
user_storage,
|
|
|
|
projects_storage,
|
|
|
|
keys_storage,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Suite {
|
|
|
|
pub fn sign_in(&self) -> users::User {
|
|
|
|
let user = users::User {
|
|
|
|
name: "test".to_string(),
|
|
|
|
email: "test@email.com".to_string(),
|
|
|
|
..Default::default()
|
|
|
|
};
|
|
|
|
self.user_storage.set(&user).expect("failed to add user");
|
|
|
|
user
|
|
|
|
}
|
|
|
|
|
2023-10-06 13:43:32 +03:00
|
|
|
fn project(&self, fs: HashMap<path::PathBuf, &str>) -> projects::Project {
|
2023-10-04 13:50:16 +03:00
|
|
|
let repository = test_repository();
|
|
|
|
for (path, contents) in fs {
|
|
|
|
if let Some(parent) = path.parent() {
|
|
|
|
fs::create_dir_all(repository.path().parent().unwrap().join(parent))
|
|
|
|
.expect("failed to create dir");
|
|
|
|
}
|
|
|
|
fs::write(
|
|
|
|
repository.path().parent().unwrap().join(&path),
|
|
|
|
contents.as_bytes(),
|
|
|
|
)
|
|
|
|
.expect("failed to write file");
|
|
|
|
}
|
|
|
|
commit_all(&repository);
|
|
|
|
|
2023-10-05 17:03:01 +03:00
|
|
|
let project = projects::Project {
|
|
|
|
id: uuid::Uuid::new_v4().to_string(),
|
|
|
|
title: repository
|
|
|
|
.path()
|
|
|
|
.parent()
|
|
|
|
.unwrap()
|
|
|
|
.to_str()
|
|
|
|
.unwrap()
|
|
|
|
.to_string(),
|
|
|
|
path: repository.path().parent().unwrap().to_path_buf(),
|
|
|
|
..Default::default()
|
|
|
|
};
|
2023-10-04 13:50:16 +03:00
|
|
|
self.projects_storage
|
2023-10-05 11:04:36 +03:00
|
|
|
.add(&project)
|
2023-10-04 13:50:16 +03:00
|
|
|
.expect("failed to add project");
|
2023-10-06 13:43:32 +03:00
|
|
|
project
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn new_case_with_files(&self, fs: HashMap<path::PathBuf, &str>) -> Case {
|
|
|
|
let project = self.project(fs);
|
|
|
|
Case::new(self, project)
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn new_case(&self) -> Case {
|
|
|
|
self.new_case_with_files(HashMap::new())
|
2023-10-04 13:50:16 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-10-04 16:53:00 +03:00
|
|
|
pub struct Case<'a> {
|
|
|
|
suite: &'a Suite,
|
2023-10-06 13:43:32 +03:00
|
|
|
pub project: projects::Project,
|
2023-10-04 13:50:16 +03:00
|
|
|
pub project_repository: project_repository::Repository,
|
|
|
|
pub gb_repository: gb_repository::Repository,
|
|
|
|
}
|
2023-09-14 13:23:25 +03:00
|
|
|
|
2023-10-06 13:43:32 +03:00
|
|
|
impl<'a> Case<'a> {
|
|
|
|
fn new(suite: &'a Suite, project: projects::Project) -> Case<'a> {
|
|
|
|
let project_repository = project_repository::Repository::open(&project)
|
|
|
|
.expect("failed to create project repository");
|
|
|
|
let gb_repository =
|
|
|
|
gb_repository::Repository::open(&suite.local_app_data, &project_repository, None)
|
|
|
|
.expect("failed to open gb repository");
|
|
|
|
Case {
|
|
|
|
suite,
|
|
|
|
project,
|
|
|
|
gb_repository,
|
|
|
|
project_repository,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn refresh(&self) -> Self {
|
|
|
|
let project = self
|
2023-10-04 16:53:00 +03:00
|
|
|
.suite
|
|
|
|
.projects_storage
|
2023-10-05 11:04:36 +03:00
|
|
|
.get(&self.project.id)
|
|
|
|
.expect("failed to get project");
|
2023-10-06 13:43:32 +03:00
|
|
|
let project_repository = project_repository::Repository::open(&project)
|
2023-10-04 16:53:00 +03:00
|
|
|
.expect("failed to create project repository");
|
2023-10-06 13:43:32 +03:00
|
|
|
Self {
|
|
|
|
suite: self.suite,
|
|
|
|
gb_repository: gb_repository::Repository::open(
|
|
|
|
&self.suite.local_app_data,
|
|
|
|
&project_repository,
|
|
|
|
self.suite
|
|
|
|
.user_storage
|
|
|
|
.get()
|
|
|
|
.expect("failed to get user")
|
|
|
|
.as_ref(),
|
|
|
|
)
|
|
|
|
.expect("failed to open gb repository"),
|
|
|
|
project_repository,
|
|
|
|
project,
|
|
|
|
}
|
2023-10-04 16:53:00 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-09-14 13:23:25 +03:00
|
|
|
pub fn test_database() -> database::Database {
|
|
|
|
let path = temp_dir().join("test.db");
|
|
|
|
database::Database::try_from(&path).unwrap()
|
|
|
|
}
|
2023-08-30 10:46:52 +03:00
|
|
|
|
2023-08-28 10:34:30 +03:00
|
|
|
pub fn temp_dir() -> path::PathBuf {
|
|
|
|
let path = tempdir().unwrap().path().to_path_buf();
|
|
|
|
fs::create_dir_all(&path).unwrap();
|
|
|
|
path
|
|
|
|
}
|
|
|
|
|
2023-08-30 10:46:52 +03:00
|
|
|
pub fn test_repository() -> git::Repository {
|
2023-08-28 10:34:30 +03:00
|
|
|
let path = temp_dir();
|
2023-08-30 10:46:52 +03:00
|
|
|
let repository = git::Repository::init(path).expect("failed to init repository");
|
2023-08-28 10:34:30 +03:00
|
|
|
let mut index = repository.index().expect("failed to get index");
|
|
|
|
let oid = index.write_tree().expect("failed to write tree");
|
2023-09-20 12:53:59 +03:00
|
|
|
let signature = git::Signature::now("test", "test@email.com").unwrap();
|
2023-08-28 10:34:30 +03:00
|
|
|
repository
|
|
|
|
.commit(
|
|
|
|
Some("HEAD"),
|
|
|
|
&signature,
|
|
|
|
&signature,
|
|
|
|
"Initial commit",
|
|
|
|
&repository.find_tree(oid).expect("failed to find tree"),
|
|
|
|
&[],
|
|
|
|
)
|
|
|
|
.expect("failed to commit");
|
|
|
|
repository
|
|
|
|
}
|
2023-08-28 11:28:26 +03:00
|
|
|
|
2023-08-30 18:03:00 +03:00
|
|
|
pub fn commit_all(repository: &git::Repository) -> git::Oid {
|
2023-08-28 11:28:26 +03:00
|
|
|
let mut index = repository.index().expect("failed to get index");
|
|
|
|
index
|
|
|
|
.add_all(["."], git2::IndexAddOption::DEFAULT, None)
|
|
|
|
.expect("failed to add all");
|
|
|
|
index.write().expect("failed to write index");
|
|
|
|
let oid = index.write_tree().expect("failed to write tree");
|
2023-09-20 12:53:59 +03:00
|
|
|
let signature = git::Signature::now("test", "test@email.com").unwrap();
|
2023-08-28 11:28:26 +03:00
|
|
|
let commit_oid = repository
|
|
|
|
.commit(
|
|
|
|
Some("HEAD"),
|
|
|
|
&signature,
|
|
|
|
&signature,
|
|
|
|
"some commit",
|
|
|
|
&repository.find_tree(oid).expect("failed to find tree"),
|
|
|
|
&[&repository
|
|
|
|
.find_commit(
|
|
|
|
repository
|
|
|
|
.refname_to_id("HEAD")
|
|
|
|
.expect("failed to get head"),
|
|
|
|
)
|
|
|
|
.expect("failed to find commit")],
|
|
|
|
)
|
|
|
|
.expect("failed to commit");
|
|
|
|
commit_oid
|
|
|
|
}
|