From ad0595b4b4e446ac66997767c81480b69dc19b48 Mon Sep 17 00:00:00 2001
From: Sara Arjona <sara@moodle.com>
Date: Mon, 11 Jul 2022 18:26:23 +0200
Subject: [PATCH] MDL-75148 mod_data: Implement preset behat generator

---
 .../generator/behat_mod_data_generator.php    | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/mod/data/tests/generator/behat_mod_data_generator.php b/mod/data/tests/generator/behat_mod_data_generator.php
index 1ae6097238d..8f39d2d02c5 100644
--- a/mod/data/tests/generator/behat_mod_data_generator.php
+++ b/mod/data/tests/generator/behat_mod_data_generator.php
@@ -49,6 +49,12 @@ class behat_mod_data_generator extends behat_generator_base {
                 'required' => ['database', 'name'],
                 'switchids' => ['database' => 'databaseid'],
             ],
+            'presets' => [
+                'singular' => 'preset',
+                'datagenerator' => 'preset',
+                'required' => ['database', 'name'],
+                'switchids' => ['database' => 'databaseid'],
+            ],
         ];
     }
 
@@ -124,6 +130,19 @@ class behat_mod_data_generator extends behat_generator_base {
         }
     }
 
+    /**
+     * Saves a preset.
+     *
+     * @param array $data Preset data.
+     */
+    protected function process_preset(array $data): void {
+        global $DB;
+
+        $instance = $DB->get_record('data', ['id' => $data['databaseid']], '*', MUST_EXIST);
+
+        $this->get_data_generator()->create_preset($instance, (object) $data);
+    }
+
     /**
      * Get the module data generator.
      *