1 files changed, 84 insertions, 0 deletions
diff --git a/spec/lib/gitlab/database/dictionary_spec.rb b/spec/lib/gitlab/database/dictionary_spec.rb
new file mode 100644
index 00000000000..6d2de41468b
--- /dev/null
+++ b/spec/lib/gitlab/database/dictionary_spec.rb
@@ -0,0 +1,84 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Gitlab::Database::Dictionary, feature_category: :database do
+  subject(:database_dictionary) { described_class.new(file_path) }
+
+  context 'for a table' do
+    let(:file_path) { 'db/docs/application_settings.yml' }
+
+    describe '#name_and_schema' do
+      it 'returns the name of the table and its gitlab schema' do
+        expect(database_dictionary.name_and_schema).to match_array(['application_settings', :gitlab_main_clusterwide])
+      end
+    end
+
+    describe '#table_name' do
+      it 'returns the name of the table' do
+        expect(database_dictionary.table_name).to eq('application_settings')
+      end
+    end
+
+    describe '#view_name' do
+      it 'returns nil' do
+        expect(database_dictionary.view_name).to be_nil
+      end
+    end
+
+    describe '#milestone' do
+      it 'returns the milestone in which the table was introduced' do
+        expect(database_dictionary.milestone).to eq('7.7')
+      end
+    end
+
+    describe '#gitlab_schema' do
+      it 'returns the gitlab_schema of the table' do
+        expect(database_dictionary.table_name).to eq('application_settings')
+      end
+    end
+
+    describe '#schema?' do
+      it 'checks if the given schema matches the schema of the table' do
+        expect(database_dictionary.schema?('gitlab_main')).to eq(false)
+        expect(database_dictionary.schema?('gitlab_main_clusterwide')).to eq(true)
+      end
+    end
+
+    describe '#key_name' do
+      it 'returns the value of the name of the table' do
+        expect(database_dictionary.key_name).to eq('application_settings')
+      end
+    end
+
+    describe '#validate!' do
+      it 'raises an error if the gitlab_schema is empty' do
+        allow(database_dictionary).to receive(:gitlab_schema).and_return(nil)
+
+        expect { database_dictionary.validate! }.to raise_error(Gitlab::Database::GitlabSchema::UnknownSchemaError)
+      end
+    end
+  end
+
+  context 'for a view' do
+    let(:file_path) { 'db/docs/views/postgres_constraints.yml' }
+
+    describe '#table_name' do
+      it 'returns nil' do
+        expect(database_dictionary.table_name).to be_nil
+      end
+    end
+
+    describe '#view_name' do
+      it 'returns the name of the view' do
+        expect(database_dictionary.view_name).to eq('postgres_constraints')
+      end
+    end
+
+    describe '#key_name' do
+      it 'returns the value of the name of the view' do
+        expect(database_dictionary.key_name).to eq('postgres_constraints')
+      end
+    end
+  end
+end