add integration test

lycheeverse · Nov 27, 2024 · 6aa76fc · 6aa76fc
1 parent b8319b0
commit 6aa76fc
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/lychee-bin/tests/cli.rs b/lychee-bin/tests/cli.rs
@@ -1880,4 +1880,28 @@ mod cli {
 
         Ok(())
     }
+
+    #[tokio::test]
+    async fn test_retry() -> Result<()> {
+        let mock_server = wiremock::MockServer::start().await;
+
+        wiremock::Mock::given(wiremock::matchers::method("GET"))
+            .respond_with(ResponseTemplate::new(429))
+            .up_to_n_times(1)
+            .mount(&mock_server)
+            .await;
+
+        wiremock::Mock::given(wiremock::matchers::method("GET"))
+            .respond_with(ResponseTemplate::new(200))
+            .mount(&mock_server)
+            .await;
+
+        let mut cmd = main_command();
+        cmd.arg("-")
+            .write_stdin(mock_server.uri())
+            .assert()
+            .success();
+
+        Ok(())
+    }
 }