serverspec-extra-types 0.1.0

data/lib/serverspec_extra_types/types/consul_node_list.rb

@@ -0,0 +1,32 @@
+require 'serverspec_extra_types/types/consul_base'
+
+
+# TODO: List and singular
+module Serverspec::Type
+  class ConsulNodeList < ConsulBase
+
+    def url
+      "#{@url_base}/v1/catalog/nodes"
+    end
+
+    def to_s
+      msg = "Consul Node List"
+      msg << %( with acl token: "#{@token}") if @token
+      msg
+    end
+    
+    def has_node?(node)
+      nodes.include? node
+    end
+
+    def nodes
+      inspection.keys
+    end
+
+    def inspection
+      @inspection ||= ::MultiJson.load(get_inspection.stdout)
+    end
+
+
+  end
+end

data/lib/serverspec_extra_types/types/consul_service.rb

@@ -0,0 +1,24 @@
+require 'serverspec_extra_types/types/consul_base'
+
+
+# TODO: List and singular
+module Serverspec::Type
+  class ConsulService < ConsulBase
+
+    def url
+      "#{@url_base}/v1/catalog/service/#{@name}"
+    end
+
+    def to_s
+      msg = "Consul Service #{@name}"
+      msg << %( with acl token: "#{@token}") if @token
+      msg
+    end
+
+    def inspection
+      @inspection ||= ::MultiJson.load(get_inspection.stdout)
+    end
+
+
+  end
+end

data/lib/serverspec_extra_types/types/consul_service_list.rb

@@ -0,0 +1,32 @@
+require 'serverspec_extra_types/types/consul_base'
+
+
+# TODO: List and singular
+module Serverspec::Type
+  class ConsulServiceList < ConsulBase
+
+    def url
+      "#{@url_base}/v1/catalog/services"
+    end
+
+    def to_s
+      msg = "Consul Service List"
+      msg << %( with acl token: "#{@token}") if @token
+      msg
+    end
+
+    def has_service?(service)
+      services.include? service
+    end
+
+    def services
+      inspection.keys
+    end
+
+    def inspection
+      @inspection ||= ::MultiJson.load(get_inspection.stdout)
+    end
+
+
+  end
+end

data/lib/serverspec_extra_types/types/docker_container.rb

