-
Notifications
You must be signed in to change notification settings - Fork 11
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add integration tests for request/response modifiers
- Loading branch information
Showing
5 changed files
with
197 additions
and
53 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1 +1,46 @@ | ||
| use std::sync::Arc; | ||
|
|
||
| use tokio::sync::Mutex; | ||
| use tracing_subscriber::{ | ||
| filter, prelude::__tracing_subscriber_SubscriberExt, util::SubscriberInitExt, Layer, | ||
| }; | ||
|
|
||
| mod e2e; | ||
| mod modifiers; | ||
|
|
||
| #[derive(Clone)] | ||
| pub struct TestScope { | ||
| pub initialized: Arc<Mutex<bool>>, | ||
| } | ||
|
|
||
| impl Default for TestScope { | ||
| fn default() -> Self { | ||
| Self { | ||
| initialized: Arc::new(Mutex::new(false)), | ||
| } | ||
| } | ||
| } | ||
|
|
||
| impl TestScope { | ||
| pub async fn init(self) { | ||
| let mut inited = self.initialized.lock().await; | ||
| if *inited == false { | ||
| if std::env::var("TEST_LOG").is_ok() { | ||
| let stdout_log = tracing_subscriber::fmt::layer().pretty(); | ||
| tracing_subscriber::registry() | ||
| .with( | ||
| stdout_log | ||
| // Add an `INFO` filter to the stdout logging layer | ||
| .with_filter(filter::LevelFilter::DEBUG), | ||
| ) | ||
| .init(); | ||
| } | ||
| *inited = true; | ||
| } | ||
| } | ||
| } | ||
|
|
||
| lazy_static::lazy_static! { | ||
| static ref SCOPE: TestScope = TestScope::default(); | ||
| static ref ADDRESS: String = String::from("http://127.0.0.1:38282"); | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,135 @@ | ||
| use std::borrow::Cow; | ||
| use std::fs; | ||
| use std::path::PathBuf; | ||
|
|
||
| use reqwest_middleware::ClientBuilder; | ||
| use reqwest_middleware::ClientWithMiddleware; | ||
| use rvcr::VCRMiddleware; | ||
| use rvcr::VCRMode; | ||
| use vcr_cassette::{Request, Response}; | ||
|
|
||
| // Replace access_token and secret header values with dummy ones | ||
| fn filter_query_params(mut uri: url::Url) -> url::Url { | ||
| let sensitive_query_params = ["access_token", "secret"]; | ||
| let cloned = uri.clone(); | ||
| let filtered_query_params = cloned.query_pairs().map(|(k, v)| { | ||
| if sensitive_query_params.contains(&k.as_ref()) { | ||
| (k.clone(), Cow::from(format!("__{}__", k.to_uppercase()))) | ||
| } else { | ||
| (k, v) | ||
| } | ||
| }); | ||
| uri.query_pairs_mut() | ||
| .clear() | ||
| .extend_pairs(filtered_query_params) | ||
| .finish(); | ||
| uri | ||
| } | ||
|
|
||
| fn request_modifier(req: &mut Request) { | ||
| // Overwrite query params with filtered ones | ||
| req.uri = filter_query_params(req.uri.clone()); | ||
| } | ||
|
|
||
| fn response_modifier(resp: &mut Response) { | ||
| for (name, value) in &mut resp.headers { | ||
| if name == "server" { | ||
| (*value).pop().unwrap(); | ||
| (*value).push("Test Server Header Emulated Expect".to_string()); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| fn saved_fixture_path(path: &str) -> PathBuf { | ||
| let mut bundle = PathBuf::from(std::env::temp_dir()); | ||
| bundle.push(path); | ||
|
|
||
| if bundle.exists() { | ||
| std::fs::remove_file(bundle.clone()).unwrap(); | ||
| } | ||
| bundle | ||
| } | ||
|
|
||
| #[tokio::test] | ||
| pub async fn test_modifier_request() { | ||
| crate::SCOPE.clone().init().await; | ||
| let bundle = saved_fixture_path("request-modifer-test-case.vcr.json"); | ||
|
|
||
| let middleware = VCRMiddleware::try_from(bundle.clone()) | ||
| .unwrap() | ||
| .with_mode(VCRMode::Record) | ||
| .with_modify_request(request_modifier); | ||
|
|
||
| let vcr_client: ClientWithMiddleware = ClientBuilder::new(reqwest::Client::new()) | ||
| .with(middleware) | ||
| .build(); | ||
|
|
||
| vcr_client | ||
| .request( | ||
| reqwest::Method::POST, | ||
| format!( | ||
| "{}{}", | ||
| crate::ADDRESS.to_string(), | ||
| "/post?access_token=s3cr3t&spam=eggs&secret=s3cr3t", | ||
| ), | ||
| ) | ||
| .send() | ||
| .await | ||
| .expect("Can not send request"); | ||
| // Drop triggers recording | ||
| drop(vcr_client); | ||
|
|
||
| let vcr_content = fs::read(bundle.clone()).expect("VCR file not created during test case"); | ||
|
|
||
| let cassette: vcr_cassette::Cassette = serde_json::from_slice(&vcr_content).unwrap(); | ||
| let interaction = cassette.http_interactions.first().unwrap(); | ||
| let recorded_url = interaction.request.uri.to_string(); | ||
| // Secret parameters were replaced | ||
| assert_eq!( | ||
| format!( | ||
| "{}/post?access_token=__ACCESS_TOKEN__&spam=eggs&secret=__SECRET__", | ||
| crate::ADDRESS.to_string() | ||
| ), | ||
| recorded_url, | ||
| ) | ||
| } | ||
|
|
||
| #[tokio::test] | ||
| pub async fn test_modifier_response() { | ||
| crate::SCOPE.clone().init().await; | ||
| let bundle = saved_fixture_path("response-modifer-test-case.vcr.json"); | ||
| let middleware = VCRMiddleware::try_from(bundle.clone()) | ||
| .unwrap() | ||
| .with_mode(VCRMode::Record) | ||
| .with_modify_response(response_modifier); | ||
|
|
||
| let vcr_client: ClientWithMiddleware = ClientBuilder::new(reqwest::Client::new()) | ||
| .with(middleware) | ||
| .build(); | ||
|
|
||
| vcr_client | ||
| .request( | ||
| reqwest::Method::POST, | ||
| format!("{}{}", crate::ADDRESS.to_string(), "/post",), | ||
| ) | ||
| .send() | ||
| .await | ||
| .expect("Can not send request"); | ||
| // Drop triggers recording | ||
| drop(vcr_client); | ||
|
|
||
| let vcr_content = fs::read(bundle.clone()).expect("VCR file not created during test case"); | ||
|
|
||
| let cassette: vcr_cassette::Cassette = serde_json::from_slice(&vcr_content).unwrap(); | ||
| let interaction = cassette.http_interactions.first().unwrap(); | ||
| let recorded_server_header = interaction | ||
| .response | ||
| .headers | ||
| .get("server") | ||
| .unwrap() | ||
| .clone() | ||
| .pop() | ||
| .unwrap(); | ||
| // Server header was replaced | ||
| assert_eq!(recorded_server_header, "Test Server Header Emulated Expect",) | ||
| } |