cli/vendor/wiremock/tests/request_header_matching.rs

use wiremock::matchers::{basic_auth, bearer_token, header, header_regex, headers, method};
use wiremock::{Mock, MockServer, ResponseTemplate};

#[async_std::test]
async fn should_match_simple_request_header() {
    // Arrange
    let mock_server = MockServer::start().await;
    let mock = Mock::given(method("GET"))
        .and(header("content-type", "application/json"))
        .respond_with(ResponseTemplate::new(200));
    mock_server.register(mock).await;

    // Act
    let client = reqwest::Client::new();
    let should_match = client
        .get(mock_server.uri())
        .header("content-type", "application/json")
        .send()
        .await
        .unwrap();
    // Assert
    assert_eq!(should_match.status(), 200);
}

#[async_std::test]
async fn should_not_match_simple_request_header_upon_wrong_key() {
    // Arrange
    let mock_server = MockServer::start().await;
    let mock = Mock::given(method("GET"))
        .and(header("content-type", "application/json"))
        .respond_with(ResponseTemplate::new(200));
    mock_server.register(mock).await;

    // Act
    let client = reqwest::Client::new();
    let should_fail_wrong_key = client
        .get(mock_server.uri())
        .header("accept", "application/json")
        .send()
        .await
        .unwrap();
    // Assert
    assert_eq!(should_fail_wrong_key.status(), 404);
}

#[async_std::test]
async fn should_not_match_simple_request_header_upon_wrong_value() {
    // Arrange
    let mock_server = MockServer::start().await;
    let mock = Mock::given(method("GET"))
        .and(header("content-type", "application/json"))
        .respond_with(ResponseTemplate::new(200));
    mock_server.register(mock).await;

    // Act
    let client = reqwest::Client::new();
    let should_fail_wrong_value = client
        .get(mock_server.uri())
        .header("content-type", "application/xml")
        .send()
        .await
        .unwrap();
    // Assert
    assert_eq!(should_fail_wrong_value.status(), 404);
}

#[async_std::test]
async fn should_match_multi_request_header() {
    // Arrange
    let mock_server = MockServer::start().await;
    let header_matcher = headers("cache-control", vec!["no-cache", "no-store"]);
    let mock = Mock::given(method("GET"))
        .and(header_matcher)
        .respond_with(ResponseTemplate::new(200));
    mock_server.register(mock).await;

    // Act
    let client = reqwest::Client::new();
    let should_match = client
        .get(mock_server.uri())
        .header("cache-control", "no-cache, no-store")
        .send()
        .await
        .unwrap();

    // Assert
    assert_eq!(should_match.status(), 200);
}

#[tokio::test]
async fn should_match_multi_request_header_x() {
    // Arrange
    let mock_server = MockServer::start().await;
    let header_matcher = headers("cache-control", vec!["no-cache", "no-store"]);
    let mock = Mock::given(method("GET"))
        .and(header_matcher)
        .respond_with(ResponseTemplate::new(200))
        .expect(1);
    mock_server.register(mock).await;

    // Act
    let should_match = reqwest::Client::new()
        .get(mock_server.uri())
        // TODO: use a dedicated headers when upgrade reqwest v0.12
        .header("cache-control", "no-cache")
        .header("cache-control", "no-store")
        .send()
        .await
        .unwrap();

    // Assert
    assert_eq!(should_match.status(), 200);
}

#[async_std::test]
async fn should_not_match_multi_request_header_upon_wrong_values() {
    // Arrange
    let mock_server = MockServer::start().await;
    let header_matcher = headers("cache-control", vec!["no-cache", "no-store"]);
    let mock = Mock::given(method("GET"))
        .and(header_matcher)
        .respond_with(ResponseTemplate::new(200));
    mock_server.register(mock).await;

    // Act
    let client = reqwest::Client::new();
    let should_fail_wrong_values = client
        .get(mock_server.uri())
        .header("cache-control", "no-cache, no-transform")
        .send()
        .await
        .unwrap();

    // Assert
    assert_eq!(should_fail_wrong_values.status(), 404);
}

#[async_std::test]
async fn should_not_match_multi_request_header_upon_incomplete_values() {
    // Arrange
    let mock_server = MockServer::start().await;
    let header_matcher = headers("cache-control", vec!["no-cache", "no-store"]);
    let mock = Mock::given(method("GET"))
        .and(header_matcher)
        .respond_with(ResponseTemplate::new(200));
    mock_server.register(mock).await;

    // Act
    let client = reqwest::Client::new();
    let should_fail_incomplete_values = client
        .get(mock_server.uri())
        .header("cache-control", "no-cache")
        .send()
        .await
        .unwrap();

    // Assert
    assert_eq!(should_fail_incomplete_values.status(), 404);
}

#[async_std::test]
async fn should_match_regex_single_header_value() {
    // Arrange
    let mock_server = MockServer::start().await;
    let mock = Mock::given(method("GET"))
        .and(header_regex("cache-control", r"no-(cache|store)"))
        .respond_with(ResponseTemplate::new(200));
    mock_server.register(mock).await;

    let client = reqwest::Client::new();

    // Act
    let should_match = client
        .get(mock_server.uri())
        .header("cache-control", "no-cache")
        .send()
        .await
        .unwrap();

    // Assert
    assert_eq!(should_match.status(), 200);
}

