gitlab-org--gitlab-foss/spec/lib/bulk_imports/pipeline_spec.rb

147 lines
4.5 KiB
Ruby

# frozen_string_literal: true
require 'spec_helper'
RSpec.describe BulkImports::Pipeline do
let(:context) { instance_double(BulkImports::Pipeline::Context, tracker: nil) }
before do
stub_const('BulkImports::Extractor', Class.new)
stub_const('BulkImports::Transformer', Class.new)
stub_const('BulkImports::Loader', Class.new)
klass = Class.new do
include BulkImports::Pipeline
abort_on_failure!
extractor BulkImports::Extractor, foo: :bar
transformer BulkImports::Transformer, foo: :bar
loader BulkImports::Loader, foo: :bar
end
stub_const('BulkImports::MyPipeline', klass)
end
describe 'pipeline attributes' do
describe 'getters' do
it 'retrieves class attributes' do
expect(BulkImports::MyPipeline.get_extractor).to eq({ klass: BulkImports::Extractor, options: { foo: :bar } })
expect(BulkImports::MyPipeline.transformers).to contain_exactly({ klass: BulkImports::Transformer, options: { foo: :bar } })
expect(BulkImports::MyPipeline.get_loader).to eq({ klass: BulkImports::Loader, options: { foo: :bar } })
expect(BulkImports::MyPipeline.abort_on_failure?).to eq(true)
end
context 'when extractor and loader are defined within the pipeline' do
before do
klass = Class.new do
include BulkImports::Pipeline
def extract; end
def load; end
end
stub_const('BulkImports::AnotherPipeline', klass)
end
it 'returns itself when retrieving extractor & loader' do
pipeline = BulkImports::AnotherPipeline.new(context)
expect(pipeline.send(:extractor)).to eq(pipeline)
expect(pipeline.send(:loader)).to eq(pipeline)
end
end
end
describe 'setters' do
it 'sets class attributes' do
klass = Class.new
options = { test: :test }
BulkImports::MyPipeline.extractor(klass, options)
BulkImports::MyPipeline.transformer(klass, options)
BulkImports::MyPipeline.loader(klass, options)
BulkImports::MyPipeline.abort_on_failure!
BulkImports::MyPipeline.file_extraction_pipeline!
expect(BulkImports::MyPipeline.get_extractor).to eq({ klass: klass, options: options })
expect(BulkImports::MyPipeline.transformers)
.to contain_exactly(
{ klass: BulkImports::Transformer, options: { foo: :bar } },
{ klass: klass, options: options })
expect(BulkImports::MyPipeline.get_loader).to eq({ klass: klass, options: options })
expect(BulkImports::MyPipeline.abort_on_failure?).to eq(true)
expect(BulkImports::MyPipeline.file_extraction_pipeline?).to eq(true)
end
end
end
describe '#instantiate' do
context 'when options are present' do
it 'instantiates new object with options' do
expect(BulkImports::Extractor).to receive(:new).with(foo: :bar)
expect(BulkImports::Transformer).to receive(:new).with(foo: :bar)
expect(BulkImports::Loader).to receive(:new).with(foo: :bar)
pipeline = BulkImports::MyPipeline.new(context)
pipeline.send(:extractor)
pipeline.send(:transformers)
pipeline.send(:loader)
end
end
context 'when options are missing' do
before do
klass = Class.new do
include BulkImports::Pipeline
extractor BulkImports::Extractor
transformer BulkImports::Transformer
loader BulkImports::Loader
end
stub_const('BulkImports::NoOptionsPipeline', klass)
end
it 'instantiates new object without options' do
expect(BulkImports::Extractor).to receive(:new).with(no_args)
expect(BulkImports::Transformer).to receive(:new).with(no_args)
expect(BulkImports::Loader).to receive(:new).with(no_args)
pipeline = BulkImports::NoOptionsPipeline.new(context)
pipeline.send(:extractor)
pipeline.send(:transformers)
pipeline.send(:loader)
end
end
end
describe '#transformers' do
before do
klass = Class.new do
include BulkImports::Pipeline
transformer BulkImports::Transformer
def transform; end
end
stub_const('BulkImports::TransformersPipeline', klass)
end
it 'has instance transform method first to run' do
transformer = double
allow(BulkImports::Transformer).to receive(:new).and_return(transformer)
pipeline = BulkImports::TransformersPipeline.new(context)
expect(pipeline.send(:transformers)).to eq([pipeline, transformer])
end
end
end