feat(mock): updates tests

src/Mock.php

@@ -4,6 +4,7 @@ declare(strict_types=1);
 
 namespace Pest;
 
+use InvalidArgumentException;
 use Mockery;
 use Pest\Exceptions\MissingDependency;
 
@@ -44,11 +45,17 @@ final class Mock
      */
     public function expect(...$methods)
     {
-        foreach ($methods as $method => $result) {
+        foreach ($methods as $method => $expectation) {
             /* @phpstan-ignore-next-line */
-            $this->mock
+            $method = $this->mock
                 ->shouldReceive((string) $method)
-                ->andReturn($result);
+                ->once();
+
+            if (!is_callable($expectation)) {
+                throw new InvalidArgumentException(sprintf('Method %s from %s expects a callable as expectation.', $method, $method->mock()->mockery_getName(), ));
+            }
+
+            $method->andReturnUsing($expectation);
         }
 
         return $this->mock;

tests/.snapshots/success.txt

@@ -79,7 +79,7 @@
 
    PASS  Tests\Features\Mocks
   ✓ it can mock methods
-  ✓ access to the mock object
+  ✓ it allows access to the underlying mockery mock
 
    PASS  Tests\Features\PendingHigherOrderTests
   ✓ get 'foo' → get 'bar' → expect true → toBeTrue 

tests/Features/Mocks.php

@@ -10,15 +10,15 @@ interface Http
 
 it('can mock methods', function () {
     $mock = mock(Http::class)->expect(
-        get: 'foo',
+        get: fn () => 'foo',
     );
 
     expect($mock->get())->toBe('foo');
 })->skip(((float) phpversion()) < 8.0);
 
-test('access to the mock object', function () {
+it('allows access to the underlying mockery mock', function () {
     $mock = mock(Http::class);
-    expect($mock->expect())->toBeInstanceOf(MockInterface::class);
 
+    expect($mock->expect())->toBeInstanceOf(MockInterface::class);
     expect($mock->shouldReceive())->toBeInstanceOf(CompositeExpectation::class);
 })->skip(((float) phpversion()) < 8.0);