fog--fog/tests/digitalocean/helper.rb

# Shortcut for Fog::Compute[:digitalocean]
def service
  Fog::Compute[:digitalocean]
end

def fog_test_server_attributes
  image = service.images.find { |i| i.name == 'Ubuntu 12.04 x64' }
  region = service.regions.find { |r| r.name == 'New York 1' }
  flavor = service.flavors.find { |r| r.name == '512MB' }

  {
    :image_id  => image.id,
    :region_id => region.id,
    :flavor_id => flavor.id
  }
end

def fog_server_name
  "fog-server-test"
end

# Create a long lived server for the tests
def fog_test_server
  server = service.servers.find { |s| s.name == fog_server_name }
  unless server
    server = service.servers.create({
      :name => fog_server_name
    }.merge(fog_test_server_attributes))
    server.wait_for { ready? }
  end
  server
end

# Destroy the long lived server
def fog_test_server_destroy
  server = service.servers.find { |s| s.name == fog_server_name }
  server.destroy if server
end

at_exit do
  unless Fog.mocking? || Fog.credentials[:digitalocean_api_key].nil?
    server = service.servers.find { |s| s.name == fog_server_name }
    if server
      server.wait_for(120) do
        reload rescue nil; ready?
      end
    end
    fog_test_server_destroy
  end
end