puppet-repl 0.0.7 → 0.0.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: efa15e795af4973b078852664154fe9e60d9b908
-  data.tar.gz: a59f074150a899b9e67c59ca9b6874651541b7c4
+  metadata.gz: b8b2d91eeef5f922cd54569176f6085199647745
+  data.tar.gz: 54320984760c3e54982a9bafda14b5c7742049c5
 SHA512:
-  metadata.gz: 174bc1f05d44735c39e55417eb05cd54bb19a6a666572afc46bc025868c3a26f3624ba4d5d691d15e7b4360ac743e0cfb24e53bcf8f77548ea9dd16e0ce4234f
-  data.tar.gz: d3b20b61ea361187b5cf646b5336a3f58207f54ed40c39b3dde5582dddeefca7e87a925f0e04f99b2046e6c5aea50e056dedb23ed5545f43968092e40e52befc
+  metadata.gz: 6f289d12177afa7d7888699f6b1e8afa65b0ada85bc1969019365af520289404969e64f8654e7ed019a7161c8d1b7278b2ef9f863740862f548e226cd2a9e267
+  data.tar.gz: 1e7434a150bb7543d64f19d26c538d67527e47856a505e43dd3795812627732e6f2a9da9b30cf6948a4f1ac7500ebf12228bca6c5cab2aa2e7fb2b104cc65f8f

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+## 0.0.8
+  * adds ability to list currently loaded classes
+  * adds formatted output using awesome print 
+  * adds verbose type output when creating resources
 ## 0.0.7
   * Added ability to list scope variables
 ## 0.0.6

data/Gemfile.lock

@@ -27,7 +27,7 @@ GEM
       multi_json (>= 1.7.5, < 2.0)
       oauth2
     hashie (3.4.3)
-    hiera (3.0.6)
+    hiera (1.3.4)
       json_pure
     highline (1.7.8)
     jeweler (2.0.1)
@@ -61,9 +61,9 @@ GEM
       coderay (~> 1.1.0)
       method_source (~> 0.8.1)
       slop (~> 3.4)
-    puppet (4.3.2)
-      facter (> 2.0, < 4)
-      hiera (>= 2.0, < 4)
+    puppet (3.8.5)
+      facter (> 1.6, < 3)
+      hiera (~> 1.0)
       json_pure
     rack (1.6.4)
     rake (10.5.0)

data/README.md

