Merge pull request #49 from anthonyoteri/replace-async-std-with-tokio

Use better async runtime
This commit is contained in:
Anthony Oteri
2023-10-02 13:15:56 -04:00
committed by GitHub
4 changed files with 14 additions and 14 deletions
+2 -2
View File
@@ -21,7 +21,6 @@ categories = [
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies] [dependencies]
async-std = { version = "1.12.0", features = ["async-attributes", "attributes", "tokio1"] }
clap = { version = "4.4.3", features = ["derive", "env", "wrap_help"] } clap = { version = "4.4.3", features = ["derive", "env", "wrap_help"] }
simple_logger = { version = "4.2.0", features = ["timestamps", "colors", "stderr"] } simple_logger = { version = "4.2.0", features = ["timestamps", "colors", "stderr"] }
http = "0.2.9" http = "0.2.9"
@@ -35,7 +34,8 @@ thiserror = "1.0.48"
toml = "0.8.0" toml = "0.8.0"
url = { version = "2.4.1", features = ["serde"] } url = { version = "2.4.1", features = ["serde"] }
xdg = "2.5.2" xdg = "2.5.2"
tokio = { version = "1.32.0", features = ["macros"] }
[dev-dependencies] [dev-dependencies]
mockito = "1.2.0" mockito = "1.2.0"
env_logger = "0.10.0" env_logger = "0.10.0"
+3 -3
View File
@@ -207,7 +207,7 @@ mod tests {
/// ///
/// Attempt to parse a valid RFC5988 header value, and ensure that the /// Attempt to parse a valid RFC5988 header value, and ensure that the
/// parsed URL was returned as expected. /// parsed URL was returned as expected.
#[async_std::test] #[tokio::test]
async fn test_parse_rfc5988_valid() { async fn test_parse_rfc5988_valid() {
// Mock a valid RFC5988 header value // Mock a valid RFC5988 header value
let valid_header_value = let valid_header_value =
@@ -225,7 +225,7 @@ mod tests {
/// ///
/// Attempt to parse an invalid string as RFC5988, ensuring that the `None` /// Attempt to parse an invalid string as RFC5988, ensuring that the `None`
/// variant is returned. /// variant is returned.
#[async_std::test] #[tokio::test]
async fn test_parse_rfc5988_invalid() { async fn test_parse_rfc5988_invalid() {
// Mock a valid RFC5988 header value // Mock a valid RFC5988 header value
let invalid_header_value = HeaderValue::from_str(r#"invalid header value"#) let invalid_header_value = HeaderValue::from_str(r#"invalid header value"#)
@@ -244,7 +244,7 @@ mod tests {
/// the digest with the proper headers set. The test then validates that /// the digest with the proper headers set. The test then validates that
/// the correct digest is returned and that the mock server had the expected /// the correct digest is returned and that the mock server had the expected
/// interactions. /// interactions.
#[async_std::test] #[tokio::test]
async fn test_get_digest() -> Result<(), ApiError> { async fn test_get_digest() -> Result<(), ApiError> {
let mut server = mockito::Server::new_async().await; let mut server = mockito::Server::new_async().await;
let path = "/v2/foo/manifests/latest"; let path = "/v2/foo/manifests/latest";
+8 -8
View File
@@ -224,7 +224,7 @@ mod tests {
/// This test spins up a mock server, and makes a request to the catalog /// This test spins up a mock server, and makes a request to the catalog
/// endpoint. It checks that the handler both called the request the /// endpoint. It checks that the handler both called the request the
/// expected number of times, and did not return an error. /// expected number of times, and did not return an error.
#[async_std::test] #[tokio::test]
async fn test_catalog_handler() { async fn test_catalog_handler() {
let mut server = mockito::Server::new_async().await; let mut server = mockito::Server::new_async().await;
let path = "/v2/_catalog"; let path = "/v2/_catalog";
@@ -253,7 +253,7 @@ mod tests {
/// should follow, resulting in the combined list. It checks that the /// should follow, resulting in the combined list. It checks that the
/// handler both called the request the expected number of times, and did /// handler both called the request the expected number of times, and did
/// not return an error. /// not return an error.
#[async_std::test] #[tokio::test]
async fn test_catalog_handler_with_pagination() { async fn test_catalog_handler_with_pagination() {
let mut server = mockito::Server::new_async().await; let mut server = mockito::Server::new_async().await;
let path = "/v2/_catalog"; let path = "/v2/_catalog";
@@ -294,7 +294,7 @@ mod tests {
/// This test spins up a mock server, and makes a request to the tags /// This test spins up a mock server, and makes a request to the tags
/// endpoint. It checks that the handler both called the request the /// endpoint. It checks that the handler both called the request the
/// expected number of times, and did not return an error. /// expected number of times, and did not return an error.
#[async_std::test] #[tokio::test]
async fn test_tags_handler() { async fn test_tags_handler() {
let mut server = mockito::Server::new_async().await; let mut server = mockito::Server::new_async().await;
let path = "/v2/some_image/tags/list"; let path = "/v2/some_image/tags/list";
@@ -324,7 +324,7 @@ mod tests {
/// should follow, resulting in the combined list. It checks that the /// should follow, resulting in the combined list. It checks that the
/// handler both called the request the expected number of times, and did /// handler both called the request the expected number of times, and did
/// not return an error. /// not return an error.
#[async_std::test] #[tokio::test]
async fn test_tags_handler_with_pagination() { async fn test_tags_handler_with_pagination() {
let mut server = mockito::Server::new_async().await; let mut server = mockito::Server::new_async().await;
let path = "/v2/some_image/tags/list"; let path = "/v2/some_image/tags/list";
@@ -366,7 +366,7 @@ mod tests {
/// This test spins up a mock server, and makes a request to the check /// This test spins up a mock server, and makes a request to the check
/// endpoint. It checks that the handler both called the request the /// endpoint. It checks that the handler both called the request the
/// expected number of times, and did not return an error. /// expected number of times, and did not return an error.
#[async_std::test] #[tokio::test]
async fn test_check_handler() { async fn test_check_handler() {
let mut server = mockito::Server::new_async().await; let mut server = mockito::Server::new_async().await;
let path = "/v2"; let path = "/v2";
@@ -392,7 +392,7 @@ mod tests {
/// ///
/// This validates that if the "Docker-Distribution-API-Version" header /// This validates that if the "Docker-Distribution-API-Version" header
/// is missing in the response, the appropriate error is returned. /// is missing in the response, the appropriate error is returned.
#[async_std::test] #[tokio::test]
async fn test_check_handler_missing_api_version() -> Result<(), Box<dyn Error>> { async fn test_check_handler_missing_api_version() -> Result<(), Box<dyn Error>> {
let mut server = mockito::Server::new_async().await; let mut server = mockito::Server::new_async().await;
let path = "/v2"; let path = "/v2";
@@ -425,7 +425,7 @@ mod tests {
/// This validates that if the "Docker-Distribution-API-Version" header /// This validates that if the "Docker-Distribution-API-Version" header
/// is present in the response but contains an unexpected value, the /// is present in the response but contains an unexpected value, the
/// appropriate error is returned. /// appropriate error is returned.
#[async_std::test] #[tokio::test]
async fn test_check_handler_invalid_api_version() -> Result<(), Box<dyn Error>> { async fn test_check_handler_invalid_api_version() -> Result<(), Box<dyn Error>> {
let mut server = mockito::Server::new_async().await; let mut server = mockito::Server::new_async().await;
let path = "/v2"; let path = "/v2";
@@ -459,7 +459,7 @@ mod tests {
/// This test spins up a mock server, and makes a request to the image /// This test spins up a mock server, and makes a request to the image
/// manifests endpoint. It checks that the handler both called the request /// manifests endpoint. It checks that the handler both called the request
/// the expected number of times, and did not return an error. /// the expected number of times, and did not return an error.
#[async_std::test] #[tokio::test]
async fn test_show_handler() { async fn test_show_handler() {
let mut server = mockito::Server::new_async().await; let mut server = mockito::Server::new_async().await;
let path = "/v2/foo/manifests/latest"; let path = "/v2/foo/manifests/latest";
+1 -1
View File
@@ -57,7 +57,7 @@ fn parse_registry_arg(host: &str) -> Result<Url, DredgeError> {
Url::parse(&host).or(Err(DredgeError::RegistryUrlError(host.to_string()))) Url::parse(&host).or(Err(DredgeError::RegistryUrlError(host.to_string())))
} }
#[async_std::main] #[tokio::main(flavor = "current_thread")]
async fn main() -> Result<(), DredgeError> { async fn main() -> Result<(), DredgeError> {
let args = Cli::parse(); let args = Cli::parse();