MDL-75913 behat: Add support for file uploads in generators

2023-06-19 16:38:18 +10:00
parent 64b701190d
commit 1f553279d2
2 changed files with 105 additions and 1 deletions
@@ -86,4 +86,95 @@ abstract class component_generator_base {

        \core\session\manager::set_user($user);
    }
+
+    /**
+     * Update the instance record, inserting any files that are referenced.
+     *
+     * @param stdClass $instance The instance record of the already-created record
+     * @param stdClass $record The data passed in to create the instance
+     * @param string $table The table that the data exists in
+     * @param context $context The context of the instance
+     * @param string $component The component of the owning plugin
+     * @param string $filearea The name of the file area
+     * @param int $targetitemid The itemid to use when saving the files
+     * @return stdClass The updated instance record
+     */
+    protected function insert_files(
+        stdClass $instance,
+        stdClass $record,
+        string $table,
+        context $context,
+        string $component,
+        string $filearea,
+        int $targetitemid
+    ): stdClass {
+        global $CFG, $DB, $USER;
+
+        $fieldname = "[[files::{$filearea}]]";
+        if (!isset($record->$fieldname)) {
+            return $instance;
+        }
+
+        preg_match('/\[\[files::(.*)\]\]/', $fieldname, $matches);
+        if (empty($matches[1])) {
+            throw new coding_exception('Invalid file field name: ' . $fieldname);
+        }
+
+        $referencedfieldname = trim($matches[1]);
+        if (!isset($record->$referencedfieldname)) {
+            throw new coding_exception("File field '{$fieldname}' references non-existent field '{$referencedfieldname}'");
+        }
+
+        $fs = get_file_storage();
+        $itemid = file_get_unused_draft_itemid();
+        $itemidfieldname = "{$referencedfieldname}[itemid]";
+        $record->$itemidfieldname = $itemid;
+
+        $filenames = explode(',', $record->$fieldname);
+        foreach ($filenames as $filename) {
+            $filename = trim($filename);
+            if (!$filename) {
+                continue;
+            }
+
+            $explodedfilename = explode('::', $filename, 3);
+            if (count($explodedfilename) === 2) {
+                [$sourcefile, $targetfile] = $explodedfilename;
+                $user = $USER;
+            } else {
+                [$sourcefile, $targetfile, $username] = $explodedfilename;
+                $user = \core_user::get_user_by_username($username);
+            }
+            $filepath = "{$CFG->dirroot}/{$sourcefile}";
+            if (!file_exists($filepath)) {
+                throw new coding_exception("File '{$filepath}' does not exist");
+            }
+            $filerecord = [
+                'userid' => $user->id,
+                'contextid' => context_user::instance($user->id)->id,
+                'component' => 'user',
+                'filearea' => 'draft',
+                'itemid' => $itemid,
+                'filepath' => '/' . dirname($targetfile),
+                'filename' => basename($targetfile),
+            ];
+            $fs->create_file_from_pathname($filerecord, $filepath);
+        }
+
+        $olduser = $USER;
+        $this->set_user($user);
+        $instance->$referencedfieldname = file_save_draft_area_files(
+            $itemid,
+            $context->id,
+            $component,
+            $referencedfieldname,
+            $targetitemid,
+            null,
+            $instance->$referencedfieldname
+        );
+        $this->set_user($olduser);
+
+        $DB->update_record($table, $instance);
+        return $instance;
+    }
 }