feat: allow for usage of associated types of generics

1 files changed, 23 insertions, 10 deletions

diff --git a/examples/generic_method.rs b/examples/generic_method.rs
index 7283810..c4e3214 100644
--- a/examples/generic_method.rs
+++ b/examples/generic_method.rs
@@ -1,12 +1,15 @@
-use std::fmt::Display;
-
 use predicates::float::is_close;
 use ridicule::mock;
 use ridicule::predicate::function;
 
+trait Haha
+{
+    type Hello;
+}
+
 trait Foo
 {
-    fn bar<Baz: Display>(&self, num: u128) -> Baz;
+    fn bar<Baz: Haha>(&self, num: u128) -> Baz::Hello;
 
     fn abc<ThingA, ThingB>(&mut self, thing_a: ThingA, thing_b: ThingB);
 }
@@ -16,18 +19,28 @@ mock! {
 
     impl Foo for MockFoo
     {
-        fn bar<Baz: Display>(&self, num: u128) -> Baz;
+        fn bar<Baz: Haha>(&self, num: u128) -> Baz::Hello;
 
         fn abc<ThingA, ThingB>(&mut self, thing_a: ThingA, thing_b: ThingB);
     }
 }
 
+impl Haha for u16
+{
+    type Hello = String;
+}
+
+impl Haha for &str
+{
+    type Hello = u8;
+}
+
 fn main()
 {
     let mut mock_foo = MockFoo::new();
 
     mock_foo
-        .expect_bar()
+        .expect_bar::<u16>()
         .returning(|_me, num| {
             println!("bar was called with {num}");
 
@@ -35,7 +48,7 @@ fn main()
         })
         .times(3);
 
-    mock_foo.expect_bar().returning(|_me, num| {
+    mock_foo.expect_bar::<&str>().returning(|_me, num| {
         println!("bar was called with {num}");
 
         128u8
@@ -52,14 +65,14 @@ fn main()
         })
         .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());
+    assert_eq!(mock_foo.bar::<u16>(123), "Hello".to_string());
+    assert_eq!(mock_foo.bar::<u16>(123), "Hello".to_string());
+    assert_eq!(mock_foo.bar::<u16>(123), "Hello".to_string());
 
     // Would panic
     // mock_foo.bar::<String>(123);
 
-    assert_eq!(mock_foo.bar::<u8>(456), 128);
+    assert_eq!(mock_foo.bar::<&str>(456), 128);
 
     mock_foo.abc(
         concat!(