wavefront-sdk 7.0.0 → 8.0.0

data/lib/wavefront-sdk/spansamplingpolicy.rb

@@ -0,0 +1,126 @@
+# frozen_string_literal: true
+
+require_relative 'defs/constants'
+require_relative 'core/api'
+
+module Wavefront
+  #
+  # View and manage the Wavefront span sampling policy.
+  #
+  class SpanSamplingPolicy < CoreApi
+    def update_keys
+      %i[name active expression description samplingPercent]
+    end
+
+    # GET /api/v2/spansamplingpolicy
+    # Get all sampling policies for a customer
+    #
+    # @param offset [Int] policy at which the list begins
+    # @param limit [Int] the number of policies to return
+    # @return [Wavefront::Response]
+    #
+    def list(offset = 0, limit = 100)
+      api.get('', offset: offset, limit: limit)
+    end
+
+    # POST /api/v2/spansamplingpolicy
+    # Create a span sampling policy
+    #
+    # @param body [Hash] description of policy
+    # @return [Wavefront::Response]
+    #
+    def create(body)
+      raise ArgumentError unless body.is_a?(Hash)
+
+      api.post('', body, 'application/json')
+    end
+
+    # DELETE /api/v2/spansamplingpolicy/{id}
+    # Delete a specific span sampling policy
+    #
+    # Deleting an active policy moves it to 'trash', from where it can
+    # be restored with an #undelete operation. Deleting an policy in
+    # 'trash' removes it for ever.
+    #
+    # @param id [String] ID of the policy
+    # @return [Wavefront::Response]
+    #
+    def delete(id)
+      wf_spansamplingpolicy_id?(id)
+      api.delete(id)
+    end
+
+    # GET /api/v2/spansamplingpolicy/deleted
+    # Get all deleted sampling policies for a customer
+    # @return [Wavefront::Response]
+    #
+    def deleted
+      api.get('deleted')
+    end
+
+    # GET /api/v2/spansamplingpolicy/{id}
+    # GET /api/v2/spansamplingpolicy/{id}/history/{version}
+    # Get a specific span sampling policy
+    # Get a specific policy / Get a specific historical version of a specific
+    # sampling policy
+    #
+    # @param id [String] ID of the policy
+    # @param version [Integer] version of policy
+    # @return [Wavefront::Response]
+    #
+    def describe(id, version = nil)
+      wf_spansamplingpolicy_id?(id)
+      wf_version?(version) if version
+      fragments = [id]
+      fragments += ['history', version] if version
+      api.get(fragments.uri_concat)
+    end
+
+    # PUT /api/v2/spansamplingpolicy/{id}
+    # Update a specific span sampling policy
+    #
+    # @param id [String] a Wavefront span-sampling policy ID
+    # @param body [Hash] key-value hash of the parameters you wish
+    #   to change
+    # @param modify [true, false] if true, use {#describe()} to get
+    #   a hash describing the existing object, and modify that with
+    #   the new body. If false, pass the new body straight through.
+    # @return [Wavefront::Response]
+
+    def update(id, body, modify = true)
+      wf_spansamplingpolicy_id?(id)
+      raise ArgumentError unless body.is_a?(Hash)
+
+      return api.put(id, body, 'application/json') unless modify
+
+      api.put(id, hash_for_update(describe(id).response, body),
+              'application/json')
+    end
+
+    # GET /api/v2/spansamplingpolicy/{id}/history
+    # Get the version history of a specific sampling policy
+    #
+    # @param id [String] ID of the policy
+    # @return [Wavefront::Response]
+    #
+    def history(id, offset = nil, limit = nil)
+      wf_spansamplingpolicy_id?(id)
+      qs = {}
+      qs[:offset] = offset if offset
+      qs[:limit] = limit if limit
+
+      api.get([id, 'history'].uri_concat, qs)
+    end
+
+    # POST /api/v2/spansamplingpolicy/{id}/undelete
+    # Restore a deleted span sampling policy
+    #
+    # @param id [String] ID of the policy
+    # @return [Wavefront::Response]
+    #
+    def undelete(id)
+      wf_spansamplingpolicy_id?(id)
+      api.post([id, 'undelete'].uri_concat)
+    end
+  end
+end

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

@@ -11,6 +11,6 @@ class String
   # @return [String] a URI path
   #
   def tagescape
