Skip to content

Commit 81e90f7

Browse files
authored
Merge pull request #2379 from input-output-hk/sfa/2360/use_initialize_dependencies_with_test_temp_dir
Reduce flakiness in tests
2 parents ba14422 + 3bf6040 commit 81e90f7

File tree

3 files changed

+25
-17
lines changed

3 files changed

+25
-17
lines changed

Cargo.lock

+1-1
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

mithril-aggregator/Cargo.toml

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[package]
22
name = "mithril-aggregator"
3-
version = "0.7.18"
3+
version = "0.7.19"
44
description = "A Mithril Aggregator server"
55
authors = { workspace = true }
66
edition = { workspace = true }

mithril-aggregator/src/runtime/runner.rs

+23-15
Original file line numberDiff line numberDiff line change
@@ -547,6 +547,7 @@ pub mod tests {
547547
use mithril_signed_entity_lock::SignedEntityTypeLock;
548548
use mockall::predicate::eq;
549549
use mockall::{mock, Sequence};
550+
use std::path::PathBuf;
550551
use std::sync::Arc;
551552
use tokio::sync::RwLock;
552553

@@ -585,8 +586,11 @@ pub mod tests {
585586
AggregatorRunner::new(Arc::new(deps))
586587
}
587588

588-
async fn build_runner(mock_certifier_service: MockCertifierService) -> AggregatorRunner {
589-
let mut deps = initialize_dependencies!().await;
589+
async fn build_runner(
590+
temp_dir: PathBuf,
591+
mock_certifier_service: MockCertifierService,
592+
) -> AggregatorRunner {
593+
let mut deps = initialize_dependencies(temp_dir).await;
590594
deps.certifier_service = Arc::new(mock_certifier_service);
591595

592596
let mut mock_signable_builder_service = MockSignableBuilderServiceImpl::new();
@@ -944,7 +948,7 @@ pub mod tests {
944948
.expect_create_open_message()
945949
.return_once(|_, _| Ok(open_message_created))
946950
.times(1);
947-
build_runner(mock_certifier_service).await
951+
build_runner(temp_dir!(), mock_certifier_service).await
948952
};
949953

950954
let open_message_returned = runner
@@ -969,7 +973,7 @@ pub mod tests {
969973
);
970974

971975
mock_certifier_service.expect_create_open_message().never();
972-
build_runner(mock_certifier_service).await
976+
build_runner(temp_dir!(), mock_certifier_service).await
973977
};
974978

975979
let open_message_returned = runner
@@ -996,7 +1000,7 @@ pub mod tests {
9961000
);
9971001

9981002
mock_certifier_service.expect_create_open_message().never();
999-
build_runner(mock_certifier_service).await
1003+
build_runner(temp_dir!(), mock_certifier_service).await
10001004
};
10011005

10021006
let open_message_returned = runner
@@ -1026,7 +1030,7 @@ pub mod tests {
10261030
.expect_create_open_message()
10271031
.return_once(|_, _| Ok(open_message_created))
10281032
.times(1);
1029-
build_runner(mock_certifier_service).await
1033+
build_runner(temp_dir!(), mock_certifier_service).await
10301034
};
10311035

10321036
let open_message_returned = runner
@@ -1051,7 +1055,7 @@ pub mod tests {
10511055
);
10521056

10531057
mock_certifier_service.expect_create_open_message().never();
1054-
build_runner(mock_certifier_service).await
1058+
build_runner(temp_dir!(), mock_certifier_service).await
10551059
};
10561060

10571061
let open_message_returned = runner
@@ -1078,7 +1082,7 @@ pub mod tests {
10781082
);
10791083

10801084
mock_certifier_service.expect_create_open_message().never();
1081-
build_runner(mock_certifier_service).await
1085+
build_runner(temp_dir!(), mock_certifier_service).await
10821086
};
10831087

10841088
let open_message_returned = runner
@@ -1122,7 +1126,7 @@ pub mod tests {
11221126
.expect_mark_open_message_if_expired()
11231127
.returning(|_| Ok(None));
11241128

1125-
let runner = build_runner(mock_certifier_service).await;
1129+
let runner = build_runner(temp_dir!(), mock_certifier_service).await;
11261130

11271131
runner
11281132
.get_current_non_certified_open_message(&TimePoint::dummy())
@@ -1203,28 +1207,32 @@ pub mod tests {
12031207
#[tokio::test]
12041208
async fn is_open_message_outdated_return_false_when_message_is_not_expired_and_no_newer_open_message(
12051209
) {
1206-
assert!(!is_outdated_returned_when(IsExpired::No, false).await);
1210+
assert!(!is_outdated_returned_when(temp_dir!(), IsExpired::No, false).await);
12071211
}
12081212

12091213
#[tokio::test]
12101214
async fn is_open_message_outdated_return_true_when_message_is_expired_and_no_newer_open_message(
12111215
) {
1212-
assert!(is_outdated_returned_when(IsExpired::Yes, false).await);
1216+
assert!(is_outdated_returned_when(temp_dir!(), IsExpired::Yes, false).await);
12131217
}
12141218

12151219
#[tokio::test]
12161220
async fn is_open_message_outdated_return_true_when_message_is_not_expired_and_exists_newer_open_message(
12171221
) {
1218-
assert!(is_outdated_returned_when(IsExpired::No, true).await);
1222+
assert!(is_outdated_returned_when(temp_dir!(), IsExpired::No, true).await);
12191223
}
12201224

12211225
#[tokio::test]
12221226
async fn is_open_message_outdated_return_true_when_message_is_expired_and_exists_newer_open_message(
12231227
) {
1224-
assert!(is_outdated_returned_when(IsExpired::Yes, true).await);
1228+
assert!(is_outdated_returned_when(temp_dir!(), IsExpired::Yes, true).await);
12251229
}
12261230

1227-
async fn is_outdated_returned_when(is_expired: IsExpired, newer_open_message: bool) -> bool {
1231+
async fn is_outdated_returned_when(
1232+
tmp_path: PathBuf,
1233+
is_expired: IsExpired,
1234+
newer_open_message: bool,
1235+
) -> bool {
12281236
let current_time_point = TimePoint {
12291237
epoch: Epoch(2),
12301238
..TimePoint::dummy()
@@ -1242,7 +1250,7 @@ pub mod tests {
12421250
};
12431251

12441252
let runner = {
1245-
let mut deps = initialize_dependencies!().await;
1253+
let mut deps = initialize_dependencies(tmp_path).await;
12461254
let mut mock_certifier_service = MockCertifierService::new();
12471255

12481256
let open_message_current = open_message_to_verify.clone();

0 commit comments

Comments
 (0)