1 files changed, 68 insertions, 0 deletions
diff --git a/spec/presenters/packages/pypi/simple_index_presenter_spec.rb b/spec/presenters/packages/pypi/simple_index_presenter_spec.rb
new file mode 100644
index 00000000000..d915706577f
--- /dev/null
+++ b/spec/presenters/packages/pypi/simple_index_presenter_spec.rb
@@ -0,0 +1,68 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe ::Packages::Pypi::SimpleIndexPresenter, :aggregate_failures do
+  using RSpec::Parameterized::TableSyntax
+
+  let_it_be(:group) { create(:group) }
+  let_it_be(:project) { create(:project, group: group) }
+  let_it_be(:package_name) { 'sample-project' }
+  let_it_be(:package1) { create(:pypi_package, project: project, name: package_name, version: '1.0.0') }
+  let_it_be(:package2) { create(:pypi_package, project: project, name: package_name, version: '2.0.0') }
+
+  let(:packages) { project.packages }
+
+  describe '#body' do
+    subject(:presenter) { described_class.new(packages, project_or_group).body }
+
+    shared_examples_for "pypi package presenter" do
+      where(:version, :expected_version) do
+        '>=2.7'                        | '&gt;=2.7'
+        '"><script>alert(1)</script>'  | '&quot;&gt;&lt;script&gt;alert(1)&lt;/script&gt;'
+        '>=2.7, !=3.0'                 | '&gt;=2.7, !=3.0'
+      end
+
+      with_them do
+        let(:python_version) { version }
+        let(:expected_python_version) { expected_version }
+
+        before do
+          package1.pypi_metadatum.update_column(:required_python, python_version)
+          package2.pypi_metadatum.update_column(:required_python, '')
+        end
+
+        it 'contains links for all packages' do
+          expect(presenter).to include(expected_link1)
+          expect(presenter).to include(expected_link2)
+        end
+      end
+    end
+
+    context 'for project' do
+      let(:project_or_group) { project }
+      let(:expected_link1) { "<a href=\"http://localhost/api/v4/projects/#{project.id}/packages/pypi/simple/#{package1.normalized_pypi_name}\" data-requires-python=\"#{expected_python_version}\">#{package1.name}</a>" } # rubocop:disable Layout/LineLength
+      let(:expected_link2) { "<a href=\"http://localhost/api/v4/projects/#{project.id}/packages/pypi/simple/#{package2.normalized_pypi_name}\" data-requires-python=\"\">#{package2.name}</a>" } # rubocop:disable Layout/LineLength
+
+      it_behaves_like 'pypi package presenter'
+    end
+
+    context 'for group' do
+      let(:project_or_group) { group }
+      let(:expected_link1) { "<a href=\"http://localhost/api/v4/groups/#{group.id}/-/packages/pypi/simple/#{package1.normalized_pypi_name}\" data-requires-python=\"#{expected_python_version}\">#{package1.name}</a>" } # rubocop:disable Layout/LineLength
+      let(:expected_link2) { "<a href=\"http://localhost/api/v4/groups/#{group.id}/-/packages/pypi/simple/#{package2.normalized_pypi_name}\" data-requires-python=\"\">#{package2.name}</a>" } # rubocop:disable Layout/LineLength
+
+      it_behaves_like 'pypi package presenter'
+    end
+
+    context 'with package files pending destruction' do
+      let_it_be(:package_pending_destruction) do
+        create(:package, :pending_destruction, project: project, name: "package_pending_destruction")
+      end
+
+      let(:project_or_group) { project }
+
+      it { is_expected.not_to include(package_pending_destruction.name) }
+    end
+  end
+end