1 files changed, 129 insertions, 0 deletions
diff --git a/spec/finders/design_management/versions_finder_spec.rb b/spec/finders/design_management/versions_finder_spec.rb
new file mode 100644
index 00000000000..11d53d0d630
--- /dev/null
+++ b/spec/finders/design_management/versions_finder_spec.rb
@@ -0,0 +1,129 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+describe DesignManagement::VersionsFinder do
+  include DesignManagementTestHelpers
+
+  let_it_be(:user) { create(:user) }
+  let_it_be(:project) { create(:project, :private) }
+  let_it_be(:issue) { create(:issue, project: project) }
+  let_it_be(:design_1) { create(:design, :with_file, issue: issue, versions_count: 1) }
+  let_it_be(:design_2) { create(:design, :with_file, issue: issue, versions_count: 1) }
+  let(:version_1) { design_1.versions.first }
+  let(:version_2) { design_2.versions.first }
+  let(:design_or_collection) { issue.design_collection }
+  let(:params) { {} }
+
+  let(:finder) { described_class.new(design_or_collection, user, params) }
+
+  subject(:versions) { finder.execute }
+
+  describe '#execute' do
+    shared_examples 'returns no results' do
+      it 'returns no results when passed a DesignCollection' do
+        expect(design_or_collection).is_a?(DesignManagement::DesignCollection)
+        is_expected.to be_empty
+      end
+
+      context 'when passed a Design' do
+        let(:design_or_collection) { design_1 }
+
+        it 'returns no results when passed a Design' do
+          is_expected.to be_empty
+        end
+      end
+    end
+
+    context 'when user cannot read designs of an issue' do
+      include_examples 'returns no results'
+    end
+
+    context 'when user can read designs of an issue' do
+      before do
+        project.add_developer(user)
+      end
+
+      context 'when design management feature is disabled' do
+        include_examples 'returns no results'
+      end
+
+      context 'when design management feature is enabled' do
+        before do
+          enable_design_management
+        end
+
+        describe 'passing a DesignCollection or a Design for the initial scoping' do
+          it 'returns the versions scoped to the DesignCollection' do
+            expect(design_or_collection).is_a?(DesignManagement::DesignCollection)
+            is_expected.to eq(issue.design_collection.versions.ordered)
+          end
+
+          context 'when passed a Design' do
+            let(:design_or_collection) { design_1 }
+
+            it 'returns the versions scoped to the Design' do
+              is_expected.to eq(design_1.versions)
+            end
+          end
+        end
+
+        describe 'returning versions earlier or equal to a version' do
+          context 'when argument is the first version' do
+            let(:params) { { earlier_or_equal_to: version_1 }}
+
+            it { is_expected.to eq([version_1]) }
+          end
+
+          context 'when argument is the second version' do
+            let(:params) { { earlier_or_equal_to: version_2 }}
+
+            it { is_expected.to contain_exactly(version_1, version_2) }
+          end
+        end
+
+        describe 'returning versions by SHA' do
+          context 'when argument is the first version' do
+            let(:params) { { sha: version_1.sha } }
+
+            it { is_expected.to contain_exactly(version_1) }
+          end
+
+          context 'when argument is the second version' do
+            let(:params) { { sha: version_2.sha } }
+
+            it { is_expected.to contain_exactly(version_2) }
+          end
+        end
+
+        describe 'returning versions by ID' do
+          context 'when argument is the first version' do
+            let(:params) { { version_id: version_1.id } }
+
+            it { is_expected.to contain_exactly(version_1) }
+          end
+
+          context 'when argument is the second version' do
+            let(:params) { { version_id: version_2.id } }
+
+            it { is_expected.to contain_exactly(version_2) }
+          end
+        end
+
+        describe 'mixing id and sha' do
+          context 'when arguments are consistent' do
+            let(:params) { { version_id: version_1.id, sha: version_1.sha } }
+
+            it { is_expected.to contain_exactly(version_1) }
+          end
+
+          context 'when arguments are in-consistent' do
+            let(:params) { { version_id: version_1.id, sha: version_2.sha } }
+
+            it { is_expected.to be_empty }
+          end
+        end
+      end
+    end
+  end
+end