diff --git a/server/src/engine/ql/tests/lexer_tests.rs b/server/src/engine/ql/tests/lexer_tests.rs index 8446cf15..17ddee5e 100644 --- a/server/src/engine/ql/tests/lexer_tests.rs +++ b/server/src/engine/ql/tests/lexer_tests.rs @@ -294,26 +294,26 @@ fn safe_query_float() { #[test] fn safe_query_binary() { - let (query, query_window) = make_safe_query(b"?", SFQ_BINARY); - let binary = lex_secure(&query, query_window).unwrap(); - assert_eq!( - binary, - vec![Token::Lit(Lit::new_bin( - "cringeπŸ˜ƒπŸ˜„πŸ˜πŸ˜†πŸ˜…πŸ˜‚πŸ€£πŸ˜ŠπŸ˜ΈπŸ˜Ί".as_bytes() - ))] - ); + for (test_payload_string, expected) in [ + (b"\x050\n".as_slice(), b"".as_slice()), + (SFQ_BINARY, "cringeπŸ˜ƒπŸ˜„πŸ˜πŸ˜†πŸ˜…πŸ˜‚πŸ€£πŸ˜ŠπŸ˜ΈπŸ˜Ί".as_bytes()), + ] { + let (query, query_window) = make_safe_query(b"?", test_payload_string); + let binary = lex_secure(&query, query_window).unwrap(); + assert_eq!(binary, vec![Token::Lit(Lit::new_bin(expected))]); + } } #[test] fn safe_query_string() { - let (query, query_window) = make_safe_query(b"?", SFQ_STRING); - let binary = lex_secure(&query, query_window).unwrap(); - assert_eq!( - binary, - vec![Token::Lit(Lit::new_string( - "cringeπŸ˜ƒπŸ˜„πŸ˜πŸ˜†πŸ˜…πŸ˜‚πŸ€£πŸ˜ŠπŸ˜ΈπŸ˜Ί".to_owned().into() - ))] - ); + for (test_payload_string, expected) in [ + (b"\x060\n".as_slice(), ""), + (SFQ_STRING, "cringeπŸ˜ƒπŸ˜„πŸ˜πŸ˜†πŸ˜…πŸ˜‚πŸ€£πŸ˜ŠπŸ˜ΈπŸ˜Ί"), + ] { + let (query, query_window) = make_safe_query(b"?", test_payload_string); + let binary = lex_secure(&query, query_window).unwrap(); + assert_eq!(binary, vec![Token::Lit(Lit::new_str(expected))]); + } } #[test]