extern crate embedded_hal_mock as hal;
use hal::i2c::Transaction as I2cTrans;
extern crate max3010x;
extern crate nb;
use max3010x::{LedPulseWidth as LedPw, SamplingRate as SR};
mod base;
use base::{destroy, new, BitFlags as BF, Register as Reg, DEV_ADDR};

#[test]
fn can_change_into_hr() {
    let transactions = [
        I2cTrans::write(DEV_ADDR, vec![Reg::MODE, 0b010]),
        I2cTrans::write(DEV_ADDR, vec![Reg::FIFO_WR_PTR, 0, 0, 0]),
    ];
    let dev = new(&transactions);
    let dev = dev.into_heart_rate().unwrap();
    destroy(dev);
}

macro_rules! set_heart_rate_test {
    ($name:ident, $method:ident, [$($arg:expr),*], $reg:ident, $expected:expr) => {
        set_in_mode_test!(
            $name,
            into_heart_rate,
            0b10,
            $method,
            [$($arg),*],
            $reg,
            $expected
        );
    };
}

set_heart_rate_test!(
    enable_new_fifo_data_ready_interrupt,
    enable_new_fifo_data_ready_interrupt,
    [],
    INT_EN1,
    BF::PPG_RDY_INT
);
set_heart_rate_test!(
    disable_new_fifo_data_ready_interrupt,
    disable_new_fifo_data_ready_interrupt,
    [],
    INT_EN1,
    0
);

macro_rules! set_test {
    ($name:ident, $method:ident, $arg:expr, $expected:expr) => {
        set_heart_rate_test!($name, $method, [$arg], SPO2_CONFIG, $expected);
    };
}

set_test!(can_set_pw_69, set_pulse_width, LedPw::Pw69, 0);
set_test!(can_set_pw_118, set_pulse_width, LedPw::Pw118, 1);
set_test!(can_set_pw_215, set_pulse_width, LedPw::Pw215, 2);
set_test!(can_set_pw_411, set_pulse_width, LedPw::Pw411, 3);

set_test!(can_set_sr_50, set_sampling_rate, SR::Sps50, 0);
set_test!(can_set_sr_100, set_sampling_rate, SR::Sps100, 1 << 2);
set_test!(can_set_sr_200, set_sampling_rate, SR::Sps200, 2 << 2);
set_test!(can_set_sr_400, set_sampling_rate, SR::Sps400, 3 << 2);
set_test!(can_set_sr_800, set_sampling_rate, SR::Sps800, 4 << 2);
set_test!(can_set_sr_1000, set_sampling_rate, SR::Sps1000, 5 << 2);
set_test!(can_set_sr_1600, set_sampling_rate, SR::Sps1600, 6 << 2);
set_test!(can_set_sr_3200, set_sampling_rate, SR::Sps3200, 7 << 2);