unsupervised-scheduler/tests/Unit/Availability/AvailabilityControllerTest.php

<?php
declare(strict_types=1);

namespace Unsupervised\Schedular\Tests\Unit\Availability;

use Brain\Monkey\Functions;
use Mockery;
use Unsupervised\Schedular\Availability\AvailabilityController;
use Unsupervised\Schedular\Availability\AvailabilityRepository;
use Unsupervised\Schedular\Availability\AvailabilitySlot;
use Unsupervised\Schedular\Availability\WindowValidator;
use Unsupervised\Schedular\Offering\OfferingRepository;
use Unsupervised\Schedular\Tests\Unit\TestCase;

class AvailabilityControllerTest extends TestCase
{
    private AvailabilityRepository&Mockery\MockInterface $repository;
    private OfferingRepository&Mockery\MockInterface $offerings;
    private AvailabilityController $controller;

    protected function setUp(): void
    {
        parent::setUp();

        $this->repository = Mockery::mock(AvailabilityRepository::class);
        $this->offerings  = Mockery::mock(OfferingRepository::class);
        $this->controller = new AvailabilityController($this->repository, $this->offerings, new WindowValidator($this->offerings));

        $_POST = [];
        $_GET  = [];

        Functions\when('current_user_can')->justReturn(true);
        Functions\when('get_current_user_id')->justReturn(3);
        Functions\when('check_admin_referer')->justReturn(true);
        Functions\when('wp_unslash')->returnArg();
        Functions\when('sanitize_text_field')->returnArg();
        Functions\when('sanitize_key')->alias(
            static fn ($key) => strtolower((string) preg_replace('/[^a-zA-Z0-9_\-]/', '', (string) $key))
        );
        Functions\when('absint')->alias(static fn ($value) => abs((int) $value));
        Functions\when('get_option')->justReturn(1);
        Functions\when('current_time')->justReturn('2026-07-06');
        Functions\when('selected')->justReturn('');
        Functions\when('admin_url')->justReturn('admin.php?page=us-availability');
        Functions\when('add_query_arg')->justReturn('admin.php?page=us-availability&usc_view=week');
        Functions\when('wp_nonce_field')->justReturn('');
        Functions\when('submit_button')->justReturn('');
        Functions\when('mysql2date')->alias(
            static fn (string $format, string $date) => date($format, (int) strtotime($date))
        );

        $this->offerings->shouldReceive('findAll')->andReturn([]);
    }

    public function testBulkDeleteRemovesOnlyOwnedExistingSlots(): void
    {
        $_POST = [
            'usc_action' => 'bulk_delete',
            'slot_ids'   => ['5', 'junk', '7', '9'],
        ];

        $owned = new AvailabilitySlot(instructorId: 3, startDt: '2026-07-08 09:00:00', endDt: '2026-07-08 10:00:00', id: 5);
        $other = new AvailabilitySlot(instructorId: 4, startDt: '2026-07-08 09:00:00', endDt: '2026-07-08 10:00:00', id: 7);

        $this->repository->shouldReceive('findById')->once()->with(5)->andReturn($owned);
        $this->repository->shouldReceive('findById')->once()->with(7)->andReturn($other);
        $this->repository->shouldReceive('findById')->once()->with(9)->andReturn(null);
        $this->repository->shouldReceive('delete')->once()->with(5)->andReturn(true);
        $this->repository->shouldReceive('findByInstructor')->once()->with(3)->andReturn([]);

        $this->render();
    }

    public function testBulkDeleteIgnoresNonArrayInput(): void
    {
        $_POST = [
            'usc_action' => 'bulk_delete',
            'slot_ids'   => '5',
        ];

        $this->repository->shouldNotReceive('findById');
        $this->repository->shouldNotReceive('delete');
        $this->repository->shouldReceive('findByInstructor')->once()->with(3)->andReturn([]);

        $this->render();
    }

    public function testSingleDeleteChecksOwnership(): void
    {
        $_POST = [
            'usc_action' => 'delete',
            'slot_id'    => '7',
        ];

        $other = new AvailabilitySlot(instructorId: 4, startDt: '2026-07-08 09:00:00', endDt: '2026-07-08 10:00:00', id: 7);

        $this->repository->shouldReceive('findById')->once()->with(7)->andReturn($other);
        $this->repository->shouldNotReceive('delete');
        $this->repository->shouldReceive('findByInstructor')->once()->with(3)->andReturn([]);

        $this->render();
    }

