mirror of
https://github.com/rust-lang/rust.git
synced 2025-02-21 11:23:03 +00:00
Fix assertion message generation
This commit is contained in:
parent
f9bfe840f4
commit
39df2231bb
@ -656,7 +656,7 @@ pub trait PrintState<'a> {
|
|||||||
style: ast::StrStyle) -> io::Result<()> {
|
style: ast::StrStyle) -> io::Result<()> {
|
||||||
let st = match style {
|
let st = match style {
|
||||||
ast::StrStyle::Cooked => {
|
ast::StrStyle::Cooked => {
|
||||||
(format!("\"{}\"", st.escape_default()))
|
(format!("\"{}\"", st.escape_debug()))
|
||||||
}
|
}
|
||||||
ast::StrStyle::Raw(n) => {
|
ast::StrStyle::Raw(n) => {
|
||||||
(format!("r{delim}\"{string}\"{delim}",
|
(format!("r{delim}\"{string}\"{delim}",
|
||||||
|
@ -42,16 +42,13 @@ pub fn expand_assert<'cx>(
|
|||||||
tts: if let Some(ts) = custom_msg_args {
|
tts: if let Some(ts) = custom_msg_args {
|
||||||
ts.into()
|
ts.into()
|
||||||
} else {
|
} else {
|
||||||
// `expr_to_string` escapes the string literals with `.escape_default()`
|
|
||||||
// which escapes all non-ASCII characters with `\u`.
|
|
||||||
let escaped_expr = escape_format_string(&unescape_printable_unicode(
|
|
||||||
&pprust::expr_to_string(&cond_expr),
|
|
||||||
));
|
|
||||||
|
|
||||||
TokenStream::from(TokenTree::Token(
|
TokenStream::from(TokenTree::Token(
|
||||||
DUMMY_SP,
|
DUMMY_SP,
|
||||||
token::Literal(
|
token::Literal(
|
||||||
token::Lit::Str_(Name::intern(&format!("assertion failed: {}", escaped_expr))),
|
token::Lit::Str_(Name::intern(&format!(
|
||||||
|
"assertion failed: {}",
|
||||||
|
pprust::expr_to_string(&cond_expr).escape_debug()
|
||||||
|
))),
|
||||||
None,
|
None,
|
||||||
),
|
),
|
||||||
)).into()
|
)).into()
|
||||||
@ -71,53 +68,3 @@ pub fn expand_assert<'cx>(
|
|||||||
);
|
);
|
||||||
MacEager::expr(if_expr)
|
MacEager::expr(if_expr)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Escapes a string for use as a formatting string.
|
|
||||||
fn escape_format_string(s: &str) -> String {
|
|
||||||
let mut res = String::with_capacity(s.len());
|
|
||||||
for c in s.chars() {
|
|
||||||
res.extend(c.escape_debug());
|
|
||||||
match c {
|
|
||||||
'{' | '}' => res.push(c),
|
|
||||||
_ => {}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
res
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_escape_format_string() {
|
|
||||||
assert!(escape_format_string(r"foo{}\") == r"foo{{}}\\");
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Unescapes the escaped unicodes (`\u{...}`) that are printable.
|
|
||||||
fn unescape_printable_unicode(mut s: &str) -> String {
|
|
||||||
use std::{char, u32};
|
|
||||||
|
|
||||||
let mut res = String::with_capacity(s.len());
|
|
||||||
|
|
||||||
loop {
|
|
||||||
if let Some(start) = s.find(r"\u{") {
|
|
||||||
res.push_str(&s[0..start]);
|
|
||||||
s = &s[start..];
|
|
||||||
s.find('}')
|
|
||||||
.and_then(|end| {
|
|
||||||
let v = u32::from_str_radix(&s[3..end], 16).ok()?;
|
|
||||||
let c = char::from_u32(v)?;
|
|
||||||
// Escape unprintable characters.
|
|
||||||
res.extend(c.escape_debug());
|
|
||||||
s = &s[end + 1..];
|
|
||||||
Some(())
|
|
||||||
})
|
|
||||||
.expect("lexer should have rejected invalid escape sequences");
|
|
||||||
} else {
|
|
||||||
res.push_str(s);
|
|
||||||
return res;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_unescape_printable_unicode() {
|
|
||||||
assert!(unescape_printable_unicode(r"\u{2603}\n\u{0}") == r"☃\n\u{0}");
|
|
||||||
}
|
|
||||||
|
19
src/test/compile-fail/issue-50471.rs
Normal file
19
src/test/compile-fail/issue-50471.rs
Normal file
@ -0,0 +1,19 @@
|
|||||||
|
// Copyright 2018 The Rust Project Developers. See the COPYRIGHT
|
||||||
|
// file at the top-level directory of this distribution and at
|
||||||
|
// http://rust-lang.org/COPYRIGHT.
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
||||||
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
||||||
|
// option. This file may not be copied, modified, or distributed
|
||||||
|
// except according to those terms.
|
||||||
|
|
||||||
|
// compile-pass
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
assert!({false});
|
||||||
|
|
||||||
|
assert!(r"\u{41}" == "A");
|
||||||
|
|
||||||
|
assert!(r"\u{".is_empty());
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user