use anyhow::Result;
use linked_hash_set::LinkedHashSet;
use crate::execution::execution_order::plan_request_order;
use crate::path_utils::CanonicalizedPathBuf;
use crate::request_sources::Preprocessed;
use crate::Config;
use crate::Profile;
use crate::RequestSource;
use crate::ResponseStore;
pub struct Requestpreprocessor {
profile: Profile,
config: Config,
requests: LinkedHashSet<RequestSource>,
response_data: ResponseStore,
}
impl Requestpreprocessor {
pub fn new(profile: Profile, requests: Vec<RequestSource>, config: Config) -> Result<Self> {
let requests_in_order = plan_request_order(requests, &profile)?;
Ok(Requestpreprocessor {
profile,
config,
requests: requests_in_order,
response_data: ResponseStore::new(),
})
}
pub fn is_empty(&self) -> bool {
self.requests.is_empty()
}
pub fn notify_response(&mut self, path: &CanonicalizedPathBuf, response: &str) {
self.response_data.store(path.clone(), response);
}
}
impl Iterator for Requestpreprocessor {
type Item = Result<RequestSource<Preprocessed>>;
fn next(&mut self) -> Option<Self::Item> {
self.requests
.pop_front()
.map(|req| req.replace_variables(&self.profile, &self.config, &self.response_data))
}
}
#[cfg(test)]
mod dependencies {
use std::env;
use crate::test_utils::root;
use crate::RequestSource;
use super::*;
#[test]
fn should_replace_dependencies_on_next_calls() -> Result<()> {
let root = root().join("resources/test/requests/nested_dependencies");
let init_path = root.join("4.http");
let dep_path = root.join("5.http");
let init_request = RequestSource::from_file(init_path, false)?;
let mut preprocessor = Requestpreprocessor::new(
Profile::empty(env::current_dir().unwrap()),
vec![init_request],
Config::default(),
)?;
preprocessor.next();
preprocessor.notify_response(&dep_path, "dependency");
let result = preprocessor.next().unwrap().unwrap();
let req = result.parse()?;
assert_eq!(req.request.url, "dependency");
Ok(())
}
}