File Manager

Viewing File: chef_run_support.rb

require "chef_zero/rspec"
require "chef/server_api"
require_relative "repository_support"
require "uri"
require_relative "../chef_run"
require_relative "recipe_run_wrapper"
require_relative "matchers"

module Cheffish
  module RSpec
    module ChefRunSupport
      include ChefZero::RSpec
      include RepositorySupport

      def self.extended(klass)
        klass.class_eval do
          include ChefRunSupportInstanceMethods
        end
      end

      def when_the_chef_12_server(*args, **options, &block)
        if Gem::Version.new(ChefZero::VERSION) >= Gem::Version.new("3.1")
          when_the_chef_server(*args, osc_compat: false, single_org: false, **options, &block)
        end
      end

      def with_converge(&recipe)
        before :each do
          r = recipe(&recipe)
          r.converge
        end
      end

      module ChefRunSupportInstanceMethods
        def rest
          ::Chef::ServerAPI.new(Chef::Config.chef_server_url, api_version: "0")
        end

        def get(path, *args)
          if path[0] == "/"
            path = URI.join(rest.url, path)
          end
          rest.get(path, *args)
        end

        def chef_config
          {}
        end

        def expect_recipe(str = nil, file = nil, line = nil, &recipe)
          r = recipe(str, file, line, &recipe)
          r.converge
          expect(r)
        end

        def expect_converge(str = nil, file = nil, line = nil, &recipe)
          expect { converge(str, file, line, &recipe) }
        end

        def recipe(str = nil, file = nil, line = nil, &recipe)
          unless recipe
            if file && line
              recipe = proc { eval(str, nil, file, line) } # rubocop:disable Security/Eval
            else
              recipe = proc { eval(str) } # rubocop:disable Security/Eval
            end
          end
          RecipeRunWrapper.new(chef_config, &recipe)
        end

        def converge(str = nil, file = nil, line = nil, &recipe)
          r = recipe(str, file, line, &recipe)
          r.converge
          r
        end

        def chef_client
          @chef_client ||= ChefRun.new(chef_config)
        end
      end
    end
  end
end