-    gsub(/"/, '\"')
+    gsub('"', '\"')
   end
 end

data/lib/wavefront-sdk/user.rb

@@ -49,7 +49,7 @@ module Wavefront
       api.post("?sendEmail=#{send_email}", body, 'application/json')
     end
 
-    # DELETE /api/v2/user/id
+    # DELETE /api/v2/user/{id}
     # Delete a specific user. See also #delete_users.
     #
     # @param id [String] ID of the user
@@ -61,7 +61,7 @@ module Wavefront
       api.delete(id)
     end
 
-    # GET /api/v2/user/id
+    # GET /api/v2/user/{id}
     # Retrieves a user by identifier (email addr).
     #
     # @param id [String] ID of the user
@@ -73,7 +73,7 @@ module Wavefront
       api.get(id)
     end
 
-    # PUT /api/v2/user/id
+    # PUT /api/v2/user/{id}
     # Update a specific user definition.
     #
     # @param id [String] a Wavefront user ID
@@ -124,7 +124,7 @@ module Wavefront
                'application/json')
     end
 
-    # PUT /api/v2/user/id/grant
+    # POST /api/v2/user/{id}/grant
     # Grants a specific user permission.
     #
     # @param id [String] ID of the user
@@ -147,7 +147,7 @@ module Wavefront
                'application/x-www-form-urlencoded')
     end
 
-    # PUT /api/v2/user/id/revoke
+    # POST /api/v2/user/{id}/revoke
     # Revokes a specific user permission.
     #
     # @param id [String] ID of the user

data/lib/wavefront-sdk/usergroup.rb

@@ -16,8 +16,8 @@ module Wavefront
 
     # GET /api/v2/usergroup
     # Get all user groups for a customer
-    # @param offset [Int] alert at which the list begins
-    # @param limit [Int] the number of alerts to return
+    # @param offset [Int] usergroup at which the list begins
+    # @param limit [Int] the number of usergroups to return
     # @return [Wavefront::Response]
     #
     def list(offset = 0, limit = 100)

data/lib/wavefront-sdk/validators.rb

@@ -164,8 +164,7 @@ module Wavefront
     #
     # @param version [Integer] the version number to validate
     # @return True if the version is valid
-    # @raise Wavefront::Exception::InvalidVersion if the alert ID is
-    #   not valid
+    # @raise Wavefront::Exception::InvalidVersion if the ID is not valid
     #
     def wf_version?(version)
       version = version.to_i if version.is_a?(String) && version =~ /^\d+$/
@@ -197,7 +196,7 @@ module Wavefront
     def wf_point_tag?(key, val)
       if key && val && (key.size + val.size < 254) &&
          key =~ /^[\w\-.:]+$/ && val !~ /\\$/
-        return
+        return false
       end
 
       raise Wavefront::Exception::InvalidTag, "#{key}=#{val}"
@@ -589,10 +588,10 @@ module Wavefront
     #   ID is not valid
     #
     def wf_account_id?(id)
-      return true if wf_user_id?(id)
+      true if wf_user_id?(id)
     rescue Wavefront::Exception::InvalidUserId
       begin
-        return true if wf_serviceaccount_id?(id)
+        true if wf_serviceaccount_id?(id)
       rescue Wavefront::Exception::InvalidServiceAccountId
         raise Wavefront::Exception::InvalidAccountId, id
       end
@@ -609,6 +608,17 @@ module Wavefront
       raise Wavefront::Exception::InvalidMonitoredClusterId, id
     end
 
+    # Ensure the given argument is a valid monitored application ID
+    # @param id [String]
+    # @raise Wavefront::Exception::InvalidMonitoredApplicationId if the ID is
+    #   not valid
+    #
+    def wf_monitoredapplication_id?(id)
+      return true if id.is_a?(String) && id.size < 256 && id =~ /^[a-z0-9\-_]+$/
+
+      raise Wavefront::Exception::InvalidMonitoredApplicationId, id
+    end
+
     # Ensure the given value is a valid sampling rate.
     # @param rate [Float]
     # @raise Wavefront::Exception::InvalidSamplingValue
@@ -654,6 +664,19 @@ module Wavefront
 
       raise Wavefront::Exception::InvalidMetricsPolicyId, id
     end
