class Gitlab::QA::Scenario::Test::Omnibus::Upgrade

def perform(image = 'CE', *rspec_args)
  ce_release = QA::Release.new(image)
  if ce_release.ee?
    raise ArgumentError, 'Only CE can be upgraded to EE!'
  end
  Docker::Volumes.new.with_temporary_volumes do |volumes|
    Scenario::Test::Instance::Image
      .perform(ce_release, *rspec_args) do |scenario|
      scenario.volumes = volumes
    end
    Scenario::Test::Instance::Image
      .perform(ce_release.to_ee, *rspec_args) do |scenario|
      scenario.volumes = volumes
    end
  end
end