rspec-puppet 0.0.1

data/lib/rspec-puppet.rb

@@ -0,0 +1,3 @@
+require 'puppet'
+require 'rspec-puppet/matchers'
+require 'rspec-puppet/example'

data/lib/rspec-puppet/example.rb

@@ -0,0 +1,16 @@
+require 'rspec-puppet/example/define_example_group'
+require 'rspec-puppet/example/class_example_group'
+
+RSpec::configure do |c|
+  def c.escaped_path(*parts)
+    Regexp.compile(parts.join('[\\\/]'))
+  end
+
+  c.include RSpec::Puppet::DefineExampleGroup, :type => :define, :example_group => {
+    :file_path => c.escaped_path(%w[spec define])
+  }
+
+  c.include RSpec::Puppet::ClassExampleGroup, :type => :class, :example_group => {
+    :file_path => c.escaped_path(%w[spec class])
+  }
+end

data/lib/rspec-puppet/example/class_example_group.rb

@@ -0,0 +1,36 @@
+module RSpec::Puppet
+  module ClassExampleGroup
+    include RSpec::Puppet::Matchers
+
+    def subject
+      @catalogue ||= catalogue
+    end
+
+    def catalogue
+      Puppet[:modulepath] = module_path
+
+      klass_name = self.class.top_level_description.downcase
+      if !self.respond_to?(:params) || params == {}
+        Puppet[:code] = "include #{klass_name}"
+      else
+        Puppet[:code] = 'class' + " { " + klass_name + ": " + params.keys.map { |r| "#{r.to_s} => '#{params[r].to_s}'"
+      }.join(', ') + " }"
+      end
+
+      nodename = self.respond_to?(:node) ? node : Puppet[:certname]
+      facts_val = self.respond_to?(:facts) ? facts : {}
+
+      node_obj = Puppet::Node.new(nodename)
+
+      node_obj.merge(facts_val)
+
+      # trying to be compatible with 2.7 as well as 2.6
+      if Puppet::Resource::Catalog.respond_to? :find
+        Puppet::Resource::Catalog.find(node_obj.name, :use_node => node_obj)
+      else
+        require 'puppet/face'
+        Puppet::Face[:catalog, :current].find(node_obj.name, :use_node => node_obj)
+      end
+    end
+  end
+end

data/lib/rspec-puppet/example/define_example_group.rb

@@ -0,0 +1,40 @@
+module RSpec::Puppet
+  module DefineExampleGroup
+    include RSpec::Puppet::Matchers
+
+    def subject
+      @catalogue ||= catalogue
+    end
+
+    def catalogue
+      define_name = self.class.top_level_description.downcase
+
+      Puppet[:modulepath] = module_path
+
+      if self.respond_to? :params
+        param_str = params.keys.map { |r|
+          "#{r.to_s} => \"#{params[r].to_s}\""
+        }.join(', ')
+      else
+        param_str = ""
+      end
+
+      Puppet[:code] = define_name + " { \"" + title + "\": " + param_str + " }"
+
+      nodename = self.respond_to?(:node) ? node : Puppet[:certname]
+      facts_val = self.respond_to?(:facts) ? facts : {}
+
+      node_obj = Puppet::Node.new(nodename)
+
+      node_obj.merge(facts_val)
+
+      # trying to be compatible with 2.7 as well as 2.6
+      if Puppet::Resource::Catalog.respond_to? :find
+        Puppet::Resource::Catalog.find(node_obj.name, :use_node => node_obj)
+      else
+        require 'puppet/face'
+        Puppet::Face[:catalog, :current].find(node_obj.name, :use_node => node)
+      end
+    end
+  end
+end

data/lib/rspec-puppet/matchers.rb

@@ -0,0 +1,3 @@
+require 'rspec-puppet/matchers/create_generic'
+require 'rspec-puppet/matchers/create_resource'
+require 'rspec-puppet/matchers/include_class'

data/lib/rspec-puppet/matchers/create_generic.rb

@@ -0,0 +1,74 @@
+module RSpec::Puppet
+  module Matchers
+    class CreateGeneric
+      def initialize(*args, &block)
+        @exp_resource_type = args.shift.to_s.gsub(/^create_/, '')
+        @args = args
+        @block = block
+        @referenced_type = referenced_type(@exp_resource_type)
+        @title = args[0]
+      end
+
+      def method_missing(method, *args, &block)
+        if method.to_s =~ /^with_/
+          param = method.to_s.gsub(/^with_/, '')
+          (@expected_params ||= []) << [param, args[0]]
+          self
+        else
+          super
+        end
+      end
+
+      def matches?(catalogue)
+        ret = true
+        resources = catalogue.resources.select { |r|
+          r.type == @referenced_type
+        }.select { |r|
+          r.title == @title if r.respond_to? :title
+        }
+
+        unless resources.length == 1
+          ret = false
+        else
+          if @expected_params
+            @expected_params.each do |name, value|
+              unless resources.first.send(:parameters)[name.to_sym].to_s == value.to_s
+                ret = false
+                (@errors ||= []) << "#{name.to_s} set to `#{value.inspect}`"
+              end
+            end
+          end
+        end
+
+        ret
+      end
+
+      def failure_message_for_should
+        "expected that the catalogue would contain #{@referenced_type}[\"#{@title}\"]#{errors}"
+      end
+
+      def failure_message_for_should_not
+        "expected that the catalogue would not contain #{@referenced_type}[\"#{@title}\"]#{errors}"
+      end
+
+      def description
+        "create #{@referenced_type}[\"#{@title}\"]"
+      end
+
+    private
+
+      def referenced_type(type)
+        type.split('-').map { |r| r.capitalize }.join('::')
+      end
+
+      def errors
+        @errors.nil? ? "" : " with #{@errors.join(', ')}"
+      end
+    end
+
+    def method_missing(method, *args, &block)
+      return RSpec::Puppet::Matchers::CreateGeneric.new(method, *args, &block) if method.to_s =~ /^create_/
+      super
+    end
+  end
+end

