1 files changed, 242 insertions, 0 deletions
diff --git a/spec/support/shared_examples/models/concerns/timebox_shared_examples.rb b/spec/support/shared_examples/models/concerns/timebox_shared_examples.rb
new file mode 100644
index 00000000000..32d502af5a2
--- /dev/null
+++ b/spec/support/shared_examples/models/concerns/timebox_shared_examples.rb
@@ -0,0 +1,242 @@
+# frozen_string_literal: true
+
+RSpec.shared_examples 'a timebox' do |timebox_type|
+  let(:project) { create(:project, :public) }
+  let(:group) { create(:group) }
+  let(:timebox) { create(timebox_type, project: project) }
+  let(:issue) { create(:issue, project: project) }
+  let(:user) { create(:user) }
+  let(:timebox_table_name) { timebox_type.to_s.pluralize.to_sym }
+
+  describe 'modules' do
+    context 'with a project' do
+      it_behaves_like 'AtomicInternalId' do
+        let(:internal_id_attribute) { :iid }
+        let(:instance) { build(timebox_type, project: build(:project), group: nil) }
+        let(:scope) { :project }
+        let(:scope_attrs) { { project: instance.project } }
+        let(:usage) { timebox_table_name }
+      end
+    end
+
+    context 'with a group' do
+      it_behaves_like 'AtomicInternalId' do
+        let(:internal_id_attribute) { :iid }
+        let(:instance) { build(timebox_type, project: nil, group: build(:group)) }
+        let(:scope) { :group }
+        let(:scope_attrs) { { namespace: instance.group } }
+        let(:usage) { timebox_table_name }
+      end
+    end
+  end
+
+  describe "Validation" do
+    before do
+      allow(subject).to receive(:set_iid).and_return(false)
+    end
+
+    describe 'start_date' do
+      it 'adds an error when start_date is greater then due_date' do
+        timebox = build(timebox_type, start_date: Date.tomorrow, due_date: Date.yesterday)
+
+        expect(timebox).not_to be_valid
+        expect(timebox.errors[:due_date]).to include("must be greater than start date")
+      end
+
+      it 'adds an error when start_date is greater than 9999-12-31' do
+        timebox = build(timebox_type, start_date: Date.new(10000, 1, 1))
+
+        expect(timebox).not_to be_valid
+        expect(timebox.errors[:start_date]).to include("date must not be after 9999-12-31")
+      end
+    end
+
+    describe 'due_date' do
+      it 'adds an error when due_date is greater than 9999-12-31' do
+        timebox = build(timebox_type, due_date: Date.new(10000, 1, 1))
+
+        expect(timebox).not_to be_valid
+        expect(timebox.errors[:due_date]).to include("date must not be after 9999-12-31")
+      end
+    end
+
+    describe 'title' do
+      it { is_expected.to validate_presence_of(:title) }
+
+      it 'is invalid if title would be empty after sanitation' do
+        timebox = build(timebox_type, project: project, title: '<img src=x onerror=prompt(1)>')
+
+        expect(timebox).not_to be_valid
+        expect(timebox.errors[:title]).to include("can't be blank")
+      end
+    end
+
+    describe '#timebox_type_check' do
+      it 'is invalid if it has both project_id and group_id' do
+        timebox = build(timebox_type, group: group)
+        timebox.project = project
+
+        expect(timebox).not_to be_valid
+        expect(timebox.errors[:project_id]).to include("#{timebox_type} should belong either to a project or a group.")
+      end
+    end
+
+    describe "#uniqueness_of_title" do
+      context "per project" do
+        it "does not accept the same title in a project twice" do
+          new_timebox = timebox.dup
+          expect(new_timebox).not_to be_valid
+        end
+
+        it "accepts the same title in another project" do
+          project = create(:project)
+          new_timebox = timebox.dup
+          new_timebox.project = project
+
+          expect(new_timebox).to be_valid
+        end
+      end
+
+      context "per group" do
+        let(:timebox) { create(timebox_type, group: group) }
+
+        before do
+          project.update(group: group)
+        end
+
+        it "does not accept the same title in a group twice" do
+          new_timebox = described_class.new(group: group, title: timebox.title)
+
+          expect(new_timebox).not_to be_valid
+        end
+
+        it "does not accept the same title of a child project timebox" do
+          create(timebox_type, project: group.projects.first)
+
+          new_timebox = described_class.new(group: group, title: timebox.title)
+
+          expect(new_timebox).not_to be_valid
+        end
+      end
+    end
+  end
+
+  describe "Associations" do
+    it { is_expected.to belong_to(:project) }
+    it { is_expected.to belong_to(:group) }
+    it { is_expected.to have_many(:issues) }
+    it { is_expected.to have_many(:merge_requests) }
+    it { is_expected.to have_many(:labels) }
+  end
+
+  describe '#timebox_name' do
+    it 'returns the name of the model' do
+      expect(timebox.timebox_name).to eq(timebox_type.to_s)
+    end
+  end
+
+  describe '#project_timebox?' do
+    context 'when project_id is present' do
+      it 'returns true' do
+        expect(timebox.project_timebox?).to be_truthy
+      end
+    end
+
+    context 'when project_id is not present' do
+      let(:timebox) { build(timebox_type, group: group) }
+
+      it 'returns false' do
+        expect(timebox.project_timebox?).to be_falsey
+      end
+    end
+  end
+
+  describe '#group_timebox?' do
+    context 'when group_id is present' do
+      let(:timebox) { build(timebox_type, group: group) }
+
+      it 'returns true' do
+        expect(timebox.group_timebox?).to be_truthy
+      end
+    end
+
+    context 'when group_id is not present' do
+      it 'returns false' do
+        expect(timebox.group_timebox?).to be_falsey
+      end
+    end
+  end
+
+  describe '#safe_title' do
+    let(:timebox) { create(timebox_type, title: "<b>foo & bar -> 2.2</b>") }
+
+    it 'normalizes the title for use as a slug' do
+      expect(timebox.safe_title).to eq('foo-bar-22')
+    end
+  end
+
+  describe '#resource_parent' do
+    context 'when group is present' do
+      let(:timebox) { build(timebox_type, group: group) }
+
+      it 'returns the group' do
+        expect(timebox.resource_parent).to eq(group)
+      end
+    end
+
+    context 'when project is present' do
+      it 'returns the project' do
+        expect(timebox.resource_parent).to eq(project)
+      end
+    end
+  end
+
+  describe "#title" do
+    let(:timebox) { create(timebox_type, title: "<b>foo & bar -> 2.2</b>") }
+
+    it "sanitizes title" do
+      expect(timebox.title).to eq("foo & bar -> 2.2")
+    end
+  end
+
+  describe '#merge_requests_enabled?' do
+    context "per project" do
+      it "is true for projects with MRs enabled" do
+        project = create(:project, :merge_requests_enabled)
+        timebox = create(timebox_type, project: project)
+
+        expect(timebox.merge_requests_enabled?).to be_truthy
+      end
+
+      it "is false for projects with MRs disabled" do
+        project = create(:project, :repository_enabled, :merge_requests_disabled)
+        timebox = create(timebox_type, project: project)
+
+        expect(timebox.merge_requests_enabled?).to be_falsey
+      end
+
+      it "is false for projects with repository disabled" do
+        project = create(:project, :repository_disabled)
+        timebox = create(timebox_type, project: project)
+
+        expect(timebox.merge_requests_enabled?).to be_falsey
+      end
+    end
+
+    context "per group" do
+      let(:timebox) { create(timebox_type, group: group) }
+
+      it "is always true for groups, for performance reasons" do
+        expect(timebox.merge_requests_enabled?).to be_truthy
+      end
+    end
+  end
+
+  describe '#to_ability_name' do
+    it 'returns timebox' do
+      timebox = build(timebox_type)
+
+      expect(timebox.to_ability_name).to eq(timebox_type.to_s)
+    end
+  end
+end