Script specs and remove unknown method

Author: Meatballs1

lib/rex/exploitation/powershell/script.rb

@@ -9,7 +9,7 @@ module Powershell
 
   class Script
     attr_accessor :code
-    attr_reader :functions
+    attr_reader :functions, :rig
 
     include Output
     include Parser
@@ -88,16 +88,6 @@ def self.to_byte_array(input_data,var_name = Rex::Text.rand_text_alpha(rand(3)+3
       return psh << lines.join("") + "\r\n"
     end
 
-    #
-    # ?? RageLtMan
-    #
-    # @param dir [String] ?
-    def self.psp_funcs(dir)
-      scripts = Dir.glob(File.expand_path(dir) + '/**/*').select {|e| e =~ /ps1$|psm1$/}
-      functions = scripts.map {|s| Script.new(s).functions}
-      return functions.flatten
-    end
-
     #
     # Return list of code modifier methods
     #

spec/lib/rex/exploitation/powershell/script_spec.rb

@@ -0,0 +1,48 @@
+# -*- coding:binary -*-
+require 'spec_helper'
+
+require 'rex/exploitation/powershell'
+
+describe Rex::Exploitation::Powershell::Output do
+
+  let(:example_script) do
+    Rex::Text.rand_text_alpha(400)
+  end
+
+  let(:subject) do
+    Rex::Exploitation::Powershell::Script.new(example_script)
+  end
+
+  describe "::initialize" do
+    it 'should create a new script object' do
+      subject.should be
+      subject.should be_kind_of Rex::Exploitation::Powershell::Script
+      subject.rig.should be
+      subject.rig.should be_kind_of Rex::RandomIdentifierGenerator
+      subject.code.should be
+      subject.code.should be_kind_of String
+      subject.code.empty?.should be_false
+      subject.functions.empty?.should be_true
+    end
+  end
+
+  describe "::to_byte_array" do
+    it 'should generate a powershell byte array' do
+      byte_array = Rex::Exploitation::Powershell::Script.to_byte_array("parp")
+      byte_array.should be
+      byte_array.should be_kind_of String
+      byte_array.include?('[Byte[]] $').should be_true
+    end
+  end
+
+  describe "::code_modifiers" do
+    it 'should return an array of modifier methods' do
+      mods = Rex::Exploitation::Powershell::Script.code_modifiers
+      mods.should be
+      mods.should be_kind_of Array
+      mods.empty?.should be_false
+    end
+  end
+
+end
+