Make structs Copy,Hash,Eq,PartialEq

This commit is contained in:
Linus Färnstrand
2016-06-20 13:18:44 +02:00
parent 17e96ae764
commit b50643fbae

View File

@ -12,19 +12,19 @@ const IPV4_BITS: u8 = 32;
const IPV6_BITS: u8 = 128;
// A network
#[derive(Debug,Clone)]
#[derive(Debug,Clone,Copy,Hash,PartialEq,Eq)]
pub enum IpNetwork {
V4(Ipv4Network),
V6(Ipv6Network),
}
#[derive(Clone)]
#[derive(Clone,Copy,Hash,PartialEq,Eq)]
pub struct Ipv4Network {
addr: Ipv4Addr,
prefix: u8,
}
#[derive(Clone)]
#[derive(Clone,Copy,Hash,PartialEq,Eq)]
pub struct Ipv6Network {
addr: Ipv6Addr,
prefix: u8,
@ -239,6 +239,8 @@ fn parse_prefix(prefix: &str, max: u8) -> Result<u8, IpNetworkError> {
#[cfg(test)]
mod test {
use std::mem;
use std::collections::HashMap;
use std::net::{Ipv4Addr, Ipv6Addr};
use super::*;
@ -345,6 +347,24 @@ mod test {
assert!(net.nth(1).is_none());
}
#[test]
fn hash_eq_compatibility_v4() {
let mut map = HashMap::new();
let net = Ipv4Network::new(Ipv4Addr::new(127, 0, 0, 1), 16).unwrap();
map.insert(net, 137);
let out = map.get(&net).unwrap();
assert_eq!(137, *out);
}
#[test]
fn copy_compatibility_v4() {
let net = Ipv4Network::new(Ipv4Addr::new(127, 0, 0, 1), 16).unwrap();
mem::drop(net);
assert_eq!(16, net.prefix());
}
#[test]
fn create_v6() {
let cidr = Ipv6Network::new(Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 1), 24).unwrap();