fog--fog/tests/openvz/helper.rb

# Shortcut for Fog::Compute[:openvz]
def openvz_service
  Fog::Compute[:openvz]
end

# Create a long lived server for the tests
def openvz_fog_test_server
  server = openvz_service.servers.find { |s| s.ctid == '104' }
  unless server
    server = openvz_service.servers.create :ctid      => '104'
    server.start
    server.reload
    # Wait for the server to come up
    begin
      server.wait_for(120) { server.reload rescue nil; server.ready? }
    rescue Fog::Errors::TimeoutError
      # Server bootstrap took more than 120 secs!
    end
  end
  server
end

# Destroy the long lived server
def openvz_fog_test_server_destroy
  server = openvz_service.servers.find { |s| s.ctid == '104' }
  server.destroy if server
end

at_exit do
  unless Fog.mocking?
    server = openvz_service.servers.find { |s| s.name == '104' }
    if server
      server.wait_for(120) do
        reload rescue nil; ready?
      end
    end
    server.stop
    openvz_fog_test_server_destroy
  end
end