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

def perform(image = 'CE')
  ce_release = 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) do |scenario|
      scenario.volumes = volumes
    end
    Scenario::Test::Instance::Image
      .perform(ce_release.to_ee) do |scenario|
      scenario.volumes = volumes
    end
  end
end