serverspec-extra-types 0.1.0

data/lib/serverspec_extra_types/matchers/be_a_worker_node.rb

@@ -0,0 +1,14 @@
+RSpec::Matchers.define :be_a_worker_node do
+  match do |actual|
+    actual.role == 'worker'
+  end
+  failure_message do |actual|
+    "expected '#{actual.role}' to be worker"
+  end
+  description do
+    'be a worker node'
+  end
+end
+RSpec::Matchers.alias_matcher :be_worker, :be_a_worker_node
+RSpec::Matchers.alias_matcher :be_a_worker, :be_a_worker_node
+RSpec::Matchers.alias_matcher :be_worker_node, :be_a_worker_node

data/lib/serverspec_extra_types/matchers/be_active.rb

@@ -0,0 +1,11 @@
+RSpec::Matchers.define :be_active do
+  match do |actual|
+    actual.active?
+  end
+  failure_message do |actual|
+    "expected '#{actual.availability}' to be active"
+  end
+  description do
+    'be active'
+  end
+end

data/lib/serverspec_extra_types/matchers/configure_queue.rb

@@ -0,0 +1,8 @@
+RSpec::Matchers.define :configure_queue do |vhost, queue|
+  match do |actual|
+   actual.configure_queue? vhost, queue
+  end
+  description do
+    "be able configure #{queue} on the #{vhost} vhost"
+  end
+end

data/lib/serverspec_extra_types/matchers/have_count.rb

@@ -0,0 +1,11 @@
+RSpec::Matchers.define :have_count do |count|
+  match do |actual|
+    actual.length ==  count
+  end
+  description do
+    "have #{count}"
+  end
+  failure_message do |actual|
+    "expected count to be #{count} was #{actual.length}"
+  end
+end

data/lib/serverspec_extra_types/matchers/have_domain_name.rb

@@ -0,0 +1,10 @@
+RSpec::Matchers.define :have_domainname do |domain|
+  match do |actual|
+    actual.has_domainname? domain
+  end
+  failure_message do |actual|
+    "expected domain name to be #{domain}, was #{actual.domain_name}"
+  end
+end
+
+RSpec::Matchers.alias_matcher :have_domain_name, :have_domainname

data/lib/serverspec_extra_types/matchers/have_engine_version.rb

@@ -0,0 +1,11 @@
+RSpec::Matchers.define :have_engine_version do |version|
+  match do |actual|
+    actual.has_engine_version? version
+  end
+  failure_message do |actual|
+    "expected version #{actual.engine_version} to be #{version}"
+  end
+  description do
+    "be running docker version #{version}"
+  end
+end

data/lib/serverspec_extra_types/matchers/have_environment_variable.rb

@@ -0,0 +1,21 @@
+RSpec::Matchers.define :have_environment_variable do |regex|
+  chain :with_value do | value|
+    @value = value
+  end
+  match do |actual|
+    actual.has_environment_variable? regex, @value
+  end
+  description do
+    message = "have an environment variable #{regex}"
+    message << %( with a value of "#{@value}") if @value
+    message
+  end
+  failure_message do |actual|
+    if !@value
+      "expected #{actual.environment_variables} to contain #{regex}"
+    else
+      "expected #{regex} to have value #{@value}, it was #{actual.environment_variable(regex)}"
+    end
+  end
+
+end

data/lib/serverspec_extra_types/matchers/have_ha_mode.rb

@@ -0,0 +1,5 @@
+RSpec::Matchers.define :have_ha_mode do |mode|
+  match do |actual|
+    actual.has_ha_mode? mode
+  end
+end

data/lib/serverspec_extra_types/matchers/have_ha_nodes.rb

@@ -0,0 +1,8 @@
+RSpec::Matchers.define :have_ha_nodes do |count|
+  match do |actual|
+    actual.has_ha_nodes? count
+  end
+  description do
+    "have #{count} ha nodes"
+  end
+end

data/lib/serverspec_extra_types/matchers/have_ha_sync_mode.rb

@@ -0,0 +1,5 @@
+RSpec::Matchers.define :have_ha_sync_mode do |mode|
+  match do |actual|
+    actual.has_ha_sync_mode? mode
+  end
+end

data/lib/serverspec_extra_types/matchers/have_host.rb

@@ -0,0 +1,8 @@
+RSpec::Matchers.define :have_host do |host|
+  match do |actual|
+    actual.has_host? host
+  end
+  failure_message do |actual|
+    "expected #{actual.hosts} to contain #{host}"
+  end
+end

data/lib/serverspec_extra_types/matchers/have_hostname.rb

@@ -0,0 +1,6 @@
+RSpec::Matchers.define :have_hostname do |hostname|
+  match do |actual|
+    actual.has_hostname? hostname
+  end
+end
+RSpec::Matchers.alias_matcher :have_host_name, :have_hostname

data/lib/serverspec_extra_types/matchers/have_image.rb

