#![feature(test)]

extern crate test;

use futures::{executor, TryStreamExt};
use futures_codec::{FramedRead, LinesCodec};
use std::io::Cursor;

#[bench]
fn short(b: &mut test::Bencher) {
    let data = [
        ["a"; 16].join("b"),
        ["b"; 16].join("c"),
        ["c"; 16].join("d"),
    ]
    .join("\n");
    b.iter(|| {
        executor::block_on(async {
            let read = Cursor::new(test::black_box(&data));
            let mut framed = FramedRead::new(read, LinesCodec {});

            framed.try_next().await.unwrap();
            framed.try_next().await.unwrap();
            framed.try_next().await.is_ok()
        })
    })
}

#[bench]
fn medium(b: &mut test::Bencher) {
    let data = [
        ["a"; 128].join("b"),
        ["b"; 128].join("c"),
        ["c"; 128].join("d"),
    ]
    .join("\n");
    b.iter(|| {
        executor::block_on(async {
            let read = Cursor::new(test::black_box(&data));
            let mut framed = FramedRead::new(read, LinesCodec {});

            framed.try_next().await.unwrap();
            framed.try_next().await.unwrap();
            framed.try_next().await.is_ok()
        })
    })
}

#[bench]
fn long(b: &mut test::Bencher) {
    let data = [
        ["a"; 2048].join("b"),
        ["b"; 2048].join("c"),
        ["c"; 2048].join("d"),
    ]
    .join("\n");
    b.iter(|| {
        executor::block_on(async {
            let read = Cursor::new(test::black_box(&data));
            let mut framed = FramedRead::new(read, LinesCodec {});

            framed.try_next().await.unwrap();
            framed.try_next().await.unwrap();
            framed.try_next().await.is_ok()
        })
    })
}