    public function testDefaultsToWeekView(): void
    {
        $slot = new AvailabilitySlot(instructorId: 3, startDt: '2026-07-08 09:00:00', endDt: '2026-07-08 10:00:00', id: 5);

        $this->repository->shouldReceive('findByInstructor')->once()->with(3)->andReturn([$slot]);

        $html = $this->render();

        // Week of Monday 2026-07-06 (start_of_week = 1, today = 2026-07-06).
        self::assertStringContainsString('Week of', $html);
        // The list table (with its bulk-delete form) is not rendered by default.
        self::assertStringNotContainsString('id="usc-bulk-delete-form"', $html);
    }

    public function testListViewRendersBulkCheckboxesOnlyForUnbookedSlots(): void
    {
        $_GET['usc_view'] = 'list';

        $unbooked = new AvailabilitySlot(instructorId: 3, startDt: '2026-07-08 09:00:00', endDt: '2026-07-08 10:00:00', id: 5);
        $booked   = new AvailabilitySlot(instructorId: 3, startDt: '2026-07-08 10:00:00', endDt: '2026-07-08 11:00:00', isBooked: true, id: 6);

        $this->repository->shouldReceive('findByInstructor')->once()->with(3)->andReturn([$unbooked, $booked]);

        $html = $this->render();

        self::assertStringContainsString('id="usc-bulk-delete-form"', $html);
        self::assertStringContainsString('value="bulk_delete"', $html);
        self::assertStringContainsString('Delete selected', $html);
        self::assertStringContainsString('name="slot_ids[]" form="usc-bulk-delete-form" value="5"', $html);
        self::assertStringNotContainsString('name="slot_ids[]" form="usc-bulk-delete-form" value="6"', $html);
    }

    /**
     * The reported bug: a 30-minute window submitted with the lesson length left
     * on 60 saved nothing and said nothing. It must now say why.
     */
    public function testAddShowsAnErrorWhenTheWindowIsShorterThanTheLessonLength(): void
    {
        $_POST = [
            'usc_action'       => 'add',
            'start_dt'         => '2026-09-10T17:30',
            'end_dt'           => '2026-09-10T18:00',
            'duration_minutes' => '60',
        ];

        $this->repository->shouldNotReceive('createFromWindow');
        $this->repository->shouldReceive('findByInstructor')->once()->with(3)->andReturn([]);

        $html = $this->render();

        self::assertStringContainsString('notice-error', $html);
        self::assertStringContainsString('60-minute lesson length', $html);
        self::assertStringNotContainsString('notice-success', $html);
    }

    /** @return array<string, array{array<string, string>, string}> */
    public static function invalidWindows(): array
    {
        return [
            'unparseable start' => [['start_dt' => 'whenever', 'end_dt' => '2026-09-10T18:00'], 'valid start and end'],
            'end before start'  => [['start_dt' => '2026-09-10T18:00', 'end_dt' => '2026-09-10T17:00'], 'after the start time'],
            'spans two days'    => [['start_dt' => '2026-09-10T23:00', 'end_dt' => '2026-09-11T01:00'], 'same day'],
        ];
    }

    /**
     * Each of these used to be a bare `return` — the page reloaded unchanged and
     * the instructor had no way to tell the save had failed.
     *
     * @dataProvider invalidWindows
     *
     * @param array<string, string> $fields
     */
    public function testAddReportsEveryRejectedWindow(array $fields, string $expected): void
    {
        $_POST = array_merge([ 'usc_action' => 'add', 'duration_minutes' => '30' ], $fields);

        $this->repository->shouldNotReceive('createFromWindow');
        $this->repository->shouldReceive('findByInstructor')->once()->with(3)->andReturn([]);

        $html = $this->render();

        self::assertStringContainsString('notice-error', $html);
        self::assertStringContainsString($expected, $html);
    }

