add tests of sys::Error

Author: molpopgen

src/error.rs

@@ -100,11 +100,7 @@ pub fn panic_on_tskit_error(code: i32) {
 /// This function must allocate a C string, which may panic
 /// if the system runs out of memory.
 pub fn get_tskit_error_message(code: i32) -> String {
-    let c_str = unsafe { std::ffi::CStr::from_ptr(crate::bindings::tsk_strerror(code)) };
-    c_str
-        .to_str()
-        .expect("failed to convert c_str to &str")
-        .to_owned()
+    sys::get_tskit_error_message(code)
 }
 
 /// Given an instance of [``TskReturnValue``](crate::TskReturnValue),

src/sys.rs

@@ -15,9 +15,9 @@ use bindings::tsk_site_table_t;
 #[non_exhaustive]
 #[derive(Error, Debug)]
 pub enum Error {
-    #[error("{}", 0)]
+    #[error("{}", *.0)]
     Message(String),
-    #[error("{}", 0)]
+    #[error("{}", get_tskit_error_message(*.0))]
     Code(i32),
 }
 
@@ -272,3 +272,46 @@ pub fn generate_slice_mut<'a, L: Into<bindings::tsk_size_t>, I, O>(
     // SAFETY: pointer is not null, length comes from C API
     unsafe { std::slice::from_raw_parts_mut(data.cast::<O>(), length.into() as usize) }
 }
+
+pub fn get_tskit_error_message(code: i32) -> String {
+    let c_str = unsafe { std::ffi::CStr::from_ptr(crate::bindings::tsk_strerror(code)) };
+    c_str
+        .to_str()
+        .expect("failed to convert c_str to &str")
+        .to_owned()
+}
+
+#[test]
+fn test_error_message() {
+    fn foo() -> Result<(), Error> {
+        Err(Error::Message("foobar".to_owned()))
+    }
+
+    let msg = "foobar".to_owned();
+    match foo() {
+        Err(Error::Message(m)) => assert_eq!(m, msg),
+        _ => panic!("unexpected match"),
+    }
+}
+
+#[test]
+fn test_error_code() {
+    fn foo() -> Result<(), Error> {
+        Err(Error::Code(-202))
+    }
+
+    match foo() {
+        Err(Error::Code(x)) => {
+            assert_eq!(x, -202);
+        }
+        _ => panic!("unexpected match"),
+    }
+
+    match foo() {
+        Err(e) => {
+            let m = format!("{}", e);
+            assert_eq!(&m, "Node out of bounds. (TSK_ERR_NODE_OUT_OF_BOUNDS)");
+        }
+        _ => panic!("unexpected match"),
+    }
+}