wavefront-cli 8.5.0 → 9.0.0

data/lib/wavefront-cli/display/write.rb

@@ -13,7 +13,8 @@ module WavefrontDisplay
     def do_point
       @not_sent = data['rejected'] + data['unsent']
       report unless nothing_to_say?
-      exit not_sent.zero? ? 0 : 1
+
+      raise unless not_sent.zero?
     end
 
     def nothing_to_say?

data/lib/wavefront-cli/event.rb

@@ -38,7 +38,7 @@ module WavefrontCli
     def do_show
       events = state.list
 
-      if events.size.zero?
+      if events.empty?
         puts 'No open events.'
       else
         events.sort.reverse_each { |e| puts e.basename }
@@ -94,7 +94,7 @@ module WavefrontCli
     end
 
     def window_start
-      parse_time((options[:start] || Time.now - 600), true)
+      parse_time((options[:start] || (Time.now - 600)), true)
     end
 
     def window_end

data/lib/wavefront-cli/event_store.rb

@@ -41,7 +41,7 @@ module WavefrontCli
     end
 
     def event_file(id)
-      id =~ /^\d{13}:.+/ ? dir + id : nil
+      /^\d{13}:.+/.match?(id) ? dir + id : nil
     end
 
     # We can override the temp directory with the WF_EVENT_STATE_DIR env var.
@@ -70,7 +70,7 @@ module WavefrontCli
     def event(id)
       if !id
         pop_event!
-      elsif id =~ /^\d{13}:.+:\d+/
+      elsif /^\d{13}:.+:\d+/.match?(id)
         id
       else
         pop_event!(id)

data/lib/wavefront-cli/exception.rb

@@ -7,26 +7,47 @@ module WavefrontCli
   #
   class Exception
     class CredentialError < RuntimeError; end
+
     class MandatoryValue < RuntimeError; end
+
     class ConfigFileNotFound < IOError; end
+
     class FileNotFound < IOError; end
+
     class ImpossibleSearch < RuntimeError; end
+
     class InsufficientData < RuntimeError; end
+
     class InvalidInput < RuntimeError; end
+
     class InvalidQuery < RuntimeError; end
+
     class InvalidValue < RuntimeError; end
+
     class ProfileExists < RuntimeError; end
+
     class ProfileNotFound < RuntimeError; end
+
     class SystemError < RuntimeError; end
+
     class UnhandledCommand < RuntimeError; end
+
     class UnparseableInput < RuntimeError; end
+
     class UnparseableResponse < RuntimeError; end
+
     class UnparseableSearchPattern < RuntimeError; end
+
     class UnsupportedFileFormat < RuntimeError; end
+
     class UnsupportedNoop < RuntimeError; end
+
     class UnsupportedOperation < RuntimeError; end
+
     class UnsupportedOutput < RuntimeError; end
+
     class UserGroupNotFound < RuntimeError; end
+
     class UserError < RuntimeError; end
   end
 end

data/lib/wavefront-cli/exception_handler.rb

@@ -8,6 +8,7 @@ module WavefrontCli
     # rubocop:disable Metrics/MethodLength
     # rubocop:disable Metrics/AbcSize
     # rubocop:disable Metrics/CyclomaticComplexity
+    # rubocop:disable Metrics/PerceivedComplexity
     def exception_handler(exception)
       case exception
       when WavefrontCli::Exception::UnhandledCommand
@@ -18,7 +19,8 @@ module WavefrontCli
         abort "\nOperation aborted at user request."
       when WavefrontCli::Exception::ConfigFileNotFound
         abort "Configuration file #{exception}' not found."
-      when WavefrontCli::Exception::CredentialError
+      when WavefrontCli::Exception::CredentialError,
+           Wavefront::Exception::CredentialError
         handle_missing_credentials(exception)
       when WavefrontCli::Exception::MandatoryValue
         abort 'A value must be supplied.'
@@ -26,6 +28,8 @@ module WavefrontCli
         abort 'Connection timed out.'
       when Wavefront::Exception::InvalidPermission
         abort "'#{exception}' is not a valid Wavefront permission."
