goose 0.19.1

//! Goose load test example demonstrating different approaches to HTTP Basic Authentication.
//!
//! This example shows multiple ways to handle Basic Auth in Goose load tests:
//! 1. Per-request basic auth using request builders
//! 2. Using a helper function for consistent auth across requests
//! 3. Setting up a custom client with default Authorization headers (recommended)
//!
//! ## Usage
//!
//! Set environment variables for authentication:
//! ```bash
//! export BASIC_AUTH_USERNAME="your_username"
//! export BASIC_AUTH_PASSWORD="your_password"
//! ```
//!
//! Or use the combined format:
//! ```bash
//! export BASIC_AUTH="username:password"
//! ```
//!
//! Run the example:
//! ```bash
//! cargo run --example basic_auth -- --host http://httpbin.org
//! ```
//!
//! ## License
//!
//! Copyright 2020-2025 Jeremy Andrews
//!
//! Licensed under the Apache License, Version 2.0 (the "License");
//! you may not use this file except in compliance with the License.
//! You may obtain a copy of the License at
//!
//! <http://www.apache.org/licenses/LICENSE-2.0>
//!
//! Unless required by applicable law or agreed to in writing, software
//! distributed under the License is distributed on an "AS IS" BASIS,
//! WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//! See the License for the specific language governing permissions and
//! limitations under the License.

#[cfg(not(feature = "cookies"))]
fn main() {
    eprintln!("ERROR: The basic_auth example requires the 'cookies' feature to be enabled.");
    eprintln!("This example demonstrates session management with Basic Authentication,");
    eprintln!("which requires cookie support to maintain authentication state across requests.");
    eprintln!();
    eprintln!("To run this example, use:");
    eprintln!("  cargo run --example basic_auth --features cookies");
    eprintln!("or:");
    eprintln!("  cargo run --example basic_auth  # (cookies enabled by default)");
    std::process::exit(1);
}

#[cfg(feature = "cookies")]
mod basic_auth_example {
    use goose::goose::GooseResponse;
    use goose::prelude::*;
    use log::{info, warn};
    use reqwest::{header, Client};
    use std::{env, time::Duration};

    #[tokio::main]
    pub async fn main() -> Result<(), GooseError> {
        GooseAttack::initialize()?
            // Scenario 1: Using custom client with default headers (recommended approach)
            .register_scenario(
                scenario!("CustomClientAuth")
                    .set_weight(3)?
                    .set_wait_time(Duration::from_secs(1), Duration::from_secs(3))?
                    // Set up custom client with Basic Auth headers on user start
                    .register_transaction(transaction!(setup_basic_auth_client).set_on_start())
                    // All subsequent requests will automatically include Basic Auth
                    .register_transaction(transaction!(test_basic_auth_endpoint))
                    .register_transaction(transaction!(test_user_agent_endpoint))
                    .register_transaction(transaction!(test_json_endpoint)),
            )
            // Scenario 2: Using helper function for per-request auth
            .register_scenario(
                scenario!("HelperFunctionAuth")
                    .set_weight(2)?
                    .set_wait_time(Duration::from_secs(1), Duration::from_secs(3))?
                    .register_transaction(transaction!(test_with_helper_function))
                    .register_transaction(transaction!(test_post_with_helper)),
            )
            // Scenario 3: Manual per-request auth (demonstrates the original problem)
            .register_scenario(
                scenario!("ManualPerRequestAuth")
                    .set_weight(1)?
                    .set_wait_time(Duration::from_secs(1), Duration::from_secs(3))?
                    .register_transaction(transaction!(test_manual_basic_auth))
                    .register_transaction(transaction!(test_manual_with_validation)),
            )
            .execute()
            .await?;

        Ok(())
    }

