nu-command 0.113.0

Nushell's built-in commands
Documentation
use std::{thread, time::Duration};

use mockito::Server;
use nu_protocol::shell_error;
use nu_test_support::prelude::*;

#[test]
fn http_put_is_success() -> Result {
    let mut server = Server::new();
    let _mock = server.mock("PUT", "/").match_body("foo").create();
    let code = r#"let url = $in; http put $url "foo""#;
    test().run_with_data(code, server.url()).expect_value_eq("")
}

#[test]
fn http_put_is_success_pipeline() -> Result {
    let mut server = Server::new();
    let _mock = server.mock("PUT", "/").match_body("foo").create();
    let code = r#"let url = $in; "foo" | http put $url "#;
    test().run_with_data(code, server.url()).expect_value_eq("")
}

#[test]
fn http_put_failed_due_to_server_error() -> Result {
    let mut server = Server::new();

    let _mock = server.mock("PUT", "/").with_status(400).create();

    let code = r#"let url = $in; http put $url "body""#;
    let err = test()
        .run_with_data(code, server.url())
        .expect_shell_error()?;
    match err {
        ShellError::NetworkFailure { msg, .. } => {
            assert_contains("Bad request (400)", msg);
            Ok(())
        }
        err => Err(err.into()),
    }
}

#[test]
fn http_put_failed_due_to_missing_body() -> Result {
    let mut server = Server::new();

    let _mock = server.mock("PUT", "/").create();

    let code = "let url = $in; http put $url";
    let err = test()
        .run_with_data(code, server.url())
        .expect_shell_error()?
        .generic_error()?;
    assert_eq!(
        err,
        "Data must be provided either through pipeline or positional argument"
    );
    Ok(())
}

#[test]
fn http_put_failed_due_to_unexpected_body() -> Result {
    let mut server = Server::new();

    let _mock = server.mock("PUT", "/").match_body("foo").create();

    let code = r#"let url = $in; http put $url "bar""#;
    let err = test()
        .run_with_data(code, server.url())
        .expect_shell_error()?;

    match err {
        ShellError::NetworkFailure { msg, .. } => {
            assert_contains("Cannot make request", msg);
            Ok(())
        }
        err => Err(err.into()),
    }
}

#[test]
fn http_put_follows_redirect() -> Result {
    let mut server = Server::new();

    let _mock = server.mock("GET", "/bar").with_body("bar").create();
    let _mock = server
        .mock("PUT", "/foo")
        .with_status(301)
        .with_header("Location", "/bar")
        .create();

    let code = "let url = $in; http put $'($url)/foo' putbody";
    test()
        .run_with_data(code, server.url())
        .expect_value_eq("bar")
}

#[test]
fn http_put_redirect_mode_manual() -> Result {
    let mut server = Server::new();

    let _mock = server
        .mock("PUT", "/foo")
        .with_status(301)
        .with_body("foo")
        .with_header("Location", "/bar")
        .create();

    let code = "let url = $in; http put --redirect-mode manual $'($url)/foo' putbody";
    test()
        .run_with_data(code, server.url())
        .expect_value_eq("foo")
}

#[test]
fn http_put_redirect_mode_error() -> Result {
    let mut server = Server::new();

    let _mock = server
        .mock("PUT", "/foo")
        .with_status(301)
        .with_body("foo")
        .with_header("Location", "/bar")
        .create();

    let code = "let url = $in; http put --redirect-mode error $'($url)/foo' putbody";

    let err = test()
        .run_with_data(code, server.url())
        .expect_shell_error()?;
    match err {
        ShellError::NetworkFailure { msg, .. } => {
            assert_eq!(
                msg,
                "Redirect encountered when redirect handling mode was 'error' (301 Moved Permanently)"
            );
            Ok(())
        }
        err => Err(err.into()),
    }
}

#[test]
fn http_put_timeout() -> Result {
    let mut server = Server::new();
    let _mock = server
        .mock("PUT", "/")
        .with_chunked_body(|w| {
            thread::sleep(Duration::from_secs(10));
            w.write_all(b"Delayed response!")
        })
        .create();

    let code = "let url = $in; http put --max-time 100ms $url putbody";
    let err = test().run_with_data(code, server.url()).expect_io_error()?;
    assert!(matches!(
        err.kind,
        shell_error::io::ErrorKind::Std(std::io::ErrorKind::TimedOut, ..)
    ));
    Ok(())
}