use std::fmt;
#[derive(Copy, Clone, PartialEq, Debug)]
pub enum CompressionMethod
{
Stored,
#[cfg(feature = "deflate")]
Deflated,
#[cfg(feature = "bzip2")]
Bzip2,
Unsupported(u16),
}
impl CompressionMethod {
pub fn from_u16(val: u16) -> CompressionMethod {
match val {
0 => CompressionMethod::Stored,
#[cfg(feature = "deflate")]
8 => CompressionMethod::Deflated,
#[cfg(feature = "bzip2")]
12 => CompressionMethod::Bzip2,
v => CompressionMethod::Unsupported(v),
}
}
pub fn to_u16(self) -> u16 {
match self {
CompressionMethod::Stored => 0,
#[cfg(feature = "deflate")]
CompressionMethod::Deflated => 8,
#[cfg(feature = "bzip2")]
CompressionMethod::Bzip2 => 12,
CompressionMethod::Unsupported(v) => v,
}
}
}
impl fmt::Display for CompressionMethod {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "{:?}", self)
}
}
#[cfg(test)]
mod test {
use super::CompressionMethod;
#[test]
fn from_eq_to() {
for v in 0..(::std::u16::MAX as u32 + 1)
{
let from = CompressionMethod::from_u16(v as u16);
let to = from.to_u16() as u32;
assert_eq!(v, to);
}
}
#[cfg(all(not(feature = "bzip2"), feature = "deflate"))]
fn methods() -> Vec<CompressionMethod> {
vec![CompressionMethod::Stored, CompressionMethod::Deflated]
}
#[cfg(all(not(feature = "deflate"), feature = "bzip2"))]
fn methods() -> Vec<CompressionMethod> {
vec![CompressionMethod::Stored, CompressionMethod::Bzip2]
}
#[cfg(all(feature = "bzip2", feature = "deflate"))]
fn methods() -> Vec<CompressionMethod> {
vec![CompressionMethod::Stored, CompressionMethod::Deflated, CompressionMethod::Bzip2]
}
#[cfg(all(not(feature = "bzip2"), not(feature = "deflate")))]
fn methods() -> Vec<CompressionMethod> {
vec![CompressionMethod::Stored]
}
#[test]
fn to_eq_from() {
fn check_match(method: CompressionMethod) {
let to = method.to_u16();
let from = CompressionMethod::from_u16(to);
let back = from.to_u16();
assert_eq!(to, back);
}
for method in methods() {
check_match(method);
}
}
#[test]
fn to_display_fmt() {
fn check_match(method: CompressionMethod) {
let debug_str = format!("{:?}", method);
let display_str = format!("{}", method);
assert_eq!(debug_str, display_str);
}
for method in methods() {
check_match(method);
}
}
}