skallwar.suckit/tests/auth.rs

//! Tests for using --auth flags for suckit

mod fixtures;

use std::fs::read_dir;
use std::process::Command;
use std::process::Stdio;
use std::sync::Once;

const PAGE: &'static str = "tests/fixtures/index.html";
const IP: &'static str = "0.0.0.0";
static START: Once = Once::new();

#[test]
fn test_auth() {
    // Spawn a single instance of a local http server usable by all tests in this module.
    START.call_once(|| {
        fixtures::spawn_local_http_server(PAGE, true, None);
    });

    // Tests below are grouped together as they depend on the local_http_server above.
    auth_different_host();
    auth_valid();
}

// Shouldn't supply credentials to a non-matching host
fn auth_different_host() {
    let output_dir = "w4";
    let mut cmd = Command::new(env!("CARGO_BIN_EXE_suckit"))
        .args(&[
            fixtures::HTTP_ADDR,
            "-o",
            "w4",
            "-a",
            "username password example.com",
            "-j",
            "16",
        ])
        .stdout(Stdio::inherit())
        .stderr(Stdio::inherit())
        .spawn()
        .unwrap();

    let status = cmd.wait().unwrap();
    assert!(status.success());
    let paths = read_dir(format!("{}/{}", output_dir, IP)).unwrap();
    // Only the initial invalid response file should be present
    assert_eq!(paths.count(), 1);

    std::fs::remove_dir_all(output_dir).unwrap();
}

// Should authenticate with credentials to host (defaulting to origin host)
fn auth_valid() {
    let output_dir = "w5";
    let mut cmd = Command::new(env!("CARGO_BIN_EXE_suckit"))
        .args(&[
            fixtures::HTTP_ADDR,
            "-o",
            "w5",
            "-a",
            "username password",
            "-j",
            "16",
        ])
        .stdout(Stdio::inherit())
        .stderr(Stdio::inherit())
        .spawn()
        .unwrap();

    let status = cmd.wait().unwrap();
    assert!(status.success());
    let paths = read_dir(format!("{}/{}", output_dir, IP)).unwrap();
    // Should load multiple paths, not just the invalid auth response
    let paths_count = paths.count();
    println!("Paths.count() = {}", paths_count);
    assert!(paths_count > 1);

    std::fs::remove_dir_all(output_dir).unwrap();
}