Source code

Revision control

Copy as Markdown

Other Tools

use std::collections::HashMap;
use ron::{
de::from_str,
error::Error,
extensions::Extensions,
ser::{to_string_pretty, PrettyConfig},
};
use serde::{Deserialize, Serialize};
#[derive(Debug, Deserialize, Serialize, Eq, PartialEq)]
enum TestEnum {
Unit,
PrimitiveNewtype(String),
Tuple(u32, bool),
Struct { a: u32, b: bool },
TupleNewtypeUnit(Unit),
TupleNewtypeNewtype(Newtype),
TupleNewtypeTuple((u32, bool)),
TupleNewtypeTupleStruct(TupleStruct),
TupleNewtypeStruct(Struct),
TupleNewtypeEnum(Enum),
TupleNewtypeOption(Option<Struct>),
TupleNewtypeSeq(Vec<Struct>),
TupleNewtypeMap(HashMap<u32, Struct>),
}
#[derive(Debug, Deserialize, Serialize, Eq, PartialEq)]
struct Unit;
#[derive(Debug, Deserialize, Serialize, Eq, PartialEq)]
struct Newtype(i32);
#[derive(Debug, Deserialize, Serialize, Eq, PartialEq)]
struct TupleStruct(u32, bool);
#[derive(Debug, Deserialize, Serialize, Eq, PartialEq)]
struct Struct {
a: u32,
b: bool,
}
#[derive(Debug, Deserialize, Serialize, Eq, PartialEq)]
enum Enum {
A,
B(Struct),
C(u32, bool),
D { a: u32, b: bool },
}
#[test]
fn test_deserialise_non_newtypes() {
assert_eq!(
from_str::<TestEnum>(r#"#![enable(unwrap_variant_newtypes)] Unit"#).unwrap(),
TestEnum::Unit,
);
assert_eq!(
from_str::<TestEnum>(r#"#![enable(unwrap_variant_newtypes)] PrimitiveNewtype("hi")"#)
.unwrap(),
TestEnum::PrimitiveNewtype(String::from("hi")),
);
assert_eq!(
from_str::<TestEnum>(r#"#![enable(unwrap_variant_newtypes)] Tuple(4, false)"#).unwrap(),
TestEnum::Tuple(4, false),
);
assert_eq!(
from_str::<TestEnum>(r#"#![enable(unwrap_variant_newtypes)] Struct(a: 4, b: false)"#)
.unwrap(),
TestEnum::Struct { a: 4, b: false },
);
}
#[test]
fn test_deserialise_tuple_newtypes() {
assert_eq!(
from_str::<TestEnum>(r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeUnit(Unit)"#)
.unwrap_err()
.code,
Error::ExpectedStructLikeEnd,
);
assert_eq!(
from_str::<TestEnum>(r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeUnit(())"#)
.unwrap_err()
.code,
Error::ExpectedStructLikeEnd,
);
assert_eq!(
from_str::<TestEnum>(r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeUnit()"#).unwrap(),
TestEnum::TupleNewtypeUnit(Unit),
);
assert_eq!(
from_str::<TestEnum>(
r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeNewtype(Newtype(4))"#
)
.unwrap_err()
.code,
Error::ExpectedInteger,
);
assert_eq!(
from_str::<TestEnum>(r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeNewtype((4))"#)
.unwrap_err()
.code,
Error::ExpectedInteger,
);
assert_eq!(
from_str::<TestEnum>(r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeNewtype(4)"#)
.unwrap(),
TestEnum::TupleNewtypeNewtype(Newtype(4)),
);
assert_eq!(
from_str::<TestEnum>(r#"#![enable(unwrap_newtypes)] TupleNewtypeNewtype(4)"#).unwrap(),
TestEnum::TupleNewtypeNewtype(Newtype(4)),
);
assert_eq!(
from_str::<TestEnum>(r#"#![enable(unwrap_newtypes)] #![enable(unwrap_variant_newtypes)] TupleNewtypeNewtype(4)"#).unwrap(),
TestEnum::TupleNewtypeNewtype(Newtype(4)),
);
assert_eq!(
from_str::<TestEnum>(
r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeTuple((4, false))"#
)
.unwrap_err()
.code,
Error::ExpectedInteger,
);
assert_eq!(
from_str::<TestEnum>(r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeTuple(4, false)"#)
.unwrap(),
TestEnum::TupleNewtypeTuple((4, false)),
);
assert_eq!(
from_str::<TestEnum>(
r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeTupleStruct(TupleStruct(4, false))"#
)
.unwrap_err()
.code,
Error::ExpectedInteger,
);
assert_eq!(
from_str::<TestEnum>(
r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeTupleStruct((4, false))"#
)
.unwrap_err()
.code,
Error::ExpectedInteger,
);
assert_eq!(
from_str::<TestEnum>(
r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeTupleStruct(4, false)"#
)
.unwrap(),
TestEnum::TupleNewtypeTupleStruct(TupleStruct(4, false)),
);
assert_eq!(
from_str::<TestEnum>(
r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeStruct(Struct(a: 4, b: false))"#
)
.unwrap_err()
.code,
Error::ExpectedMapColon,
);
assert_eq!(
from_str::<TestEnum>(
r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeStruct((a: 4, b: false))"#
)
.unwrap_err()
.code,
Error::ExpectedIdentifier,
);
assert_eq!(
from_str::<TestEnum>(
r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeStruct(a: 4, b: false)"#
)
.unwrap(),
TestEnum::TupleNewtypeStruct(Struct { a: 4, b: false }),
);
assert_eq!(
from_str::<TestEnum>(r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeEnum(A)"#).unwrap(),
TestEnum::TupleNewtypeEnum(Enum::A),
);
assert_eq!(
from_str::<TestEnum>(
r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeEnum(B(a: 4, b: false))"#
)
.unwrap(),
TestEnum::TupleNewtypeEnum(Enum::B(Struct { a: 4, b: false })),
);
assert_eq!(
from_str::<TestEnum>(r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeEnum(C 4, false)"#)
.unwrap_err()
.code,
Error::ExpectedStructLike,
);
assert_eq!(
from_str::<TestEnum>(
r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeEnum(C(4, false))"#
)
.unwrap(),
TestEnum::TupleNewtypeEnum(Enum::C(4, false)),
);
assert_eq!(
from_str::<TestEnum>(
r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeEnum(D a: 4, b: false)"#
)
.unwrap_err()
.code,
Error::ExpectedStructLike,
);
assert_eq!(
from_str::<TestEnum>(
r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeEnum(D(a: 4, b: false))"#
)
.unwrap(),
TestEnum::TupleNewtypeEnum(Enum::D { a: 4, b: false }),
);
assert_eq!(
from_str::<TestEnum>(r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeOption(None)"#)
.unwrap(),
TestEnum::TupleNewtypeOption(None),
);
assert_eq!(
from_str::<TestEnum>(
r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeOption(Some(a: 4, b: false))"#
)
.unwrap(),
TestEnum::TupleNewtypeOption(Some(Struct { a: 4, b: false })),
);
assert_eq!(
from_str::<TestEnum>(
r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeOption(a: 4, b: false)"#
)
.unwrap_err()
.code,
Error::ExpectedOption,
);
assert_eq!(
from_str::<TestEnum>(r#"#![enable(unwrap_variant_newtypes, implicit_some)] TupleNewtypeOption(a: 4, b: false)"#).unwrap(),
TestEnum::TupleNewtypeOption(Some(Struct { a: 4, b: false })),
);
assert_eq!(
from_str::<TestEnum>(r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeSeq([])"#).unwrap(),
TestEnum::TupleNewtypeSeq(vec![]),
);
assert_eq!(
from_str::<TestEnum>(
r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeSeq([(a: 4, b: false)])"#
)
.unwrap(),
TestEnum::TupleNewtypeSeq(vec![Struct { a: 4, b: false }]),
);
assert_eq!(
from_str::<TestEnum>(
r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeSeq([Struct(a: 4, b: false)])"#
)
.unwrap(),
TestEnum::TupleNewtypeSeq(vec![Struct { a: 4, b: false }]),
);
assert_eq!(
from_str::<TestEnum>(r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeMap({})"#).unwrap(),
TestEnum::TupleNewtypeMap(vec![].into_iter().collect()),
);
assert_eq!(
from_str::<TestEnum>(
r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeMap({2: (a: 4, b: false)})"#
)
.unwrap(),
TestEnum::TupleNewtypeMap(vec![(2, Struct { a: 4, b: false })].into_iter().collect()),
);
assert_eq!(
from_str::<TestEnum>(
r#"#![enable(unwrap_variant_newtypes)] TupleNewtypeMap({8: Struct(a: 4, b: false)})"#
)
.unwrap(),
TestEnum::TupleNewtypeMap(vec![(8, Struct { a: 4, b: false })].into_iter().collect()),
);
}
#[test]
fn test_serialise_non_newtypes() {
assert_eq_serialize_roundtrip(TestEnum::Unit, Extensions::UNWRAP_VARIANT_NEWTYPES);
assert_eq_serialize_roundtrip(
TestEnum::PrimitiveNewtype(String::from("hi")),
Extensions::UNWRAP_VARIANT_NEWTYPES,
);
assert_eq_serialize_roundtrip(
TestEnum::Tuple(4, false),
Extensions::UNWRAP_VARIANT_NEWTYPES,
);
assert_eq_serialize_roundtrip(
TestEnum::Struct { a: 4, b: false },
Extensions::UNWRAP_VARIANT_NEWTYPES,
);
}
#[test]
fn test_serialise_tuple_newtypes() {
assert_eq_serialize_roundtrip(
TestEnum::TupleNewtypeUnit(Unit),
Extensions::UNWRAP_VARIANT_NEWTYPES,
);
assert_eq_serialize_roundtrip(
TestEnum::TupleNewtypeNewtype(Newtype(4)),
Extensions::UNWRAP_VARIANT_NEWTYPES,
);
assert_eq_serialize_roundtrip(
TestEnum::TupleNewtypeNewtype(Newtype(4)),
Extensions::UNWRAP_NEWTYPES,
);
assert_eq_serialize_roundtrip(
TestEnum::TupleNewtypeNewtype(Newtype(4)),
Extensions::UNWRAP_VARIANT_NEWTYPES | Extensions::UNWRAP_NEWTYPES,
);
assert_eq_serialize_roundtrip(
TestEnum::TupleNewtypeTuple((4, false)),
Extensions::UNWRAP_VARIANT_NEWTYPES,
);
assert_eq_serialize_roundtrip(
TestEnum::TupleNewtypeTupleStruct(TupleStruct(4, false)),
Extensions::UNWRAP_VARIANT_NEWTYPES,
);
assert_eq_serialize_roundtrip(
TestEnum::TupleNewtypeStruct(Struct { a: 4, b: false }),
Extensions::UNWRAP_VARIANT_NEWTYPES,
);
assert_eq_serialize_roundtrip(
TestEnum::TupleNewtypeEnum(Enum::A),
Extensions::UNWRAP_VARIANT_NEWTYPES,
);
assert_eq_serialize_roundtrip(
TestEnum::TupleNewtypeEnum(Enum::B(Struct { a: 4, b: false })),
Extensions::UNWRAP_VARIANT_NEWTYPES,
);
assert_eq_serialize_roundtrip(
TestEnum::TupleNewtypeEnum(Enum::C(4, false)),
Extensions::UNWRAP_VARIANT_NEWTYPES,
);
assert_eq_serialize_roundtrip(
TestEnum::TupleNewtypeEnum(Enum::D { a: 4, b: false }),
Extensions::UNWRAP_VARIANT_NEWTYPES,
);
assert_eq_serialize_roundtrip(
TestEnum::TupleNewtypeOption(None),
Extensions::UNWRAP_VARIANT_NEWTYPES,
);
assert_eq_serialize_roundtrip(
TestEnum::TupleNewtypeOption(Some(Struct { a: 4, b: false })),
Extensions::UNWRAP_VARIANT_NEWTYPES,
);
assert_eq_serialize_roundtrip(
TestEnum::TupleNewtypeOption(Some(Struct { a: 4, b: false })),
Extensions::IMPLICIT_SOME,
);
assert_eq_serialize_roundtrip(
TestEnum::TupleNewtypeOption(Some(Struct { a: 4, b: false })),
Extensions::UNWRAP_VARIANT_NEWTYPES | Extensions::IMPLICIT_SOME,
);
assert_eq_serialize_roundtrip(
TestEnum::TupleNewtypeSeq(vec![]),
Extensions::UNWRAP_VARIANT_NEWTYPES,
);
assert_eq_serialize_roundtrip(
TestEnum::TupleNewtypeSeq(vec![Struct { a: 4, b: false }]),
Extensions::UNWRAP_VARIANT_NEWTYPES,
);
assert_eq_serialize_roundtrip(
TestEnum::TupleNewtypeMap(vec![].into_iter().collect()),
Extensions::UNWRAP_VARIANT_NEWTYPES,
);
assert_eq_serialize_roundtrip(
TestEnum::TupleNewtypeMap(vec![(2, Struct { a: 4, b: false })].into_iter().collect()),
Extensions::UNWRAP_VARIANT_NEWTYPES,
);
}
fn assert_eq_serialize_roundtrip<
S: Serialize + serde::de::DeserializeOwned + PartialEq + std::fmt::Debug,
>(
value: S,
extensions: Extensions,
) {
assert_eq!(
from_str::<S>(
&to_string_pretty(&value, PrettyConfig::default().extensions(extensions)).unwrap()
)
.unwrap(),
value,
);
}