use assert_cli;

use std::env;

fn get_test_dir(final_piece: &str) -> String {
    let mut cwd = env::current_dir().unwrap();
    cwd.push("tests");
    cwd.push(final_piece);
    String::from(cwd.to_str().unwrap())
}

#[test]
fn test_dog_decode() {
    assert_cli::Assert::main_binary()
        .with_env(assert_cli::Environment::inherit().insert("FDSET_PATH", get_test_dir("fdsets")))
        .with_args(&["--msgtype=com.example.dog.Dog"])
        .stdin(include_str!("samples/dog"))
        .succeeds()
        .and()
        .stdout()
        .contains("{\"breed\":\"gsd\",\"age\":3,\"temperament\":\"excited\"}")
        .unwrap();
}

#[test]
fn test_dog_decode_stream() {
    assert_cli::Assert::main_binary()
        .with_env(assert_cli::Environment::inherit().insert("FDSET_PATH", get_test_dir("fdsets")))
        .with_args(&["--msgtype=com.example.dog.Dog", "--stream=varint"])
        .stdin(include_str!("samples/dog_stream"))
        .succeeds()
        .and()
        .stdout()
        .contains("{\"breed\":\"rottweiler\",\"age\":2,\"temperament\":\"chill\"}")
        .unwrap();
}

#[test]
fn test_dog_decode_i32be_stream() {
    assert_cli::Assert::main_binary()
        .with_env(assert_cli::Environment::inherit().insert("FDSET_PATH", get_test_dir("fdsets")))
        .with_args(&["--msgtype=com.example.dog.Dog", "--stream=i32be"])
        .stdin(include_str!("samples/dog_i32be_stream"))
        .succeeds()
        .and()
        .stdout()
        .contains("{\"breed\":\"gsd\",\"age\":3,\"temperament\":\"excited\"}")
        .unwrap();
}

#[test]
fn test_nonexistent_fdset_dir() {
    assert_cli::Assert::main_binary()
        .with_env(
            assert_cli::Environment::inherit()
                .insert("FDSET_PATH", get_test_dir("fdsets-doesnt-exist")),
        )
        .with_args(&["--msgtype=com.example.dog.Dog"])
        .stdin(include_str!("samples/dog"))
        .fails()
        .and()
        .stderr()
        .contains("No valid fdset files found. Checked:")
        .unwrap();
}

#[test]
fn test_no_fdset_files() {
    assert_cli::Assert::main_binary()
        .with_env(
            assert_cli::Environment::inherit().insert("FDSET_PATH", get_test_dir("fdsets-invalid")),
        )
        .with_args(&["--msgtype=com.example.dog.Dog"])
        .stdin(include_str!("samples/dog"))
        .fails()
        .and()
        .stderr()
        .contains("No valid fdset files found. Checked:")
        .unwrap();
}

#[test]
fn test_person_decode() {
    assert_cli::Assert::main_binary()
        .with_env(assert_cli::Environment::inherit().insert("FDSET_PATH", get_test_dir("fdsets")))
        .with_args(&["--msgtype=com.example.person.Person"])
        .stdin(include_str!("samples/person"))
        .succeeds()
        .and()
        .stdout()
        .contains("{\"name\":\"khosrov\",\"id\":0}")
        .unwrap();
}

#[test]
fn test_bad_input() {
    assert_cli::Assert::main_binary()
        .with_env(assert_cli::Environment::inherit().insert("FDSET_PATH", get_test_dir("fdsets")))
        .with_args(&["--msgtype=com.example.dog.Dog"])
        .stdin(include_str!("samples/bad"))
        .fails()
        .and()
        .stderr()
        .contains("protobuf error")
        .unwrap();
}

#[test]
fn test_person_decode_with_command_line_fdset_dir() {
    assert_cli::Assert::main_binary()
        .with_args(&[
            "--msgtype=com.example.person.Person",
            &format!("--fdsetdir={0}", get_test_dir("fdsets")),
        ])
        .stdin(include_str!("samples/person"))
        .succeeds()
        .and()
        .stdout()
        .contains("{\"name\":\"khosrov\",\"id\":0}")
        .unwrap();
}

#[test]
fn test_person_decode_with_command_line_fdset_file() {
    assert_cli::Assert::main_binary()
        .with_args(&[
            "--msgtype=com.example.person.Person",
            &format!("--fdsetfile={0}", get_test_dir("fdsets/person.fdset")),
        ])
        .stdin(include_str!("samples/person"))
        .succeeds()
        .and()
        .stdout()
        .contains("{\"name\":\"khosrov\",\"id\":0}")
        .unwrap();
}

#[test]
fn test_no_args() {
    assert_cli::Assert::main_binary()
        .fails()
        .and()
        .stderr()
        .contains("No valid fdset files found")
        .unwrap();
}

#[test]
fn test_cat_noncanonical_decode() {
    assert_cli::Assert::main_binary()
        .with_args(&[
            "--msgtype=com.example.cat.Cat",
            &format!("--fdsetfile={0}", get_test_dir("fdsets/cat.fdset")),
        ])
        .stdin(include_str!("samples/cat"))
        .succeeds()
        .and()
        .stdout()
        .contains("{\"is_lazy\":false}")
        .unwrap();
}

#[test]
#[ignore]
fn test_cat_canonical_decode() {
    assert_cli::Assert::main_binary()
        .with_args(&[
            "--msgtype=com.example.cat.Cat",
            &format!("--fdsetfile={0}", get_test_dir("fdsets/cat.fdset")),
        ])
        .stdin(include_str!("samples/cat"))
        .succeeds()
        .and()
        .stdout()
        .contains("{\"isLazy\":false}")
        .unwrap();
}

#[test]
fn test_dog_decode_from_proto() {
    assert_cli::Assert::main_binary()
        .with_env(
            assert_cli::Environment::inherit().insert("PROTOC_INCLUDE", get_test_dir("protos")),
        )
        .with_args(&[
            "--msgtype=com.example.dog.Dog",
            &format!("--protofile={0}", get_test_dir("protos/dog.proto")),
        ])
        .stdin(include_str!("samples/dog"))
        .succeeds()
        .and()
        .stdout()
        .contains("{\"breed\":\"gsd\",\"age\":3,\"temperament\":\"excited\"}")
        .unwrap();
}