puppet-repl 0.1.1 → 0.2.0

data/spec/puppet-repl_spec.rb

@@ -1,5 +1,5 @@
 require 'spec_helper'
-
+require 'stringio'
 describe "PuppetRepl" do
 
   let(:resource) do
@@ -10,8 +10,12 @@ describe "PuppetRepl" do
     repl.handle_input('reset')
   end
 
+  let(:output) do
+    StringIO.new('', 'w')
+  end
+
   let(:repl) do
-    PuppetRepl::Cli.new
+    PuppetRepl::Cli.new(:out_buffer => output)
   end
 
   let(:input) do
@@ -22,6 +26,39 @@ describe "PuppetRepl" do
     repl.parser.evaluate_string(repl.scope, input)
   end
 
+  describe 'multiple lines of input' do
+    describe '3 lines' do
+      let(:input) do
+        "$var1 = 'test'\nfile{\"/tmp/${var1}.txt\": ensure => present, mode => '0755'}\nvars"
+      end
+      it do
+        repl.handle_input(input)
+        expect(output.string).to match(/server_facts/) if Puppet.version.to_f >= 4.1
+        expect(output.string).to match(/test/)
+        expect(output.string).to match(/Puppet::Type::File/)
+      end
+    end
+    describe '2 lines' do
+      let(:input) do
+        "$var1 = 'test'\n $var2 = 'test2'"
+      end
+      it do
+        repl.handle_input(input)
+        expect(output.string).to eq("\n => \e[0;33m\"test\"\e[0m\n => \e[0;33m\"test2\"\e[0m\n")
+      end
+    end
+    describe '1 lines' do
+      let(:input) do
+        "$var1 = 'test'"
+      end
+      it do
+        repl.handle_input(input)
+        expect(output.string).to eq("\n => \e[0;33m\"test\"\e[0m\n")
+      end
+    end
+
+  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')
@@ -33,12 +70,13 @@ describe "PuppetRepl" do
       'help'
     end
     it 'can show the help screen' do
