Add tokio test utilities

Cargo.lock

@@ -66,6 +66,27 @@ version = "0.5.2"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "23b62fc65de8e4e7f52534fb52b0f3ed04746ae267519eef2a83941e8085068b"
 
+[[package]]
+name = "async-stream"
+version = "0.3.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3670df70cbc01729f901f94c887814b3c68db038aad1329a418bae178bc5295c"
+dependencies = [
+ "async-stream-impl",
+ "futures-core",
+]
+
+[[package]]
+name = "async-stream-impl"
+version = "0.3.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a3548b8efc9f8e8a5a0a2808c5bd8451a9031b9e5b879a79590304ae928b0a70"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn",
+]
+
 [[package]]
 name = "autocfg"
 version = "1.0.1"
@@ -1179,6 +1200,7 @@ dependencies = [
  "serde_json",
  "sha2",
  "tokio",
+ "tokio-test",
  "warp",
  "x25519-dalek",
 ]
@@ -1374,6 +1396,19 @@ dependencies = [
  "tokio",
 ]
 
+[[package]]
+name = "tokio-test"
+version = "0.4.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f58403903e94d4bc56805e46597fced893410b2e753e229d3f7f22423ea03f67"
+dependencies = [
+ "async-stream",
+ "bytes",
+ "futures-core",
+ "tokio",
+ "tokio-stream",
+]
+
 [[package]]
 name = "tokio-tungstenite"
 version = "0.13.0"

Cargo.toml

@@ -22,3 +22,6 @@ sha2 = "0.9"
 tokio = { version = "1.3", features = ["full"] }
 warp = { version = "0.3", features = ["tls"] }
 x25519-dalek = "1.1"
+
+[dev-dependencies]
+tokio-test = "*"

src/tests.rs

@@ -1,4 +1,10 @@
 
+macro_rules! aw {
+    ($e:expr) => {
+        tokio_test::block_on($e)
+    };
+  }
+
 #[test]
 fn dummy_test() {
     assert!(true);