use super::*;

#[test]
fn no_test_rejects_non_close_open_reply() {
    let mut config = default_test_config();
    config.request.run_mode = RunMode::NoTest;
    let params = default_params();
    let server = start_fake_server(move |socket, tx| {
        let (_, peer) = recv_request(&socket, &tx);
        socket
            .send_to(
                &open_reply(FLAG_OPEN | FLAG_REPLY, TOKEN, &params, None),
                peer,
            )
            .unwrap();
        let _ = recv_request(&socket, &tx);
    });
    let mut client = Client::connect(server.addr.to_string(), config).unwrap();
    assert!(matches!(
        client.open(),
        Err(ClientError::UnexpectedNoTestReply)
    ));
    let packets: Vec<_> = server.rx.iter().take(2).collect();
    assert_eq!(packets[1][3], flags::FLAG_CLOSE);
    assert_eq!(
        u64::from_le_bytes(packets[1][4..12].try_into().unwrap()),
        TOKEN
    );
    server.join();
}

#[test]
fn no_test_rejects_non_zero_token_with_close_reply() {
    let mut config = default_test_config();
    config.request.run_mode = RunMode::NoTest;
    let params = default_params();
    let server = no_test_server(params, TOKEN);
    let mut client = Client::connect(server.addr.to_string(), config).unwrap();
    assert!(matches!(
        client.open(),
        Err(ClientError::NonZeroNoTestToken { token: TOKEN })
    ));
    server.join();
}

#[test]
fn no_test_strict_negotiation_rejects_changed_params() {
    let mut config = default_test_config();
    config.request.run_mode = RunMode::NoTest;
    let mut params = default_params();
    params.dscp = 1;
    let server = no_test_server(params, 0);
    let mut client = Client::connect(server.addr.to_string(), config).unwrap();
    assert!(matches!(
        client.open(),
        Err(ClientError::NegotiationRejected { .. })
    ));
    server.join();
}

#[test]
fn no_test_loose_negotiation_accepts_restricted_params() {
    let mut config = default_test_config();
    config.request.run_mode = RunMode::NoTest;
    config.open.negotiation = NegotiationPolicy::Loose;
    let requested = default_params();
    let mut params = requested.clone();
    params.duration_ns /= 2;
    let server = no_test_server(params.clone(), 0);
    let mut client = Client::connect(server.addr.to_string(), config).unwrap();
    let negotiated = assert_no_test_completed(client.open().unwrap());
    assert_eq!(negotiated.params, params);
    assert_eq!(
        negotiated.restrictions,
        vec![irtt_client::NegotiationRestriction::DurationReduced {
            requested_ns: requested.duration_ns,
            negotiated_ns: params.duration_ns,
        }]
    );
    server.join();
}

#[test]
fn send_probe_fails_after_no_test_completed() {
    let mut config = default_test_config();
    config.request.run_mode = RunMode::NoTest;
    let params = default_params();
    let server = no_test_server(params, 0);
    let mut client = Client::connect(server.addr.to_string(), config).unwrap();
    assert_no_test_completed(client.open().unwrap());
    assert!(matches!(
        client.send_probe(),
        Err(ClientError::AlreadyCompleted)
    ));
    server.join();
}

#[test]
fn operations_fail_after_no_test_completed_without_datagrams() {
    let mut config = default_test_config();
    config.request.run_mode = RunMode::NoTest;

    let params = default_params();
    let server = no_test_server(params, 0);
    let mut client = Client::connect(server.addr.to_string(), config).unwrap();
    client
        .set_recv_timeout(Some(Duration::from_millis(50)))
        .unwrap();
    assert_no_test_completed(client.open().unwrap());
    assert!(matches!(client.open(), Err(ClientError::AlreadyCompleted)));
    let results = [
        client.recv_once(),
        client.recv_available(RecvBudget { max_packets: 1 }),
    ];
    assert!(
        matches!(
            results,
            [
                Err(ClientError::AlreadyCompleted),
                Err(ClientError::AlreadyCompleted)
            ]
        ),
        "unexpected receive results: {results:?}"
    );
    server.join();
}