1 files changed, 111 insertions, 0 deletions
diff --git a/tests/encoder.rs b/tests/encoder.rs
new file mode 100644
index 0000000..808a20c
--- /dev/null
+++ b/tests/encoder.rs
@@ -0,0 +1,111 @@
+#[derive(Copy, Clone)]
+pub struct Encoder;
+
+impl toml_test_harness::Encoder for Encoder {
+    fn name(&self) -> &str {
+        "toml_edit"
+    }
+
+    fn encode(&self, data: toml_test_harness::Decoded) -> Result<String, toml_test_harness::Error> {
+        let doc = decoded_to_document(&data)?;
+        Ok(doc.to_string())
+    }
+}
+
+fn decoded_to_document(
+    decoded: &toml_test_harness::Decoded,
+) -> Result<toml_edit::Document, toml_test_harness::Error> {
+    let item = root_from_decoded(decoded)?;
+    let mut doc = toml_edit::Document::new();
+    *doc = item;
+    Ok(doc)
+}
+
+fn root_from_decoded(
+    decoded: &toml_test_harness::Decoded,
+) -> Result<toml_edit::Table, toml_test_harness::Error> {
+    match decoded {
+        toml_test_harness::Decoded::Value(_) => {
+            Err(toml_test_harness::Error::new("Root cannot be a value"))
+        }
+        toml_test_harness::Decoded::Table(value) => value
+            .iter()
+            .map(|(k, v)| {
+                let k = k.as_str();
+                let v = from_decoded(v)?;
+                Ok((k, v))
+            })
+            .collect(),
+        toml_test_harness::Decoded::Array(_) => {
+            Err(toml_test_harness::Error::new("Root cannot be an array"))
+        }
+    }
+}
+
+fn from_decoded(
+    decoded: &toml_test_harness::Decoded,
+) -> Result<toml_edit::Value, toml_test_harness::Error> {
+    let value = match decoded {
+        toml_test_harness::Decoded::Value(value) => from_decoded_value(value)?,
+        toml_test_harness::Decoded::Table(value) => {
+            toml_edit::Value::InlineTable(from_table(value)?)
+        }
+        toml_test_harness::Decoded::Array(value) => toml_edit::Value::Array(from_array(value)?),
+    };
+    Ok(value)
+}
+
+fn from_decoded_value(
+    decoded: &toml_test_harness::DecodedValue,
+) -> Result<toml_edit::Value, toml_test_harness::Error> {
+    let value: toml_edit::Value = match decoded {
+        toml_test_harness::DecodedValue::String(value) => value.into(),
+        toml_test_harness::DecodedValue::Integer(value) => value
+            .parse::<i64>()
+            .map_err(toml_test_harness::Error::new)?
+            .into(),
+        toml_test_harness::DecodedValue::Float(value) => value
+            .parse::<f64>()
+            .map_err(toml_test_harness::Error::new)?
+            .into(),
+        toml_test_harness::DecodedValue::Bool(value) => value
+            .parse::<bool>()
+            .map_err(toml_test_harness::Error::new)?
+            .into(),
+        toml_test_harness::DecodedValue::Datetime(value) => value
+            .parse::<toml_edit::Datetime>()
+            .map_err(toml_test_harness::Error::new)?
+            .into(),
+        toml_test_harness::DecodedValue::DatetimeLocal(value) => value
+            .parse::<toml_edit::Datetime>()
+            .map_err(toml_test_harness::Error::new)?
+            .into(),
+        toml_test_harness::DecodedValue::DateLocal(value) => value
+            .parse::<toml_edit::Datetime>()
+            .map_err(toml_test_harness::Error::new)?
+            .into(),
+        toml_test_harness::DecodedValue::TimeLocal(value) => value
+            .parse::<toml_edit::Datetime>()
+            .map_err(toml_test_harness::Error::new)?
+            .into(),
+    };
+    Ok(value)
+}
+
+fn from_table(
+    decoded: &std::collections::HashMap<String, toml_test_harness::Decoded>,
+) -> Result<toml_edit::InlineTable, toml_test_harness::Error> {
+    decoded
+        .iter()
+        .map(|(k, v)| {
+            let v = from_decoded(v)?;
+            Ok((k, v))
+        })
+        .collect()
+}
+
+fn from_array(
+    decoded: &[toml_test_harness::Decoded],
+) -> Result<toml_edit::Array, toml_test_harness::Error> {
+    decoded.iter().map(from_decoded).collect()
+}