-      repl_output = /Type \"exit\", \"functions\", \"vars\", \"krt\", \"facts\", \"resources\", \"classes\",\n     \"play\",\"reset\", or \"help\" for more information.\n\n/
-      expect{repl.handle_input(input)}.to output(/Ruby Version: #{RUBY_VERSION}\n/).to_stdout
-      expect{repl.handle_input(input)}.to output(/Puppet Version: \d.\d.\d\n/).to_stdout
-      expect{repl.handle_input(input)}.to output(/Puppet Repl Version: \d.\d.\d\n/).to_stdout
-      expect{repl.handle_input(input)}.to output(/Created by: NWOps <corey@nwops.io>\n/).to_stdout
-      expect{repl.handle_input(input)}.to output(repl_output).to_stdout
+      expected_repl_output = /Type \"exit\", \"functions\", \"vars\", \"krt\", \"facts\", \"resources\", \"classes\",\n     \"play\", \"classification\", \"reset\", or \"help\" for more information.\n\n/
+      repl.handle_input(input)
+      expect(output.string).to match(/Ruby Version: #{RUBY_VERSION}\n/)
+      expect(output.string).to match(/Puppet Version: \d.\d.\d\n/)
+      expect(output.string).to match(/Puppet Repl Version: \d.\d.\d\n/)
+      expect(output.string).to match(/Created by: NWOps <corey@nwops.io>\n/)
+      expect(output.string).to match(expected_repl_output)
     end
   end
 
@@ -47,16 +85,18 @@ describe "PuppetRepl" do
       ""
     end
     it 'can run' do
-      repl_output = " => \n"
-      expect{repl.handle_input(input)}.to output(repl_output).to_stdout
+      repl_output = "\n"
+      repl.handle_input(input)
+      expect(output.string).to eq(repl_output)
     end
     describe 'space' do
       let(:input) do
         " "
       end
       it 'can run' do
-        repl_output = " => \n"
-        expect{repl.handle_input(input)}.to output(repl_output).to_stdout
+        repl_output = "\n => \n"
+        repl.handle_input(input)
+        expect(output.string).to eq(repl_output)
       end
     end
   end
@@ -67,7 +107,8 @@ describe "PuppetRepl" do
     end
     it 'can run' do
       repl_output = /hostclasses/
-      expect{repl.handle_input(input)}.to output(repl_output).to_stdout
+      repl.handle_input(input)
+      expect(output.string).to match(repl_output)
     end
   end
 
@@ -80,12 +121,13 @@ describe "PuppetRepl" do
       'https://gist.githubusercontent.com/logicminds/f9b1ac65a3a440d562b0/raw'
     end
     it 'file' do
-      expect{repl.handle_input("play #{fixtures_file}")}.to output(/Puppet::Type::File/).to_stdout
+      repl.handle_input("play #{fixtures_file}")
+      expect(output.string).to match(/Puppet::Type::File/)
     end
     it 'url' do
-      expect{repl.handle_input("play #{file_url}")}.to output(/Puppet::Type::File/).to_stdout
+      repl.handle_input("play #{file_url}")
+      expect(output.string).to match(/Puppet::Type::File/)
     end
-
   end
 
   describe 'variables' do
@@ -93,8 +135,9 @@ describe "PuppetRepl" do
       "$file_path = '/tmp/test2.txt'"
     end
     it 'can process a variable' do
-      repl_output = " => \e[0;33m\"/tmp/test2.txt\"\e[0m\n"
-      expect{repl.handle_input(input)}.to output(repl_output).to_stdout
+      repl_output = "\n => \e[0;33m\"/tmp/test2.txt\"\e[0m\n"
+      repl.handle_input(input)
+      expect(output.string).to eq(repl_output)
     end
   end
 
@@ -104,7 +147,8 @@ describe "PuppetRepl" do
     end
     it 'can process a resource' do
       repl_output = /Puppet::Type::File/
-      expect{repl.handle_input(input)}.to output(repl_output).to_stdout
+      repl.handle_input(input)
+      expect(output.string).to match(repl_output)
     end
   end
 
@@ -113,18 +157,20 @@ describe "PuppetRepl" do
       "Service{"
     end
     it 'can process' do
-      repl_output = " => \e[31mSyntax error at end of file\e[0m\n"
-      expect{repl.handle_input(input)}.to output(repl_output).to_stdout
+      repl_output = "\n => \e[31mSyntax error at end of file\e[0m\n"
+      repl.handle_input(input)
+      expect(output.string).to eq(repl_output)
     end
   end
 
-  describe 'bad resources' do
+  describe 'classification' do
     let(:input) do
-      "file{'/tmp/test': ensure => present, mode => 755}"
+      "classification"
     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
+
+    it 'can process a file' do
+      repl.handle_input(input)
+      expect(output.string).to eq("\n[]\n")
     end
   end
 
@@ -135,19 +181,21 @@ describe "PuppetRepl" do
 
     it 'can process a file' do
       repl_output = /Puppet::Type::File/
-      expect{repl.handle_input(input)}.to output(repl_output).to_stdout
+      repl.handle_input(input)
+      expect(output.string).to match(repl_output)
       repl.handle_input('reset')
-      expect{repl.handle_input(input)}.to output(repl_output).to_stdout
+      repl.handle_input(input)
+      expect(output.string).to match(repl_output)
     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)
+        expect(Puppet::Util::Log.destinations[:buffer].name).to eq(:buffer)
         repl.handle_input('reset')
         expect(Puppet::Util::Log.level).to eq(:debug)
-        expect(Puppet::Util::Log.destinations[:console].name).to eq(:console)
+        expect(Puppet::Util::Log.destinations[:buffer].name).to eq(:buffer)
       end
     end
   end
@@ -158,7 +206,8 @@ describe "PuppetRepl" do
     end
     it 'can process a each block' do
       repl_output = /Puppet::Type::File/
-      expect{repl.handle_input(input)}.to output(repl_output).to_stdout
+      repl.handle_input(input)
+      expect(output.string).to match(repl_output)
     end
   end
 
@@ -167,10 +216,11 @@ describe "PuppetRepl" 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"
+      "\n => [\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
-      expect{repl.handle_input(input)}.to output(repl_output).to_stdout
+      repl.handle_input(input)
+      expect(output.string).to eq(repl_output)
     end
   end
 
@@ -179,8 +229,9 @@ describe "PuppetRepl" do
       "$::fqdn"
     end
     it 'should be able to resolve fqdn' do
-      repl_output = " => \e[0;33m\"foo.example.com\"\e[0m\n"
-      expect{repl.handle_input(input)}.to output(repl_output).to_stdout
+      repl_output = "\n => \e[0;33m\"foo.example.com\"\e[0m\n"
+      repl.handle_input(input)
+      expect(output.string).to eq(repl_output)
     end
   end
 
@@ -189,7 +240,9 @@ describe "PuppetRepl" do
       "facts"
     end
     it 'should be able to print facts' do
-      expect{repl.handle_input(input)}.to output(/kernel/).to_stdout
+      repl_output = /kernel/
+      repl.handle_input(input)
+      expect(output.string).to match(repl_output)
     end
   end
 
@@ -198,7 +251,20 @@ describe "PuppetRepl" do
       'resources'
     end
     it 'should be able to print resources' do
-      expect{repl.handle_input(input)}.to output(/main/).to_stdout
+      repl_output = /main/
+      repl.handle_input(input)
+      expect(output.string).to match(repl_output)
+    end
+  end
+
+  describe 'print class' do
+    let(:input) do
+      "Class['settings']"
+    end
+    it 'should be able to print classes' do
+      repl_output = /Settings/
+      repl.handle_input(input)
+      expect(output.string).to match(repl_output)
     end
   end
 
@@ -207,7 +273,9 @@ describe "PuppetRepl" do
       'resources'
     end
     it 'should be able to print classes' do
-      expect{repl.handle_input(input)}.to output(/Settings/).to_stdout
+      repl_output = /Settings/
+      repl.handle_input(input)
+      expect(output.string).to match(repl_output)
     end
   end
 
@@ -216,10 +284,11 @@ describe "PuppetRepl" do
       ":set loglevel debug"
     end
     it 'should set the loglevel' do
-      output = /loglevel debug is set/
-      expect{repl.handle_input(input)}.to output(output).to_stdout
+      repl_output = /loglevel debug is set/
+      repl.handle_input(input)
+      expect(output.string).to match(repl_output)
       expect(Puppet::Util::Log.level).to eq(:debug)
-      expect(Puppet::Util::Log.destinations[:console].name).to eq(:console)
+      expect(Puppet::Util::Log.destinations[:buffer].name).to eq(:buffer)
     end
   end
 
@@ -228,16 +297,29 @@ describe "PuppetRepl" do
       "vars"
     end
     it 'display facts variable' do
-      output = /facts/
-      expect{repl.handle_input(input)}.to output(output).to_stdout
+      repl_output = /facts/
+      repl.handle_input(input)
+      expect(output.string).to match(repl_output)
+    end
+    it 'display server facts variable' do
+      repl_output = /server_facts/
+      repl.handle_input(input)
+      expect(output.string).to match(repl_output) if Puppet.version.to_f >= 4.1
+    end
+    it 'display serverversion variable' do
+      repl_output = /serverversion/
+      repl.handle_input(input)
+      expect(output.string).to match(repl_output) if Puppet.version.to_f >= 4.1
     end
     it 'display local variable' do
-      expect{repl.handle_input("$var1 = 'value1'")}.to output(/value1/).to_stdout
-      expect{repl.handle_input("$var1")}.to output(/value1/).to_stdout
-
+      repl.handle_input("$var1 = 'value1'")
+      expect(output.string).to match(/value1/)
+      repl.handle_input("$var1")
+      expect(output.string).to match(/value1/)
     end
     it 'display productname variable' do
-      expect{repl.handle_input("$productname")}.to output(/VirtualBox/).to_stdout
+      repl.handle_input("$productname")
+      expect(output.string).to match(/VirtualBox/)
     end
   end
 
@@ -246,13 +328,181 @@ describe "PuppetRepl" do
       "md5('hello')"
     end
     it 'execute md5' do
-      sum = " => \e[0;33m\"5d41402abc4b2a76b9719d911017c592\"\e[0m\n"
-      expect{repl.handle_input(input)}.to output(sum).to_stdout
+      repl_output =  "\n => \e[0;33m\"5d41402abc4b2a76b9719d911017c592\"\e[0m\n"
+      repl.handle_input(input)
+      expect(output.string).to eq(repl_output)
     end
     it 'execute swapcase' do
-      output = /HELLO/
-      expect{repl.handle_input("swapcase('hello')")}.to output(output).to_stdout
+      repl_output =  /HELLO/
+      repl.handle_input("swapcase('hello')")
+      expect(output.string).to match(repl_output)
+    end
+  end
+
+  describe 'unidentified object' do
+    let(:repl_output) { "\n => \n" }
+    describe "Node['foot']" do
+      let(:input) { subject }
+      it 'returns string' do
+        repl.handle_input(input)
+        expect(output.string).to eq(repl_output)
+      end
+    end
+    describe "Puppet::Pops::Types::PStringType" do
+      let(:input) { subject }
+      it 'returns string' do
+        repl.handle_input(input)
+        expect(output.string).to eq(repl_output)
+      end
+    end
+    describe 'Facts' do
+      let(:input) { subject }
+      it 'returns string' do
+        repl.handle_input(input)
+        expect(output.string).to eq(repl_output)
+      end
+    end
+  end
+
+  describe 'remote node' do
+    let(:node_obj) do
+      YAML.load_file(File.join(fixtures_dir, 'node_obj.yaml'))
+    end
+    let(:node_name) do
+      'puppetdev.localdomain'
+    end
+    before :each do
+      allow(repl).to receive(:get_remote_node).with(node_name).and_return(node_obj)
+      repl.handle_input(":set node #{node_name}")
+    end
+
+    describe 'set' do
+      it 'sends message about resetting' do
+        expect(output.string).to eq("\n => Resetting to use node puppetdev.localdomain\n")
+      end
+
+      it "return node name" do
+        output.reopen # removes previous message
+        repl.handle_input('$::hostname')
+        expect(output.string).to match(/puppetdev.localdomain/)
+      end
+
+      it "return classification" do
+        output.reopen # removes previous message
+        repl.handle_input('classification')
+        expect(output.string).to match(/certificate_authority_host/)
+      end
+    end
+
+    describe 'facts' do
+      let(:input) do
+        "$::facts['os']['family'].downcase == 'debian'"
+      end
+      it 'fact evaulation should return false' do
+        repl_output = /false/
+        repl.handle_input(input)
+        expect(output.string).to match(repl_output)
+      end
+
+    end
+    describe 'use defaults when invalid' do
+      let(:node_obj) do
+        YAML.load_file(File.join(fixtures_dir, 'invalid_node_obj.yaml'))
+      end
+      let(:node_name) do
+        'invalid.localdomain'
+      end
+      it 'name' do
+        expect{repl.node.name}.to raise_error(PuppetRepl::Exception::UndefinedNode)
+      end
+    end
+
+    it 'set node name' do
+      expect(repl.remote_node_name = 'puppetdev.localdomain').to eq("puppetdev.localdomain")
+    end
+
+    describe 'print classes' do
+      let(:input) do
+        'resources'
+      end
+      it 'should be able to print classes' do
+        repl_output = /Settings/
+        repl.handle_input(input)
+        expect(output.string).to match(repl_output)
+      end
     end
 
+    describe 'vars' do
+      let(:input) do
+        "vars"
+      end
+      it 'display facts variable' do
+        repl_output = /facts/
+        repl.handle_input(input)
+        expect(output.string).to match(repl_output)
+      end
+      it 'display server facts variable' do
+        repl_output = /server_facts/
+        repl.handle_input(input)
+        expect(output.string).to match(repl_output) if Puppet.version.to_f >= 4.1
+      end
+      it 'display server facts variable' do
+        repl_output = /server_facts/
+        repl.handle_input(input)
+        expect(output.string).to match(repl_output) if Puppet.version.to_f >= 4.1
+      end
+      it 'display local variable' do
+        repl.handle_input("$var1 = 'value1'")
+        expect(output.string).to match(/value1/)
+        repl.handle_input("$var1")
+        expect(output.string).to match(/value1/)
+      end
+      it 'display productname variable' do
+        repl.handle_input("$productname")
+        expect(output.string).to match(/VMware Virtual Platform/)
+      end
+    end
+
+    describe 'execute functions' do
+      let(:input) do
+        "md5('hello')"
+      end
+      it 'execute md5' do
+        repl_output =  /5d41402abc4b2a76b9719d911017c592/
+        repl.handle_input(input)
+        expect(output.string).to match(repl_output)
+      end
+      it 'execute swapcase' do
+        repl_output =  /HELLO/
+        repl.handle_input("swapcase('hello')")
+        expect(output.string).to match(repl_output)
+      end
+    end
+
+    describe 'reset' do
+      let(:input) do
+        "file{'/tmp/reset': ensure => present}"
+      end
+
+      it 'can process a file' do
+        repl_output = /Puppet::Type::File/
+        repl.handle_input(input)
+        expect(output.string).to match(repl_output)
+        repl.handle_input('reset')
+        repl.handle_input(input)
+        expect(output.string).to match(repl_output)
+      end
+    end
+
+    describe 'classification' do
+      let(:input) do
+        "classification"
+      end
+
+      it 'shows certificate_authority_host' do
+        repl.handle_input(input)
+        expect(output.string).to match(/certificate_authority_host/)
+      end
+    end
   end
 end

data/spec/spec_helper.rb

@@ -1,20 +1,22 @@
 require 'simplecov'
 require_relative '../lib/puppet-repl'
-
-module SimpleCov::Configuration
-  def clean_filters
-    @filters = []
-  end
-end
-
-SimpleCov.configure do
-  clean_filters
-  load_adapter 'test_frameworks'
-end
-
-ENV["COVERAGE"] && SimpleCov.start do
-  add_filter "/.rvm/"
-end
+require 'yaml'
+# module SimpleCov::Configuration
+#   def clean_filters
+#     @filters = []
+#   end
+# end
+#
+# SimpleCov.configure do
+#   clean_filters
+#   load_profile 'test_frameworks'
+# end
+#
+# SimpleCov.start do
+#   add_filter "/.rvm/"
+#   add_filter "vendor"
+#   add_filter "bundler"
+# end
 $LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
 $LOAD_PATH.unshift(File.dirname(__FILE__))
 

data/spec/support_spec.rb

@@ -3,14 +3,38 @@ require 'tempfile'
 
 describe 'support' do
 
+  let(:output) do
+    StringIO.new('', 'w')
+  end
+
   let(:repl) do
-    PuppetRepl::Cli.new
+    PuppetRepl::Cli.new(:out_buffer => output)
   end
 
   let(:scope) do
     repl.scope
   end
 
+  describe 'play' do
+    let(:url) do
+      'https://gist.github.com/logicminds/f9b1ac65a3a440d562b0'
+    end
+    let(:input) do
+      "play #{url}"
+    end
+
+    let(:expected) do
+      ''
+    end
+
+    it do
+      repl.handle_input(input)
+      expect(output.string).to match(/server_facts/) if Puppet.version.to_f >= 4.1
+      expect(output.string).to match(/test/)
+      expect(output.string).to match(/Puppet::Type::File/)
+    end
+  end
+
   let(:puppet_version) do
     repl.mod_finder.match(repl.puppet_lib_dir)[1]
   end
@@ -29,7 +53,6 @@ describe 'support' do
     service{'httpd': ensure => running}\n
 
     EOF
-
   end
 
   after(:each) do
@@ -68,4 +91,93 @@ describe 'support' do
     expect(repl.node.facts.values['fqdn']).to eq('foo.example.com')
   end
 
+  describe 'convert  url' do
+
+    describe 'unsupported' do
+      let(:url) { 'https://bitbuck.com/master/lib/log_helper.rb'}
+      let(:converted) { 'https://bitbuck.com/master/lib/log_helper.rb' }
+      it do
+        expect(repl.convert_to_text(url)).to eq(converted)
+      end
+    end
+    describe 'gitlab' do
+      describe 'blob' do
+        let(:url) { 'https://gitlab.com/nwops/prepl-web/blob/master/lib/log_helper.rb'}
+        let(:converted) { 'https://gitlab.com/nwops/prepl-web/raw/master/lib/log_helper.rb' }
+        it do
+          expect(repl.convert_to_text(url)).to eq(converted)
+        end
+      end
+
+      describe 'raw' do
+        let(:url) { 'https://gitlab.com/nwops/prepl-web/raw/master/lib/log_helper.rb'}
+        let(:converted) { 'https://gitlab.com/nwops/prepl-web/raw/master/lib/log_helper.rb' }
+        it do
+          expect(repl.convert_to_text(url)).to eq(converted)
+        end
+      end
+
+      describe 'snippet' do
+
+        describe 'not raw' do
+          let(:url) { 'https://gitlab.com/snippets/19471'}
+          let(:converted) { 'https://gitlab.com/snippets/19471/raw'}
+          it do
+            expect(repl.convert_to_text(url)).to eq(converted)
+          end
+        end
+
+        describe 'raw' do
+          let(:url) { 'https://gitlab.com/snippets/19471/raw'}
+          let(:converted) { 'https://gitlab.com/snippets/19471/raw'}
+          it do
+            expect(repl.convert_to_text(url)).to eq(converted)
+          end
+        end
+      end
+    end
+
+    describe 'github' do
+      describe 'raw' do
+        let(:url) { 'https://gist.githubusercontent.com/logicminds/f9b1ac65a3a440d562b0/raw'}
+        let(:converted) { 'https://gist.githubusercontent.com/logicminds/f9b1ac65a3a440d562b0/raw' }
+        it do
+          expect(repl.convert_to_text(url)).to eq(converted)
+        end
+      end
+
+      describe 'raw gist' do
+        let(:url) {'https://gist.githubusercontent.com/logicminds/f9b1ac65a3a440d562b0/raw/c8f6be52da5b2b0eeaabb9aa75832b75793d35d1/controls.pp'}
+        let(:converted) {'https://gist.githubusercontent.com/logicminds/f9b1ac65a3a440d562b0/raw/c8f6be52da5b2b0eeaabb9aa75832b75793d35d1/controls.pp'}
+        it do
+          expect(repl.convert_to_text(url)).to eq(converted)
+        end
+      end
+      describe 'raw non gist' do
+        let(:url) { 'https://raw.githubusercontent.com/nwops/puppet-repl/master/lib/puppet-repl.rb'}
+        let(:converted) { 'https://raw.githubusercontent.com/nwops/puppet-repl/master/lib/puppet-repl.rb' }
+        it do
+          expect(repl.convert_to_text(url)).to eq(converted)
+        end
+
+      end
+
+      describe 'blob' do
+        let(:url) { 'https://github.com/nwops/puppet-repl/blob/master/lib/puppet-repl.rb'}
+        let(:converted) { 'https://github.com/nwops/puppet-repl/raw/master/lib/puppet-repl.rb' }
+        it do
+          expect(repl.convert_to_text(url)).to eq(converted)
+        end
+      end
+
+      describe 'gist' do
+        let(:url) { 'https://gist.github.com/logicminds/f9b1ac65a3a440d562b0'}
+        let(:converted) { 'https://gist.github.com/logicminds/f9b1ac65a3a440d562b0.txt' }
+        it do
+          expect(repl.convert_to_text(url)).to eq(converted)
+        end
+      end
+    end
+  end
+
 end