@@ -0,0 +1,8 @@
+RSpec::Matchers.define :have_image do |image|
+  match do |actual|
+    actual.has_image?(image)
+  end
+  failure_message do |actual|
+    "expected image to be #{image} was #{actual.image}"
+  end
+end

data/lib/serverspec_extra_types/matchers/have_image_sha.rb

@@ -0,0 +1,5 @@
+RSpec::Matchers.define :have_image_sha do |sha|
+  match do |actual|
+    actual['Image'] == sha
+  end
+end

data/lib/serverspec_extra_types/matchers/have_label.rb

@@ -0,0 +1,20 @@
+RSpec::Matchers.define :have_label do |label|
+  match do |actual|
+    actual.labels.key? label
+  end
+  failure_message do |actual|
+    "expected #{actual.labels} to have key #{label}"
+  end
+end
+
+RSpec::Matchers.define :have_label_with_value do |label, value|
+  match do |actual|
+    actual.labels[label] == value
+  end
+  failure_message do |actual|
+    "expected #{actual.labels[label]} to be #{value}"
+  end
+  description do
+    "have label #{label} with value #{value}"
+  end
+end

data/lib/serverspec_extra_types/matchers/have_mount.rb

@@ -0,0 +1,11 @@
+RSpec::Matchers.define :have_mount do |source, target, type = 'bind'|
+  match do |actual|
+    actual.has_mount?(source, target, type)
+  end
+  description do
+    "mount #{source} on host to #{target} in container as a #{type} mount"
+  end
+  failure_message do |actual|
+    "expected #{actual.mounts} to contain {'Target' => #{target}, 'Destination' => #{source} , 'Type' => #{type}}"
+  end
+end

data/lib/serverspec_extra_types/matchers/have_network.rb

@@ -0,0 +1,8 @@
+RSpec::Matchers.define :have_network do |name|
+  match do |actual|
+    actual.has_network? name
+  end
+  failure_message do |actual|
+    "expected #{actual.networks} to include #{name}"
+  end
+end

data/lib/serverspec_extra_types/matchers/have_placement_constraint.rb

@@ -0,0 +1,8 @@
+RSpec::Matchers.define :have_placement_constraint do |constraint|
+  match do |actual|
+    actual.has_placement_constraint? constraint
+  end
+  failure_message do |actual|
+    "expected #{actual.placement_constraint} to include #{constraint}"
+  end
+end

data/lib/serverspec_extra_types/matchers/have_replica_count.rb

@@ -0,0 +1,8 @@
+RSpec::Matchers.define :have_replica_count do |count|
+  match do |actual|
+    actual.replicas == count
+  end
+  description do
+    "have #{count} replica#{count != 1 ? 's' : ''}"
+  end
+end

data/lib/serverspec_extra_types/matchers/have_restart_limit.rb

@@ -0,0 +1,8 @@
+RSpec::Matchers.define :have_restart_limit do |count|
+  match do |actual|
+    actual.has_restart_limit? count
+  end
+  failure_message do |actual|
+    "expected restart limit to be #{count}, was #{actual.restart_limit}"
+  end
+end

data/lib/serverspec_extra_types/matchers/have_restart_policy.rb

@@ -0,0 +1,8 @@
+RSpec::Matchers.define :have_restart_policy do |policy|
+  match do |actual|
+    actual.has_restart_policy?(policy)
+  end
+  failure_message do |actual|
+    "expected restart policy to be #{policy}, was #{actual.restart_policy}"
+  end
+end

data/lib/serverspec_extra_types/matchers/have_user.rb

@@ -0,0 +1,10 @@
+RSpec::Matchers.define :have_user do |user|
+  match do |actual|
+    actual.has_user? user
+  end
+  failure_message do |actual|
+    "expected user to be #{user} was #{actual.user}"
+  end
+end
+
+RSpec::Matchers.alias_matcher :run_as_user, :have_user

data/lib/serverspec_extra_types/matchers/have_vhost.rb

@@ -0,0 +1,5 @@
+RSpec::Matchers.define :have_vhost do |vhost|
+  match do |actual|
+    actual.has_vhost? vhost
+  end
+end

data/lib/serverspec_extra_types/matchers/include_regex.rb

@@ -0,0 +1,5 @@
+RSpec::Matchers.define :include_regex do |regex|
+  match do |actual|
+    actual.include_regex? regex
+  end
+end

data/lib/serverspec_extra_types/matchers/map_port.rb

@@ -0,0 +1,25 @@
+RSpec::Matchers.define :map_port do |host, container|
+
+  match do |actual|
+    @protocol = 'tcp' unless @protocol
+    if @mode
+      actual.map_port? host, container, @protocol, @mode
+    else
+      actual.map_port? host, container, @protocol
+    end
+  end
+  description do
+    msg = "map container port #{container} to host port #{host}"
+    msg << %( using the "#{@protocol}" protocol) if @protocol
+    msg << %( in "#{@mode}" mode) if @mode
+    msg
+  end
+  failure_message do |actual|
+    @protocol = 'tcp' unless @protocol
+    "expected #{actual.port_map} to contain {\"#{container}/#{@protocol}\"=>[{\"HostPort\"=>\"#{host}\"}] }"
+  end
+
+  chain :using_protocol, :protocol
+  chain :with_mode, :mode
+
+end