    public function testAddRejectsAnOfferingTheInstructorDoesNotOwn(): void
    {
        // The REST endpoint always checked this; the admin form never did, so a
        // crafted POST could tie a slot to another instructor's offering.
        $_POST = [
            'usc_action'       => 'add',
            'start_dt'         => '2026-09-10T17:00',
            'end_dt'           => '2026-09-10T18:00',
            'duration_minutes' => '60',
            'offering_id'      => '8',
        ];

        $this->offerings->shouldReceive('findById')->once()->with(8)->andReturn(null);
        $this->repository->shouldNotReceive('createFromWindow');
        $this->repository->shouldReceive('findByInstructor')->once()->with(3)->andReturn([]);

        $html = $this->render();

        self::assertStringContainsString('notice-error', $html);
        self::assertStringContainsString('not available', $html);
    }

    public function testAddReportsHowManySlotsWereCreated(): void
    {
        $_POST = [
            'usc_action'       => 'add',
            'start_dt'         => '2026-09-10T17:00',
            'end_dt'           => '2026-09-10T19:00',
            'duration_minutes' => '60',
            'recurrence'       => 'weekly',
            'weeks'            => '41',
        ];

        $this->repository->shouldReceive('createFromWindow')->once()
            ->with(Mockery::type(AvailabilitySlot::class), true, 41)
            ->andReturn(range(1, 82));
        $this->repository->shouldReceive('findByInstructor')->once()->with(3)->andReturn([]);

        $html = $this->render();

        self::assertStringContainsString('notice-success', $html);
        self::assertStringContainsString('Added 82 bookable slots.', $html);
    }

    public function testAddReportsAFailedWrite(): void
    {
        // A valid window always splits into at least one slot, so an empty result
        // means the inserts themselves failed.
        $_POST = [
            'usc_action'       => 'add',
            'start_dt'         => '2026-09-10T17:00',
            'end_dt'           => '2026-09-10T18:00',
            'duration_minutes' => '60',
        ];

        $this->repository->shouldReceive('createFromWindow')->once()->andReturn([]);
        $this->repository->shouldReceive('findByInstructor')->once()->with(3)->andReturn([]);

        $html = $this->render();

        self::assertStringContainsString('notice-error', $html);
        self::assertStringContainsString('could not be saved', $html);
    }

    public function testSingleDeleteReportsAFailureToDelete(): void
    {
        $_POST = [ 'usc_action' => 'delete', 'slot_id' => '7' ];

        $other = new AvailabilitySlot(instructorId: 4, startDt: '2026-07-08 09:00:00', endDt: '2026-07-08 10:00:00', id: 7);

        $this->repository->shouldReceive('findById')->once()->with(7)->andReturn($other);
        $this->repository->shouldReceive('findByInstructor')->once()->with(3)->andReturn([]);

        $html = $this->render();

        self::assertStringContainsString('notice-error', $html);
        self::assertStringContainsString('could not be deleted', $html);
    }

    public function testBulkDeleteReportsBothHalvesOfAPartialResult(): void
    {
        $_POST = [ 'usc_action' => 'bulk_delete', 'slot_ids' => ['5', '7'] ];

        $owned  = new AvailabilitySlot(instructorId: 3, startDt: '2026-07-08 09:00:00', endDt: '2026-07-08 10:00:00', id: 5);
        $booked = new AvailabilitySlot(instructorId: 3, startDt: '2026-07-08 10:00:00', endDt: '2026-07-08 11:00:00', isBooked: true, id: 7);

        $this->repository->shouldReceive('findById')->once()->with(5)->andReturn($owned);
        $this->repository->shouldReceive('findById')->once()->with(7)->andReturn($booked);
        $this->repository->shouldReceive('delete')->once()->with(5)->andReturn(true);
        // The repository refuses a booked row, reporting it by returning false.
        $this->repository->shouldReceive('delete')->once()->with(7)->andReturn(false);
        $this->repository->shouldReceive('findByInstructor')->once()->with(3)->andReturn([]);

        $html = $this->render();

        self::assertStringContainsString('1 slot deleted.', $html);
        self::assertStringContainsString('1 slot could not be deleted', $html);
    }

    private function render(): string
    {
        ob_start();
        $this->controller->renderPage();

        return (string) ob_get_clean();
    }
}