From 9459246bc95a4f2fdddb92a91dbf3dfa79f723df Mon Sep 17 00:00:00 2001
From: epi <epibar052@gmail.com>
Date: Tue, 2 Feb 2021 15:09:53 -0600
Subject: [PATCH] added banner test

---
 tests/test_banner.rs | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/tests/test_banner.rs b/tests/test_banner.rs
index 6c2b390..2292fc8 100644
--- a/tests/test_banner.rs
+++ b/tests/test_banner.rs
@@ -842,3 +842,30 @@ fn banner_prints_similarity_filter() {
                 .and(predicate::str::contains("─┴─")),
         );
 }
+
+#[test]
+/// test allows non-existent wordlist to trigger the banner printing to stderr
+/// expect to see all mandatory prints + rate limit
+fn banner_prints_rate_limit() {
+    Command::cargo_bin("feroxbuster")
+        .unwrap()
+        .arg("--url")
+        .arg("http://localhost")
+        .arg("--rate-limit")
+        .arg("6735")
+        .assert()
+        .success()
+        .stderr(
+            predicate::str::contains("─┬─")
+                .and(predicate::str::contains("Target Url"))
+                .and(predicate::str::contains("http://localhost"))
+                .and(predicate::str::contains("Threads"))
+                .and(predicate::str::contains("Wordlist"))
+                .and(predicate::str::contains("Status Codes"))
+                .and(predicate::str::contains("Timeout (secs)"))
+                .and(predicate::str::contains("User-Agent"))
+                .and(predicate::str::contains("Requests per Second"))
+                .and(predicate::str::contains("│ 6735"))
+                .and(predicate::str::contains("─┴─")),
+        );
+}