Alex Crichton | ccbb45d | 2017-05-23 10:58:24 -0700 | [diff] [blame] | 1 | #![cfg(feature = "extra-traits")] |
| 2 | |
Arnavion | 02ef13f | 2017-04-25 00:54:31 -0700 | [diff] [blame] | 3 | extern crate syn; |
| 4 | use syn::*; |
| 5 | |
| 6 | macro_rules! assert_let { |
| 7 | ($p:pat = $e:expr) => { |
| 8 | assert_let!($p = $e; {}) |
| 9 | }; |
| 10 | |
| 11 | ($p:pat = $e:expr; $body:block) => { |
| 12 | if let $p = $e |
| 13 | $body |
| 14 | else { |
| 15 | panic!("Expected to match {} but got {:?}", stringify!($p), $e) |
| 16 | } |
| 17 | }; |
| 18 | } |
| 19 | |
| 20 | #[test] |
| 21 | #[cfg(feature = "full")] |
| 22 | fn test_catch_expr() { |
| 23 | // Taken from tests/rust/src/test/run-pass/catch-expr.rs |
| 24 | let raw = r#" |
| 25 | struct catch {} |
| 26 | |
| 27 | pub fn main() { |
| 28 | let catch_result = do catch { |
| 29 | let x = 5; |
| 30 | x |
| 31 | }; |
| 32 | assert_eq!(catch_result, 5); |
| 33 | |
| 34 | let mut catch = true; |
| 35 | while catch { catch = false; } |
| 36 | assert_eq!(catch, false); |
| 37 | |
| 38 | catch = if catch { false } else { true }; |
| 39 | assert_eq!(catch, true); |
| 40 | |
| 41 | match catch { |
| 42 | _ => {} |
| 43 | }; |
| 44 | |
| 45 | let catch_err = do catch { |
| 46 | Err(22)?; |
| 47 | Ok(1) |
| 48 | }; |
| 49 | assert_eq!(catch_err, Err(22)); |
| 50 | |
| 51 | let catch_okay: Result<i32, i32> = do catch { |
| 52 | if false { Err(25)?; } |
| 53 | Ok::<(), i32>(())?; |
| 54 | Ok(28) |
| 55 | }; |
| 56 | assert_eq!(catch_okay, Ok(28)); |
| 57 | |
| 58 | let catch_from_loop: Result<i32, i32> = do catch { |
| 59 | for i in 0..10 { |
| 60 | if i < 5 { Ok::<i32, i32>(i)?; } else { Err(i)?; } |
| 61 | } |
| 62 | Ok(22) |
| 63 | }; |
| 64 | assert_eq!(catch_from_loop, Err(5)); |
| 65 | |
| 66 | let cfg_init; |
| 67 | let _res: Result<(), ()> = do catch { |
| 68 | cfg_init = 5; |
| 69 | Ok(()) |
| 70 | }; |
| 71 | assert_eq!(cfg_init, 5); |
| 72 | |
| 73 | let cfg_init_2; |
| 74 | let _res: Result<(), ()> = do catch { |
| 75 | cfg_init_2 = 6; |
| 76 | Err(())?; |
| 77 | Ok(()) |
| 78 | }; |
| 79 | assert_eq!(cfg_init_2, 6); |
| 80 | |
| 81 | let my_string = "test".to_string(); |
| 82 | let res: Result<&str, ()> = do catch { |
| 83 | Ok(&my_string) |
| 84 | }; |
| 85 | assert_eq!(res, Ok("test")); |
| 86 | } |
| 87 | "#; |
| 88 | |
David Tolnay | c7a5d3d | 2017-06-04 12:11:05 -0700 | [diff] [blame] | 89 | let actual: File = syn::parse_str(raw).unwrap(); |
Arnavion | 02ef13f | 2017-04-25 00:54:31 -0700 | [diff] [blame] | 90 | |
David Tolnay | c6b55bc | 2017-11-09 22:48:38 -0800 | [diff] [blame] | 91 | assert_let!(Item::Struct(ItemStruct { ref ident, .. }) = actual.items[0]; { |
David Tolnay | 63e3dee | 2017-06-03 20:13:17 -0700 | [diff] [blame] | 92 | assert_eq!(ident, "catch"); |
| 93 | }); |
Arnavion | 02ef13f | 2017-04-25 00:54:31 -0700 | [diff] [blame] | 94 | |
David Tolnay | c6b55bc | 2017-11-09 22:48:38 -0800 | [diff] [blame] | 95 | assert_let!(Item::Fn(ItemFn { ref block, .. }) = actual.items[1]; { |
Alex Crichton | 62a0a59 | 2017-05-22 13:58:53 -0700 | [diff] [blame] | 96 | assert_let!(Stmt::Local(ref local) = block.stmts[0]; { |
David Tolnay | 8b4d302 | 2017-12-29 12:11:10 -0500 | [diff] [blame] | 97 | assert_let!(Local { init: Some((_, ref init_expr)), .. } = **local; { |
David Tolnay | 8c91b88 | 2017-12-28 23:04:32 -0500 | [diff] [blame] | 98 | assert_let!(Expr::Catch(..) = **init_expr); |
Arnavion | 02ef13f | 2017-04-25 00:54:31 -0700 | [diff] [blame] | 99 | }); |
| 100 | }); |
| 101 | |
Alex Crichton | 62a0a59 | 2017-05-22 13:58:53 -0700 | [diff] [blame] | 102 | assert_let!(Stmt::Local(ref local) = block.stmts[2]; { |
David Tolnay | 24237fb | 2017-12-29 02:15:26 -0500 | [diff] [blame] | 103 | assert_let!(Pat::Ident(PatIdent { by_ref: None, mutability: Some(_), ref ident, .. }) = *local.pat; { |
Arnavion | 02ef13f | 2017-04-25 00:54:31 -0700 | [diff] [blame] | 104 | assert_eq!(ident, "catch"); |
| 105 | }); |
| 106 | }); |
| 107 | |
Alex Crichton | 62a0a59 | 2017-05-22 13:58:53 -0700 | [diff] [blame] | 108 | assert_let!(Stmt::Expr(ref expr) = block.stmts[3]; { |
David Tolnay | 8c91b88 | 2017-12-28 23:04:32 -0500 | [diff] [blame] | 109 | assert_let!(Expr::While(ExprWhile { ref cond, .. }) = **expr; { |
| 110 | assert_let!(Expr::Path(ExprPath { qself: None, ref path, .. }) = **cond; { |
Alex Crichton | 62a0a59 | 2017-05-22 13:58:53 -0700 | [diff] [blame] | 111 | assert_eq!(*path, "catch".into()); |
Arnavion | 02ef13f | 2017-04-25 00:54:31 -0700 | [diff] [blame] | 112 | }); |
| 113 | }); |
| 114 | }); |
| 115 | |
Alex Crichton | ccbb45d | 2017-05-23 10:58:24 -0700 | [diff] [blame] | 116 | assert_let!(Stmt::Semi(ref expr, _) = block.stmts[5]; { |
David Tolnay | 8c91b88 | 2017-12-28 23:04:32 -0500 | [diff] [blame] | 117 | assert_let!(Expr::Assign(ExprAssign { ref left, ref right, .. }) = **expr; { |
| 118 | assert_let!(Expr::Path(ExprPath { qself: None, ref path, .. }) = **left; { |
Alex Crichton | 62a0a59 | 2017-05-22 13:58:53 -0700 | [diff] [blame] | 119 | assert_eq!(*path, "catch".into()); |
Arnavion | 02ef13f | 2017-04-25 00:54:31 -0700 | [diff] [blame] | 120 | }); |
| 121 | |
David Tolnay | 8c91b88 | 2017-12-28 23:04:32 -0500 | [diff] [blame] | 122 | assert_let!(Expr::If(ExprIf { ref cond, .. }) = **right; { |
| 123 | assert_let!(Expr::Path(ExprPath { qself: None, ref path, .. }) = **cond; { |
Alex Crichton | 62a0a59 | 2017-05-22 13:58:53 -0700 | [diff] [blame] | 124 | assert_eq!(*path, "catch".into()); |
Arnavion | 02ef13f | 2017-04-25 00:54:31 -0700 | [diff] [blame] | 125 | }); |
| 126 | }); |
| 127 | }); |
| 128 | }); |
| 129 | |
Alex Crichton | ccbb45d | 2017-05-23 10:58:24 -0700 | [diff] [blame] | 130 | assert_let!(Stmt::Semi(ref expr, _) = block.stmts[7]; { |
David Tolnay | 8c91b88 | 2017-12-28 23:04:32 -0500 | [diff] [blame] | 131 | assert_let!(Expr::Match(ExprMatch { ref expr, .. }) = **expr; { |
| 132 | assert_let!(Expr::Path(ExprPath { qself: None, ref path, .. }) = **expr; { |
Alex Crichton | 62a0a59 | 2017-05-22 13:58:53 -0700 | [diff] [blame] | 133 | assert_eq!(*path, "catch".into()); |
Arnavion | 02ef13f | 2017-04-25 00:54:31 -0700 | [diff] [blame] | 134 | }); |
| 135 | }); |
| 136 | }); |
| 137 | }); |
| 138 | } |