BigW Consortium Gitlab

base_event_fetcher_spec.rb 1.29 KB
Newer Older
James Lopez committed
1 2 3 4
require 'spec_helper'

describe Gitlab::CycleAnalytics::BaseEventFetcher do
  let(:max_events) { 2 }
5
  let(:project) { create(:project, :repository) }
James Lopez committed
6 7 8
  let(:user) { create(:user, :admin) }
  let(:start_time_attrs) { Issue.arel_table[:created_at] }
  let(:end_time_attrs) { [Issue::Metrics.arel_table[:first_associated_with_milestone_at]] }
9
  let(:options) do
James Lopez committed
10 11
    { start_time_attrs: start_time_attrs,
      end_time_attrs: end_time_attrs,
12
      from: 30.days.ago }
James Lopez committed
13
  end
James Lopez committed
14 15 16 17 18 19 20 21 22

  subject do
    described_class.new(project: project,
                        stage: :issue,
                        options: options).fetch
  end

  before do
    allow_any_instance_of(Gitlab::ReferenceExtractor).to receive(:issues).and_return(Issue.all)
23
    allow_any_instance_of(described_class).to receive(:serialize) do |event|
James Lopez committed
24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45
      event
    end

    stub_const('Gitlab::CycleAnalytics::BaseEventFetcher::MAX_EVENTS', max_events)

    setup_events(count: 3)
  end

  it 'limits the rows to the max number' do
    expect(subject.count).to eq(max_events)
  end

  def setup_events(count:)
    count.times do
      issue = create(:issue, project: project, created_at: 2.days.ago)
      milestone = create(:milestone, project: project)

      issue.update(milestone: milestone)
      create_merge_request_closing_issue(issue)
    end
  end
end