<?php
// This file is part of Moodle - http://moodle.org/
//
// Moodle is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// Moodle is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Moodle.  If not, see <http://www.gnu.org/licenses/>.

/**
 * Behat data generator for mod_assign.
 *
 * @package   mod_assign
 * @category  test
 * @copyright 2021 Andrew Lyons
 * @license   http://www.gnu.org/copyleft/gpl.html GNU GPL v3 or later
 */
class behat_mod_assign_generator extends behat_generator_base {

    /**
     * Get a list of the entities that Behat can create using the generator step.
     *
     * @return array
     */
    protected function get_creatable_entities(): array {
        return [
            'submissions' => [
                'singular' => 'submission',
                'datagenerator' => 'submission',
                'required' => ['assign', 'user'],
                'switchids' => ['assign' => 'cmid', 'user' => 'userid'],
            ],
            'extensions' => [
                'singular' => 'extension',
                'datagenerator' => 'extension',
                'required' => ['assign', 'user', 'extensionduedate'],
                'switchids' => ['assign' => 'cmid', 'user' => 'userid'],
            ],
            'user overrides' => [
                'singular' => 'user override',
                'datagenerator' => 'override',
                'required' => ['assignment', 'user'],
                'switchids' => ['assignment' => 'assignid', 'user' => 'userid'],
            ],
            'group overrides' => [
                'singular' => 'group override',
                'datagenerator' => 'override',
                'required' => ['assignment', 'group'],
                'switchids' => ['assignment' => 'assignid', 'group' => 'groupid'],
            ],
        ];
    }

    /**
     * Get the assignment cmid using an activity name or idnumber.
     *
     * @param string $identifier activity name or idnumber
     * @return int The cmid
     */
    protected function get_assign_id(string $identifier): int {
        return $this->get_cm_by_activity_name('assign', $identifier)->id;
    }

    /**
     * Get the assignment instance ID using an activity name or idnumber.
     *
     * @param string $identifier activity name or idnumber
     * @return int The assignment instance ID
     */
    protected function get_assignment_id(string $identifier): int {
        return $this->get_cm_by_activity_name('assign', $identifier)->instance;
    }
}
