Contents

require 'serverspec'
require 'serverspec/type/base'
require 'net/http'
require 'json'

module Serverspec::Type
  class OctopusDeployWorkerPool < Base
    @worker_pool = nil
    @serverUrl = nil
    @apiKey = nil

    def initialize(serverUrl, apiKey, worker_pool_name)
      @name = "Octopus Deploy Worker Pool #{worker_pool_name}"
      @runner = Specinfra::Runner
      @serverUrl = serverUrl
      @apiKey = apiKey

      if (serverUrl.nil?)
        raise "'serverUrl' was not provided. Unable to connect to Octopus server to validate configuration."
      end
      if (apiKey.nil?)
        raise "'apiKey' was not provided. Unable to connect to Octopus server to validate configuration."
      end
      if (worker_pool_name.nil?)
        raise "'worker_pool_name' was not provided. Unable to connect to Octopus server to validate configuration."
      end

      @worker_pool = get_worker_pool_via_api(serverUrl, apiKey, worker_pool_name)
    end

    def exists?
      (!@worker_pool.nil?) && (@worker_pool != [])
    end
  end

  def octopus_deploy_worker_pool(serverUrl, apiKey, worker_pool_name)
    OctopusDeployWorkerPool.new(serverUrl, apiKey, worker_pool_name)
  end

  private

  def get_worker_pool_via_api(serverUrl, apiKey, worker_pool_name)
    worker_pool = nil
    url = "#{serverUrl}/api/workerpools/all?api-key=#{apiKey}"

    begin
      resp = Net::HTTP.get_response(URI.parse(url))
      body = JSON.parse(resp.body)
      worker_pool = body.select {|i| i['Name'] == worker_pool_name } unless body.nil?
    rescue => e
      raise "Unable to connect to #{url}: #{e}"
    end

    worker_pool
  end
end

include Serverspec::Type

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
octopus-serverspec-extensions-0.15.5	lib/octopus_serverspec_extensions/type/octopus_deploy_worker_pool.rb
octopus-serverspec-extensions-0.15.4	lib/octopus_serverspec_extensions/type/octopus_deploy_worker_pool.rb
octopus-serverspec-extensions-0.15.3	lib/octopus_serverspec_extensions/type/octopus_deploy_worker_pool.rb
octopus-serverspec-extensions-0.15.2	lib/octopus_serverspec_extensions/type/octopus_deploy_worker_pool.rb
octopus-serverspec-extensions-0.15.1	lib/octopus_serverspec_extensions/type/octopus_deploy_worker_pool.rb
octopus-serverspec-extensions-0.15.0	lib/octopus_serverspec_extensions/type/octopus_deploy_worker_pool.rb