public function it_count_notification_not_read(NotificationDB $notificationRepo) { $entity_id = 1; $notificationCount = 10; $notificationRepo->countNotRead($entity_id, null, null)->shouldBeCalled()->willReturn($notificationCount); $this->countNotRead($entity_id)->shouldReturn($notificationCount); }