Add --puppetdb-api-version commandline

Author: Kevin Paulisse

lib/octocatalog-diff/catalog-diff/cli/options/puppetdb_api_version.rb

@@ -0,0 +1,14 @@
+# Specify the API version to use for the PuppetDB. The current values supported are '3' or '4', and '4' is
+# the default.
+# @param parser [OptionParser object] The OptionParser argument
+# @param options [Hash] Options hash being constructed; this is modified in this method.
+OctocatalogDiff::CatalogDiff::Cli::Options::Option.newoption(:puppetdb_api_version) do
+  has_weight 319
+
+  def parse(parser, options)
+    parser.on('--puppetdb-api-version N', OptionParser::DecimalInteger, 'Version of PuppetDB API (3 or 4)') do |x|
+      options[:puppetdb_api_version] = x
+      raise ArgumentError, 'Only PuppetDB versions 3 and 4 are supported' unless [3, 4].include?(x)
+    end
+  end
+end

spec/octocatalog-diff/tests/catalog-diff/cli/options/puppetdb_api_version_spec.rb

@@ -0,0 +1,19 @@
+require_relative '../options_helper'
+
+describe OctocatalogDiff::CatalogDiff::Cli::Options do
+  describe '#opt_puppetdb_api_version' do
+    it 'should handle --puppetdb-api-version with API version 3' do
+      result = run_optparse(['--puppetdb-api-version', '3'])
+      expect(result[:puppetdb_api_version]).to eq(3)
+    end
+
+    it 'should handle --puppetdb-api-version with API version 4' do
+      result = run_optparse(['--puppetdb-api-version', '4'])
+      expect(result[:puppetdb_api_version]).to eq(4)
+    end
+
+    it 'should fail if --puppetdb-api-version is unsupported' do
+      expect { run_optparse(['--puppetdb-api-version', '9000']) }.to raise_error(ArgumentError)
+    end
+  end
+end