+
+    # Ensure the given argument is a valid Wavefront Span Sampling Policy ID.
+    # So far as I can tell they're just strings.
+    #
+    # @param id [String] the span sampling policy ID to validate
+    # @return true if the ID is valid
+    # @raise Wavefront::Exception::InvalidSpanSamplingPolicyId if the ID is
+    #   not valid
+    def wf_spansamplingpolicy_id?(id)
+      return true if id.is_a?(String)
+
+      raise Wavefront::Exception::InvalidSpanSamplingPolicyId, id
+    end
   end
   # rubocop:enable Metrics/ModuleLength
 end

data/lib/wavefront-sdk/webhook.rb

@@ -35,7 +35,7 @@ module Wavefront
       api.post('', body, 'application/json')
     end
 
-    # DELETE /api/v2/webhook/id
+    # DELETE /api/v2/webhook/{id}
     # Delete a specific webhook.
     #
     # @param id [String, Integer] ID of the webhook
@@ -46,7 +46,7 @@ module Wavefront
       api.delete(id)
     end
 
-    # GET /api/v2/webhook/id
+    # GET /api/v2/webhook/{id}
     # Get a specific webhook.
     #
     # @param id [String, Integer] ID of the webhook
@@ -57,7 +57,7 @@ module Wavefront
       api.get(id)
     end
 
-    # PUT /api/v2/webhook/id
+    # PUT /api/v2/webhook/{id}
     # Update a specific webhook.
     #
     # @param id [String] a Wavefront webhook ID

data/lib/wavefront-sdk/write.rb

@@ -88,6 +88,7 @@ module Wavefront
       writer.close
     end
 
+    # POST /report
     # A wrapper to the writer class's #write method.
     # Writers implement this method differently, Check the
     # appropriate class documentation for @return information etc.

data/lib/wavefront-sdk/writers/api.rb

@@ -6,10 +6,23 @@ require_relative '../core/api_caller'
 module Wavefront
   module Writer
     #
-    # Send points direct to Wavefront's API. This requires an
-    # endpoint, a token, and HTTPS egress.
+    # Send points direct to Wavefront's API. This requires an endpoint, a
+    # token, and HTTPS egress.
+    #
+    # Points are sent in batches of BATCH_SIZE. We attempt to make a summary
+    # of how many points are sent or rejected, but it's quantized by the batch
+    # size.
+    #
+    # TODO I think this needs a composite response. It makes one or more API
+    # calls depending on the amount of metrics to be sent, and the CLI needs
+    # to know if there was anything other than a 200. Options are to return
+    # the first failure when it happens, or to try all the chunks and return
+    # the last non-200, or the highest numbered return code, or some other
+    # indication of failure.
     #
     class Api < Core
+      BATCH_SIZE = 100
+
       def open
         @conn = Wavefront::ApiCaller.new(self, creds, opts)
       end
@@ -32,7 +45,6 @@ module Wavefront
 
       def send_point(body)
         _send_point(body)
-        summary.sent += body.size
         true
       rescue StandardError => e
         summary.unsent += body.size
@@ -52,15 +64,26 @@ module Wavefront
         send_point(body)
       end
 
-      # Send points in batches of a hundred. I'm not sure exactly
-      # how much the API can cope with in a single call, so this
-      # might change.
+      # Send points in batches of BATCH_SIZE. I'm not sure exactly how much the
+      # API can cope with in a single call, so this might change.
+      # @return [Nil]
       #
+      # rubocop:disable Metrics/MethodLength
       def _send_point(body)
-        body.each_slice(100) do |p|
-          conn.post('/?f=wavefront', p.join("\n"), 'application/octet-stream')
+        body.each_slice(BATCH_SIZE) do |p|
+          ret = conn.post('/?f=wavefront', p.join("\n"),
+                          'application/octet-stream')
+
+          if ret.ok?
+            logger.log("sent #{BATCH_SIZE} points", :debug)
+            summary.sent += p.count
+          else
+            logger.log("failed to send #{BATCH_SIZE} points", :debug)
+            summary.unsent += p.count
+          end
         end
       end
+      # rubocop:enable Metrics/MethodLength
     end
   end
 end

data/spec/spec_helper.rb

@@ -10,7 +10,7 @@ require_relative 'constants'
 
 # Abstract class which sets up everything needed by the API tests
 #
-class WavefrontTestBase < MiniTest::Test
+class WavefrontTestBase < Minitest::Test
   attr_reader :wf, :wf_noop, :headers, :invalid_id, :valid_id
 
   def initialize(args)

data/spec/wavefront-sdk/accesspolicy_spec.rb

