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(())
}