diff options
Diffstat (limited to 'lib/bulk_imports/stage.rb')
-rw-r--r-- | lib/bulk_imports/stage.rb | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/lib/bulk_imports/stage.rb b/lib/bulk_imports/stage.rb index 9c19e9ea60b..6cf394c5df0 100644 --- a/lib/bulk_imports/stage.rb +++ b/lib/bulk_imports/stage.rb @@ -2,10 +2,13 @@ module BulkImports class Stage - def initialize(bulk_import) - raise(ArgumentError, 'Expected an argument of type ::BulkImport') unless bulk_import.is_a?(::BulkImport) + def initialize(bulk_import_entity) + unless bulk_import_entity.is_a?(::BulkImports::Entity) + raise(ArgumentError, 'Expected an argument of type ::BulkImports::Entity') + end - @bulk_import = bulk_import + @bulk_import_entity = bulk_import_entity + @bulk_import = bulk_import_entity.bulk_import end def pipelines |