| David Tolnay | 3fd7f56 | 2020-01-26 17:47:11 -0800 | [diff] [blame] | 1 | use cxx_test_suite::ffi; |
| 2 | use std::cell::Cell; |
| David Tolnay | f306da4 | 2020-02-22 19:55:43 -0800 | [diff] [blame] | 3 | use std::ffi::CStr; |
| David Tolnay | 3fd7f56 | 2020-01-26 17:47:11 -0800 | [diff] [blame] | 4 | |
| 5 | thread_local! { |
| 6 | static CORRECT: Cell<bool> = Cell::new(false); |
| 7 | } |
| 8 | |
| 9 | #[no_mangle] |
| 10 | extern "C" fn cxx_test_suite_set_correct() { |
| 11 | CORRECT.with(|correct| correct.set(true)); |
| 12 | } |
| 13 | |
| David Tolnay | f306da4 | 2020-02-22 19:55:43 -0800 | [diff] [blame] | 14 | macro_rules! check { |
| 15 | ($run:expr) => {{ |
| 16 | CORRECT.with(|correct| correct.set(false)); |
| 17 | $run; |
| 18 | assert!(CORRECT.with(|correct| correct.get()), stringify!($run)); |
| 19 | }}; |
| 20 | } |
| 21 | |
| David Tolnay | 3fd7f56 | 2020-01-26 17:47:11 -0800 | [diff] [blame] | 22 | #[test] |
| 23 | fn test_c_return() { |
| 24 | let shared = ffi::Shared { z: 2020 }; |
| 25 | |
| 26 | assert_eq!(2020, ffi::c_return_primitive()); |
| 27 | assert_eq!(2020, ffi::c_return_shared().z); |
| David Tolnay | be13d8a | 2020-03-06 15:45:39 -0800 | [diff] [blame] | 28 | assert_eq!(2020, *ffi::c_return_box()); |
| David Tolnay | 3fd7f56 | 2020-01-26 17:47:11 -0800 | [diff] [blame] | 29 | ffi::c_return_unique_ptr(); |
| 30 | assert_eq!(2020, *ffi::c_return_ref(&shared)); |
| 31 | assert_eq!("2020", ffi::c_return_str(&shared)); |
| Adrian Taylor | ec9430e | 2020-04-14 16:09:58 -0700 | [diff] [blame] | 32 | assert_eq!(b"2020\0", ffi::c_return_sliceu8(&shared)); |
| David Tolnay | 3fd7f56 | 2020-01-26 17:47:11 -0800 | [diff] [blame] | 33 | assert_eq!("2020", ffi::c_return_rust_string()); |
| David Tolnay | 5e29b21 | 2020-04-17 15:02:16 -0700 | [diff] [blame] | 34 | assert_eq!("2020", ffi::c_return_unique_ptr_string().to_str().unwrap()); |
| David Tolnay | 9964262 | 2020-03-25 13:07:35 -0700 | [diff] [blame] | 35 | } |
| David Tolnay | ebef4a2 | 2020-03-17 15:33:47 -0700 | [diff] [blame] | 36 | |
| David Tolnay | 9964262 | 2020-03-25 13:07:35 -0700 | [diff] [blame] | 37 | #[test] |
| 38 | fn test_c_try_return() { |
| David Tolnay | ebef4a2 | 2020-03-17 15:33:47 -0700 | [diff] [blame] | 39 | assert_eq!((), ffi::c_try_return_void().unwrap()); |
| 40 | assert_eq!(2020, ffi::c_try_return_primitive().unwrap()); |
| 41 | assert_eq!( |
| 42 | "logic error", |
| 43 | ffi::c_fail_return_primitive().unwrap_err().what(), |
| 44 | ); |
| David Tolnay | 9964262 | 2020-03-25 13:07:35 -0700 | [diff] [blame] | 45 | assert_eq!(2020, *ffi::c_try_return_box().unwrap()); |
| 46 | assert_eq!("2020", *ffi::c_try_return_ref(&"2020".to_owned()).unwrap()); |
| 47 | assert_eq!("2020", ffi::c_try_return_str("2020").unwrap()); |
| Adrian Taylor | ec9430e | 2020-04-14 16:09:58 -0700 | [diff] [blame] | 48 | assert_eq!(b"2020", ffi::c_try_return_sliceu8(b"2020").unwrap()); |
| David Tolnay | 9964262 | 2020-03-25 13:07:35 -0700 | [diff] [blame] | 49 | assert_eq!("2020", ffi::c_try_return_rust_string().unwrap()); |
| David Tolnay | 5e29b21 | 2020-04-17 15:02:16 -0700 | [diff] [blame] | 50 | assert_eq!("2020", &*ffi::c_try_return_unique_ptr_string().unwrap()); |
| David Tolnay | 3fd7f56 | 2020-01-26 17:47:11 -0800 | [diff] [blame] | 51 | } |
| 52 | |
| 53 | #[test] |
| 54 | fn test_c_take() { |
| David Tolnay | 3fd7f56 | 2020-01-26 17:47:11 -0800 | [diff] [blame] | 55 | let unique_ptr = ffi::c_return_unique_ptr(); |
| 56 | |
| 57 | check!(ffi::c_take_primitive(2020)); |
| 58 | check!(ffi::c_take_shared(ffi::Shared { z: 2020 })); |
| David Tolnay | be13d8a | 2020-03-06 15:45:39 -0800 | [diff] [blame] | 59 | check!(ffi::c_take_box(Box::new(2020))); |
| David Tolnay | 5e29b21 | 2020-04-17 15:02:16 -0700 | [diff] [blame] | 60 | check!(ffi::c_take_ref_c(&unique_ptr)); |
| David Tolnay | 3fd7f56 | 2020-01-26 17:47:11 -0800 | [diff] [blame] | 61 | check!(ffi::c_take_unique_ptr(unique_ptr)); |
| 62 | check!(ffi::c_take_str("2020")); |
| Adrian Taylor | f5dd552 | 2020-04-13 16:50:14 -0700 | [diff] [blame] | 63 | check!(ffi::c_take_sliceu8(b"2020")); |
| David Tolnay | 3fd7f56 | 2020-01-26 17:47:11 -0800 | [diff] [blame] | 64 | check!(ffi::c_take_rust_string("2020".to_owned())); |
| 65 | check!(ffi::c_take_unique_ptr_string( |
| 66 | ffi::c_return_unique_ptr_string() |
| 67 | )); |
| 68 | } |
| David Tolnay | f306da4 | 2020-02-22 19:55:43 -0800 | [diff] [blame] | 69 | |
| 70 | #[test] |
| David Tolnay | 75dca2e | 2020-03-25 20:17:52 -0700 | [diff] [blame] | 71 | fn test_c_callback() { |
| 72 | fn callback(s: String) -> usize { |
| 73 | if s == "2020" { |
| 74 | cxx_test_suite_set_correct(); |
| 75 | } |
| 76 | 0 |
| 77 | } |
| 78 | |
| 79 | check!(ffi::c_take_callback(callback)); |
| 80 | } |
| 81 | |
| 82 | #[test] |
| David Tolnay | f306da4 | 2020-02-22 19:55:43 -0800 | [diff] [blame] | 83 | fn test_c_call_r() { |
| 84 | fn cxx_run_test() { |
| 85 | extern "C" { |
| 86 | fn cxx_run_test() -> *const i8; |
| 87 | } |
| 88 | let failure = unsafe { cxx_run_test() }; |
| 89 | if !failure.is_null() { |
| 90 | let msg = unsafe { CStr::from_ptr(failure) }; |
| 91 | eprintln!("{}", msg.to_string_lossy()); |
| 92 | } |
| 93 | } |
| 94 | check!(cxx_run_test()); |
| 95 | } |
| David Tolnay | be13d8a | 2020-03-06 15:45:39 -0800 | [diff] [blame] | 96 | |
| Joel Galenson | 3d4f612 | 2020-04-07 15:54:05 -0700 | [diff] [blame] | 97 | #[test] |
| 98 | fn test_c_method_calls() { |
| 99 | let mut unique_ptr = ffi::c_return_unique_ptr(); |
| 100 | |
| David Tolnay | 5e29b21 | 2020-04-17 15:02:16 -0700 | [diff] [blame] | 101 | let old_value = unique_ptr.get(); |
| Joel Galenson | 3d4f612 | 2020-04-07 15:54:05 -0700 | [diff] [blame] | 102 | assert_eq!(2020, old_value); |
| David Tolnay | 5e29b21 | 2020-04-17 15:02:16 -0700 | [diff] [blame] | 103 | assert_eq!(2021, unique_ptr.set(2021)); |
| 104 | assert_eq!(2021, unique_ptr.get()); |
| Joel Galenson | e1e969d | 2020-04-21 12:50:20 -0700 | [diff] [blame^] | 105 | assert_eq!(old_value, unique_ptr.set2(old_value)); |
| 106 | assert_eq!(old_value, unique_ptr.get2()) |
| Joel Galenson | 3d4f612 | 2020-04-07 15:54:05 -0700 | [diff] [blame] | 107 | } |
| 108 | |
| David Tolnay | be13d8a | 2020-03-06 15:45:39 -0800 | [diff] [blame] | 109 | #[no_mangle] |
| 110 | extern "C" fn cxx_test_suite_get_box() -> *mut cxx_test_suite::R { |
| 111 | Box::into_raw(Box::new(2020usize)) |
| 112 | } |
| David Tolnay | a7d00e8 | 2020-03-06 15:50:14 -0800 | [diff] [blame] | 113 | |
| 114 | #[no_mangle] |
| 115 | unsafe extern "C" fn cxx_test_suite_r_is_correct(r: *const cxx_test_suite::R) -> bool { |
| 116 | *r == 2020 |
| 117 | } |