#![allow(unused_assignments)]
use cbor::CborTagEncode;
use rustc_serialize::{Decodable, Decoder, Encodable, Encoder};
use types;
#[derive(PartialEq, Eq, Clone, Debug)]
pub struct ConnectRequest {
pub local : types::EndPoint,
pub external : types::EndPoint,
pub request_id : types::DhtId,
pub receiver_id : types::DhtId,
pub requester_fob : types::PublicPmid
}
impl ConnectRequest {
pub fn generate_random() -> ConnectRequest {
ConnectRequest {
local: types::EndPoint::generate_random(),
external: types::EndPoint::generate_random(),
request_id: types::DhtId::generate_random(),
receiver_id: types::DhtId::generate_random(),
requester_fob: types::PublicPmid::generate_random(),
}
}
}
impl Encodable for ConnectRequest {
fn encode<E: Encoder>(&self, e: &mut E)->Result<(), E::Error> {
CborTagEncode::new(5483_001, &(&self.local, &self.external, &self.request_id,
&self.receiver_id, &self.requester_fob)).encode(e)
}
}
impl Decodable for ConnectRequest {
fn decode<D: Decoder>(d: &mut D)->Result<ConnectRequest, D::Error> {
try!(d.read_u64());
let (local, external, request_id, receiver_id, requester_fob) = try!(Decodable::decode(d));
Ok(ConnectRequest { local: local, external: external, request_id: request_id,
receiver_id: receiver_id, requester_fob: requester_fob})
}
}
#[cfg(test)]
mod test {
use super::*;
use cbor;
#[test]
fn connect_request_serialisation() {
let obj_before = ConnectRequest::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: ConnectRequest = d.decode().next().unwrap().unwrap();
assert_eq!(obj_before, obj_after);
}
}