summaryrefslogtreecommitdiff
path: root/spec/lib/csv_builders/stream_spec.rb
blob: 204baf965d02ba66bf51094b67405fe5f31f0da1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
# frozen_string_literal: true

require 'spec_helper'

RSpec.describe CsvBuilders::Stream do
  let(:event_1) { double(title: 'Added salt', description: 'A teaspoon') }
  let(:event_2) { double(title: 'Added sugar', description: 'Just a pinch') }
  let(:fake_relation) { FakeRelation.new([event_1, event_2]) }

  subject(:builder) { described_class.new(fake_relation, 'Title' => 'title', 'Description' => 'description') }

  describe '#render' do
    before do
      stub_const('FakeRelation', Array)

      FakeRelation.class_eval do
        def find_each(&block)
          each(&block)
        end
      end
    end

    it 'returns a lazy enumerator' do
      expect(builder.render).to be_an(Enumerator::Lazy)
    end

    it 'returns all rows up to default max value' do
      expect(builder.render.to_a).to eq([
        "Title,Description\n",
        "Added salt,A teaspoon\n",
        "Added sugar,Just a pinch\n"
      ])
    end

    it 'truncates to max rows' do
      expect(builder.render(1).to_a).to eq([
        "Title,Description\n",
        "Added salt,A teaspoon\n"
      ])
    end
  end
end