use std; // These tests used to be separate files, but I wanted to refactor all // the common code. import std::ebml; import io::writer; import std::prettyprint::serializer; import std::ebml::serializer; import std::ebml::deserializer; import std::serialization::{serialize_uint, deserialize_uint}; fn test_ser_and_deser(a1: A, expected: str, ebml_ser_fn: fn(ebml::writer, A), ebml_deser_fn: fn(ebml::ebml_deserializer) -> A, io_ser_fn: fn(io::writer, A)) { // check the pretty printer: io_ser_fn(io::stdout(), a1); let s = io::with_str_writer(|w| io_ser_fn(w, a1) ); #debug["s == %?", s]; assert s == expected; // check the EBML serializer: let buf = io::mem_buffer(); let w = ebml::writer(buf as io::writer); ebml_ser_fn(w, a1); let d = ebml::doc(@io::mem_buffer_buf(buf)); let a2 = ebml_deser_fn(ebml::ebml_deserializer(d)); io::print("\na1 = "); io_ser_fn(io::stdout(), a1); io::print("\na2 = "); io_ser_fn(io::stdout(), a2); io::print("\n"); assert a1 == a2; } #[auto_serialize] enum expr { val(uint), plus(@expr, @expr), minus(@expr, @expr) } #[auto_serialize] type spanned = {lo: uint, hi: uint, node: T}; #[auto_serialize] type spanned_uint = spanned; #[auto_serialize] type some_rec = {v: uint_vec}; #[auto_serialize] enum an_enum = some_rec; #[auto_serialize] type uint_vec = ~[uint]; #[auto_serialize] type point = {x: uint, y: uint}; #[auto_serialize] enum quark { top(T), bottom(T) } #[auto_serialize] type uint_quark = quark; #[auto_serialize] enum c_like { a, b, c } fn main() { test_ser_and_deser(plus(@minus(@val(3u), @val(10u)), @plus(@val(22u), @val(5u))), "plus(@minus(@val(3u), @val(10u)), \ @plus(@val(22u), @val(5u)))", serialize_expr, deserialize_expr, serialize_expr); test_ser_and_deser({lo: 0u, hi: 5u, node: 22u}, "{lo: 0u, hi: 5u, node: 22u}", serialize_spanned_uint, deserialize_spanned_uint, serialize_spanned_uint); test_ser_and_deser(an_enum({v: ~[1u, 2u, 3u]}), "an_enum({v: [1u, 2u, 3u]})", serialize_an_enum, deserialize_an_enum, serialize_an_enum); test_ser_and_deser({x: 3u, y: 5u}, "{x: 3u, y: 5u}", serialize_point, deserialize_point, serialize_point); test_ser_and_deser(~[1u, 2u, 3u], "[1u, 2u, 3u]", serialize_uint_vec, deserialize_uint_vec, serialize_uint_vec); test_ser_and_deser(top(22u), "top(22u)", serialize_uint_quark, deserialize_uint_quark, serialize_uint_quark); test_ser_and_deser(bottom(222u), "bottom(222u)", serialize_uint_quark, deserialize_uint_quark, serialize_uint_quark); test_ser_and_deser(a, "a", serialize_c_like, deserialize_c_like, serialize_c_like); test_ser_and_deser(b, "b", serialize_c_like, deserialize_c_like, serialize_c_like); }