+      when Wavefront::Exception::InvalidTimestamp
+        abort "'#{exception}' is not a parseable time."
       when Wavefront::Exception::InvalidUserGroupId
         abort "'#{exception}' is not a valid user group ID."
       when Wavefront::Exception::InvalidAccountId
@@ -38,6 +42,8 @@ module WavefrontCli
         abort "'#{exception}' is not a valid API token ID."
       when Wavefront::Exception::InvalidIngestionPolicyId
         abort "'#{exception}' is not a valid ingestion policy ID."
+      when Wavefront::Exception::InvalidVersion
+        abort "'#{exception}' is not a valid version."
       when WavefrontCli::Exception::InvalidValue
         abort "Invalid value for #{exception}."
       when WavefrontCli::Exception::ProfileExists
@@ -80,6 +86,7 @@ module WavefrontCli
         abort
       end
     end
+    # rubocop:enable Metrics/PerceivedComplexity
     # rubocop:enable Metrics/MethodLength
     # rubocop:enable Metrics/AbcSize
     # rubocop:enable Metrics/CyclomaticComplexity

data/lib/wavefront-cli/helpers/load_file.rb

@@ -40,11 +40,11 @@ module WavefrontCli
       private
 
       def load_json(file)
-        read_json(IO.read(file))
+        read_json(File.read(file))
       end
 
       def load_yaml(file)
-        read_yaml(IO.read(file))
+        read_yaml(File.read(file))
       end
 
       # Read STDIN and return a Ruby object, assuming that STDIN is

data/lib/wavefront-cli/maintenancewindow.rb

@@ -27,7 +27,7 @@ module WavefrontCli
 
       [%i[CustomerTags atag], %i[HostTags htag],
        %i[HostNames host]].each do |key, opt|
-        k = ('relevant' + key.to_s).to_sym
+        k = "relevant#{key}".to_sym
         body[k] = options[opt] unless options[opt].empty?
       end
 

data/lib/wavefront-cli/metricspolicy.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+require 'wavefront-sdk/support/mixins'
+require_relative 'base'
+require_relative 'helpers/load_file'
+
+module WavefrontCli
+  #
+  # CLI coverage for the metricspolicy part of the v2 'usage' API.
+  #
+  class MetricsPolicy < WavefrontCli::Base
+    def do_describe
+      wf.describe(options[:version])
+    end
+
+    def do_history
+      wf.history(options[:offset] || 0, options[:limit] || 100)
+    end
+
+    def do_revert
+      wf.revert(options[:'<version>'])
+    end
+
+    def do_update
+      raw = WavefrontCli::Helper::LoadFile.new(options[:'<file>']).load
+      rules = process_update(raw)
+      wf.update(policyRules: rules)
+    end
+
+    # It looks like the API expects arrays of ID strings for accounts, groups,
+    # and roles, but when you export one, those fields are objects with name
+    # and ID.
+    #
+    def process_update(raw)
+      raw[:policyRules].tap do |rule|
+        rule[:accounts] = rule[:accounts].map { |r| r[:id] }
+        rule[:userGroups] = rule[:userGroups].map { |r| r[:id] }
+        rule[:roles] = rule[:roles].map { |r| r[:id] }
+      end
+    end
+  end
+end

data/lib/wavefront-cli/opt_handler.rb

@@ -29,7 +29,7 @@ module WavefrontCli
 
     def initialize(cli_opts = {})
       cred_opts = setup_cred_opts(cli_opts)
-      cli_opts.reject! { |_k, v| v.nil? }
+      cli_opts.compact!
       @opts = DEFAULT_OPTS.merge(load_profile(cred_opts)).merge(cli_opts)
     rescue WavefrontCli::Exception::ConfigFileNotFound => e
       abort "Configuration file '#{e}' not found."
@@ -43,11 +43,10 @@ module WavefrontCli
     # constructor.
     # @param cli_opts [Hash] options from docopt, which may include
     #   the location of the config file and the stanza within it
-    # @return [Hash] keys are none, one, or both of :file and
-    #   :profile
+    # @return [Hash] keys are none, one, or both of :file and :profile
     #
     def setup_cred_opts(cli_opts)
