use a trait default implementation to avoid repeating the code

glehmann · Feb 8, 2024 · d3cc000 · d3cc000
1 parent 21adb7c
commit d3cc000
Showing 1 changed file with 10 additions and 15 deletions.
diff --git a/tests/common.rs b/tests/common.rs
@@ -1,3 +1,5 @@
+use std::ops::Deref;
+
 use predicates_tree::CaseTreeExt;
 
 pub trait TestPathChild {
@@ -11,26 +13,19 @@ impl TestPathChild for assert_fs::fixture::ChildPath {
 }
 
 pub trait TestString {
-    fn assert(&self, predicate: impl predicates::Predicate<str>) -> &Self;
-}
-
-impl TestString for str {
-    fn assert(&self, predicate: impl predicates::Predicate<str>) -> &Self {
-        if let Some(case) = predicate.find_case(false, self.as_ref()) {
-            panic!("{}", case.tree(),);
+    fn assert(&self, predicate: impl predicates::Predicate<str>) -> &Self
+    where
+        Self: Deref<Target = str>,
+    {
+        if let Some(case) = predicate.find_case(false, self) {
+            panic!("{}", case.tree());
         }
         self
     }
 }
 
-impl TestString for String {
-    fn assert(&self, predicate: impl predicates::Predicate<str>) -> &Self {
-        if let Some(case) = predicate.find_case(false, self.as_ref()) {
-            panic!("{}", case.tree(),);
-        }
-        self
-    }
-}
+impl TestString for str {}
+impl TestString for String {}
 
 #[macro_export]
 macro_rules! yage {