@@ -0,0 +1,53 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require_relative '../spec_helper'
+require_relative '../test_mixins/general'
+
+# Unit tests for AccessPolicy class
+#
+class WavefrontAccessPolicyTest < WavefrontTestBase
+  def test_describe
+    assert_gets('/api/v2/accesspolicy') { wf.describe }
+  end
+
+  def test_update
+    assert_puts('/api/v2/accesspolicy', payload.to_json) do
+      wf.update(payload)
+    end
+  end
+
+  def test_validate
+    assert_gets('/api/v2/accesspolicy/validate?ip=1.2.3.4') do
+      wf.validate('1.2.3.4')
+    end
+  end
+
+  private
+
+  def api_class
+    'accesspolicy'
+  end
+
+  def payload
+    {
+      status: {
+        result: 'OK',
+        message: 'string',
+        code: 0
+      },
+      response: {
+        customer: 'string',
+        lastUpdatedMs: 0,
+        policyRules: [
+          {
+            name: 'string',
+            description: 'string',
+            subnet: 'string',
+            action: 'ALLOW'
+          }
+        ]
+      }
+    }
+  end
+end

data/spec/wavefront-sdk/account_spec.rb

@@ -179,6 +179,18 @@ class WavefrontAccountTest < WavefrontTestBase
     assert_raises(ArgumentError) { wf.validate_accounts }
   end
 
+  def test_admins
+    assert_gets('/api/v2/account/user/admin') { wf.admins }
+  end
+
+  def test_update_perms
+    assert_invalid_id { wf.update_perms(invalid_id, perms_payload) }
+
+    assert_puts("/api/v2/account/user/#{user_id}", perms_payload.to_json) do
+      wf.update_perms(user_id, perms_payload, false)
+    end
+  end
+
   private
 
   def api_class
@@ -203,6 +215,10 @@ class WavefrontAccountTest < WavefrontTestBase
        2659191e-aad4-4302-a94e-9667e1515678]
   end
 
+  def user_id
+    'user@example.com'
+  end
+
   def invalid_role
     %w[bad_role]
   end
@@ -212,7 +228,7 @@ class WavefrontAccountTest < WavefrontTestBase
   end
 
   def id_list
-    %w[sa:test user@example.com]
+    ['sa:test', user_id]
   end
 
   def invalid_permission
@@ -235,4 +251,19 @@ class WavefrontAccountTest < WavefrontTestBase
     { emailAddress: id,
       groups: %w[browse] }
   end
+
+  def perms_payload
+    {
+      identifier: 'user@example.com',
+      groups: [
+        'user_management'
+      ],
+      userGroups: [
+        '8b23136b-ecd2-4cb5-8c92-62477dcc4090'
+      ],
+      roles: [
+        'Role'
+      ]
+    }
+  end
 end

data/spec/wavefront-sdk/alert_spec.rb

@@ -35,6 +35,23 @@ class WavefrontAlertTest < WavefrontTestBase
     assert_invalid_id { wf.unsnooze(invalid_id) }
   end
 
+  def test_check_query
+    query = {
+      inputQuery: 'string',
+      translatedInput: 'sum(http_requests_total{method="GET"})',
+      queryType: 'PromQL'
+    }
+    assert_posts('/api/v2/alert/checkQuery', query.to_json) do
+      wf.check_query(query)
+    end
+  end
+
+  def test_preview
+    assert_posts('/api/v2/alert/preview', payload.to_json) do
+      wf.preview(payload)
+    end
+  end
+
   def test_summary
     assert_gets('/api/v2/alert/summary') { wf.summary }
   end

data/spec/wavefront-sdk/api_mixins/user_mixins_spec.rb

@@ -8,7 +8,7 @@ require_relative '../../../lib/wavefront-sdk/core/exception'
 
 # Test user mixins
 #
-class WavefrontUserMixinsTest < MiniTest::Test
+class WavefrontUserMixinsTest < Minitest::Test
   include Wavefront::Mixin::User
   include Wavefront::Validators
 

data/spec/wavefront-sdk/apitoken_spec.rb

@@ -10,6 +10,18 @@ class WavefrontApiTokenTest < WavefrontTestBase
     assert_gets('/api/v2/apitoken') { wf.list }
   end
 
+  def test_list_customer_tokens
+    assert_gets('/api/v2/apitoken/customertokens') { wf.list_customer_tokens }
+  end
+
+  def test_describe_customer_token
+    assert_invalid_id { wf.describe_customer_token(invalid_token_id) }
+
+    assert_gets("/api/v2/apitoken/customertokens/#{token_id}") do
+      wf.describe_customer_token(token_id)
+    end
+  end
+
   def test_create
     assert_posts('/api/v2/apitoken', 'null') { wf.create }
     assert_raises(ArgumentError) { wf.create('test') }

