1 files changed, 82 insertions, 0 deletions
diff --git a/spec/models/design_management/design_collection_spec.rb b/spec/models/design_management/design_collection_spec.rb
new file mode 100644
index 00000000000..bd48f742042
--- /dev/null
+++ b/spec/models/design_management/design_collection_spec.rb
@@ -0,0 +1,82 @@
+# frozen_string_literal: true
+require 'spec_helper'
+
+describe DesignManagement::DesignCollection do
+  include DesignManagementTestHelpers
+
+  let_it_be(:issue, reload: true) { create(:issue) }
+
+  subject(:collection) { described_class.new(issue) }
+
+  describe ".find_or_create_design!" do
+    it "finds an existing design" do
+      design = create(:design, issue: issue, filename: 'world.png')
+
+      expect(collection.find_or_create_design!(filename: 'world.png')).to eq(design)
+    end
+
+    it "creates a new design if one didn't exist" do
+      expect(issue.designs.size).to eq(0)
+
+      new_design = collection.find_or_create_design!(filename: 'world.png')
+
+      expect(issue.designs.size).to eq(1)
+      expect(new_design.filename).to eq('world.png')
+      expect(new_design.issue).to eq(issue)
+    end
+
+    it "only queries the designs once" do
+      create(:design, issue: issue, filename: 'hello.png')
+      create(:design, issue: issue, filename: 'world.jpg')
+
+      expect do
+        collection.find_or_create_design!(filename: 'hello.png')
+        collection.find_or_create_design!(filename: 'world.jpg')
+      end.not_to exceed_query_limit(1)
+    end
+  end
+
+  describe "#versions" do
+    it "includes versions for all designs" do
+      version_1 = create(:design_version)
+      version_2 = create(:design_version)
+      other_version = create(:design_version)
+      create(:design, issue: issue, versions: [version_1])
+      create(:design, issue: issue, versions: [version_2])
+      create(:design, versions: [other_version])
+
+      expect(collection.versions).to contain_exactly(version_1, version_2)
+    end
+  end
+
+  describe "#repository" do
+    it "builds a design repository" do
+      expect(collection.repository).to be_a(DesignManagement::Repository)
+    end
+  end
+
+  describe '#designs_by_filename' do
+    let(:designs) { create_list(:design, 5, :with_file, issue: issue) }
+    let(:filenames) { designs.map(&:filename) }
+    let(:query) { subject.designs_by_filename(filenames) }
+
+    it 'finds all the designs with those filenames on this issue' do
+      expect(query).to have_attributes(size: 5)
+    end
+
+    it 'only makes a single query' do
+      designs.each(&:id)
+      expect { query }.not_to exceed_query_limit(1)
+    end
+
+    context 'some are deleted' do
+      before do
+        delete_designs(*designs.sample(2))
+      end
+
+      it 'takes deletion into account' do
+        expect(query).to have_attributes(size: 3)
+      end
+    end
+  end
+end