Function test_fuzz_authentication_messages

tests/protocol_fuzzing_test.rs:60–94

()

#[tokio::test]
async fn test_fuzz_authentication_messages() {
    let inputs = generate_auth_message_fuzzing();

    for input in inputs {
        let result = parse_auth_message_safely(&input).await;

        match result {
            Ok(auth_req) => {
                // Valid auth request should have reasonable parameters
                match auth_req {
                    AuthenticationRequest::Ok => {},
                    AuthenticationRequest::Password => {},
                    AuthenticationRequest::MD5Password { salt } => {
                        assert_eq!(salt.len(), 4);
                    },
                    AuthenticationRequest::SASL { mechanisms } => {
                        assert!(mechanisms.len() <= 10);
                        for mechanism in mechanisms {
                            assert!(mechanism.len() <= 100);
                        }
                    },
                    AuthenticationRequest::SASLContinue { data } => {
                        assert!(data.len() <= 4096);
                    },
                    AuthenticationRequest::SASLFinal { data } => {
                        assert!(data.len() <= 4096);
                    },
                }
            }
            Err(_) => {
                // Error handling is acceptable
            }
        }
    }
}

#[tokio::test]
async fn test_fuzz_query_messages() {