From cc274ee33d1bc63ca783343af6a74af7d26c281d Mon Sep 17 00:00:00 2001
From: zhoupeiheng <zhoupeiheng@bytedance.com>
Date: Wed, 20 Jul 2022 18:04:50 +0800
Subject: [PATCH] test: check_static_file_route_pattern

---
 src/config.rs   | 10 ++++++++--
 tests/config.rs | 28 ++++++++++++++++++++++++++++
 2 files changed, 36 insertions(+), 2 deletions(-)
 create mode 100644 tests/config.rs

diff --git a/src/config.rs b/src/config.rs
index 0a5c945c3..d3aa845c4 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -152,9 +152,15 @@ impl MiniserveConfig {
         // Generate some random routes for the favicon and css so that they are very unlikely to conflict with
         // real files.
         let (favicon_route, css_route) = if args.random_route {
-            (format!("/{}", nanoid::nanoid!(10, &ROUTE_ALPHABET)), format!("/{}", nanoid::nanoid!(10, &ROUTE_ALPHABET)))
+            (
+                format!("/{}", nanoid::nanoid!(10, &ROUTE_ALPHABET)),
+                format!("/{}", nanoid::nanoid!(10, &ROUTE_ALPHABET)),
+            )
         } else {
-            (format!("{}/{}", route_prefix, nanoid::nanoid!(10, &ROUTE_ALPHABET)), format!("{}/{}", route_prefix, nanoid::nanoid!(10, &ROUTE_ALPHABET)))
+            (
+                format!("{}/{}", route_prefix, nanoid::nanoid!(10, &ROUTE_ALPHABET)),
+                format!("{}/{}", route_prefix, nanoid::nanoid!(10, &ROUTE_ALPHABET)),
+            )
         };
 
         let default_color_scheme = args.color_scheme;
diff --git a/tests/config.rs b/tests/config.rs
new file mode 100644
index 000000000..136b3b19d
--- /dev/null
+++ b/tests/config.rs
@@ -0,0 +1,28 @@
+mod fixtures;
+
+use fixtures::{server_no_stderr, Error, TestServer};
+use regex::Regex;
+use rstest::rstest;
+use select::{document::Document, predicate::Attr};
+
+#[rstest]
+#[case(server_no_stderr(&[] as &[&str]), "/[a-f0-9]+")]
+#[case(server_no_stderr(&["--random-route"]), "/[a-f0-9]+")]
+#[case(server_no_stderr(&["--route-prefix", "foo"]), "/foo/[a-f0-9]+")]
+fn check_static_file_route_pattern(
+    #[case] server: TestServer,
+    #[case] route_pattern: String,
+) -> Result<(), Error> {
+    let body = reqwest::blocking::get(server.url())?;
+    let parsed = Document::from_read(body)?;
+    let re = Regex::new(&route_pattern).unwrap();
+
+    assert!(parsed
+        .find(Attr("rel", "stylesheet"))
+        .all(|x| re.is_match(x.attr("href").unwrap())));
+    assert!(parsed
+        .find(Attr("rel", "icon"))
+        .all(|x| re.is_match(x.attr("href").unwrap())));
+
+    Ok(())
+}