(MAINT) Make Port Check Configurable

This change makes the port and port timeout configurable via the YAML
config file. Before these were hard coded to 22 and 5, respectively.

Tests have been added to verify the default is chosen when not
configured and the configuration is picked up when supplied.

lib/vmpooler/pool_manager.rb

@@ -1,5 +1,12 @@
 module Vmpooler
   class PoolManager
+
+    # Defaults
+    # port number to verify pending VM is ready
+    DEFAULT_PORT_NO      = 22
+    # timeout, in seconds, when connecting to DEFAULT_PORT_NO
+    DEFAULT_PORT_TIMEOUT = 5
+
     def initialize(config, logger, redis, graphite=nil)
       $config = config
 
@@ -32,8 +39,8 @@ module Vmpooler
 
       if host
         begin
-          Timeout.timeout(5) do
+          Timeout.timeout($config[:config]['port_timeout'] || DEFAULT_PORT_TIMEOUT) do
-            TCPSocket.new vm, 22
+            TCPSocket.new vm, ($config[:config]['port'] || DEFAULT_PORT_NO)
           end
           move_pending_vm_to_ready(vm, pool, host)
         rescue

spec/vmpooler/pool_manager_spec.rb

@@ -33,16 +33,43 @@ describe 'Pool Manager' do
 
     end
 
-    context 'host is in pool' do
+    context 'host is in pool (defaults)' do
       let(:vm_finder) { double('vm_finder') }
       let(:tcpsocket) { double('TCPSocket') }
+      let(:config) { {config: {}} }
 
       it 'calls move_pending_vm_to_ready' do
         stub_const("TCPSocket", tcpsocket)
 
         allow(pool_helper).to receive(:find_vm).and_return(vm_finder)
         allow(vm_finder).to receive(:summary).and_return(nil)
-        allow(tcpsocket).to receive(:new).and_return(true)
+        # ensure the default port was used
+        allow(tcpsocket).to receive(:new).with(String, 22).and_return(true)
+
+        expect(vm_finder).to receive(:summary).once
+        expect(redis).not_to receive(:hget).with(String, 'clone')
+
+        subject._check_pending_vm(vm, pool, timeout)
+      end
+    end
+
+    context 'host is in pool (port config)' do
+      let(:vm_finder) { double('vm_finder') }
+      let(:tcpsocket) { double('TCPSocket') }
+      # port and port timeout are now configurable.
+      # update the config mock
+      let(:port_no) { 99 }
+      let(:config) { {config: {
+          'port'         => port_no,
+          'port_timeout' => 2}} }
+
+      it 'calls move_pending_vm_to_ready' do
+        stub_const("TCPSocket", tcpsocket)
+
+        allow(pool_helper).to receive(:find_vm).and_return(vm_finder)
+        allow(vm_finder).to receive(:summary).and_return(nil)
+        # ensure the port config made it to the TCPSocket call:
+        allow(tcpsocket).to receive(:new).with(String, port_no).and_return(true)
 
         expect(vm_finder).to receive(:summary).once
         expect(redis).not_to receive(:hget).with(String, 'clone')

vmpooler.yaml.example

@@ -161,6 +161,16 @@
 #
 #   - domain
 #     If set, returns a top-level 'domain' JSON key in POST requests
+#
+#   - port
+#     The port number used to verify a host is ready and available.
+#     (optional; default: 22)
+#
+#   - port_timeout
+#     How long (in seconds) to wait for connection to port before canceling.
+#     This is not used in marking a VM dead or stale.
+#     (optional; default: 5)
+#
 
 # Example:
 
@@ -176,6 +186,8 @@
     - 'created_by'
     - 'project'
   domain: 'company.com'
+  port: 22
+  port_timeout: 5
 
 # :pools:
 #