Add CLI option for Puppet Master timeout

Author: Kevin Paulisse

lib/octocatalog-diff/cli/options/puppet_master_timeout.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+# Specify a timeout for retrieving a catalog from a Puppet master / Puppet server.
+# This timeout is specified in seconds.
+# @param parser [OptionParser object] The OptionParser argument
+# @param options [Hash] Options hash being constructed; this is modified in this method.
+OctocatalogDiff::Cli::Options::Option.newoption(:puppet_master_timeout) do
+  has_weight 329
+
+  def parse(parser, options)
+    OctocatalogDiff::Cli::Options.option_globally_or_per_branch(
+      parser: parser,
+      options: options,
+      cli_name: 'puppet-master-timeout',
+      option_name: 'puppet_master_timeout',
+      desc: 'Puppet Master catalog retrieval timeout in seconds',
+      validator: ->(x) { x.to_i > 0 || raise(ArgumentError, 'Specify timeout as a integer greater than 0') },
+      translator: ->(x) { x.to_i }
+    )
+  end
+end

spec/octocatalog-diff/tests/cli/options/puppet_master_timeout_spec.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+require_relative '../options_helper'
+
+describe OctocatalogDiff::Cli::Options do
+  describe '#opt_puppet_master_timeout' do
+    it 'should handle --puppet-master-timeout with timeout as a string' do
+      result = run_optparse(['--puppet-master-timeout', '20'])
+      expect(result[:to_puppet_master_timeout]).to eq(20)
+      expect(result[:from_puppet_master_timeout]).to eq(20)
+    end
+
+    it 'should raise error if --puppet-master-timeout == 0' do
+      expect do
+        run_optparse(['--puppet-master-timeout', '0'])
+      end.to raise_error(ArgumentError, 'Specify timeout as a integer greater than 0')
+    end
+
+    it 'should raise error if --puppet-master-timeout evaluates to 0' do
+      expect do
+        run_optparse(['--puppet-master-timeout', 'chickens'])
+      end.to raise_error(ArgumentError, 'Specify timeout as a integer greater than 0')
+    end
+
+    it 'should handle --to-puppet-master-timeout' do
+      result = run_optparse(['--to-puppet-master-timeout', '3'])
+      expect(result[:to_puppet_master_timeout]).to eq(3)
+    end
+
+    it 'should handle --from-puppet-master-timeout' do
+      result = run_optparse(['--from-puppet-master-timeout', '3'])
+      expect(result[:from_puppet_master_timeout]).to eq(3)
+    end
+  end
+end