data/lib/rspec-puppet/matchers/create_resource.rb

@@ -0,0 +1,53 @@
+module RSpec::Puppet
+  module Matchers
+    extend RSpec::Matchers::DSL
+
+    matcher :create_resource do |expected_type, expected_title|
+      match do |catalogue|
+        ret = true
+        resources = catalogue.resources.select { |r|
+          r.type == referenced_type(expected_type)
+        }.select { |r|
+          r.title == expected_title if r.respond_to? :title
+        }
+
+        unless resources.length == 1
+          ret = false
+        end
+
+        if @expected_params
+          @expected_params.each do |name, value|
+            unless resources.first.send(:parameters)[name.to_sym] == value
+              ret = false
+              (@errors ||= []) << "the parameter #{name.to_s} set to `#{value}`"
+            end
+          end
+        end
+
+        ret
+      end
+
+      def errors
+        @errors.nil? ? "" : " with #{@errors.join(', ')}"
+      end
+
+      def referenced_type(type)
+        type.split('::').map { |r| r.capitalize }.join('::')
+      end
+
+      chain :with_param do |param_name,param_value|
+        (@expected_params ||= []) << [param_name, param_value]
+      end
+
+      description do
+        type = referenced_type(expected_type)
+        "create #{type}['#{expected_title}']"
+      end
+
+      failure_message_for_should do |actual|
+        type = referenced_type(expected_type)
+        "expected that the catalogue would contain #{type}['#{expected_title}']#{errors}"
+      end
+    end
+  end
+end

data/lib/rspec-puppet/matchers/include_class.rb

@@ -0,0 +1,19 @@
+module RSpec::Puppet
+  module Matchers
+    extend RSpec::Matchers::DSL
+
+    matcher :include_class do |expected_class|
+      match do |catalogue|
+        catalogue.classes.include? expected_class
+      end
+
+      description do
+        "include Class['#{expected_class}']"
+      end
+
+      failure_message_for_should do |actual|
+        "expected that the catalogue would include Class['#{expected_class}']"
+      end
+    end
+  end
+end

data/rspec-puppet.gemspec

@@ -0,0 +1,24 @@
+Gem::Specification.new do |s|
+  s.name = 'rspec-puppet'
+  s.version = '0.0.1'
+  s.homepage = 'https://github.com/rodjek/rspec-puppet/'
+  s.summary = 'RSpec tests for your Puppet manifests'
+  s.description = 'RSpec tests for your Puppet manifests'
+
+  s.files = [
+    'rspec-puppet.gemspec',
+    'lib/rspec-puppet.rb',
+    'lib/rspec-puppet/matchers.rb',
+    'lib/rspec-puppet/matchers/create_generic.rb',
+    'lib/rspec-puppet/matchers/create_resource.rb',
+    'lib/rspec-puppet/matchers/include_class.rb',
+    'lib/rspec-puppet/example.rb',
+    'lib/rspec-puppet/example/define_example_group.rb',
+    'lib/rspec-puppet/example/class_example_group.rb',
+  ]
+
+  s.add_dependency 'rspec'
+
+  s.authors = ['Tim Sharpe']
+  s.email = 'tim@sharpe.id.au'
+end

metadata

@@ -0,0 +1,88 @@
+--- !ruby/object:Gem::Specification 
+name: rspec-puppet
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Tim Sharpe
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-07-19 00:00:00 +10:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+description: RSpec tests for your Puppet manifests
+email: tim@sharpe.id.au
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- rspec-puppet.gemspec
+- lib/rspec-puppet.rb
+- lib/rspec-puppet/matchers.rb
+- lib/rspec-puppet/matchers/create_generic.rb
+- lib/rspec-puppet/matchers/create_resource.rb
+- lib/rspec-puppet/matchers/include_class.rb
+- lib/rspec-puppet/example.rb
+- lib/rspec-puppet/example/define_example_group.rb
+- lib/rspec-puppet/example/class_example_group.rb
+has_rdoc: true
+homepage: https://github.com/rodjek/rspec-puppet/
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.10
+signing_key: 
+specification_version: 3
+summary: RSpec tests for your Puppet manifests
+test_files: []
+