humansize/tests/test.rs

use humansize::{
    file_size_opts::{self, BINARY, CONVENTIONAL, DECIMAL},
    FileSize,
};

#[test]
fn test_sizes() {
    assert_eq!(0.file_size(BINARY).unwrap(), "0 B");
    assert_eq!(999.file_size(BINARY).unwrap(), "999 B");
    assert_eq!(1000.file_size(BINARY).unwrap(), "1000 B");
    assert_eq!(1000.file_size(DECIMAL).unwrap(), "1 KB");
    assert_eq!(1023.file_size(BINARY).unwrap(), "1023 B");
    assert_eq!(1023.file_size(DECIMAL).unwrap(), "1.02 KB");
    assert_eq!(1024.file_size(BINARY).unwrap(), "1 KiB");
    assert_eq!(1024.file_size(CONVENTIONAL).unwrap(), "1 KB");

    let semi_custom_options = file_size_opts::FormatSizeOptions {
        space: false,
        ..file_size_opts::DECIMAL
    };
    assert_eq!(1000.file_size(semi_custom_options).unwrap(), "1KB");

    let semi_custom_options2 = file_size_opts::FormatSizeOptions {
        suffix: "/s",
        ..file_size_opts::BINARY
    };
    assert_eq!(999.file_size(semi_custom_options2).unwrap(), "999 B/s");

    let semi_custom_options3 = file_size_opts::FormatSizeOptions {
        suffix: "/day",
        space: false,
        ..file_size_opts::DECIMAL
    };
    assert_eq!(1000.file_size(semi_custom_options3).unwrap(), "1KB/day");

    let semi_custom_options4 = file_size_opts::FormatSizeOptions {
        fixed_at: file_size_opts::FixedAt::Byte,
        ..file_size_opts::BINARY
    };
    assert_eq!(2048.file_size(semi_custom_options4).unwrap(), "2048 B");

    let semi_custom_options5 = file_size_opts::FormatSizeOptions {
        fixed_at: file_size_opts::FixedAt::Kilo,
        ..file_size_opts::BINARY
    };
    assert_eq!(
        16584975.file_size(semi_custom_options5).unwrap(),
        "16196.26 KiB"
    );

    let semi_custom_options6 = file_size_opts::FormatSizeOptions {
        fixed_at: file_size_opts::FixedAt::Tera,
        decimal_places: 10,
        ..file_size_opts::BINARY
    };
    assert_eq!(
        15284975.file_size(semi_custom_options6).unwrap(),
        "0.0000139016 TiB"
    );

    let semi_custom_options7 = file_size_opts::FormatSizeOptions {
        allow_negative: true,
        ..file_size_opts::DECIMAL
    };
    assert_eq!(
        (-5500).file_size(&semi_custom_options7).unwrap(),
        "-5.50 KB"
    );
    assert_eq!((5500).file_size(&semi_custom_options7).unwrap(), "5.50 KB");
}

#[test]
fn use_custom_option_struct_twice() {
    let options = file_size_opts::FormatSizeOptions {
        long_units: true,
        ..file_size_opts::DECIMAL
    };

    assert_eq!(1500.file_size(&options).unwrap(), "1.50 Kilobyte",);

    assert_eq!(2500.file_size(&options).unwrap(), "2.50 Kilobytes",);
}

#[test]
fn pluralization_works() {
    let options = file_size_opts::FormatSizeOptions {
        long_units: true,
        decimal_zeroes: 2,
        ..file_size_opts::DECIMAL
    };

    assert_eq!(1.file_size(&options).unwrap(), "1.00 Byte",);

    assert_eq!(1000.file_size(&options).unwrap(), "1.00 Kilobyte",);

    assert_eq!(1000000.file_size(&options).unwrap(), "1.00 Megabyte",);

    assert_eq!(1000000000.file_size(&options).unwrap(), "1.00 Gigabyte",);

    assert_eq!(
        1000000000000_i64.file_size(&options).unwrap(),
        "1.00 Terabyte",
    );

    assert_eq!(
        1000000000000000_i64.file_size(&options).unwrap(),
        "1.00 Petabyte",
    );

    assert_eq!(
        1000000000000000000_i64.file_size(&options).unwrap(),
        "1.00 Exabyte",
    );
}

#[test]
fn max_value_decimal() {
    let options = file_size_opts::FormatSizeOptions {
        long_units: true,
        decimal_places: 7,
        ..file_size_opts::DECIMAL
    };

    assert_eq!(
        (std::u64::MAX).file_size(&options).unwrap(),
        "18.4467441 Exabytes",
    );
}

#[test]
fn max_value_binary() {
    let options = file_size_opts::FormatSizeOptions {
        long_units: true,
        decimal_places: 7,
        ..file_size_opts::BINARY
    };

    assert_eq!((std::u64::MAX).file_size(&options).unwrap(), "16 Exbibytes",);
}