#![allow(unused_assignments)]
use cbor::CborTagEncode;
use rustc_serialize::{Decodable, Decoder, Encodable, Encoder};
use types;
#[derive(PartialEq, Eq, Clone, Debug)]
pub struct ConnectResponse {
pub requester_local : types::EndPoint,
pub requester_external : types::EndPoint,
pub receiver_local : types::EndPoint,
pub receiver_external : types::EndPoint,
pub requester_id : types::DhtId,
pub receiver_id : types::DhtId,
pub receiver_fob : types::PublicPmid
}
impl ConnectResponse {
pub fn generate_random() -> ConnectResponse {
ConnectResponse {
requester_local: types::EndPoint::generate_random(),
requester_external: types::EndPoint::generate_random(),
receiver_local: types::EndPoint::generate_random(),
receiver_external: types::EndPoint::generate_random(),
requester_id: types::DhtId::generate_random(),
receiver_id: types::DhtId::generate_random(),
receiver_fob: types::PublicPmid::generate_random(),
}
}
}
impl Encodable for ConnectResponse {
fn encode<E: Encoder>(&self, e: &mut E)->Result<(), E::Error> {
CborTagEncode::new(5483_001, &(&self.requester_local, &self.requester_external,
&self.receiver_local, &self.receiver_external,
&self.requester_id, &self.receiver_id, &self.receiver_fob)).encode(e)
}
}
impl Decodable for ConnectResponse {
fn decode<D: Decoder>(d: &mut D)->Result<ConnectResponse, D::Error> {
try!(d.read_u64());
let (requester_local, requester_external, receiver_local, receiver_external,
requester_id, receiver_id, receiver_fob) = try!(Decodable::decode(d));
Ok(ConnectResponse { requester_local: requester_local, requester_external: requester_external,
receiver_local: receiver_local, receiver_external: receiver_external,
requester_id: requester_id, receiver_id: receiver_id, receiver_fob: receiver_fob})
}
}
#[cfg(test)]
mod test {
use super::*;
use cbor;
#[test]
fn connect_response_serialisation() {
let obj_before = ConnectResponse::generate_random();
let mut e = cbor::Encoder::from_memory();
e.encode(&[&obj_before]).unwrap();
let mut d = cbor::Decoder::from_bytes(e.as_bytes());
let obj_after: ConnectResponse = d.decode().next().unwrap().unwrap();
assert_eq!(obj_before, obj_after);
}
}