#[async_std::test]
async fn should_match_regex_multiple_header_values() {
    // Arrange
    let mock_server = MockServer::start().await;
    let mock = Mock::given(method("GET"))
        .and(header_regex("cache-control", r"no-(cache|store)"))
        .respond_with(ResponseTemplate::new(200));
    mock_server.register(mock).await;

    let client = reqwest::Client::new();

    // Act
    let should_match = client
        .get(mock_server.uri())
        .header("cache-control", "no-cache")
        .header("cache-control", "no-store")
        .send()
        .await
        .unwrap();

    // Assert
    assert_eq!(should_match.status(), 200);
}

#[async_std::test]
async fn should_not_match_regex_with_wrong_header_value() {
    // Arrange
    let mock_server = MockServer::start().await;
    let mock = Mock::given(method("GET"))
        .and(header_regex("cache-control", r"no-(cache|store)"))
        .respond_with(ResponseTemplate::new(200));
    mock_server.register(mock).await;

    let client = reqwest::Client::new();

    // Act
    let should_fail_wrong_value = client
        .get(mock_server.uri())
        .header("cache-control", "no-junk")
        .send()
        .await
        .unwrap();

    // Assert
    assert_eq!(should_fail_wrong_value.status(), 404);
}

#[async_std::test]
async fn should_not_match_regex_with_at_least_one_wrong_header_value() {
    // Arrange
    let mock_server = MockServer::start().await;
    let mock = Mock::given(method("GET"))
        .and(header_regex("cache-control", r"no-(cache|store)"))
        .respond_with(ResponseTemplate::new(200));
    mock_server.register(mock).await;

    let client = reqwest::Client::new();

    // Act
    let should_fail_wrong_value = client
        .get(mock_server.uri())
        .header("cache-control", "no-cache")
        .header("cache-control", "no-junk")
        .send()
        .await
        .unwrap();

    // Assert
    assert_eq!(should_fail_wrong_value.status(), 404);
}

#[async_std::test]
async fn should_not_match_regex_with_no_values_for_header() {
    // Arrange
    let mock_server = MockServer::start().await;
    let mock = Mock::given(method("GET"))
        .and(header_regex("cache-control", r"no-(cache|store)"))
        .respond_with(ResponseTemplate::new(200));
    mock_server.register(mock).await;

    // Act
    let should_fail_wrong_value = reqwest::get(mock_server.uri()).await.unwrap();

    // Assert
    assert_eq!(should_fail_wrong_value.status(), 404);
}

#[async_std::test]
async fn should_match_basic_auth_header() {
    // Arrange
    let mock_server = MockServer::start().await;
    let mock = Mock::given(method("GET"))
        .and(basic_auth("Aladdin", "open sesame"))
        .respond_with(ResponseTemplate::new(200));
    mock_server.register(mock).await;

    let client = reqwest::Client::new();

    // Act
    let should_match = client
        .get(mock_server.uri())
        .header("Authorization", "Basic QWxhZGRpbjpvcGVuIHNlc2FtZQ==")
        .send()
        .await
        .unwrap();
    // Assert
    assert_eq!(should_match.status(), 200);
}

#[async_std::test]
async fn should_not_match_bad_basic_auth_header() {
    // Arrange
    let mock_server = MockServer::start().await;
    let mock = Mock::given(method("GET"))
        .and(basic_auth("Aladdin", "close sesame"))
        .respond_with(ResponseTemplate::new(200));
    mock_server.register(mock).await;

    let client = reqwest::Client::new();

    // Act
    let should_not_match = client
        .get(mock_server.uri())
        .header("Authorization", "Basic QWxhZGRpbjpvcGVuIHNlc2FtZQ==")
        .send()
        .await
        .unwrap();
    // Assert
    assert_eq!(should_not_match.status(), 404);
}

#[async_std::test]
async fn should_match_bearer_token_header() {
    // Arrange
    let mock_server = MockServer::start().await;
    let mock = Mock::given(method("GET"))
        .and(bearer_token("delightful"))
        .respond_with(ResponseTemplate::new(200));
    mock_server.register(mock).await;

    let client = reqwest::Client::new();

    // Act
    let should_match = client
        .get(mock_server.uri())
        .header("Authorization", "Bearer delightful")
        .send()
        .await
        .unwrap();
    // Assert
    assert_eq!(should_match.status(), 200);
}

#[async_std::test]
async fn should_not_match_bearer_token_header() {
    // Arrange
    let mock_server = MockServer::start().await;
    let mock = Mock::given(method("GET"))
        .and(bearer_token("expired"))
        .respond_with(ResponseTemplate::new(200));
    mock_server.register(mock).await;

    let client = reqwest::Client::new();

    // Act
    let should_not_match = client
        .get(mock_server.uri())
        .header("Authorization", "Bearer delightful")
        .send()
        .await
        .unwrap();
    // Assert
    assert_eq!(should_not_match.status(), 404);
}