initial commit

lib/inferium/benches/client.rs

@@ -0,0 +1,60 @@
+#![feature(test)]
+extern crate test;
+use std::collections::HashMap;
+
+use test::Bencher;
+
+extern crate inferium;
+use inferium::h1::{SyncClient, Response, ResponseHead, ProtocolVariant};
+use inferium::{Status, HeaderValue};
+use inferium::TestSyncStream;
+
+fn parse_response_sync_inner() {
+    let src = "HTTP/1.1 200 OK\r\nserver: inferium\r\n\r\n".as_bytes().to_vec();
+    let stream = TestSyncStream::<4>::new(&src);
+    let mut client = SyncClient::<TestSyncStream<4>>::new(stream);
+    let target = Response::HeadersOnly(ResponseHead::new(
+        Status::Ok,
+        ProtocolVariant::HTTP1_1,
+        HashMap::from([
+            ("server".into(), HeaderValue::new(vec!["inferium".to_string()]))
+        ])
+    ));
+    assert_eq!(client.receive_response().unwrap(), target);
+}
+
+fn parse_response_sync_inner_body() {
+    let mut src = "HTTP/1.1 200 OK\r\ncontent-length: 50\r\n\r\n".as_bytes().to_vec();
+    src.extend_from_slice(b"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa");
+    let stream = TestSyncStream::<4>::new(&src);
+    let mut client = SyncClient::<TestSyncStream<4>>::new(stream);
+    let target_head = ResponseHead::new(
+        Status::Ok,
+        ProtocolVariant::HTTP1_1,
+        HashMap::from([
+            ("content-length".into(), HeaderValue::new(vec!["50".to_string()]))
+        ])
+    );
+    let Response::WithSizedBody((h, mut b)) = client.receive_response().unwrap() else {
+        panic!();
+    };
+    let b = b.recv_all().unwrap();
+    assert_eq!(h, target_head);
+    assert_eq!(b, b"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa");
+}
+
+#[bench]
+fn parse_response_sync(b: &mut Bencher) {
+    b.bytes = 37;
+    b.iter(|| {
+        parse_response_sync_inner();
+    });
+}
+
+#[bench]
+fn parse_response_sync_with_body(b: &mut Bencher) {
+    b.bytes = 39 + 13;
+    b.iter(|| {
+        parse_response_sync_inner_body();
+    });
+}

lib/inferium/benches/status_code_parsing.rs

@@ -0,0 +1,43 @@
+#![feature(test)]
+extern crate test;
+use test::Bencher;
+
+extern crate inferium;
+use inferium::Status;
+
+#[bench]
+fn baseline(b: &mut Bencher) {
+    fn status_test_from_slice(raw: &[u8]) -> Result<Status, ()> {
+        match raw {
+            b"200" => Ok(Status::Ok),
+            b"404" => Ok(Status::NotFound),
+            _ => Err(()),
+        }
+    }
+
+    b.iter(|| {
+        assert_eq!(status_test_from_slice(b"200".as_slice()), Ok(Status::Ok));
+    });
+}
+
+#[bench]
+fn valid_ok(b: &mut Bencher) {
+    b.iter(|| {
+        assert_eq!(Status::try_from(b"200".as_slice()), Ok(Status::Ok));
+    });
+
+}
+
+#[bench]
+fn valid_internal_server_error(b: &mut Bencher) {
+    b.iter(|| {
+        assert_eq!(Status::try_from(b"500".as_slice()), Ok(Status::InternalServerError));
+    });
+}
+
+#[bench]
+fn invalid(b: &mut Bencher) {
+    b.iter(|| {
+        assert!(Status::try_from(b"690".as_slice()).is_err());
+    })
+}