data/lib/serverspec_extra_types/matchers/mirror_all.rb

@@ -0,0 +1,5 @@
+RSpec::Matchers.define :mirror_all do
+  match do |actual|
+    actual.mirror_all?
+  end
+end

data/lib/serverspec_extra_types/matchers/publish_all_ports.rb

@@ -0,0 +1,3 @@
+RSpec::Matchers.define :publish_all_ports do
+  match { |actual| actual.publishes_all_ports? }
+end

data/lib/serverspec_extra_types/matchers/read_from_queue.rb

@@ -0,0 +1,8 @@
+RSpec::Matchers.define :read_from_queue do |vhost, queue|
+  match do |actual|
+    actual.read_from_queue?(vhost, queue)
+  end
+  description do
+    "be able to read from #{queue} on the #{vhost} vhost"
+  end
+end

data/lib/serverspec_extra_types/matchers/write_to_queue.rb

@@ -0,0 +1,8 @@
+RSpec::Matchers.define :write_to_queue do |vhost, queue|
+  match do |actual|
+    actual.write_to_queue? vhost, queue
+  end
+  description do
+    "be able to write to #{queue} on the #{vhost} vhost"
+  end
+end

data/lib/serverspec_extra_types/types.rb

@@ -0,0 +1,18 @@
+require 'serverspec_extra_types/types/docker_container'
+require 'serverspec/helper/type'
+
+module Serverspec
+  module Helper
+    module Type
+      types = %w[docker_service rabbitmq_vhost_policy rabbitmq_node_list rabbitmq_vhost_list
+                 rabbitmq_user_permission consul_service consul_service_list consul_node consul_node_list]
+
+      types.each do |type|
+        require "serverspec_extra_types/types/#{type}"
+        define_method type do |*_args|
+          eval "Serverspec::Type::#{type.to_camel_case}.new(*_args)"
+        end
+      end
+    end
+  end
+end

data/lib/serverspec_extra_types/types/api_base.rb

@@ -0,0 +1,36 @@
+require 'serverspec'
+require 'serverspec/type/base'
+require 'multi_json'
+require 'serverspec_extra_types/helpers/properties'
+
+module Serverspec::Type
+  class ApiBase < Base
+    def initialize(name = nil, options = {})
+      super(name, options)
+    end
+
+    def [](key)
+      value = inspection
+      key.split('.').each do |k|
+        is_index = k.start_with?('[') && k.end_with?(']')
+        value = value[is_index ? k[1..-2].to_i : k]
+      end
+      value
+    end
+
+    def url
+      @url_base
+    end
+
+    def inspection
+      @inspection ||= ::MultiJson.load(get_inspection.stdout)
+    end
+
+    private
+
+    def get_inspection
+      command = "curl -s #{url}"
+      @get_inspection ||= @runner.run_command(command)
+    end
+  end
+end

data/lib/serverspec_extra_types/types/consul_base.rb

@@ -0,0 +1,40 @@
+require 'serverspec'
+require 'serverspec/type/base'
+require 'multi_json'
+require 'serverspec_extra_types/helpers/properties'
+require 'serverspec_extra_types/types/api_base'
+
+module Serverspec::Type
+  class ConsulBase < ApiBase
+    def initialize(name = nil, acl_token = nil, options = {})
+      super(name, options)
+      @token = acl_token
+      @url_base = property[:variables][:consul_url] || 'http://localhost:8500'
+    end
+
+    def [](key)
+      value = inspection
+      key.split('.').each do |k|
+        is_index = k.start_with?('[') && k.end_with?(']')
+        value = value[is_index ? k[1..-2].to_i : k]
+      end
+      value
+    end
+
+    def url
+      @url_base
+    end
+
+    def inspection
+      @inspection ||= ::MultiJson.load(get_inspection.stdout)[0]
+    end
+
+    private
+
+    def get_inspection
+      headers = @token ? "--header 'X-Consul-Token: #{@token}'" : ''
+      command = "curl -s #{headers} #{url}"
+      @get_inspection ||= @runner.run_command(command)
+    end
+  end
+end

data/lib/serverspec_extra_types/types/consul_node.rb

@@ -0,0 +1,20 @@
+require 'serverspec_extra_types/types/consul_base'
+
+
+# TODO: List and singular
+module Serverspec::Type
+  class ConsulNode < ConsulBase
+
+    def url
+      "#{@url_base}/v1/catalog/node/#{@name}"
+    end
+
+
+
+    def inspection
+      @inspection ||= ::MultiJson.load(get_inspection.stdout) #Need a find
+    end
+
+
+  end
+end