data/spec/wavefront-sdk/cluster_spec.rb

@@ -0,0 +1,13 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require_relative '../spec_helper'
+require_relative '../test_mixins/general'
+
+# Unit tests for Cluster class
+#
+class WavefrontClusterTest < WavefrontTestBase
+  def test_describe
+    assert_gets('/api/v2/cluster/info') { wf.describe }
+  end
+end

data/spec/wavefront-sdk/core/api_caller_spec.rb

@@ -19,7 +19,7 @@ end
 
 # Test Wavefront API caller
 #
-class WavefrontApiCallerTest < MiniTest::Test
+class WavefrontApiCallerTest < Minitest::Test
   attr_reader :wf, :wf_noop, :uri_base, :headers
 
   def setup

data/spec/wavefront-sdk/core/api_spec.rb

@@ -6,7 +6,7 @@ require_relative '../../../lib/wavefront-sdk/core/api'
 
 # Test SDK core API class
 #
-class WavefrontCoreApiTest < MiniTest::Test
+class WavefrontCoreApiTest < Minitest::Test
   attr_reader :wf
 
   def setup

data/spec/wavefront-sdk/core/logger_spec.rb

@@ -7,7 +7,7 @@ require_relative '../../../lib/wavefront-sdk/core/logger'
 
 # Test SDK logger class
 #
-class WavefrontLoggerTest < MiniTest::Test
+class WavefrontLoggerTest < Minitest::Test
   attr_reader :wfl
 
   def setup

data/spec/wavefront-sdk/core/response_spec.rb

@@ -20,7 +20,7 @@ ERR_RESP = { error: 'HTTP 415 Unsupported Media Type',
 # Unit tests for Response class. Also indirectly tests the Status
 # type.
 #
-class WavefrontResponseTest < MiniTest::Test
+class WavefrontResponseTest < Minitest::Test
   attr_reader :wfg, :wfb, :wfe
 
   def setup

data/spec/wavefront-sdk/credentials_spec.rb

@@ -10,7 +10,7 @@ CONF2 = RESOURCE_DIR.join('test2.conf')
 
 # Test SDK base class end-to-end
 #
-class WavefrontCredentialsTest < MiniTest::Test
+class WavefrontCredentialsTest < Minitest::Test
   def test_initialize_1
     ENV.delete('WAVEFRONT_ENDPOINT')
     ENV.delete('WAVEFRONT_TOKEN')
@@ -64,7 +64,7 @@ class Giblets < Wavefront::Credentials; end
 
 # And here are the tests
 #
-class GibletsTest < MiniTest::Test
+class GibletsTest < Minitest::Test
   attr_reader :wf, :raw
 
   def setup

data/spec/wavefront-sdk/distribution_spec.rb

@@ -7,7 +7,7 @@ require_relative '../../lib/wavefront-sdk/distribution'
 # Most of the distribution methods are inherited from the Write
 # class so they aren't tested again here.
 #
-class WavefrontDistributionTest < MiniTest::Test
+class WavefrontDistributionTest < Minitest::Test
   attr_reader :wf
 
   def setup

data/spec/wavefront-sdk/ingestionpolicy_spec.rb

@@ -11,6 +11,17 @@ class WavefrontIngestionPolicyTest < WavefrontTestBase
   include WavefrontTest::Describe
   include WavefrontTest::Update
   include WavefrontTest::Delete
+  include WavefrontTest::History
+
+  def test_revert
+    assert_posts("/api/v2/usage/ingestionpolicy/#{id}/revert/5", nil, :json) do
+      wf.revert(id, 5)
+    end
+
+    assert_raises(Wavefront::Exception::InvalidVersion) { wf.revert(id, 'v5') }
+
+    assert_invalid_id { wf.revert(invalid_id, 5) }
+  end
 
   private
 

data/spec/wavefront-sdk/integration_spec.rb

@@ -47,6 +47,10 @@ class WavefrontIntegrationTest < WavefrontTestBase
     assert_gets('/api/v2/integration/status') { wf.statuses }
   end
 
+  def test_manifests_min
+    assert_gets('/api/v2/integration/manifests/min') { wf.manifests_min }
+  end
+
   private
 
   def api_class