    /// Approach 1: Set up a custom client with Basic Auth headers (RECOMMENDED)
    ///
    /// This approach sets up the HTTP client once per user with default Basic Auth headers.
    /// All subsequent requests will automatically include the authentication headers,
    /// including requests for static assets (CSS, JS, images) when using
    /// `validate_and_load_static_assets`.
    ///
    /// By setting up authentication at the client level, we solve the static asset
    /// authentication problem. When Goose loads a page with CSS, JS, or image files,
    /// those secondary requests will automatically inherit the Basic Auth headers from the
    /// custom client configuration, unlike manual per-request authentication approaches.
    async fn setup_basic_auth_client(user: &mut GooseUser) -> TransactionResult {
        // Get credentials from environment variables
        let (username, password) = get_basic_auth_credentials()?;

        // Use reqwest's built-in basic_auth() to generate a properly encoded
        // Authorization header, then extract it for use as a default header.
        let temp_request = reqwest::Client::new()
            .get("http://localhost")
            .basic_auth(&username, Some(&password))
            .build()
            .map_err(TransactionError::Reqwest)?;

        let mut headers = header::HeaderMap::new();
        if let Some(auth_header) = temp_request.headers().get(header::AUTHORIZATION) {
            headers.insert(header::AUTHORIZATION, auth_header.clone());
        }

        // Set up the client builder with default headers and other common settings
        // Enable cookie store for maintaining session state with Basic Auth
        let builder = Client::builder()
            .user_agent("Goose/1.0 BasicAuth Example")
            .default_headers(headers)
            .timeout(Duration::from_secs(30))
            .cookie_store(true);

        // Apply the custom client to this user
        user.set_client_builder(builder).await?;

        Ok(())
    }

    /// Test endpoint that requires Basic Auth - using custom client approach
    async fn test_basic_auth_endpoint(user: &mut GooseUser) -> TransactionResult {
        // Since we set up the client with default headers, this request will
        // automatically include the Basic Auth header
        let goose_metrics = user.get("/basic-auth/user/passwd").await?;

        // Since we're using httpbin.org for testing, we can validate the response
        if let Ok(response) = &goose_metrics.response {
            if response.status() == 200 {
                // Success! The Basic Auth worked
                info!("Basic Auth successful!");
            }
        }

        Ok(())
    }

    /// Test user-agent endpoint to show other headers work too
    async fn test_user_agent_endpoint(user: &mut GooseUser) -> TransactionResult {
        let _goose_metrics = user.get("/user-agent").await?;
        Ok(())
    }

    /// Test JSON endpoint with Basic Auth
    async fn test_json_endpoint(user: &mut GooseUser) -> TransactionResult {
        let _goose_metrics = user.get("/json").await?;
        Ok(())
    }

    /// Approach 2: Helper function for consistent Basic Auth across requests
    ///
    /// This approach uses a helper function that adds Basic Auth to each request.
    /// It's more flexible than the custom client approach but requires calling
    /// the helper for every request.
    async fn send_authenticated_request(
        user: &mut GooseUser,
        method: &GooseMethod,
        path: &str,
    ) -> Result<GooseResponse, Box<TransactionError>> {
        let mut reqwest_request_builder = user.get_request_builder(method, path)?;

        // Check for Basic Auth credentials in environment
        if let Ok((username, password)) = get_basic_auth_credentials() {
            reqwest_request_builder = reqwest_request_builder.basic_auth(username, Some(password));
        }

        // Build the GooseRequest and make the request. The path names the request in
        // the metrics, as the RequestBuilder already has its URL.
        let goose_request = GooseRequest::builder()
            .path(path)
            .set_request_builder(reqwest_request_builder)
            .build();

        let goose_response = user.request(goose_request).await?;
        Ok(goose_response)
    }

    /// Test using the helper function approach
    async fn test_with_helper_function(user: &mut GooseUser) -> TransactionResult {
        let _goose_response =
            send_authenticated_request(user, &GooseMethod::Get, "/basic-auth/user/passwd").await?;
        Ok(())
    }