-      cred_opts = { raise_on_no_profile: true }
+      cred_opts = cli_opts[:config] ? { raise_on_no_profile: true } : {}
 
       if cli_opts[:config]
         cred_opts[:file] = Pathname.new(cli_opts[:config])

data/lib/wavefront-cli/output/csv/query.rb

@@ -32,7 +32,7 @@ module WavefrontCsvOutput
     def raw_output
       resp.each_with_object([]) do |point, a|
         point[:points].each do |p|
-          a.<< csv_format(options[:'<metric>'],
+          a << csv_format(options[:'<metric>'],
                           p[:value],
                           p[:timestamp],
                           options[:host],
@@ -48,7 +48,7 @@ module WavefrontCsvOutput
 
       resp[:timeseries].each_with_object([]) do |ts, a|
         ts[:data].each do |point|
-          a.<< csv_format(ts[:label],
+          a << csv_format(ts[:label],
                           point[1],
                           point[0],
                           ts[:host],

data/lib/wavefront-cli/output/hcl/dashboard.rb

@@ -32,19 +32,19 @@ module WavefrontHclOutput
     # @return [String] HCL list of vals
     #
     def listmaker(vals, method)
-      vals.each_with_object([]) { |v, a| a.<< send(method, v) }.to_hcl_list
+      vals.each_with_object([]) { |v, a| a << send(method, v) }.to_hcl_list
     end
 
     def vhandle_sections(vals)
       vals.each_with_object([]) do |section, a|
-        a.<< ("name = \"#{section[:name]}\"\n      row = " +
+        a << ("name = \"#{section[:name]}\"\n      row = " +
               handle_rows(section[:rows])).braced(4)
       end.to_hcl_list
     end
 
     def handle_rows(rows)
       rows.each_with_object([]) do |row, a|
-        a.<< ('chart = ' + handle_charts(row[:charts]).to_s).braced(8)
+        a << "chart = #{handle_charts(row[:charts])}".braced(8)
       end.to_hcl_list
     end
 
@@ -58,10 +58,10 @@ module WavefrontHclOutput
       lines = chart.each_with_object([]) do |(k, v), a|
         next unless fields.include?(k)
 
-        a.<< format('%<key>s = %<value>s', key: k, value: quote_value(v))
+        a << format('%<key>s = %<value>s', key: k, value: quote_value(v))
       end
 
-      lines.<< "source = #{handle_sources(chart[:sources])}"
+      lines << "source = #{handle_sources(chart[:sources])}"
       lines.to_hcl_obj(10)
     end
 
@@ -75,7 +75,7 @@ module WavefrontHclOutput
 
         k = 'queryBuilderEnabled' if k == 'querybuilderEnabled'
 
-        a.<< format('%<key>s = %<value>s',
+        a << format('%<key>s = %<value>s',
                     key: k.to_snake,
                     value: quote_value(v))
       end.to_hcl_obj(14)

data/lib/wavefront-cli/output/hcl/stdlib/array.rb

@@ -9,7 +9,7 @@ class Array
   # @return [String]
   #
   def to_hcl_list
-    '[' + join(',') + ']'
+    "[#{join(',')}]"
   end
 
   # Turn an array into a string which represents an HCL object

data/lib/wavefront-cli/output/wavefront/query.rb

@@ -20,11 +20,11 @@ module WavefrontWavefrontOutput
     def raw_output
       resp.each_with_object('') do |point, a|
         point[:points].each do |p|
-          a.<< wavefront_format(options[:'<metric>'],
-                                p[:value],
-                                p[:timestamp],
-                                options[:host],
-                                point[:tags]) + "\n"
+          a << "#{wavefront_format(options[:'<metric>'],
+                                   p[:value],
+                                   p[:timestamp],
+                                   options[:host],
+                                   point[:tags])}\n"
         end
       end
     end
@@ -34,7 +34,7 @@ module WavefrontWavefrontOutput
 
       resp[:timeseries].each_with_object([]) do |ts, a|
         ts[:data].each do |point|
-          a.<< wavefront_format(ts[:label],
+          a << wavefront_format(ts[:label],
                                 point[1],
                                 point[0],
                                 ts[:host],
@@ -46,7 +46,7 @@ module WavefrontWavefrontOutput
     def wavefront_format(path, value, timestamp, source, tags = nil)
       arr = [path, value, timestamp, format('source=%<source>s',
                                             source: source)]
-      arr.<< tags.to_wf_tag if tags && !tags.empty?
+      arr << tags.to_wf_tag if tags && !tags.empty?
       arr.join(' ')
     end
   end

data/lib/wavefront-cli/query.rb

@@ -126,7 +126,7 @@ module WavefrontCli
       end
     end
 
-    def handle_errcode_404(_status)
+    def handle_errcode404(_status)
       'Perhaps metric does not exist for given host.'
     end
   end

data/lib/wavefront-cli/settings.rb

@@ -7,6 +7,8 @@ module WavefrontCli
   # CLI coverage for the v2 'settings' API.
   #
   class Settings < WavefrontCli::Base
+    JOBS = %w[invitePermissions defaultUserGroups].freeze
+
     def do_list_permissions
       wf.permissions
     end
@@ -24,10 +26,7 @@ module WavefrontCli
         k, v = o.split('=', 2)
         next unless v && !v.empty?
 
-        if %w[invitePermissions defaultUserGroups].include?(k)
-          v = v.include?(',') ? v.split(',') : [v]
-        end
-
+        v = v.include?(',') ? v.split(',') : [v] if JOBS.include?(k)
         a[k] = v
       end
 

data/lib/wavefront-cli/stdlib/string.rb

@@ -10,7 +10,7 @@ class String
   # @param indent [Integer] size of hanging indent, in chars
   #
   def cmd_fold(twidth = TW, indent = 10)
-    gsub(/(-\w) /, '\\1^').scan_line(twidth - 12).join("\n" + ' ' * indent)
+    gsub(/(-\w) /, '\\1^').scan_line(twidth - 12).join("\n#{' ' * indent}")
                           .restored
   end
 
@@ -50,7 +50,7 @@ class String
   # undesirable line breaking. This puts it back
   #
   def restored
-    tr('^', ' ').chomp("\n")
+    tr('^', ' ').chomp
   end
 
   # Fold long value lines in two-column output. The returned string
@@ -58,7 +58,7 @@ class String
   #
   def value_fold(indent = 0, twidth = TW)
     max_line_length = twidth - indent - 4
-    scan_line(max_line_length).join("\n" + ' ' * indent)
+    scan_line(max_line_length).join("\n#{' ' * indent}")
   end
 
   # @param width [Integer] length of longest string (width of
@@ -94,7 +94,7 @@ class String
   #
   def to_snake
     gsub(/(.)([A-Z])/) do
-      Regexp.last_match[1] + '_' + Regexp.last_match[2].downcase
+      "#{Regexp.last_match[1]}_#{Regexp.last_match[2].downcase}"
     end
   end
 
@@ -102,7 +102,7 @@ class String
 
   def indent_folded_lines(chunks, twidth, indent, prefix)
     chunks.join(' ').scan_line(twidth - indent - 5).map do |line|
-      prefix + ' ' * indent + line
+      "#{prefix}#{' ' * indent}#{line}"
     end
   end
 end

data/lib/wavefront-cli/usage.rb

@@ -18,7 +18,7 @@ module WavefrontCli
     end
 
     def default_start
-      parse_time(Time.now - 60 * 60 * 24)
+      parse_time(Time.now - 86_400)
     end
   end
 end

data/lib/wavefront-cli/version.rb

@@ -1,3 +1,3 @@
 # frozen_string_literal: true
 
-WF_CLI_VERSION = '8.5.0'
+WF_CLI_VERSION = '9.0.0'

data/lib/wavefront-cli/write.rb

@@ -74,8 +74,8 @@ module WavefrontCli
     end
     # rubocop:enable Metrics/AbcSize
 
-    # Turn our user's representation of a distribution into one
-    # which suits Wavefront. The SDK can do this for us.
+    # Turn our user's representation of a distribution into one which suits
+    # Wavefront. The SDK can do this for us.
     #
     def mk_dist
       xpanded = expand_dist(options[:'<val>'])
@@ -83,7 +83,11 @@ module WavefrontCli
     end
 
     def extra_options
-      options[:using] ? { writer: options[:using] } : {}
+      if options[:using]
+        { writer: options[:using] }
+      else
+        { writer: :http }
+      end
     end
 
     # I chose to prioritise UI consistency over internal elegance
@@ -108,6 +112,7 @@ module WavefrontCli
         port: options[:port] || default_port,
         socket: options[:socket],
         endpoint: options[:endpoint],
+        agent: "wavefront-cli-#{WF_CLI_VERSION}",
         token: options[:token] }
     end
 
@@ -115,18 +120,9 @@ module WavefrontCli
       distribution? ? 40_000 : 2878
     end
 
+    # The SDK writer plugins validate the credentials they need
     def validate_opts
       validate_opts_file if options[:file]
-
-      if options[:using] == 'unix'
-        return true if options[:socket]
-
-        raise(WavefrontCli::Exception::CredentialError, 'No socket path.')
-      end
-
-      return true if options[:proxy]
-
-      raise(WavefrontCli::Exception::CredentialError, 'No proxy address.')
     end
 
     def validate_opts_file
@@ -167,7 +163,7 @@ module WavefrontCli
     #
     def process_input_file(data)
       data.each_with_object([]) do |l, a|
-        a.<< process_line(l)
+        a << process_line(l)
       rescue WavefrontCli::Exception::UnparseableInput => e
         puts "Bad input. #{e.message}."
         next
@@ -375,7 +371,7 @@ module WavefrontCli
     end
 
     def format_string_is_all_valid_chars?(fmt)
-      return true if fmt =~ /^[dmstTv]+$/
+      return true if /^[dmstTv]+$/.match?(fmt)
 
       raise(WavefrontCli::Exception::UnparseableInput,
             'unsupported field in format string')
@@ -426,17 +422,17 @@ module WavefrontCli
     #
     def valid_timestamp?(timestamp)
       (timestamp.is_a?(Integer) ||
-        timestamp.is_a?(String) && timestamp.match(/^\d+$/)) &&
+        (timestamp.is_a?(String) && timestamp.match(/^\d+$/))) &&
         timestamp.to_i > 946_684_800 &&
         timestamp.to_i < (Time.now.to_i + 31_557_600)
     end
 
     def setup_fmt(fmt)
-      @fmt = fmt.split('')
+      @fmt = fmt.chars
     end
 
     def load_data(file)
-      IO.read(file)
+      File.read(file)
     rescue StandardError
       raise WavefrontCli::Exception::FileNotFound
     end

data/spec/.rubocop.yml

@@ -1,23 +1,8 @@
 ---
-AllCops:
-  NewCops: enable
+inherit_from:
+  - ../.rubocop.yml
 
 Metrics/MethodLength:
   Max: 30
-
 Metrics/AbcSize:
   Max: 45
-
-Metrics/ClassLength:
-  Max: 300
-
-# Is nothing sacred?
-Layout/LineLength:
-  Max: 80
-
-Style/IfUnlessModifier:
-  Enabled: false # because it wants to make lines >80 chars
-Style/StringConcatenation:
-  Enabled: false
-Style/OptionalBooleanParameter:
-  Enabled: false

data/spec/constants.rb

@@ -12,10 +12,9 @@ TW = 80
 
 ENDPOINT = 'metrics.wavefront.com'
 TOKEN = '0123456789-ABCDEF'
-RES_DIR = ROOT + 'spec' + 'wavefront-cli' + 'resources'
-CF = RES_DIR + 'wavefront.conf'
+RES_DIR = ROOT.join('spec', 'wavefront-cli', 'resources')
+CF = RES_DIR.join('wavefront.conf')
 CF_VAL =  IniFile.load(CF)
-JSON_POST_HEADERS = {
-  'Content-Type': 'application/json', Accept: 'application/json'
-}.freeze
+JSON_POST_HEADERS = { 'Content-Type': 'application/json',
+                      Accept: 'application/json' }.freeze
 TEE_ZERO = Time.now.freeze

data/spec/support/command_base.rb

@@ -9,11 +9,13 @@ require_relative '../../lib/wavefront-cli/controller'
 # commands.
 #
 class EndToEndTest < MiniTest::Test
+  attr_accessor :single_perm
   attr_reader :wf
 
   def setup
     before_setup if respond_to?(:before_setup)
     @wf = WavefrontCliController
+    @single_perm = false
   end
 
   def api_path
@@ -79,4 +81,14 @@ class EndToEndTest < MiniTest::Test
     { status: { result: 'OK', message: '', code: 200 },
       items: [] }.to_json
   end
+
+  def blank_envvars
+    %w[WAVEFRONT_ENDPOINT WAVEFRONT_PROXY WAVEFRONT_TOKEN].each do |v|
+      ENV[v] = nil
+    end
+  end
+
+  def config?
+    Pathname.new(Dir.home).join('.wavefront').exist?
+  end
 end

data/spec/support/minitest_assertions.rb

@@ -147,10 +147,13 @@ module Minitest
     end
 
     def assert_cmd_posts(command, api_path, payload = 'null',
-                         response = dummy_response)
+                         response = nil, extra_headers = {})
+      response ||= dummy_response
       all_permutations do |p|
         assert_posts("https://#{p[:endpoint]}#{api_path}",
-                     mk_headers(p[:token]), payload, response) do
+                     mk_headers(p[:token], extra_headers),
+                     payload,
+                     response) do
           wf.new("#{cmd_word} #{command} #{p[:cmdline]}".split)
         end
       end
@@ -166,7 +169,7 @@ module Minitest
     end
 
     def assert_cmd_deletes(command, api_path, response = dummy_response)
-      permutations.each do |p|
+      all_permutations do |p|
         assert_deletes("https://#{p[:endpoint]}#{api_path}",
                        mk_headers(p[:token]), response) do
           wf.new("#{cmd_word} #{command} #{p[:cmdline]}".split)
@@ -188,13 +191,14 @@ module Minitest
       assert_empty(err)
     end
 
-    # Run tests with all available permutations. We'll always need
-    # to mock out display, unless we don't, when even if we do, it
-    # won't matter.
+    # Run tests with all available permutations, unless the single_perm class
+    # variable is set. This lets us run tests faster by running fewer (but
+    # still a good random selection) and lets us run tests which must only be
+    # run once, like tests which pop stuff off the event stack.
     #
     def all_permutations
       perms = permutations
-      perms = [perms[2]]
+      perms = perms.shuffle.take(1) if @single_perm || ENV['FAST_TESTS']
 
       perms.each do |p|
         yield(p)
@@ -213,11 +217,11 @@ module Minitest
 
     private
 
-    def mk_headers(token = nil)
-      { 'Accept': /.*/,
+    def mk_headers(token = nil, extra_headers = {})
+      { Accept: /.*/,
         'Accept-Encoding': /.*/,
-        'Authorization': 'Bearer ' + (token || '0123456789-ABCDEF'),
-        'User-Agent': "wavefront-cli-#{WF_CLI_VERSION}" }
+        Authorization: "Bearer #{token || '0123456789-ABCDEF'}",
+        'User-Agent': "wavefront-cli-#{WF_CLI_VERSION}" }.merge(extra_headers)
     end
 
     # Every command we simulate running is done under the following

data/spec/support/output_tester.rb

@@ -14,7 +14,7 @@ class OutputTester
   # @return [Object] canned raw responses used to test outputs
   #
   def load_input(file, only_items = true)
-    ret = JSON.parse(IO.read(RES_DIR + 'display' + file),
+    ret = JSON.parse(File.read(RES_DIR.join('display', file)),
                      symbolize_names: true)
     only_items ? ret[:items] : ret
   end
@@ -23,7 +23,7 @@ class OutputTester
   # @return [String]
   #
   def load_expected(file)
-    IO.read(RES_DIR + 'display' + file)
+    File.read(RES_DIR.join('display', file))
   end
 
   def in_and_out(input, expected, only_items = true)