json-tests init

json-tests/src/lib.rs

@@ -0,0 +1,68 @@
+extern crate rustc_serialize;
+extern crate glob;
+
+use std::str::from_utf8;
+use std::path::*;
+use std::io::prelude::*;
+use std::fs::File;
+use glob::glob;
+use rustc_serialize::*;
+
+pub mod trie;
+
+pub trait JsonTest: Sized {
+	type Input;
+	type Output;
+
+	fn new(data: &[u8]) -> Self;
+	fn input(&self) -> Self::Input;
+	fn output(&self) -> Self::Output;
+}
+
+pub struct JsonLoader {
+	json: json::Json
+}
+
+impl JsonTest for JsonLoader {
+	type Input = json::Json;
+	type Output = json::Json;
+
+	fn new(data: &[u8]) -> Self {
+		JsonLoader {
+			json: json::Json::from_str(from_utf8(data).unwrap()).unwrap()
+		}
+	}
+	fn input(&self) -> Self::Input {
+		self.json.as_object().unwrap()["input"].clone()
+	}
+
+	fn output(&self) -> Self::Output {
+		self.json.as_object().unwrap()["output"].clone()
+	}
+}
+
+pub fn execute_test<T, F>(data: &[u8], f: &mut F) where T: JsonTest, F: FnMut(T::Input, T::Output) {
+	let test = T::new(data);
+	f(test.input(), test.output())
+}
+
+pub fn execute_test_from_file<T, F>(path: &Path, f: &mut F) where T: JsonTest, F: FnMut(T::Input, T::Output) {
+	let mut file = File::open(path).unwrap();
+	let mut buffer = vec![];
+	let _  = file.read_to_end(&mut buffer);
+	let test = T::new(&buffer);
+	f(test.input(), test.output())
+}
+
+pub fn execute_tests_in_directory<T, F>(pattern: &str, f: &mut F) where T: JsonTest, F: FnMut(String, T::Input, T::Output) {
+	for path in glob(pattern).unwrap().filter_map(Result::ok) {
+		execute_test_from_file::<T, _>(&path, &mut | input, output | {
+			f(path.to_str().unwrap().to_string(), input, output);
+		});
+	}
+}
+
+#[test]
+fn it_works() {
+}
+

json-tests/src/trie.rs

@@ -0,0 +1,113 @@
+//! json trie tests
+use std::collections::HashMap;
+use rustc_serialize::*;
+use rustc_serialize::hex::FromHex;
+use super::{JsonTest, JsonLoader};
+
+pub enum OperationType {
+	Insert,
+	Remove
+}
+
+impl Decodable for OperationType {
+	fn decode<D>(d: &mut D) -> Result<OperationType, D::Error> where D: Decoder {
+		match try!(String::decode(d)).as_ref() {
+			"insert" => Ok(OperationType::Insert),
+			"remove" => Ok(OperationType::Remove),
+			other => panic!("invalid operation type: {}", other)
+		}
+	}
+}
+
+#[derive(RustcDecodable)]
+struct RawOperation {
+	operation: OperationType,
+	key: String,
+	value: Option<String>
+}
+
+pub struct Operation {
+	pub operation: OperationType,
+	pub key: Vec<u8>,
+	pub value: Option<Vec<u8>>
+}
+
+fn hex_or_string(s: &str) -> Vec<u8> {
+	match s.starts_with("0x") {
+		true => s[2..].from_hex().unwrap(),
+		false => From::from(s)
+	}
+}
+
+impl Into<Operation> for RawOperation {
+	fn into(self) -> Operation {
+		Operation {
+			operation: self.operation,
+			key: hex_or_string(&self.key),
+			value: self.value.map(|v| {
+				hex_or_string(&v)
+			})
+		}
+	}
+}
+
+pub struct TrieTest {
+	loader: JsonLoader
+}
+
+impl JsonTest for TrieTest {
+	type Input = Vec<Operation>;
+	type Output = Vec<u8>;
+	
+	fn new(data: &[u8]) -> Self {
+		TrieTest {
+			loader: JsonLoader::new(data) 
+		}
+	}
+
+	fn input(&self) -> Self::Input {
+		let mut decoder = json::Decoder::new(self.loader.input());
+		let raw: Vec<RawOperation> = Decodable::decode(&mut decoder).unwrap();
+		raw.into_iter()
+			.map(|i| i.into())
+			.collect()
+	}
+	fn output(&self) -> Self::Output {
+		hex_or_string(self.loader.output().as_string().unwrap())
+	}
+}
+
+pub struct TriehashTest {
+	trietest: TrieTest
+}
+
+impl JsonTest for TriehashTest {
+	type Input = Vec<(Vec<u8>, Vec<u8>)>;
+	type Output = Vec<u8>;
+
+	fn new(data: &[u8]) -> Self {
+		TriehashTest {
+			trietest: TrieTest::new(data)
+		}
+	}
+
+	fn input(&self) -> Self::Input {
+		self.trietest.input()
+			.into_iter()
+			.fold(HashMap::new(), | mut map, o | {
+				match o.operation {
+					OperationType::Insert => map.insert(o.key, o.value.unwrap()),
+					OperationType::Remove => map.remove(&o.key)
+				};
+				map
+			})
+			.into_iter()
+			.map(|p| { p })
+			.collect()
+	}
+
+	fn output(&self) -> Self::Output {
+		self.trietest.output()
+	}
+}
+