@@ -8,6 +8,7 @@
   - [Load path](#load-path)
   - [Usage](#usage)
   - [Using Variables](#using-variables)
+    - [Listing variables](#listing-variables)
   - [Using functions](#using-functions)
   - [Duplicate resource error](#duplicate-resource-error)
   - [Setting the puppet log level](#setting-the-puppet-log-level)
@@ -46,17 +47,18 @@ and try to enforce the catalog. This has a few side affects.
 Example Usage
 ```
 MacBook-Pro-2/tmp % prepl
-Puppet Version: 4.2.2
-Puppet Repl Version: 0.0.1
+Ruby Version: 2.0.0
+Puppet Version: 3.8.5
+Puppet Repl Version: 0.0.7
 Created by: NWOps <corey@nwops.io>
-Type "exit", "functions", "types", "reset", "help" for more information.
+Type "exit", "functions", "vars", "krt", "facts", "reset", "help" for more information.
 
->> file{'/tmp/test2': ensure => present}
- => File['/tmp/test2']
 >> ['/tmp/test3', '/tmp/test4'].each |String $path| { file{$path: ensure => present} }
- =>
-/tmp/test3
-/tmp/test4
+  => [
+     [0] "/tmp/test3",
+     [1] "/tmp/test4"
+ ]
+ >>
 
 ```
 
@@ -64,16 +66,37 @@ Type "exit", "functions", "types", "reset", "help" for more information.
 
 ```
 MacBook-Pro-2/tmp % prepl
-Puppet Version: 4.2.2
-Puppet Repl Version: 0.0.1
+Ruby Version: 2.0.0
+Puppet Version: 3.8.5
+Puppet Repl Version: 0.0.7
 Created by: NWOps <corey@nwops.io>
-Type "exit", "functions", "types", "reset", "help" for more information.
+Type "exit", "functions", "vars", "krt", "facts", "reset", "help" for more information.
 
->> $config_file = '/etc/httpd/httpd.conf'
- => /etc/httpd/httpd.conf
->> file{$config_file: ensure => present, content => 'hello'}
- => File['/etc/httpd/httpd.conf']
 >>
+
+>> $config_file = '/etc/httpd/httpd.conf'
+ => "/etc/httpd/httpd.conf"
+ >> file{$config_file: ensure => present, content => 'hello'}
+  => Puppet::Type::File {
+                        path => "/etc/httpd/httpd.conf",
+                    provider => posix,
+                      ensure => present,
+                     content => "{md5}5d41402abc4b2a76b9719d911017c592",
+                    checksum => nil,
+                      backup => "puppet",
+                     replace => true,
+                       links => manage,
+                       purge => false,
+                sourceselect => first,
+                   show_diff => true,
+        validate_replacement => "%",
+          source_permissions => use,
+     selinux_ignore_defaults => false,
+                    loglevel => notice,
+                        name => "/etc/httpd/httpd.conf",
+                       title => "/etc/httpd/httpd.conf"
+ }
+ >>
 ```
 ### Listing variables
 To see the current variables in the scope use the  `vars` keyword.

data/VERSION

@@ -1 +1 @@
-0.0.7
+0.0.8

data/lib/puppet-repl/cli.rb

@@ -2,39 +2,70 @@ require 'puppet'
 require 'readline'
 require 'json'
 require_relative 'support'
+
 module PuppetRepl
   class Cli
     include PuppetRepl::Support
 
+    attr_accessor :settings, :log_level
+
+    def initialize
+      @log_level = 'notice'
+    end
+
+    def ap_formatter
+      unless @ap_formatter
+        inspector = AwesomePrint::Inspector.new({:sort_keys => true, :indent => 2})
+        @ap_formatter = AwesomePrint::Formatter.new(inspector)
+      end
+      @ap_formatter
+    end
+
     def puppet_eval(input)
-      begin
-        parser.evaluate_string(scope, input)
-      rescue ArgumentError => e
-        e.message
-      rescue Puppet::ParseErrorWithIssue => e
-        e.message
-      rescue Exception => e
-        e.message
+      parser.evaluate_string(scope, input)
+    end
+
+    def to_resource_declaration(type)
+      res = scope.catalog.resource(type.type_name, type.title)
+      res.to_ral
+    end
+
+    # ruturns a formatted array
+    def expand_resource_type(types)
+      output = [types].flatten.map do |t|
+        if t.class.to_s == 'Puppet::Pops::Types::PResourceType'
+          to_resource_declaration(t)
+        else
+          t
+        end
       end
+      output
     end
 
     def normalize_output(result)
       if result.instance_of?(Array)
-        if result.count == 1
-          return result.first
+        output = expand_resource_type(result)
+        if output.count == 1
+          return output.first
         end
+        return output
       end
       result
     end
 
+    def set_log_level(level)
+      Puppet::Util::Log.level = level.to_sym
+      Puppet::Util::Log.newdestination(:console)
+    end
+
     def handle_set(input)
       args = input.split(' ')
       args.shift # throw away the set
       case args.shift
       when /loglevel/
         if level = args.shift
-          Puppet::Util::Log.level = level.to_sym
-          Puppet::Util::Log.newdestination(:console)
+          @log_level = level
+          set_log_level(level)
           puts "loglevel #{Puppet::Util::Log.level} is set"
         end
       end
@@ -60,16 +91,40 @@ module PuppetRepl
         ap(vars, {:sort_keys => true, :indent => -1})
       when 'environment'
         puts "Puppet Environment: #{puppet_env_name}"
+      when 'vars'
+        ap(scope.to_hash, {:sort_keys => true, :indent => 0})
       when 'exit'
         exit 0
       when 'reset'
         @scope = nil
-      when 'current_resources'
-        compiler.known_resource_types
+        # initilize scope again
+        scope
+        set_log_level(log_level)
+      when 'krt'
+        ap(known_resource_types, {:sort_keys => true, :indent => -1})
       else
-        result = puppet_eval(input)
-        @last_item = result
-        puts(" => #{normalize_output(result)}")
+        begin
+          result = puppet_eval(input)
+          @last_item = result
+          print " => "
+          output = normalize_output(result)
+          if output.nil?
+            puts ""
+          else
+            ap(output)
+          end
+        rescue ArgumentError => e
+          print " => "
+          puts e.message.fatal
+        rescue Puppet::ResourceError => e
+          print " => "
+          puts e.message.fatal
+        rescue Puppet::ParseErrorWithIssue => e
+          print " => "
+          puts e.message.fatal
+        rescue Exception => e
+          puts e.message.fatal
+        end
       end
     end
 
@@ -79,7 +134,7 @@ Ruby Version: #{RUBY_VERSION}
 Puppet Version: #{Puppet.version}
 Puppet Repl Version: #{PuppetRepl::VERSION}
 Created by: NWOps <corey@nwops.io>
-Type "exit", "functions", "vars", "facts", "reset", "help" for more information.
+Type "exit", "functions", "vars", "krt", "facts", "reset", "help" for more information.
 
       EOT
     end

data/lib/puppet-repl/support/compiler.rb

@@ -0,0 +1,11 @@
+require 'tempfile'
+
+module PuppetRepl
+  module Support
+    module Compilier
+      def create_compiler(node)
+        Puppet::Parser::Compiler.new(node)
+      end
+    end
+  end
+end

data/lib/puppet-repl/support/environment.rb

@@ -0,0 +1,30 @@
+module PuppetRepl
+  module Support
+    module Environment
+      # creates a puppet environment given a module path and environment name
+      # this is cached
+      def puppet_environment
+        unless @puppet_environment
+          do_initialize
+          @puppet_environment = Puppet::Node::Environment.create(
+          puppet_env_name,
+          module_dirs,
+          manifests_dir
+          )
+        end
+        @puppet_environment
+      end
+
+      # the cached name of the environment
+      def puppet_env_name
+        @penv ||= ENV['PUPPET_ENV'] || Puppet[:environment]
+      end
+
+      # currently this is not being used
+      def environment_loaders
+        name = compiler.loaders.public_environment_loader.loader_name
+      end
+
+    end
+  end
+end

data/lib/puppet-repl/support/facts.rb

@@ -0,0 +1,19 @@
+module PuppetRepl
+  module Support
+    module Facts
+      # in the future we will want to grab real facts from real systems via puppetdb
+      # or enc data
+      def facterdb_filter
+        'operatingsystem=RedHat and operatingsystemrelease=/^7/ and architecture=x86_64 and facterversion=/^2.4\./'
+      end
+
+      # uses facterdb (cached facts) and retrives the facts given a filter
+      def facts
+        unless @facts
+          @facts ||= FacterDB.get_facts(facterdb_filter).first
+        end
+        @facts
+      end
+    end
+  end
+end

data/lib/puppet-repl/support/functions.rb

@@ -0,0 +1,41 @@
+module PuppetRepl
+  module Support
+    module Functions
+      # returns a array of function files
+      def function_files
+        search_dirs = lib_dirs.map do |lib_dir|
+          [File.join(lib_dir, 'puppet', 'functions', '**', '*.rb'),
+            File.join(lib_dir, 'functions', '**', '*.rb'),
+           File.join(lib_dir, 'puppet', 'parser', 'functions', '*.rb')
+           ]
+        end
+        # add puppet lib directories
+        search_dirs << [File.join(puppet_lib_dir, 'puppet', 'functions', '**', '*.rb'),
+          File.join(puppet_lib_dir, 'puppet', 'parser', 'functions', '*.rb')
+         ]
+        Dir.glob(search_dirs.flatten)
+      end
+
+      # returns a map of functions
+      def function_map
+        unless @functions
+          do_initialize
+          @functions = {}
+          function_files.each do |file|
+            obj = {}
+            name = File.basename(file, '.rb')
+            obj[:name] = name
+            obj[:parent] = mod_finder.match(file)[1]
+            @functions["#{obj[:parent]}::#{name}"] = obj
+          end
+        end
+        @functions
+      end
+    
+      # def functions
+      #   @functions = []
+      #   @functions << compiler.loaders.static_loader.loaded.keys.find_all {|l| l.type == :function}
+      # end
+    end
+  end
+end

data/lib/puppet-repl/support/node.rb

@@ -0,0 +1,15 @@
+module PuppetRepl
+  module Support
+    module Node
+      # creates a node object
+      def create_node
+        options = {}
+        options[:parameters] = facts
+        options[:facts] = facts
+        options[:classes] = []
+        options[:environment] = puppet_environment
+        Puppet::Node.new(facts[:fqdn], options)
+      end
+    end
+  end
+end

data/lib/puppet-repl/support/scope.rb

@@ -0,0 +1,23 @@
+module PuppetRepl
+  module Support
+    module Scope
+      def create_scope(node)
+        @compiler = create_compiler(node) # creates a new compiler for each scope
+        scope = Puppet::Parser::Scope.new(@compiler)
+        scope.source = Puppet::Resource::Type.new(:node, node.name)
+        scope.parent = @compiler.topscope
+        load_lib_dirs
+        # compiling will load all the facts into the scope
+        # without this step facts will not get resolved
+        scope.compiler.compile # this will load everything into the scope
+        scope
+      end
+
+      # returns a hash of varaibles that are currently in scope
+      def scope_vars
+        vars = scope.to_hash.delete_if {| key, value | node.facts.key?(key.to_sym) }
+        vars['facts'] = 'removed by the puppet-repl'
+      end
+    end
+  end
+end

data/lib/puppet-repl/support.rb

@@ -1,17 +1,47 @@
 require 'puppet/pops'
 require 'facterdb'
+# load all the generators found in the generators directory
+Dir.glob(File.join(File.dirname(__FILE__),'support', '*.rb')).each do |file|
+  require_relative File.join('support', File.basename(file, '.rb'))
+end
 
 module PuppetRepl
   module Support
+    include PuppetRepl::Support::Compilier
+    include PuppetRepl::Support::Environment
+    include PuppetRepl::Support::Facts
+    include PuppetRepl::Support::Scope
+    include PuppetRepl::Support::Functions
+    include PuppetRepl::Support::Node
 
     # returns an array of module directories
     def module_dirs
       dirs = []
+      do_initialize if Puppet[:codedir].nil?
       dirs << File.join(Puppet[:environmentpath],puppet_env_name,'modules') unless Puppet[:environmentpath].empty?
       dirs << Puppet.settings[:basemodulepath].split(':')
       dirs.flatten
     end
 
+    def known_resource_types
+      res = {
+        :hostclasses => scope.known_resource_types.hostclasses.keys,
+        :definitions => scope.known_resource_types.definitions.keys,
+        :nodes => scope.known_resource_types.nodes.keys,
+      }
+      if sites = scope.known_resource_types.instance_variable_get(:@sites)
+        res.merge!(:sites => scope.known_resource_types.instance_variable_get(:@sites).first)
+      end
+      if scope.known_resource_types.respond_to?(:applications)
+        res.merge!(:applications => scope.known_resource_types.applications.keys)
+      end
+      # some versions of puppet do not support capabilities
+      if scope.known_resource_types.respond_to?(:capability_mappings)
+        res.merge!(:capability_mappings => scope.known_resource_types.capability_mappings.keys)
+      end
+      res
+    end
+
     # this is required in order to load things only when we need them
     def do_initialize
       begin
@@ -26,42 +56,11 @@ module PuppetRepl
       @puppet_lib_dir ||= File.dirname(Puppet.method(:[]).source_location.first)
     end
 
-    # returns a array of function files
-    def function_files
-      search_dirs = lib_dirs.map do |lib_dir|
-        [File.join(lib_dir, 'puppet', 'functions', '**', '*.rb'),
-          File.join(lib_dir, 'functions', '**', '*.rb'),
-         File.join(lib_dir, 'puppet', 'parser', 'functions', '*.rb')
-         ]
-      end
-      # add puppet lib directories
-      search_dirs << [File.join(puppet_lib_dir, 'puppet', 'functions', '**', '*.rb'),
-        File.join(puppet_lib_dir, 'puppet', 'parser', 'functions', '*.rb')
-       ]
-      Dir.glob(search_dirs.flatten)
-    end
-
     # returns either the module name or puppet version
     def mod_finder
       @mod_finder ||= Regexp.new('\/([\w\-\.]+)\/lib')
     end
 
-    # returns a map of functions
-    def function_map
-      unless @functions
-        do_initialize
-        @functions = {}
-        function_files.each do |file|
-          obj = {}
-          name = File.basename(file, '.rb')
-          obj[:name] = name
-          obj[:parent] = mod_finder.match(file)[1]
-          @functions["#{obj[:parent]}::#{name}"] = obj
-        end
-      end
-      @functions
-    end
-
     def lib_dirs
       module_dirs.map do |mod_dir|
         Dir["#{mod_dir}/*/lib"].entries
@@ -79,82 +78,20 @@ module PuppetRepl
       @parser || ::Puppet::Pops::Parser::EvaluatingParser.new
     end
 
-    # the cached name of the environment
-    def puppet_env_name
-      @penv ||= ENV['PUPPET_ENV'] || Puppet[:environment]
-    end
-
-    # creates a puppet environment given a module path and environment name
-    # this is cached
-    def puppet_environment
-      unless @puppet_environment
-        do_initialize
-        @puppet_environment = Puppet::Node::Environment.create(
-          puppet_env_name,
-          module_dirs,
-          manifests_dir
-          )
-      end
-      @puppet_environment
-    end
-
-    # def functions
-    #   @functions = []
-    #   @functions << compiler.loaders.static_loader.loaded.keys.find_all {|l| l.type == :function}
-    # end
-
-    def environment_loaders
-      name = compiler.loaders.public_environment_loader.loader_name
-    end
-
     def compiler
       @compiler
     end
 
+    # @return [node] puppet node object
     def node
       @node ||= create_node
     end
 
-    def create_scope
-      @compiler = create_compiler(node) # creates a new compiler for each scope
-      scope = Puppet::Parser::Scope.new(compiler)
-      scope.source = Puppet::Resource::Type.new(:node, node.name)
-      scope.parent = compiler.topscope
-      load_lib_dirs
-      compiler.compile # this will load everything into the scope
-      scope
-    end
-
-    def create_compiler(node)
-      Puppet::Parser::Compiler.new(node)
-    end
-
-    def facterdb_filter
-      'operatingsystem=RedHat and operatingsystemrelease=/^7/ and architecture=x86_64 and facterversion=/^2.4\./'
-    end
-
-    # uses facterdb (cached facts) and retrives the facts given a filter
-    def facts
-      unless @facts
-        @facts ||= FacterDB.get_facts(facterdb_filter).first
-      end
-      @facts
-    end
-
-    # creates a node object
-    def create_node
-      options = {}
-      options[:parameters] = facts
-      options[:facts] = facts
-      options[:classes] = []
-      options[:environment] = puppet_environment
-      Puppet::Node.new(facts[:fqdn], options)
-    end
-
+    # @return [Scope] puppet scope object
     def scope
       unless @scope
         do_initialize
-        @scope ||= create_scope
+        @scope ||= create_scope(node)
       end
       @scope
     end
@@ -165,4 +102,3 @@ module PuppetRepl
 
   end
 end
-#scope.environment.known_resource_types

data/lib/puppet-repl.rb

@@ -2,3 +2,19 @@ require_relative 'puppet-repl/cli'
 require_relative 'version'
 require 'awesome_print'
 require_relative 'awesome_print/ext/awesome_puppet'
+
+# monkey patch in some color effects string methods
+class String
+  def red;            "\033[31m#{self}\033[0m" end
+  def green;          "\033[32m#{self}\033[0m" end
+  def cyan;           "\033[36m#{self}\033[0m" end
+  def yellow;         "\033[33m#{self}\033[0m" end
+  def warning;        yellow                   end
+  def fatal;          red                      end
+  def info;           green                    end
+
+  def camel_case
+    return self if self !~ /_/ && self =~ /[A-Z]+.*/
+    split('_').map(&:capitalize).join
+  end
+end

data/lib/version.rb

@@ -1,3 +1,3 @@
 module PuppetRepl
-  VERSION = '0.0.7'
+  VERSION = '0.0.8'
 end

data/puppet-repl.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = "puppet-repl"
-  s.version = "0.0.7"
+  s.version = "0.0.8"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Corey Osman"]
-  s.date = "2016-04-02"
+  s.date = "2016-04-03"
   s.description = "A interactive command line tool for evaluating the puppet language"
   s.email = "corey@nwops.io"
   s.executables = ["prepl"]
@@ -33,8 +33,15 @@ Gem::Specification.new do |s|
     "lib/puppet-repl.rb",
     "lib/puppet-repl/cli.rb",
     "lib/puppet-repl/support.rb",
+    "lib/puppet-repl/support/compiler.rb",
+    "lib/puppet-repl/support/environment.rb",
+    "lib/puppet-repl/support/facts.rb",
+    "lib/puppet-repl/support/functions.rb",
+    "lib/puppet-repl/support/node.rb",
+    "lib/puppet-repl/support/scope.rb",
     "lib/version.rb",
     "puppet-repl.gemspec",
+    "spec/fixtures/environments/production/manifests/site.pp",
     "spec/puppet-repl_spec.rb",
     "spec/spec_helper.rb",
     "spec/support_spec.rb"

data/spec/fixtures/environments/production/manifests/site.pp

@@ -0,0 +1 @@
+fail('test_error')

data/spec/puppet-repl_spec.rb

@@ -6,16 +6,44 @@ describe "PuppetRepl" do
     "service{'httpd': ensure => running}"
   end
 
+  before(:each) do
+    repl.handle_input('reset')
+  end
+
   let(:repl) do
     PuppetRepl::Cli.new
   end
 
+  let(:input) do
+    "file{'/tmp/test2.txt': ensure => present, mode => '0755'}"
+  end
+
+  let(:resource_types) do
+    repl.parser.evaluate_string(repl.scope, input)
+  end
+
+  describe 'returns a array of resource_types' do
+    it 'returns resource type' do
+      expect(resource_types.first.class.to_s).to eq('Puppet::Pops::Types::PResourceType')
+    end
+  end
+
   describe 'help' do
     let(:input) do
       'help'
     end
     it 'can show the help screen' do
-      repl_output = /Ruby Version: #{RUBY_VERSION}\nPuppet Version: \d.\d.\d\nPuppet Repl Version: \d.\d.\d\nCreated by: NWOps <corey@nwops.io>\nType \"exit\", \"functions\", \"vars\", \"facts\", \"reset\", \"help\" for more information.\n\n/
+      repl_output = /Ruby Version: #{RUBY_VERSION}\nPuppet Version: \d.\d.\d\nPuppet Repl Version: \d.\d.\d\nCreated by: NWOps <corey@nwops.io>\nType \"exit\", \"functions\", \"vars\", \"krt\", \"facts\", \"reset\", \"help\" for more information.\n\n/
+      expect{repl.handle_input(input)}.to output(repl_output).to_stdout
+    end
+  end
+
+  describe 'krt' do
+    let(:input) do
+      "krt"
+    end
+    it 'can run' do
+      repl_output = /hostclasses/
       expect{repl.handle_input(input)}.to output(repl_output).to_stdout
     end
   end
@@ -25,7 +53,7 @@ describe "PuppetRepl" do
       "$file_path = '/tmp/test2.txt'"
     end
     it 'can process a variable' do
-      repl_output = " => /tmp/test2.txt\n"
+      repl_output = " => \e[0;33m\"/tmp/test2.txt\"\e[0m\n"
       expect{repl.handle_input(input)}.to output(repl_output).to_stdout
     end
   end
@@ -35,7 +63,7 @@ describe "PuppetRepl" do
       "file{'/tmp/test2.txt': ensure => present, mode => '0755'}"
     end
     it 'can process a resource' do
-      repl_output = " => File['/tmp/test2.txt']\n"
+      repl_output = /Puppet::Type::File/
       expect{repl.handle_input(input)}.to output(repl_output).to_stdout
     end
   end
@@ -45,7 +73,17 @@ describe "PuppetRepl" do
       "Service{"
     end
     it 'can process' do
-      repl_output = " => Syntax error at end of file\n"
+      repl_output = " => \e[31mSyntax error at end of file\e[0m\n"
+      expect{repl.handle_input(input)}.to output(repl_output).to_stdout
+    end
+  end
+
+  describe 'bad resources' do
+    let(:input) do
+      "file{'/tmp/test': ensure => present, mode => 755}"
+    end
+    xit 'can process' do  #this fails with puppet 3.8 and passes with others
+      repl_output = /must be a string/
       expect{repl.handle_input(input)}.to output(repl_output).to_stdout
     end
   end
@@ -60,7 +98,27 @@ describe "PuppetRepl" do
 
     it 'can process a each block' do
       repl.handle_input('reset')
-      repl_output = " => File['/tmp/reset']\n"
+      repl_output = /Puppet::Type::File/
+      expect{repl.handle_input(input)}.to output(repl_output).to_stdout
+    end
+    describe 'loglevel' do
+      it 'has not changed' do
+        repl.handle_input(":set loglevel debug")
+        expect(Puppet::Util::Log.level).to eq(:debug)
+        expect(Puppet::Util::Log.destinations[:console].name).to eq(:console)
+        repl.handle_input('reset')
+        expect(Puppet::Util::Log.level).to eq(:debug)
+        expect(Puppet::Util::Log.destinations[:console].name).to eq(:console)
+      end
+    end
+  end
+
+  describe 'map block' do
+    let(:input) do
+      "['/tmp/test3', '/tmp/test4'].map |String $path| { file{$path: ensure => present} }"
+    end
+    it 'can process a each block' do
+      repl_output = /Puppet::Type::File/
       expect{repl.handle_input(input)}.to output(repl_output).to_stdout
     end
   end
@@ -69,8 +127,10 @@ describe "PuppetRepl" do
     let(:input) do
       "['/tmp/test3', '/tmp/test4'].each |String $path| { file{$path: ensure => present} }"
     end
+    let(:repl_output) do
+      " => [\n    \e[1;37m[0] \e[0m\e[0;33m\"/tmp/test3\"\e[0m,\n    \e[1;37m[1] \e[0m\e[0;33m\"/tmp/test4\"\e[0m\n]\n"
+    end
     it 'can process a each block' do
-      repl_output = " => [\"/tmp/test3\", \"/tmp/test4\"]\n"
       expect{repl.handle_input(input)}.to output(repl_output).to_stdout
     end
   end
@@ -80,7 +140,7 @@ describe "PuppetRepl" do
       "$::fqdn"
     end
     it 'should be able to resolve fqdn' do
-      repl_output = " => foo.example.com\n"
+      repl_output = " => \e[0;33m\"foo.example.com\"\e[0m\n"
       expect{repl.handle_input(input)}.to output(repl_output).to_stdout
     end
   end
@@ -125,14 +185,13 @@ describe "PuppetRepl" do
       "md5('hello')"
     end
     it 'execute md5' do
-      sum = " => 5d41402abc4b2a76b9719d911017c592\n"
+      sum = " => \e[0;33m\"5d41402abc4b2a76b9719d911017c592\"\e[0m\n"
       expect{repl.handle_input(input)}.to output(sum).to_stdout
     end
     it 'execute swapcase' do
-      output = " => HELLO\n"
+      output = /HELLO/
       expect{repl.handle_input("swapcase('hello')")}.to output(output).to_stdout
     end
 
   end
-
 end

data/spec/spec_helper.rb

@@ -35,6 +35,13 @@ end
 #
 # install_stdlib
 
-RSpec.configure do |config|
+def fixtures_dir
+  File.join(File.dirname(__FILE__), 'fixtures')
+end
 
+def environments_dir
+  File.join(fixtures_dir, 'environments')
+end
+
+RSpec.configure do |config|
 end

data/spec/support_spec.rb

@@ -1,5 +1,5 @@
 require 'spec_helper'
-
+require 'tempfile'
 
 describe 'support' do
 
@@ -15,6 +15,27 @@ describe 'support' do
     repl.mod_finder.match(repl.puppet_lib_dir)[1]
   end
 
+  let(:manifest_file) do
+    file = File.open('/tmp/repl_puppet_manifest.pp', 'w') do |f|
+      f.write(manifest_code)
+    end
+    '/tmp/repl_puppet_manifest.pp'
+  end
+
+  let(:manifest_code) do
+    <<-EOF
+    file{'/tmp/test.txt': ensure => absent } \n
+    notify{'hello_there':} \n
+    service{'httpd': ensure => running}\n
+
+    EOF
+
+  end
+
+  after(:each) do
+    #manifest_file.close
+  end
+
   context '#function_map' do
 
     it 'should list functions' do
@@ -42,6 +63,4 @@ describe 'support' do
     expect(repl.facts[:fqdn]).to eq('foo.example.com')
   end
 
-
-
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: puppet-repl
 version: !ruby/object:Gem::Version
-  version: 0.0.7
+  version: 0.0.8
 platform: ruby
 authors:
 - Corey Osman
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-04-02 00:00:00.000000000 Z
+date: 2016-04-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: puppet
@@ -104,8 +104,15 @@ files:
 - lib/puppet-repl.rb
 - lib/puppet-repl/cli.rb
 - lib/puppet-repl/support.rb
+- lib/puppet-repl/support/compiler.rb
+- lib/puppet-repl/support/environment.rb
+- lib/puppet-repl/support/facts.rb
+- lib/puppet-repl/support/functions.rb
+- lib/puppet-repl/support/node.rb
+- lib/puppet-repl/support/scope.rb
 - lib/version.rb
 - puppet-repl.gemspec
+- spec/fixtures/environments/production/manifests/site.pp
 - spec/puppet-repl_spec.rb
 - spec/spec_helper.rb
 - spec/support_spec.rb