123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395 |
- <?php
- namespace Illuminate\Support\Testing\Fakes;
- use Closure;
- use Exception;
- use Illuminate\Contracts\Notifications\Dispatcher as NotificationDispatcher;
- use Illuminate\Contracts\Notifications\Factory as NotificationFactory;
- use Illuminate\Contracts\Queue\ShouldQueue;
- use Illuminate\Contracts\Translation\HasLocalePreference;
- use Illuminate\Notifications\AnonymousNotifiable;
- use Illuminate\Support\Collection;
- use Illuminate\Support\Str;
- use Illuminate\Support\Traits\Macroable;
- use Illuminate\Support\Traits\ReflectsClosures;
- use PHPUnit\Framework\Assert as PHPUnit;
- class NotificationFake implements Fake, NotificationDispatcher, NotificationFactory
- {
- use Macroable, ReflectsClosures;
- /**
- * All of the notifications that have been sent.
- *
- * @var array
- */
- protected $notifications = [];
- /**
- * Locale used when sending notifications.
- *
- * @var string|null
- */
- public $locale;
- /**
- * Indicates if notifications should be serialized and restored when pushed to the queue.
- *
- * @var bool
- */
- protected $serializeAndRestore = false;
- /**
- * Assert if a notification was sent on-demand based on a truth-test callback.
- *
- * @param string|\Closure $notification
- * @param callable|null $callback
- * @return void
- *
- * @throws \Exception
- */
- public function assertSentOnDemand($notification, $callback = null)
- {
- $this->assertSentTo(new AnonymousNotifiable, $notification, $callback);
- }
- /**
- * Assert if a notification was sent based on a truth-test callback.
- *
- * @param mixed $notifiable
- * @param string|\Closure $notification
- * @param callable|null $callback
- * @return void
- *
- * @throws \Exception
- */
- public function assertSentTo($notifiable, $notification, $callback = null)
- {
- if (is_array($notifiable) || $notifiable instanceof Collection) {
- if (count($notifiable) === 0) {
- throw new Exception('No notifiable given.');
- }
- foreach ($notifiable as $singleNotifiable) {
- $this->assertSentTo($singleNotifiable, $notification, $callback);
- }
- return;
- }
- if ($notification instanceof Closure) {
- [$notification, $callback] = [$this->firstClosureParameterType($notification), $notification];
- }
- if (is_numeric($callback)) {
- return $this->assertSentToTimes($notifiable, $notification, $callback);
- }
- PHPUnit::assertTrue(
- $this->sent($notifiable, $notification, $callback)->count() > 0,
- "The expected [{$notification}] notification was not sent."
- );
- }
- /**
- * Assert if a notification was sent on-demand a number of times.
- *
- * @param string $notification
- * @param int $times
- * @return void
- */
- public function assertSentOnDemandTimes($notification, $times = 1)
- {
- return $this->assertSentToTimes(new AnonymousNotifiable, $notification, $times);
- }
- /**
- * Assert if a notification was sent a number of times.
- *
- * @param mixed $notifiable
- * @param string $notification
- * @param int $times
- * @return void
- */
- public function assertSentToTimes($notifiable, $notification, $times = 1)
- {
- $count = $this->sent($notifiable, $notification)->count();
- PHPUnit::assertSame(
- $times, $count,
- "Expected [{$notification}] to be sent {$times} times, but was sent {$count} times."
- );
- }
- /**
- * Determine if a notification was sent based on a truth-test callback.
- *
- * @param mixed $notifiable
- * @param string|\Closure $notification
- * @param callable|null $callback
- * @return void
- *
- * @throws \Exception
- */
- public function assertNotSentTo($notifiable, $notification, $callback = null)
- {
- if (is_array($notifiable) || $notifiable instanceof Collection) {
- if (count($notifiable) === 0) {
- throw new Exception('No notifiable given.');
- }
- foreach ($notifiable as $singleNotifiable) {
- $this->assertNotSentTo($singleNotifiable, $notification, $callback);
- }
- return;
- }
- if ($notification instanceof Closure) {
- [$notification, $callback] = [$this->firstClosureParameterType($notification), $notification];
- }
- PHPUnit::assertCount(
- 0, $this->sent($notifiable, $notification, $callback),
- "The unexpected [{$notification}] notification was sent."
- );
- }
- /**
- * Assert that no notifications were sent.
- *
- * @return void
- */
- public function assertNothingSent()
- {
- PHPUnit::assertEmpty($this->notifications, 'Notifications were sent unexpectedly.');
- }
- /**
- * Assert that no notifications were sent to the given notifiable.
- *
- * @param mixed $notifiable
- * @return void
- *
- * @throws \Exception
- */
- public function assertNothingSentTo($notifiable)
- {
- if (is_array($notifiable) || $notifiable instanceof Collection) {
- if (count($notifiable) === 0) {
- throw new Exception('No notifiable given.');
- }
- foreach ($notifiable as $singleNotifiable) {
- $this->assertNothingSentTo($singleNotifiable);
- }
- return;
- }
- PHPUnit::assertEmpty(
- $this->notifications[get_class($notifiable)][$notifiable->getKey()] ?? [],
- 'Notifications were sent unexpectedly.',
- );
- }
- /**
- * Assert the total amount of times a notification was sent.
- *
- * @param string $notification
- * @param int $expectedCount
- * @return void
- */
- public function assertSentTimes($notification, $expectedCount)
- {
- $actualCount = collect($this->notifications)
- ->flatten(1)
- ->reduce(fn ($count, $sent) => $count + count($sent[$notification] ?? []), 0);
- PHPUnit::assertSame(
- $expectedCount, $actualCount,
- "Expected [{$notification}] to be sent {$expectedCount} times, but was sent {$actualCount} times."
- );
- }
- /**
- * Assert the total count of notification that were sent.
- *
- * @param int $expectedCount
- * @return void
- */
- public function assertCount($expectedCount)
- {
- $actualCount = collect($this->notifications)->flatten(3)->count();
- PHPUnit::assertSame(
- $expectedCount, $actualCount,
- "Expected {$expectedCount} notifications to be sent, but {$actualCount} were sent."
- );
- }
- /**
- * Get all of the notifications matching a truth-test callback.
- *
- * @param mixed $notifiable
- * @param string $notification
- * @param callable|null $callback
- * @return \Illuminate\Support\Collection
- */
- public function sent($notifiable, $notification, $callback = null)
- {
- if (! $this->hasSent($notifiable, $notification)) {
- return collect();
- }
- $callback = $callback ?: fn () => true;
- $notifications = collect($this->notificationsFor($notifiable, $notification));
- return $notifications->filter(
- fn ($arguments) => $callback(...array_values($arguments))
- )->pluck('notification');
- }
- /**
- * Determine if there are more notifications left to inspect.
- *
- * @param mixed $notifiable
- * @param string $notification
- * @return bool
- */
- public function hasSent($notifiable, $notification)
- {
- return ! empty($this->notificationsFor($notifiable, $notification));
- }
- /**
- * Get all of the notifications for a notifiable entity by type.
- *
- * @param mixed $notifiable
- * @param string $notification
- * @return array
- */
- protected function notificationsFor($notifiable, $notification)
- {
- return $this->notifications[get_class($notifiable)][$notifiable->getKey()][$notification] ?? [];
- }
- /**
- * Send the given notification to the given notifiable entities.
- *
- * @param \Illuminate\Support\Collection|array|mixed $notifiables
- * @param mixed $notification
- * @return void
- */
- public function send($notifiables, $notification)
- {
- $this->sendNow($notifiables, $notification);
- }
- /**
- * Send the given notification immediately.
- *
- * @param \Illuminate\Support\Collection|array|mixed $notifiables
- * @param mixed $notification
- * @param array|null $channels
- * @return void
- */
- public function sendNow($notifiables, $notification, ?array $channels = null)
- {
- if (! $notifiables instanceof Collection && ! is_array($notifiables)) {
- $notifiables = [$notifiables];
- }
- foreach ($notifiables as $notifiable) {
- if (! $notification->id) {
- $notification->id = Str::uuid()->toString();
- }
- $notifiableChannels = $channels ?: $notification->via($notifiable);
- if (method_exists($notification, 'shouldSend')) {
- $notifiableChannels = array_filter(
- $notifiableChannels,
- fn ($channel) => $notification->shouldSend($notifiable, $channel) !== false
- );
- }
- if (empty($notifiableChannels)) {
- continue;
- }
- $this->notifications[get_class($notifiable)][$notifiable->getKey()][get_class($notification)][] = [
- 'notification' => $this->serializeAndRestore && $notification instanceof ShouldQueue
- ? $this->serializeAndRestoreNotification($notification)
- : $notification,
- 'channels' => $notifiableChannels,
- 'notifiable' => $notifiable,
- 'locale' => $notification->locale ?? $this->locale ?? value(function () use ($notifiable) {
- if ($notifiable instanceof HasLocalePreference) {
- return $notifiable->preferredLocale();
- }
- }),
- ];
- }
- }
- /**
- * Get a channel instance by name.
- *
- * @param string|null $name
- * @return mixed
- */
- public function channel($name = null)
- {
- //
- }
- /**
- * Set the locale of notifications.
- *
- * @param string $locale
- * @return $this
- */
- public function locale($locale)
- {
- $this->locale = $locale;
- return $this;
- }
- /**
- * Specify if notification should be serialized and restored when being "pushed" to the queue.
- *
- * @param bool $serializeAndRestore
- * @return $this
- */
- public function serializeAndRestore(bool $serializeAndRestore = true)
- {
- $this->serializeAndRestore = $serializeAndRestore;
- return $this;
- }
- /**
- * Serialize and unserialize the notification to simulate the queueing process.
- *
- * @param mixed $notification
- * @return mixed
- */
- protected function serializeAndRestoreNotification($notification)
- {
- return unserialize(serialize($notification));
- }
- /**
- * Get the notifications that have been sent.
- *
- * @return array
- */
- public function sentNotifications()
- {
- return $this->notifications;
- }
- }
|