kitchen-digital_ocean 0.3.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: fc1701a957915027c014f74ee178a5222744d634
+  data.tar.gz: c5355d1367f8c36c9144aa7239798cdec06f8f88
+SHA512:
+  metadata.gz: 209df6a181b18ab1a2977ba447e7eab39700b13b74c3b28e15b500ee0346799d3b04005a07cc9bb04e2751b7d4f3cfa2c0ed9943940e709ccbd66068336d381b
+  data.tar.gz: 092cee75876360102fc19761d7f00b013271d7d7baa561b411c5069d920e4795bb8b0fee375dda022053a05c93d924fa9f1efde5cab78a9ba31b81f6b05f6f04

data/.gitignore

@@ -0,0 +1,20 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+.kitchen
+*.sw?
+*~

data/Gemfile

@@ -0,0 +1,6 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in kitchen-digitalocean.gemspec
+gemspec
+
+# vim: ai et ts=2 sts=2 sw=2 ft=ruby

data/LICENSE.txt

@@ -0,0 +1,17 @@
+Author:: Greg Fitzgerald (<greg@gregf.org>)
+
+Copyright (C) 2013, Greg Fitzgerald
+Copyright (C) 2014, Will Farrington
+
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+   http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.

data/README.md

@@ -0,0 +1,48 @@
+# Kitchen::DigitalOcean
+
+A Test Kitchen driver for Digital Ocean API v2.
+
+Forked from the official [test-kitchen/kitchen-digitalocean driver](https://github.com/test-kitchen/kitchen-digitalocean).
+
+## Requirements
+
+* Obviously, this gem depends on Test Kitchen.
+* It also depends on `rest_client`, but that'll automatically be handled via RubyGems.
+* You also need a Digital Ocean account.
+* You need to define the following environment variables:
+  * `DIGITALOCEAN_API_TOKEN` -- your API token (requires read+write)
+  * `DIGITALOCEAN_SSH_KEYS` -- comma-separated list of SSH key IDs (get from Digital Ocean API)
+
+## Installation and Setup
+
+1. Use ChefDK.
+1. Be happy.
+
+```
+chef gem install kitchen-digital_ocean
+```
+
+In your `.kitchen.yml` or `.kitchen.local.yml`:
+
+``` yaml
+---
+driver:
+  name: digitalocean
+
+platforms:
+  - name: ubuntu-12.10
+```
+
+## Additional configuration options
+
+* `username` -- user name to SSH with
+* `port` -- SSH port to SSH into
+* `private_networking` -- enable private networking on the drpolet
+* `region` -- the region to provision the droplet in; you can use the short slugs (eg. `nyc2`)
+* `size` -- the size to provision the droplet as; you can use the short slugs (eg. `2gb`)
+* `image` -- the image to provision the droplet with; you can use the short slugs
+* `server_name` -- you probably want to leave this alone
+* `digitalocean_api_token` -- you can set this, but seriously use the environment variable above
+* `digitalocean_ssh_keys` -- you can set this, but seriously use the environment variable above
+
+For more details, see the [Digital Ocean API documentation](https://developers.digitalocean.com/).

data/kitchen-digital_ocean.gemspec

@@ -0,0 +1,28 @@
+# coding: utf-8
+
+Gem::Specification.new do |spec|
+  spec.name          = 'kitchen-digital_ocean'
+  spec.version       = '0.3.0'
+  spec.authors       = ['Will Farrington', 'Greg Fitzgerald']
+  spec.email         = ['wfarr@digitalocean.com']
+  spec.description   = 'A Test Kitchen Driver for Digital Ocean using apiv2'
+  spec.summary       = spec.description
+  spec.homepage      = 'https://github.com/wfarr/kitchen-digital_ocean'
+  spec.license       = 'Apache 2.0'
+
+  spec.files         = `git ls-files`.split($INPUT_RECORD_SEPARATOR)
+  spec.executables   = []
+  spec.test_files    = []
+  spec.require_paths = ['lib']
+
+  spec.add_dependency 'test-kitchen', '~> 1.0'
+  spec.add_dependency 'rest_client'
+
+  spec.add_development_dependency 'rubocop'
+  spec.add_development_dependency 'rake'
+  spec.add_development_dependency 'cane'
+  spec.add_development_dependency 'countloc'
+  spec.add_development_dependency 'rspec'
+end
+
+# vim: ai et ts=2 sts=2 sw=2 ft=ruby

data/lib/kitchen/driver/digital_ocean.rb

@@ -0,0 +1,184 @@
+# -*- encoding: utf-8 -*-
+#
+# Author:: Greg Fitzgerald (<greg@gregf.org>)
+#
+# Copyright (C) 2013, Greg Fitzgerald
+# Copyright (C) 2014, Will Farrington
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'rest_client'
+require 'json'
+require 'kitchen'
+require 'etc'
+require 'socket'
+
+module Kitchen
+  module Driver
+    # Digital Ocean driver for Kitchen.
+    #
+    # @author Greg Fitzgerald <greg@gregf.org>
+    class DigitalOcean < Kitchen::Driver::SSHBase
+      IMAGES = {
+        "ubuntu-12.04" => "ubuntu-12-04-x64",
+        "ubuntu-14.04" => "ubuntu-14-04-x64",
+      }
+
+      default_config :username, 'root'
+      default_config :port, '22'
+
+      default_config :private_networking, true
+
+      default_config :region, 'nyc2'
+      default_config :size, '2gb'
+
+      default_config :image do |driver|
+        driver.default_image
+      end
+
+      default_config :server_name do |driver|
+        driver.default_name
+      end
+
+      default_config :digitalocean_api_token do
+        ENV['DIGITALOCEAN_API_TOKEN']
+      end
+
+      default_config :ssh_keys do
+        ENV['DIGITALOCEAN_SSH_KEYS']
+      end
+
+      required_config :digitalocean_api_token
+      required_config :ssh_keys
+
+      def create(state)
+        droplet = create_droplet
+        state[:server_id] = droplet['id']
+
+        info("Digital Ocean instance <#{state[:server_id]}> created.")
+
+        while true
+          sleep 10
+          droplet = get_droplet(state[:server_id])
+
+          break if droplet \
+            && droplet['networks'] \
+            && droplet['networks']['v4'] \
+            && droplet['networks']['v4'].any? { |n| n['type'] == 'public' }
+        end
+
+        state[:hostname] = droplet['networks']['v4'].detect { |n| n['type'] == 'public' }['ip_address']
+
+        wait_for_sshd(state[:hostname]) ; print "(ssh ready)\n"
+
+        debug("digitalocean:create #{state[:hostname]}")
+      rescue  RestClient::Exception => e
+        raise ActionFailed, e.message
+      end
+
+      def destroy(state)
+        return if state[:server_id].nil?
+
+        if get_droplet(state[:server_id])
+          destroy_droplet state[:server_id]
+        end
+
+        info("Digital Ocean instance <#{state[:server_id]}> destroyed.")
+
+        state.delete(:server_id)
+        state.delete(:hostname)
+      rescue  RestClient::Exception => e
+        raise ActionFailed, e.message
+      end
+
+      def default_image
+        IMAGES.fetch(instance.platform.name) { 'ubuntu-14-04-x64' }
+      end
+
+      def default_name
+        # Generate what should be a unique server name
+        rand_str = Array.new(8) { rand(36).to_s(36) }.join
+        "#{instance.name}-"\
+        "#{Etc.getlogin.gsub('_', '-')}-"\
+        "#{rand_str}-"\
+        "#{Socket.gethostname}"
+      end
+
+      private
+
+      def get_droplet(id)
+        api_request(:get, "droplets/#{id}")['droplet']
+      rescue RestClient::Exception => e
+        nil
+      end
+
+      def create_droplet
+        debug_droplet_config
+
+        droplet = api_request :post, 'droplets', {
+          name: config[:server_name],
+          region: config[:region],
+          size: config[:size],
+          image: config[:image],
+          ssh_keys: config[:ssh_keys].split(','),
+        }
+
+        droplet['droplet']
+      end
+
+      def destroy_droplet(id)
+        api_request :delete, "droplets/#{id}"
+      end
+
+      def debug_droplet_config
+        debug("digitalocean_api_token #{config[:digitalocean_api_token]}")
+        debug("digitalocean:name #{config[:server_name]}")
+        debug("digitalocean:image_id #{config[:image_id]}")
+        debug("digitalocean:flavor_id #{config[:flavor_id]}")
+        debug("digitalocean:region_id #{config[:region_id]}")
+        debug("digitalocean:ssh_key_ids #{config[:ssh_key_ids]}")
+        debug("digitalocean:private_networking #{config[:private_networking]}")
+      end
+
+      def api_request(method, url, content=nil)
+        url = "https://api.digitalocean.com/v2/#{url}" unless url =~ /^http/
+        headers = {'Authorization' => "Bearer #{config[:digitalocean_api_token]}"}
+        if %w(get head delete).include? method.to_s
+          res = RestClient.send(method.to_sym, url, headers)
+          json = JSON.parse(res) unless res.empty?
+
+          if json
+            if json['links'] && json['links']['pages'] && json['links']['pages']['next']
+              json.deep_merge!(api_request(method, json['links']['pages']['next']))
+            else
+              json
+            end
+          else
+            ""
+          end
+        elsif %w(post put).include? method.to_s
+          begin
+            JSON.parse(RestClient.send(method.to_sym, url, content, headers.merge(:content_type => 'application/json')))
+          rescue RestClient::Exception => rce
+            raise rce
+          end
+        else
+          raise "What kind of HTTP method actually is #{method}"
+        end
+      end
+
+    end
+  end
+end
+
+# vim: ai et ts=2 sts=2 sw=2 ft=ruby

metadata

@@ -0,0 +1,150 @@
+--- !ruby/object:Gem::Specification
+name: kitchen-digital_ocean
+version: !ruby/object:Gem::Version
+  version: 0.3.0
+platform: ruby
+authors:
+- Will Farrington
+- Greg Fitzgerald
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-07-26 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: test-kitchen
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: rest_client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: cane
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: countloc
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: A Test Kitchen Driver for Digital Ocean using apiv2
+email:
+- wfarr@digitalocean.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- kitchen-digital_ocean.gemspec
+- lib/kitchen/driver/digital_ocean.rb
+homepage: https://github.com/wfarr/kitchen-digital_ocean
+licenses:
+- Apache 2.0
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.1
+signing_key: 
+specification_version: 4
+summary: A Test Kitchen Driver for Digital Ocean using apiv2
+test_files: []
+has_rdoc: