Add latest changes from gitlab-org/gitlab@16-0-stable-eev16.0.0-rc42

1 files changed, 31 insertions, 0 deletions

diff --git a/db/post_migrate/20230328030101_add_secureflag_training_provider.rb b/db/post_migrate/20230328030101_add_secureflag_training_provider.rb
new file mode 100644
index 00000000000..4b32570ea56
--- /dev/null
+++ b/db/post_migrate/20230328030101_add_secureflag_training_provider.rb
@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+class AddSecureflagTrainingProvider < Gitlab::Database::Migration[2.1]
+  disable_ddl_transaction!
+
+  restrict_gitlab_migration gitlab_schema: :gitlab_main
+
+  SECUREFLAG_DATA = {
+    name: 'SecureFlag',
+    description: "Get remediation advice with example code and recommended hands-on labs in a fully
+    interactive virtualised environment.",
+    url: "https://knowledge-base-api.secureflag.com/gitlab"
+  }
+
+  class TrainingProvider < MigrationRecord
+    self.table_name = 'security_training_providers'
+  end
+
+  def up
+    current_time = Time.current
+    timestamps = { created_at: current_time, updated_at: current_time }
+
+    TrainingProvider.reset_column_information
+    TrainingProvider.upsert(SECUREFLAG_DATA.merge(timestamps))
+  end
+
+  def down
+    TrainingProvider.reset_column_information
+    TrainingProvider.find_by(name: SECUREFLAG_DATA[:name])&.destroy
+  end
+end