Commit f7373f8b authored by Nick Mathewson's avatar Nick Mathewson 🦞
Browse files

tor-rtcompat: Add some miscellaneous tests 

These probably aren't for things that will fail IRL, but it's nice
to have coverage on the code, just in case.
parent af852622

crates/tor-rtcompat/src/async_std.rs

+34 −0
Original line number Diff line number Diff line
@@ -137,3 +137,37 @@ impl AsyncStdRustlsRuntime { 
        runtime.clone().block_on(func(runtime))

    }

}



#[cfg(test)]

mod test {

    #![allow(clippy::unwrap_used)]

    use super::*;



    #[test]

    fn current() {

        // We should actually have to run this inside a runtime with async_std,

        // but let's do it anyway to make sure that "current" works.

        let runtime = PreferredRuntime::create().unwrap();

        runtime.block_on(async {

            #[cfg(feature = "native-tls")]

            assert!(AsyncStdNativeTlsRuntime::current().is_ok());



            #[cfg(feature = "rustls")]

            assert!(AsyncStdRustlsRuntime::current().is_ok());

        });

    }



    #[test]

    fn debug() {

        #[cfg(feature = "native-tls")]

        assert_eq!(

            format!("{:?}", AsyncStdNativeTlsRuntime::create().unwrap()),

            "AsyncStdNativeTlsRuntime { .. }"

        );

        #[cfg(feature = "rustls")]

        assert_eq!(

            format!("{:?}", AsyncStdRustlsRuntime::create().unwrap()),

            "AsyncStdRustlsRuntime { .. }"

        );

    }

}

crates/tor-rtcompat/src/impls/rustls.rs

+1 −0
Original line number Diff line number Diff line
@@ -262,6 +262,7 @@ mod test { 
        check_cvt!(ECDSA_NISTP384_SHA384);

        check_cvt!(RSA_PSS_SHA256);

        check_cvt!(RSA_PSS_SHA384);

        check_cvt!(RSA_PSS_SHA512);

        check_cvt!(ED25519);

        check_cvt!(ED448);

crates/tor-rtcompat/src/lib.rs

+6 −0
Original line number Diff line number Diff line
@@ -611,6 +611,9 @@ mod test { 
            mod async_std_runtime_tests {

                tests_with_runtime! { &crate::async_std::PreferredRuntime::create()? => $($id),* }

            }

            mod default_runtime_tests {

                tests_with_runtime! { &crate::create_runtime()? => $($id),* }

            }

        }

    }
@@ -633,6 +636,9 @@ mod test { 
            mod async_std_rustls_tests {

                tests_with_runtime! {  &crate::async_std::AsyncStdRustlsRuntime::create()? => $($id),* }

            }

            mod default_runtime_tls_tests {

                tests_with_runtime! { &crate::create_runtime()? => $($id),* }

            }

        }

    }

crates/tor-rtcompat/src/tokio.rs

+50 −0
Original line number Diff line number Diff line
@@ -183,3 +183,53 @@ impl TokioRustlsRuntime { 
fn current_handle() -> std::io::Result<tokio_crate::runtime::Handle> {

    tokio_crate::runtime::Handle::try_current().map_err(|e| IoError::new(ErrorKind::Other, e))

}



#[cfg(test)]

mod test {

    #![allow(clippy::unwrap_used)]

    use super::*;



    #[test]

    fn no_current() {

        // There should be no running tokio runtime in this context.



        #[cfg(feature = "native-tls")]

        assert!(TokioNativeTlsRuntime::current().is_err());



        #[cfg(feature = "rustls")]

        assert!(TokioRustlsRuntime::current().is_err());

    }



    #[test]

    fn current() {

        // Now start a tokio runtime and make sure that the "current" functions do work in that case.

        let runtime = PreferredRuntime::create().unwrap();

        runtime.block_on(async {

            #[cfg(feature = "native-tls")]

            assert!(TokioNativeTlsRuntime::current().is_ok());



            #[cfg(feature = "rustls")]

            assert!(TokioRustlsRuntime::current().is_ok());

        });

    }



    #[test]

    fn debug() {

        #[cfg(feature = "native-tls")]

        assert_eq!(

            format!("{:?}", TokioNativeTlsRuntime::create().unwrap()),

            "TokioNativeTlsRuntime { .. }"

        );

        #[cfg(feature = "rustls")]

        assert_eq!(

            format!("{:?}", TokioRustlsRuntime::create().unwrap()),

            "TokioRustlsRuntime { .. }"

        );



        // Just for fun, let's try the Debug output for the Compound.

        assert_eq!(

            format!("{:?}", PreferredRuntime::create().unwrap().inner),

            "CompoundRuntime { .. }"

        );

    }

}