@@ -0,0 +1,124 @@
+require 'serverspec'
+
+module Serverspec::Type
+  # This class  monkey patches serverspec's docker container type with some more method to be used in matchers
+  class DockerContainer
+    def include_regex?
+      inspection.find { |str| str =~ regex }
+    end
+
+    def has_image?(image)
+      self.image == image
+    end
+
+    def has_image_sha?(image)
+      self.image_sha == image
+    end
+
+    def image
+      inspection['Config']['Image'].split('@')[0]
+    end
+
+    def image_sha
+      inspection['Config']['Image'].split('@')[1]
+    end
+
+    def has_environment_variable?(regex, value = nil)
+      if value
+        environment_variable(regex) == value
+      else
+        environment_variable(regex)
+      end
+
+    end
+
+    def environment_variable(regex)
+      environment_variables.find { |str| str =~ /^#{regex}=/ }.split('=')[1]
+    end
+
+    def environment_variables
+      inspection['Config']['Env']
+    end
+
+    def has_user?(user)
+      self.user == user
+    end
+
+    def user
+      inspection['Config']['User']
+    end
+
+    def has_hostname?(hostname)
+      self.hostname == hostname
+    end
+
+    def hostname
+      inspection['Config']['Hostname']
+    end
+
+    def has_domainname?(domain)
+      self.domain_name == domain
+    end
+
+    def domain_name
+      inspection['Config']['Domainname']
+    end
+
+    def has_restart_limit?(limit)
+      restart_limit == limit
+    end
+
+    def restart_limit
+      inspection['HostConfig']['RestartPolicy']['MaximumRetryCount']
+    end
+
+    def has_restart_policy?(policy)
+      restart_policy == policy
+    end
+
+    def restart_policy
+      inspection['HostConfig']['RestartPolicy']['Name']
+    end
+    def has_host?(host)
+      hosts.include? host
+    end
+    def hosts
+      inspection['HostConfig']['ExtraHosts'].map {|itm| itm.split(':')[1] + ' ' + itm.split(':')[0]}
+    end
+
+    def privileged?
+      inspection['HostConfig']['Privileged']
+    end
+
+    # TODO: matcher for this
+    def publishes_all_ports?
+      inspection['HostConfig']['PublishAllPorts']
+    end
+
+
+    def map_port?(host, container, protocol = 'tcp')
+      inspection['NetworkSettings']['Ports']["#{container}/#{protocol}"][0]['HostPort'] == host
+    end
+
+    def port_map
+      inspection['HostConfig']['PortBindings']
+    end
+
+    def has_mount?(source, target, type)
+      mounts.find { |mount| mount['Source'] == source && mount['Destination'] == target && mount['Type'] == type }
+    end
+
+    def mounts
+      inspection['Mounts']
+    end
+
+    # TODO: matcher for volumes
+
+    private
+
+    def get_inspection
+      @containers ||=  @name.include?('=') ? @runner.run_command("docker ps -qa -f #{@name}").stdout : @name
+      @get_inspection ||= @runner.run_command("docker inspect #{@containers}")
+    end
+  end
+end

data/lib/serverspec_extra_types/types/docker_node.rb

@@ -0,0 +1,41 @@
+module Serverspec::Type
+  class DockerNode < DockerBase
+
+    def active?
+      availability == 'active'
+    end
+
+    def availability
+      inspection['Spec']['Availability']
+    end
+
+    def role
+      inspection['Spec']['Role']
+    end
+
+    def manager?
+      role == 'manager'
+    end
+
+    def worker?
+      role == 'worker'
+    end
+
+    def labels
+      inspection['Spec']['Labels']
+    end
+
+    def has_engine_version?(version)
+      engine_version == version
+    end
+
+    def engine_version
+      inspection['Description']['Engine']['EngineVersion']
+    end
+    private
+
+    def get_inspection
+      @get_inspection ||= @runner.run_command("docker service inspect #{@name}")
+    end
+  end
+end

data/lib/serverspec_extra_types/types/docker_service.rb

@@ -0,0 +1,121 @@
+require 'serverspec'
+require 'serverspec/type/base'
+require 'serverspec/type/docker_base'
+
+module Serverspec::Type
+  class DockerService < DockerBase
+    def exist?
+      get_inspection.success?
+    end
+
+    def has_restart_policy?(policy)
+      restart_policy == policy
+    end
+
+    def has_restart_limit?(limit)
+      restart_limit == limit
+    end
+
+    def restart_policy
+      inspection['Spec']['TaskTemplate']['RestartPolicy']['Condition']
+    end
+
+    def restart_limit
+      inspection['Spec']['TaskTemplate']['RestartPolicy']['MaxAttempts']
+    end
+
+    def has_image?(image)
+      self.image == image
+    end
+
+    def has_image_sha?(image)
+      self.image == image
+    end
+
+    def image
+      inspection['Spec']['TaskTemplate']['ContainerSpec']['Image'].split('@')[0]
+    end
+
+    def image_sha
+      inspection['Spec']['TaskTemplate']['ContainerSpec']['Image'].split('@')[1]
+    end
+
+    def has_user?(user)
+      self.user == user
+    end
+
+    def user
+      inspection['Spec']['TaskTemplate']['ContainerSpec']['User']
+    end
+
+    def has_mount?(source, target, type)
+      mounts.find do |mount|
+        mount['Source'] == source && mount['Target'] == target && mount['Type'] == type
+      end
+    end
+
+    def mounts
+      inspection['Spec']['TaskTemplate']['ContainerSpec']['Mounts']
+    end
+
+    def has_host?(host)
+      hosts.include? host
+    end
+
+    def hosts
+      inspection['Spec']['TaskTemplate']['ContainerSpec']['Hosts']
+    end
+
+    def has_placement_constraint?(constraint)
+      placement_constraints.include? constraint
+    end
+
+    def placement_constraints
+      inspection['Spec']['TaskTemplate']['Placement']['Constraints']
+    end
+
+    def replicated?
+      inspection['Spec']['Mode']['Replicated']
+    end
+
+    def replicas
+      ['Spec']['Mode']['Replicated']['Replicas']
+    end
+    def global?
+      inspection['Spec']['Mode']['Global']
+    end
+
+    def replicas
+      inspection['Spec']['Mode']['Replicated']['Replicas']
+    end
+
+    def networks
+      inspection['Spec']['TaskTemplate']['Networks']
+    end
+
+    def has_network?(name)
+      networks.find { |network| network['Aliases'].include? name }
+    end
+
+    def port_map
+      inspection['Spec']['EndpointSpec']['Ports']
+    end
+
+    def map_port?(published, target, protocol = 'tcp',mode = 'ingress')
+      port_map.find do |port|
+        port['PublishedPort'] == published.to_i &&
+          port['TargetPort'] == target.to_i && port['PublishMode'] == mode &&
+          port['Protocol'] == protocol
+      end
+    end
+
+
+    private
+
+    def get_inspection
+      @get_inspection ||= @runner.run_command("docker service inspect #{@name}")
+    end
+  end
+end
+
+include Serverspec::Type

data/lib/serverspec_extra_types/types/rabbitmq_base.rb

@@ -0,0 +1,38 @@
+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 RabbitmqBase < ApiBase
+    def initialize(name = nil, options = {})
+      super(name, options)
+      @user = ENV['RABBITMQ_USER'] || 'guest'
+      @password = ENV['RABBITMQ_PASSWORD'] || 'guest'
+      @url_base = property[:variables][:rabbitmq_url] || 'http://localhost:15672'
+    end
+
+
+    def inspection
+      @inspection ||= ::MultiJson.load(get_inspection.stdout)[0]
+    end
+
+    def length
+      if inspection.is_a? String
+        inspection.length
+      elsif inspection.kind_of? Array
+        inspection.length
+      else
+        1
+      end
+    end
+
+    private
+
+    def get_inspection
+      command = "curl -s -u #{@user}:#{@password} #{url}"
+      @get_inspection ||= @runner.run_command(command)
+    end
+  end
+end

data/lib/serverspec_extra_types/types/rabbitmq_node_list.rb

@@ -0,0 +1,18 @@
+require 'serverspec_extra_types/types/rabbitmq_base'
+
+module Serverspec::Type
+  class RabbitmqNodeList   < RabbitmqBase
+
+    def url
+      "#{@url_base}/api/vhosts"
+    end
+
+
+
+    def inspection
+      @inspection ||= ::MultiJson.load(get_inspection.stdout)
+    end
+
+
+  end
+end

data/lib/serverspec_extra_types/types/rabbitmq_user_permission.rb

@@ -0,0 +1,39 @@
+require 'serverspec_extra_types/types/rabbitmq_base'
+
+module Serverspec::Type
+  class RabbitmqUserPermission < RabbitmqBase
+
+    def url
+      "#{@url_base}/api/users/#{name}/permissions"
+    end
+
+    def read_from_queue?(vhost, queue)
+      queue.match read_permissions(vhost)
+    end
+
+    def write_to_queue?(vhost, queue)
+      queue.match write_permissions(vhost)
+    end
+
+    def configure_queue?(vhost, queue)
+      queue.match configure_permissions(vhost)
+    end
+
+    def read_permissions(vhost)
+      inspection.find { |item| item['vhost'] == vhost}['read']
+    end
+
+    def write_permissions(vhost)
+      inspection.find { |item| item['vhost'] == vhost}['write']
+    end
+
+    def configure_permissions(vhost)
+      inspection.find { |item| item['vhost'] == vhost}['configure']
+    end
+
+    def inspection
+      @inspection ||= ::MultiJson.load(get_inspection.stdout)
+    end
+
+  end
+end