summaryrefslogtreecommitdiff
path: root/solutions/23_conversions/from_str.rs
diff options
context:
space:
mode:
authormo8it <mo8it@proton.me>2024-07-05 13:39:50 +0200
committermo8it <mo8it@proton.me>2024-07-05 13:39:50 +0200
commit7123c7ae3a9605fbe962e4ef0a0f1424cd16fef8 (patch)
treec67f7e62bb9a179ae4fdbab492501cb6847e64c7 /solutions/23_conversions/from_str.rs
parent77b687d501771c24bd83294d97b8e6f9ffa92d6b (diff)
parent4d9c346a173bb722b929f3ea3c00f84954483e24 (diff)
Merge remote-tracking branch 'upstream/main' into fix-enum-variant-inconsistency
Diffstat (limited to 'solutions/23_conversions/from_str.rs')
-rw-r--r--solutions/23_conversions/from_str.rs117
1 files changed, 117 insertions, 0 deletions
diff --git a/solutions/23_conversions/from_str.rs b/solutions/23_conversions/from_str.rs
new file mode 100644
index 0000000..005b501
--- /dev/null
+++ b/solutions/23_conversions/from_str.rs
@@ -0,0 +1,117 @@
+// This is similar to the previous `from_into` exercise. But this time, we'll
+// implement `FromStr` and return errors instead of falling back to a default
+// value. Additionally, upon implementing `FromStr`, you can use the `parse`
+// method on strings to generate an object of the implementor type. You can read
+// more about it in the documentation:
+// https://doc.rust-lang.org/std/str/trait.FromStr.html
+
+use std::num::ParseIntError;
+use std::str::FromStr;
+
+#[derive(Debug, PartialEq)]
+struct Person {
+ name: String,
+ age: u8,
+}
+
+// We will use this error type for the `FromStr` implementation.
+#[derive(Debug, PartialEq)]
+enum ParsePersonError {
+ // Incorrect number of fields
+ BadLen,
+ // Empty name field
+ NoName,
+ // Wrapped error from parse::<u8>()
+ ParseInt(ParseIntError),
+}
+
+impl FromStr for Person {
+ type Err = ParsePersonError;
+
+ fn from_str(s: &str) -> Result<Self, Self::Err> {
+ let mut split = s.split(',');
+ let (Some(name), Some(age), None) = (split.next(), split.next(), split.next()) else {
+ // ^^^^ there should be no third element
+ return Err(ParsePersonError::BadLen);
+ };
+
+ if name.is_empty() {
+ return Err(ParsePersonError::NoName);
+ }
+
+ let age = age.parse().map_err(ParsePersonError::ParseInt)?;
+
+ Ok(Self {
+ name: name.into(),
+ age,
+ })
+ }
+}
+
+fn main() {
+ let p = "Mark,20".parse::<Person>();
+ println!("{p:?}");
+}
+
+#[cfg(test)]
+mod tests {
+ use super::*;
+ use ParsePersonError::*;
+
+ #[test]
+ fn empty_input() {
+ assert_eq!("".parse::<Person>(), Err(BadLen));
+ }
+
+ #[test]
+ fn good_input() {
+ let p = "John,32".parse::<Person>();
+ assert!(p.is_ok());
+ let p = p.unwrap();
+ assert_eq!(p.name, "John");
+ assert_eq!(p.age, 32);
+ }
+
+ #[test]
+ fn missing_age() {
+ assert!(matches!("John,".parse::<Person>(), Err(ParseInt(_))));
+ }
+
+ #[test]
+ fn invalid_age() {
+ assert!(matches!("John,twenty".parse::<Person>(), Err(ParseInt(_))));
+ }
+
+ #[test]
+ fn missing_comma_and_age() {
+ assert_eq!("John".parse::<Person>(), Err(BadLen));
+ }
+
+ #[test]
+ fn missing_name() {
+ assert_eq!(",1".parse::<Person>(), Err(NoName));
+ }
+
+ #[test]
+ fn missing_name_and_age() {
+ assert!(matches!(",".parse::<Person>(), Err(NoName | ParseInt(_))));
+ }
+
+ #[test]
+ fn missing_name_and_invalid_age() {
+ assert!(matches!(
+ ",one".parse::<Person>(),
+ Err(NoName | ParseInt(_)),
+ ));
+ }
+
+ #[test]
+ fn trailing_comma() {
+ assert_eq!("John,32,".parse::<Person>(), Err(BadLen));
+ }
+
+ #[test]
+ fn trailing_comma_and_some_string() {
+ assert_eq!("John,32,man".parse::<Person>(), Err(BadLen));
+ }
+}