diff options
Diffstat (limited to 'spec/lib/gitlab/database/postgresql_database_tasks/load_schema_versions_mixin_spec.rb')
-rw-r--r-- | spec/lib/gitlab/database/postgresql_database_tasks/load_schema_versions_mixin_spec.rb | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/spec/lib/gitlab/database/postgresql_database_tasks/load_schema_versions_mixin_spec.rb b/spec/lib/gitlab/database/postgresql_database_tasks/load_schema_versions_mixin_spec.rb new file mode 100644 index 00000000000..3e675a85999 --- /dev/null +++ b/spec/lib/gitlab/database/postgresql_database_tasks/load_schema_versions_mixin_spec.rb @@ -0,0 +1,32 @@ +# frozen_string_literal: true + +require 'spec_helper' + +RSpec.describe Gitlab::Database::PostgresqlDatabaseTasks::LoadSchemaVersionsMixin do + let(:instance_class) do + klass = Class.new do + def structure_load + original_structure_load + end + + def original_structure_load + end + end + + klass.prepend(described_class) + + klass + end + + let(:instance) { instance_class.new } + + it 'calls SchemaMigrations load_all' do + connection = double('connection') + allow(instance).to receive(:connection).and_return(connection) + + expect(instance).to receive(:original_structure_load).ordered + expect(Gitlab::Database::SchemaMigrations).to receive(:load_all).with(connection).ordered + + instance.structure_load + end +end |