Function parser_parses_authentication_messages

src/protocol/tests.rs:353–422

()

#[test]
fn parser_parses_authentication_messages() -> Result<()> {
    match parse_single(buffers::authentication_ok())? {
        BackendMessage::Authentication(AuthenticationMessage::Ok(msg)) => {
            assert_eq!(msg.length, 8);
        }
        other => panic!("unexpected message: {:?}", other.name()),
    }

    match parse_single(buffers::authentication_cleartext_password())? {
        BackendMessage::Authentication(AuthenticationMessage::Cleartext(msg)) => {
            assert_eq!(msg.length, 8);
        }
        other => panic!("unexpected message: {:?}", other.name()),
    }

    match parse_single(buffers::authentication_md5_password())? {
        BackendMessage::Authentication(AuthenticationMessage::Md5(msg)) => {
            assert_eq!(msg.length, 12);
            assert_eq!(msg.salt, vec![1, 2, 3, 4]);
        }
        other => panic!("unexpected message: {:?}", other.name()),
    }

    match parse_single(buffers::authentication_sasl())? {
        BackendMessage::Authentication(AuthenticationMessage::Sasl(msg)) => {
            assert_eq!(msg.length, buffers::authentication_sasl().len() - 1);
            assert_eq!(msg.mechanisms, vec!["SCRAM-SHA-256".to_string()]);
        }
        other => panic!("unexpected message: {:?}", other.name()),
    }

    match parse_single(buffers::authentication_sasl_continue())? {
        BackendMessage::Authentication(AuthenticationMessage::SaslContinue(msg)) => {
            assert_eq!(
                msg.length,
                buffers::authentication_sasl_continue().len() - 1
            );
            assert_eq!(msg.data, "data");
        }
        other => panic!("unexpected message: {:?}", other.name()),
    }

    let mut extended_continue = buffers::authentication_sasl_continue();
    extended_continue.extend_from_slice(&[1, 2, 3, 4]);
    match parse_single(extended_continue)? {
        BackendMessage::Authentication(AuthenticationMessage::SaslContinue(msg)) => {
            assert_eq!(msg.data, "data");
        }
        other => panic!("unexpected message: {:?}", other.name()),
    }

    match parse_single(buffers::authentication_sasl_final())? {
        BackendMessage::Authentication(AuthenticationMessage::SaslFinal(msg)) => {
            assert_eq!(msg.length, buffers::authentication_sasl_final().len() - 1);
            assert_eq!(msg.data, "data");
        }
        other => panic!("unexpected message: {:?}", other.name()),
    }