From aef63bfecb7731c0307cc65eab0b9a06b8a7363d Mon Sep 17 00:00:00 2001
From: HampusM <hampus@hampusmat.com>
Date: Sat, 25 Mar 2023 11:58:16 +0100
Subject: feat: add argument matching

---
 examples/generic_method.rs | 25 +++++++++++++++++++++++--
 1 file changed, 23 insertions(+), 2 deletions(-)

(limited to 'examples/generic_method.rs')

diff --git a/examples/generic_method.rs b/examples/generic_method.rs
index 4262ee2..7283810 100644
--- a/examples/generic_method.rs
+++ b/examples/generic_method.rs
@@ -1,12 +1,14 @@
 use std::fmt::Display;
 
+use predicates::float::is_close;
 use ridicule::mock;
+use ridicule::predicate::function;
 
 trait Foo
 {
     fn bar<Baz: Display>(&self, num: u128) -> Baz;
 
-    fn abc<Baz: Display>(&mut self, baz: Baz);
+    fn abc<ThingA, ThingB>(&mut self, thing_a: ThingA, thing_b: ThingB);
 }
 
 mock! {
@@ -16,7 +18,7 @@ mock! {
     {
         fn bar<Baz: Display>(&self, num: u128) -> Baz;
 
-        fn abc<Baz: Display>(&mut self, baz_baz: Baz);
+        fn abc<ThingA, ThingB>(&mut self, thing_a: ThingA, thing_b: ThingB);
     }
 }
 
@@ -39,6 +41,17 @@ fn main()
         128u8
     });
 
+    mock_foo
+        .expect_abc::<&str, f64>()
+        .with(
+            function(|thing_a: &&str| thing_a.starts_with("Good morning")),
+            is_close(7.13081),
+        )
+        .returning(|_me, _thing_a, _thing_b| {
+            println!("abc was called");
+        })
+        .times(1);
+
     assert_eq!(mock_foo.bar::<String>(123), "Hello".to_string());
     assert_eq!(mock_foo.bar::<String>(123), "Hello".to_string());
     assert_eq!(mock_foo.bar::<String>(123), "Hello".to_string());
@@ -47,4 +60,12 @@ fn main()
     // mock_foo.bar::<String>(123);
 
     assert_eq!(mock_foo.bar::<u8>(456), 128);
+
+    mock_foo.abc(
+        concat!(
+            "Good morning, and in case I don't see ya, good afternoon,",
+            " good evening, and good night!"
+        ),
+        7.13081f64,
+    );
 }
-- 
cgit v1.2.3-18-g5258