1 files changed, 64 insertions, 0 deletions
diff --git a/spec/graphql/resolvers/design_management/version_in_collection_resolver_spec.rb b/spec/graphql/resolvers/design_management/version_in_collection_resolver_spec.rb
new file mode 100644
index 00000000000..ef50598d241
--- /dev/null
+++ b/spec/graphql/resolvers/design_management/version_in_collection_resolver_spec.rb
@@ -0,0 +1,64 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+describe Resolvers::DesignManagement::VersionInCollectionResolver do
+  include GraphqlHelpers
+  include DesignManagementTestHelpers
+
+  let(:resolver) { described_class }
+
+  describe '#resolve' do
+    let_it_be(:issue) { create(:issue) }
+    let_it_be(:current_user) { create(:user) }
+    let_it_be(:first_version) { create(:design_version, issue: issue) }
+
+    let(:project) { issue.project }
+    let(:params) { {} }
+
+    before do
+      enable_design_management
+      project.add_developer(current_user)
+    end
+
+    let(:appropriate_error) { ::Gitlab::Graphql::Errors::ArgumentError }
+
+    subject(:result) { resolve_version(issue.design_collection) }
+
+    context 'Neither id nor sha is passed as parameters' do
+      it 'raises an appropriate error' do
+        expect { result }.to raise_error(appropriate_error)
+      end
+    end
+
+    context 'we pass an id' do
+      let(:params) { { id: global_id_of(first_version) } }
+
+      it { is_expected.to eq(first_version) }
+    end
+
+    context 'we pass a sha' do
+      let(:params) { { sha: first_version.sha } }
+
+      it { is_expected.to eq(first_version) }
+    end
+
+    context 'we pass an inconsistent mixture of sha and version id' do
+      let(:params) { { sha: first_version.sha, id: global_id_of(create(:design_version)) } }
+
+      it { is_expected.to be_nil }
+    end
+
+    context 'we pass the id of something that is not a design_version' do
+      let(:params) { { id: global_id_of(project) } }
+
+      it 'raises an appropriate error' do
+        expect { result }.to raise_error(appropriate_error)
+      end
+    end
+  end
+
+  def resolve_version(obj, context = { current_user: current_user })
+    resolve(resolver, obj: obj, args: params, ctx: context)
+  end
+end