Merge pull request #878 from chef/ssd/issue-867

Allow port to be specified as a string

lib/resources/port.rb

@@ -36,29 +36,15 @@ module Inspec::Resources
     def initialize(*args)
       args.unshift(nil) if args.length <= 1 # add the ip address to the front
       @ip = args[0]
-      @port = args[1]
-
-      @port_manager = nil
-      @cache = nil
-      os = inspec.os
-      if os.linux?
-        @port_manager = LinuxPorts.new(inspec)
-      elsif %w{darwin aix}.include?(os[:family])
-        # AIX: see http://www.ibm.com/developerworks/aix/library/au-lsof.html#resources
-        #      and https://www-01.ibm.com/marketing/iwm/iwm/web/reg/pick.do?source=aixbp
-        # Darwin: https://developer.apple.com/library/mac/documentation/Darwin/Reference/ManPages/man8/lsof.8.html
-        @port_manager = LsofPorts.new(inspec)
-      elsif os.windows?
-        @port_manager = WindowsPorts.new(inspec)
-      elsif ['freebsd'].include?(os[:family])
-        @port_manager = FreeBsdPorts.new(inspec)
-      elsif os.solaris?
-        @port_manager = SolarisPorts.new(inspec)
-      elsif os.hpux?
-        @port_manager = HpuxPorts.new(inspec)
+      @port = if args[1].nil?
+                nil
               else
-        return skip_resource 'The `port` resource is not supported on your OS yet.'
+                args[1].to_i
               end
+
+      @cache = nil
+      @port_manager = port_manager_for_os
+      return skip_resource 'The `port` resource is not supported on your OS yet.' if @port_manager.nil?
     end
 
     filter = FilterTable.create
@@ -78,6 +64,26 @@ module Inspec::Resources
 
     private
 
+    def port_manager_for_os
+      os = inspec.os
+      if os.linux?
+        LinuxPorts.new(inspec)
+      elsif %w{darwin aix}.include?(os[:family])
+        # AIX: see http://www.ibm.com/developerworks/aix/library/au-lsof.html#resources
+        #      and https://www-01.ibm.com/marketing/iwm/iwm/web/reg/pick.do?source=aixbp
+        # Darwin: https://developer.apple.com/library/mac/documentation/Darwin/Reference/ManPages/man8/lsof.8.html
+        LsofPorts.new(inspec)
+      elsif os.windows?
+        WindowsPorts.new(inspec)
+      elsif ['freebsd'].include?(os[:family])
+        FreeBsdPorts.new(inspec)
+      elsif os.solaris?
+        SolarisPorts.new(inspec)
+      elsif os.hpux?
+        HpuxPorts.new(inspec)
+      end
+    end
+
     def info
       return @cache if !@cache.nil?
       # abort if os detection has not worked

test/unit/resources/port_test.rb

@@ -44,6 +44,15 @@ describe 'Inspec::Resources::Port' do
     _(resource.addresses).must_equal ["0.0.0.0"]
   end
 
+  it 'accepts the port as a string' do
+    resource = MockLoader.new(:ubuntu1404).load_resource('port', '111')
+    _(resource.listening?).must_equal true
+    _(resource.protocols).must_equal %w{ udp }
+    _(resource.pids).must_equal [545]
+    _(resource.processes).must_equal ['rpcbind']
+    _(resource.addresses).must_equal ["0.0.0.0"]
+  end
+
   it 'verify port on MacOs x' do
     resource = MockLoader.new(:osx104).load_resource('port', 2022)
     _(resource.listening?).must_equal true