1 files changed, 42 insertions, 0 deletions
diff --git a/spec/lib/gitlab/database/migrations/observers/batch_details_spec.rb b/spec/lib/gitlab/database/migrations/observers/batch_details_spec.rb
new file mode 100644
index 00000000000..5b3c23736ed
--- /dev/null
+++ b/spec/lib/gitlab/database/migrations/observers/batch_details_spec.rb
@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Gitlab::Database::Migrations::Observers::BatchDetails, feature_category: :database do
+  subject(:observe) { described_class.new(observation, path, connection) }
+
+  let(:connection) { ActiveRecord::Migration.connection }
+  let(:observation) { Gitlab::Database::Migrations::Observation.new(meta: meta) }
+  let(:path) { Dir.mktmpdir }
+  let(:file_name) { 'batch-details.json' }
+  let(:file_path) { Pathname.new(path).join(file_name) }
+  let(:json_file) { Gitlab::Json.parse(File.read(file_path)) }
+  let(:job_meta) do
+    { "min_value" => 1, "max_value" => 19, "batch_size" => 20, "sub_batch_size" => 5, "pause_ms" => 100 }
+  end
+
+  where(:meta, :expected_keys) do
+    [
+      [lazy { { job_meta: job_meta } }, %w[time_spent min_value max_value batch_size sub_batch_size pause_ms]],
+      [nil, %w[time_spent]],
+      [{ job_meta: nil }, %w[time_spent]]
+    ]
+  end
+
+  with_them do
+    before do
+      observe.before
+      observe.after
+    end
+
+    after do
+      FileUtils.remove_entry(path)
+    end
+
+    it 'records expected information to file' do
+      observe.record
+
+      expect(json_file.keys).to match_array(expected_keys)
+    end
+  end
+end