    /// Test POST request using helper function
    async fn test_post_with_helper(user: &mut GooseUser) -> TransactionResult {
        let _goose_response = send_authenticated_request(user, &GooseMethod::Post, "/post").await?;
        Ok(())
    }

    /// Approach 3: Manual per-request Basic Auth (demonstrates the original problem)
    ///
    /// This approach manually adds Basic Auth to each individual request.
    /// Note: This does NOT propagate to static asset requests, which is the
    /// original problem described in issue #608.
    async fn test_manual_basic_auth(user: &mut GooseUser) -> TransactionResult {
        let (username, password) = get_basic_auth_credentials()?;

        let reqwest_request_builder = user
            .get_request_builder(&GooseMethod::Get, "/basic-auth/user/passwd")?
            .basic_auth(username, Some(password));

        let goose_request = GooseRequest::builder()
            .path("/basic-auth/user/passwd")
            .set_request_builder(reqwest_request_builder)
            .build();

        let _goose_metrics = user.request(goose_request).await?;

        Ok(())
    }

    /// Manual approach with validation (shows the static asset problem)
    async fn test_manual_with_validation(user: &mut GooseUser) -> TransactionResult {
        let (username, password) = get_basic_auth_credentials()?;

        let reqwest_request_builder = user
            .get_request_builder(&GooseMethod::Get, "/basic-auth/user/passwd")?
            .basic_auth(username, Some(password));

        let goose_request = GooseRequest::builder()
            .path("/basic-auth/user/passwd")
            .set_request_builder(reqwest_request_builder)
            .build();

        let goose_metrics = user.request(goose_request).await?;

        // WARNING: If this page had static assets (CSS, JS, images), they would
        // fail to load because they won't have the Basic Auth header!
        // This is the problem that the custom client approach solves.
        if let Ok(response) = &goose_metrics.response {
            if response.status() == 200 {
                warn!("Manual Basic Auth successful, but static assets would fail!");
            }
        }

        Ok(())
    }

    /// Helper function to get Basic Auth credentials from environment variables
    ///
    /// Supports two formats:
    /// 1. BASIC_AUTH_USERNAME and BASIC_AUTH_PASSWORD (separate variables)
    /// 2. BASIC_AUTH in "username:password" format (combined variable)
    fn get_basic_auth_credentials() -> Result<(String, String), Box<TransactionError>> {
        // Try the separate username/password format first
        if let (Ok(username), Ok(password)) = (
            env::var("BASIC_AUTH_USERNAME"),
            env::var("BASIC_AUTH_PASSWORD"),
        ) {
            return Ok((username, password));
        }

        // Try the combined format
        if let Ok(basic_auth) = env::var("BASIC_AUTH") {
            // Use splitn(2, ':') instead of split(':') to handle passwords containing colons
            // splitn(2, ':') splits into at most 2 parts: username and everything after the first colon
            // This prevents passwords like "my:complex:password" from being incorrectly split
            // Example: "user:pass:word" -> ["user", "pass:word"] (correct)
            // vs split(':') -> ["user", "pass", "word"] (incorrect - would truncate password)
            let parts: Vec<&str> = basic_auth.splitn(2, ':').collect();
            if parts.len() == 2 {
                return Ok((parts[0].to_string(), parts[1].to_string()));
            } else {
                return Err(Box::new(TransactionError::Custom(format!(
                    "BASIC_AUTH must be in format 'username:password', got: {}",
                    basic_auth
                ))));
            }
        }

        // Default credentials for testing with httpbin.org
        warn!("No Basic Auth credentials found in environment variables.");
        warn!("Using default credentials 'user:passwd' for httpbin.org testing.");
        warn!("Set BASIC_AUTH_USERNAME/BASIC_AUTH_PASSWORD or BASIC_AUTH environment variables for real usage.");

        Ok(("user".to_string(), "passwd".to_string()))
    }
}

#[cfg(feature = "cookies")]
use basic_auth_example::main;