fix tests

Signed-off-by: Yoshua Wuyts <[email protected]>

Author: yoshuawuyts

tests/common/mod.rs

@@ -1,14 +1,19 @@
-use async_std::io::Read;
 use async_std::fs::File;
+use async_std::io::{self, Read, Write};
 use async_std::path::PathBuf;
+use async_std::sync::Arc;
+use async_std::task::{Context, Poll};
+use std::pin::Pin;
 
 #[macro_export]
 macro_rules! assert {
-    ($actual:expr, $expected:expr, $block:expr) => {
+    ($actual:expr, $expected_file:expr, $block:expr) => {
         task::block_on(async {
+            use async_std::io::prelude::*;
             $block.await.unwrap();
             let mut actual = std::string::String::from_utf8($actual).unwrap();
-            let mut expected = std::string::String::from_utf8($expected).unwrap();
+            let mut expected = std::string::String::new();
+            $expected_file.read_to_string(&mut expected).await.unwrap();
             match expected.find("{DATE}") {
                 Some(i) => {
                     expected.replace_range(i..i + 6, "");
@@ -27,9 +32,38 @@ macro_rules! assert {
     };
 }
 
-pub fn read_fixture(name: &str) -> Arc<File> {
+pub async fn read_fixture(name: &str) -> TestFile {
     let directory: PathBuf = env!("CARGO_MANIFEST_DIR").into();
     let path: PathBuf = format!("tests/fixtures/{}.txt", name).into();
-    let mut file = File::open(directory.join(path)).expect("Reading fixture file didn't work");
-    Arc::new(file)
+    let file = File::open(directory.join(path))
+        .await
+        .expect("Reading fixture file didn't work");
+    TestFile(Arc::new(file))
+}
+
+#[derive(Clone)]
+pub struct TestFile(Arc<File>);
+
+impl Read for TestFile {
+    fn poll_read(
+        self: Pin<&mut Self>,
+        cx: &mut Context,
+        buf: &mut [u8],
+    ) -> Poll<io::Result<usize>> {
+        Pin::new(&mut &*self.0).poll_read(cx, buf)
+    }
+}
+
+impl Write for TestFile {
+    fn poll_write(self: Pin<&mut Self>, cx: &mut Context, buf: &[u8]) -> Poll<io::Result<usize>> {
+        Pin::new(&mut &*self.0).poll_write(cx, buf)
+    }
+
+    fn poll_flush(self: Pin<&mut Self>, cx: &mut Context) -> Poll<io::Result<()>> {
+        Pin::new(&mut &*self.0).poll_flush(cx)
+    }
+
+    fn poll_close(self: Pin<&mut Self>, cx: &mut Context) -> Poll<io::Result<()>> {
+        Pin::new(&mut &*self.0).poll_close(cx)
+    }
 }

tests/server.rs

@@ -6,20 +6,22 @@ use http_types::{Response, StatusCode};
 
 #[test]
 fn test_basic_request() {
-    let request = read_fixture("request1");
-    let expected = read_fixture("response1");
-    let mut actual = Vec::new();
-    let addr = "http://example.com";
+    task::block_on(async {
+        let request = read_fixture("request1").await;
+        let mut expected = read_fixture("response1").await;
+        let mut actual = Vec::new();
+        let addr = "http://example.com";
 
-    assert!(
-        actual,
-        expected,
-        server::accept(addr, request, &mut actual, |_req| {
-            async {
-                let mut resp = Response::new(StatusCode::Ok);
-                resp.set_body("");
-                Ok(resp)
-            }
-        })
-    );
+        assert!(
+            actual,
+            expected,
+            server::accept(addr, request.clone(), &mut actual, |_req| {
+                async {
+                    let mut resp = Response::new(StatusCode::Ok);
+                    resp.set_body("");
+                    Ok(resp)
+                }
+           })
+        );
+    })
 }