feat: implement type inclusion for expectFilter mocks

10up · Feb 1, 2025 · 34b2e0f · 34b2e0f
1 parent 010d556
commit 34b2e0f
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/php/WP_Mock/Functions.php b/php/WP_Mock/Functions.php
@@ -413,6 +413,9 @@ public static function anyOf(): AnyOf
      */
     public static function type(string $expected): Type
     {
-        return Mockery::type($expected);
+        $type = Mockery::type($expected);
+        Filter::$objects[ $expected ] = spl_object_hash($type);
+
+        return $type;
     }
 }
diff --git a/php/WP_Mock/Hook.php b/php/WP_Mock/Hook.php
@@ -22,6 +22,9 @@ abstract class Hook
     /** @var array<mixed> collection of processors */
     protected $processors = [];
 
+    /** @var array<mixed> collection of objects mapped to their Type hashes */
+    public static array $objects = [];
+
     /**
      * Hook constructor.
      *
@@ -62,6 +65,14 @@ protected function safe_offset($value): string
         }
 
         if (is_object($value)){
+            if (! $value instanceof Type) {
+                $class = get_class($value);
+
+                if (isset(static::$objects[ $class ])) {
+                    return static::$objects[ $class ];
+                }
+            }
+
             return spl_object_hash($value);
         }