Skip to content

Commit

Permalink
Remove unused things and single use things from src/test.rs.
Browse files Browse the repository at this point in the history
  • Loading branch information
partim committed Jul 25, 2024
1 parent ce41f73 commit e164be6
Show file tree
Hide file tree
Showing 4 changed files with 28 additions and 1,246 deletions.
2 changes: 1 addition & 1 deletion src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -9,5 +9,5 @@ pub mod constants;
pub mod daemon;
pub mod pubd;
pub mod ta;
pub mod test;
pub(crate) mod test;
pub mod upgrades;
51 changes: 19 additions & 32 deletions src/pubd/manager.rs
Original file line number Diff line number Diff line change
Expand Up @@ -370,7 +370,7 @@ mod tests {
constants::*,
daemon::config::{SignerConfig, SignerType},
pubd::{Publisher, RrdpServer},
test::{self, https, init_config, rsync},
test::{self, https, rsync},
};

fn publisher_alice(storage_uri: &Url) -> Publisher {
Expand Down Expand Up @@ -416,17 +416,22 @@ mod tests {
)
}

fn make_server(storage_uri: &Url, data_dir: &Path) -> RepositoryManager {
fn make_server(
storage_uri: &Url
) -> (RepositoryManager, tempfile::TempDir) {
let data_dir = tempfile::tempdir().unwrap();

enable_test_mode();
let mut config = Config::test(
storage_uri,
Some(data_dir),
Some(data_dir.path()),
true,
false,
false,
false,
);
init_config(&mut config);
let _ = config.init_logging();
config.process().unwrap();

let signer = KrillSignerBuilder::new(
storage_uri,
Expand All @@ -451,15 +456,14 @@ mod tests {

repository_manager.init(uris).unwrap();

repository_manager
(repository_manager, data_dir)
}

#[test]
fn should_add_publisher() {
// we need a disk, as repo_dir, etc. use data_dir by default
let (data_dir, cleanup) = test::tmp_dir();
let storage_uri = test::mem_storage();
let server = make_server(&storage_uri, &data_dir);
let (server, _data_dir) = make_server(&storage_uri);

let alice = publisher_alice(&storage_uri);

Expand All @@ -476,17 +480,12 @@ mod tests {
assert_eq!(alice_found.base_uri(), alice.base_uri());
assert_eq!(alice_found.id_cert(), alice.id_cert());
assert!(alice_found.current_files().is_empty());

cleanup();
}

#[test]
fn should_not_add_publisher_twice() {
// we need a disk, as repo_dir, etc. use data_dir by default
let (data_dir, cleanup) = test::tmp_dir();
let storage_uri = test::mem_storage();

let server = make_server(&storage_uri, &data_dir);
let (server, _data_dir) = make_server(&storage_uri);

let alice = publisher_alice(&storage_uri);

Expand All @@ -505,16 +504,12 @@ mod tests {
}
_ => panic!("Expected error"),
}

cleanup();
}

#[test]
fn should_list_files() {
// we need a disk, as repo_dir, etc. use data_dir by default
let (data_dir, cleanup) = test::tmp_dir();
let storage_uri = test::mem_storage();
let server = make_server(&storage_uri, &data_dir);
let (server, _data_dir) = make_server(&storage_uri);

let alice = publisher_alice(&storage_uri);

Expand All @@ -527,18 +522,15 @@ mod tests {

let list_reply = server.list(&alice_handle).unwrap();
assert_eq!(0, list_reply.elements().len());

cleanup();
}

#[tokio::test]
async fn should_publish_files() {
// we need a disk, as repo_dir, etc. use data_dir by default
let (data_dir, cleanup) = test::tmp_dir();
let storage_uri = test::mem_storage();
let server = make_server(&storage_uri, &data_dir);
let (server, data_dir) = make_server(&storage_uri);

let session = session_dir(&data_dir);
let session = session_dir(data_dir.path());

// Check that the server starts with dir for serial 1 for RRDP
// and does not use 0 (RFC 8182)
Expand Down Expand Up @@ -774,15 +766,12 @@ mod tests {
&session,
RRDP_FIRST_SERIAL + 3
));

cleanup();
}

#[test]
pub fn repository_session_reset() {
let (data_dir, cleanup) = test::tmp_dir();
let storage_uri = test::mem_storage();
let server = make_server(&storage_uri, &data_dir);
let (server, data_dir) = make_server(&storage_uri);

// set up server with default repository, and publisher alice
let alice = publisher_alice(&storage_uri);
Expand Down Expand Up @@ -839,7 +828,7 @@ mod tests {
let stats_before = server.repo_stats().unwrap();
let session_before = stats_before.session();
let snapshot_before_session_reset = find_in_session_and_serial_dir(
&data_dir,
data_dir.path(),
&session_before,
RRDP_FIRST_SERIAL + 1,
"snapshot.xml",
Expand All @@ -855,7 +844,7 @@ mod tests {
let session_after = stats_after.session();

let snapshot_after_session_reset = find_in_session_and_serial_dir(
&data_dir,
&data_dir.path(),
&session_after,
RRDP_FIRST_SERIAL,
"snapshot.xml",
Expand All @@ -869,15 +858,13 @@ mod tests {

// and clean up old dir
let snapshot_before_session_reset = find_in_session_and_serial_dir(
&data_dir,
data_dir.path(),
&session_before,
RRDP_FIRST_SERIAL + 1,
"snapshot.xml",
);

assert!(snapshot_before_session_reset.is_none());

cleanup();
}

fn session_dir(base_dir: &Path) -> PathBuf {
Expand Down
4 changes: 3 additions & 1 deletion src/ta/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -82,7 +82,9 @@ mod tests {

let timing = TaTimingConfig::default();

let actor = test::test_actor();
let actor = crate::commons::actor::Actor::actor_from_def(
crate::constants::ACTOR_DEF_KRILL,
);

let proxy_handle = TrustAnchorHandle::new("proxy".into());
let proxy_init = TrustAnchorProxyInitCommand::make(
Expand Down
Loading

0 comments on commit e164be6

Please sign in to comment.