extern crate gluster;

use std::fs::File;
use std::io::Read;

use gluster::*;

#[test]
fn test_translate_to_bytes() {
    let kb_result: u64 = gluster::translate_to_bytes("100KB").unwrap();
    println!("kb_result: {}", kb_result);
    assert_eq!(kb_result, 102400);

    let mb_result: u64 = gluster::translate_to_bytes("100MB").unwrap();
    println!("mb_result: {}", mb_result);
    assert_eq!(mb_result, 104857600);

    let gb_result: u64 = gluster::translate_to_bytes("100GB").unwrap();
    println!("gb_result: {}", gb_result);
    assert_eq!(gb_result, 107374182400);

    let tb_result: u64 = gluster::translate_to_bytes("100TB").unwrap();
    println!("tb_result: {}", tb_result);
    assert_eq!(tb_result, 109951162777600);

    let pb_result: u64 = gluster::translate_to_bytes("100PB").unwrap();
    println!("pb_result: {}", pb_result);
    assert_eq!(pb_result, 112589990684262400);

    let pb_float_result: f64 = gluster::translate_to_bytes("100.5PB").unwrap();
    println!("pb_float_result: {}", pb_float_result);
    assert_eq!(pb_float_result, 113152940637683710.0);
}

#[test]
fn test_read_fop_stats() {
    let aggr_test_data = {
        let mut f = File::open("tests/fop_aggr_sample.json").unwrap();
        let mut s = String::new();
        f.read_to_string(&mut s).unwrap();
        s
    };
    let inter_test_data = {
        let mut f = File::open("tests/fop_inter_sample.json").unwrap();
        let mut s = String::new();
        f.read_to_string(&mut s).unwrap();
        s
    };

    let aggr_result = gluster::fop::read_aggr_fop(&aggr_test_data, "_mnt_xvdf").unwrap();
    let inter_result = gluster::fop::read_inter_fop(&inter_test_data, "_mnt_xvdf").unwrap();
    println!("Aggr result: {:?}", aggr_result);
    println!("Inter result: {:?}", inter_result);
}

#[test]
fn test_get_local_ip() {
    let result = gluster::get_local_ip();
    println!("local ip: {:?}", result);
}