wavefront-sdk 3.6.1 → 5.0.1

data/spec/wavefront-sdk/metric_helper_spec.rb

@@ -118,7 +118,7 @@ class WavefrontMetricHelperTest < MiniTest::Test
     assert_equal(1, out.select do |o|
                       o[:value] == [[2, 10.0], [1, 11.0],
                                     [1, 12.0]]
-                    end .size)
+                    end.size)
     assert_equal(1, out.select { |o| o[:tags] == WH_TAGS }.size)
     assert_equal(3, out.select { |o| o[:path] == 'test.dist1' }.size)
   end

data/spec/wavefront-sdk/role_spec.rb

@@ -0,0 +1,96 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require_relative '../spec_helper'
+require_relative '../test_mixins/general'
+
+# Unit tests for Wavefront::Role
+#
+class WavefrontRoleTest < WavefrontTestBase
+  include WavefrontTest::Create
+  include WavefrontTest::Delete
+  include WavefrontTest::Describe
+  include WavefrontTest::List
+  include WavefrontTest::Update
+
+  def test_add_assignees
+    assert_posts("/api/v2/role/#{id}/addAssignees", assignees.to_json) do
+      wf.add_assignees(id, assignees)
+    end
+
+    assert_raises(Wavefront::Exception::InvalidRoleId) do
+      wf.add_assignees(invalid_id, assignees)
+    end
+  end
+
+  def test_remove_assignees
+    assert_posts("/api/v2/role/#{id}/removeAssignees", assignees.to_json) do
+      wf.remove_assignees(id, assignees)
+    end
+
+    assert_raises(Wavefront::Exception::InvalidRoleId) do
+      wf.remove_assignees(invalid_id, assignees)
+    end
+  end
+
+  def test_grant
+    assert_posts("/api/v2/role/grant/#{permission}", roles.to_json) do
+      wf.grant(permission, roles)
+    end
+
+    assert_raises(Wavefront::Exception::InvalidRoleId) do
+      wf.grant(permission, [invalid_id])
+    end
+
+    assert_raises(Wavefront::Exception::InvalidPermission) do
+      wf.grant('made_up_permission', roles)
+    end
+  end
+
+  def test_revoke
+    assert_posts("/api/v2/role/revoke/#{permission}", roles.to_json) do
+      wf.revoke(permission, roles)
+    end
+
+    assert_raises(Wavefront::Exception::InvalidRoleId) do
+      wf.revoke(permission, [invalid_id])
+    end
+
+    assert_raises(Wavefront::Exception::InvalidPermission) do
+      wf.revoke('made_up_permission', roles)
+    end
+  end
+
+  private
+
+  def api_class
+    :role
+  end
+
+  def id
+    'f8dc0c14-91a0-4ca9-8a2a-7d47f4db4672'
+  end
+
+  def roles
+    %w[f8dc0c14-91a0-4ca9-8a2a-7d47f4db4672
+       2659191e-aad4-4302-a94e-9667e1517127]
+  end
+
+  def assignees
+    roles.push('sa::test')
+  end
+
+  def invalid_id
+    '__BAD_ID__'
+  end
+
+  def payload
+    { name: 'test role',
+      permissions: %w[alerts_management events_management],
+      description: 'dummy role for unit tests' }
+  end
+
+  def permission
+    'alerts_management'
+  end
+end

data/spec/wavefront-sdk/spy_spec.rb

@@ -0,0 +1,113 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require_relative '../spec_helper'
+require_relative '../../lib/wavefront-sdk/spy'
+
+# Unit tests for Spy class
+#
+class WavefrontSpyTest < MiniTest::Test
+  attr_reader :wf
+
+  def setup
+    @wf = Wavefront::Spy.new(CREDS)
+  end
+
+  def test_points
+    capture_io do
+      assert_gets('/api/spy/points?sampling=0.01') { wf.points }
+      assert_gets('/api/spy/points?sampling=0.05') { wf.points(0.05) }
+
+      assert_gets('/api/spy/points?sampling=0.05&metric=my_prefix') do
+        wf.points(0.05, prefix: 'my_prefix')
+      end
+
+      assert_gets('/api/spy/points?sampling=0.05&metric=my_prefix&host=h1') do
+        wf.points(0.05, prefix: 'my_prefix', host: 'h1')
+      end
+
+      assert_gets('/api/spy/points?sampling=0.02&metric=my_prefix&' \
+                  'pointTagKey=mytag') do
+        wf.points(0.02, prefix: 'my_prefix', tag_key: 'mytag')
+      end
+
+      assert_gets('/api/spy/points?sampling=0.02&metric=my_prefix&' \
+                  'pointTagKey=tag1&pointTagKey=tag2') do
+        wf.points(0.02, prefix: 'my_prefix', tag_key: %w[tag1 tag2])
+      end
+    end
+  end
+
+  def test_histograms
+    capture_io do
+      assert_gets('/api/spy/histograms?sampling=0.01') { wf.histograms }
+      assert_gets('/api/spy/histograms?sampling=0.05') { wf.histograms(0.05) }
+
+      assert_gets('/api/spy/histograms?sampling=0.05&histogram=my_prefix') do
+        wf.histograms(0.05, prefix: 'my_prefix')
+      end
+
+      assert_gets(
+        '/api/spy/histograms?sampling=0.05&histogram=my_prefix&host=h1'
+      ) do
+        wf.histograms(0.05, prefix: 'my_prefix', host: 'h1')
+      end
+
+      assert_gets('/api/spy/histograms?sampling=0.02&histogram=my_prefix&' \
+                  'histogramTagKey=the_tag') do
+        wf.histograms(0.02, prefix: 'my_prefix', tag_key: 'the_tag')
+      end
+
+      assert_gets('/api/spy/histograms?sampling=0.02&histogram=my_prefix&' \
+                  'histogramTagKey=tag1&histogramTagKey=tag2') do
+        wf.histograms(0.02, prefix: 'my_prefix', tag_key: %w[tag1 tag2])
+      end
+    end
+  end
+
+  def test_spans
+    capture_io do
+      assert_gets('/api/spy/spans?sampling=0.01') { wf.spans }
+      assert_gets('/api/spy/spans?sampling=0.05') { wf.spans(0.05) }
+
+      assert_gets('/api/spy/spans?sampling=0.05&name=my_prefix') do
+        wf.spans(0.05, prefix: 'my_prefix')
+      end
+
+      assert_gets(
+        '/api/spy/spans?sampling=0.05&name=my_prefix&host=h1'
+      ) do
+        wf.spans(0.05, prefix: 'my_prefix', host: 'h1')
+      end
+
+      assert_gets('/api/spy/spans?sampling=0.02&name=my_prefix&' \
+                  'spanTagKey=the_tag') do
+        wf.spans(0.02, prefix: 'my_prefix', tag_key: 'the_tag')
+      end
+
+      assert_gets('/api/spy/spans?sampling=0.02&name=my_prefix&' \
+                  'spanTagKey=tag1&spanTagKey=tag2') do
+        wf.spans(0.02, prefix: 'my_prefix', tag_key: %w[tag1 tag2])
+      end
+    end
+  end
+
+  def test_ids
+    capture_io do
+      assert_gets('/api/spy/ids?sampling=0.01') { wf.ids }
+      assert_gets('/api/spy/ids?sampling=0.05') { wf.ids(0.05) }
+
+      assert_gets('/api/spy/ids?sampling=0.05&type=METRIC') do
+        wf.ids(0.05, type: 'METRIC')
+      end
+
+      assert_gets('/api/spy/ids?sampling=0.05&type=SPAN&name=my_prefix') do
+        wf.ids(0.05, type: 'SPAN', prefix: 'my_prefix')
+      end
+    end
+  end
+
+  def dummy_response
+    DUMMY_RESPONSE
+  end
+end

data/spec/wavefront-sdk/unstable/chart_spec.rb

@@ -0,0 +1,39 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require_relative '../../spec_helper'
+require_relative '../../../lib/wavefront-sdk/unstable/chart'
+
+# Unit tests for Chart class
+#
+class WavefrontChartTest < MiniTest::Test
+  attr_reader :wf
+
+  def setup
+    @wf = Wavefront::Unstable::Chart.new(CREDS)
+  end
+
+  def test_all_metrics
+    assert_gets('/chart/metrics/all?l=100&q=&trie=true') do
+      wf.metrics_under('')
+    end
+  end
+
+  def test_metrics_under
+    assert_gets('/chart/metrics/all?l=100&q=test.path&trie=true') do
+      wf.metrics_under('test.path')
+    end
+
+    assert_gets('/chart/metrics/all?l=10&q=test.path&trie=true') do
+      wf.metrics_under('test.path', nil, 10)
+    end
+
+    assert_gets('/chart/metrics/all?l=100&p=last.one&q=test.path&trie=true') do
+      wf.metrics_under('test.path', 'last.one')
+    end
+  end
+
+  def dummy_response
+    { metrics: ['test data'] }.to_json
+  end
+end

data/spec/wavefront-sdk/usage_spec.rb

@@ -0,0 +1,33 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require_relative '../spec_helper'
+require_relative '../test_mixins/general'
+
+# Unit tests for Usage class
+#
+class WavefrontUsageTest < WavefrontTestBase
+  def test_export_csv
+    assert_raises(ArgumentError) { wf.export_csv }
+
+    assert_gets("/api/v2/usage/exportcsv?startTime=#{t_start}") do
+      wf.export_csv(t_start)
+    end
+
+    assert_gets(
+      "/api/v2/usage/exportcsv?startTime=#{t_start}&endTime=#{t_end}"
+    ) do
+      wf.export_csv(t_start, t_end)
+    end
+  end
+
+  private
+
+  def t_start
+    1_577_890_000
+  end
+
+  def t_end
+    1_577_899_999
+  end
+end

data/spec/wavefront-sdk/user_spec.rb

@@ -1,6 +1,7 @@
 #!/usr/bin/env ruby
 # frozen_string_literal: true
 
+require 'logger'
 require_relative '../spec_helper'
 require_relative '../test_mixins/general'
 
@@ -13,6 +14,13 @@ class WavefrontUserTest < WavefrontTestBase
   include WavefrontTest::Describe
   include WavefrontTest::Update
 
+  # Override the parent constructor so we can suppress all the 'deprecated'
+  # log messages
+  #
+  def setup
+    @wf = Wavefront::User.new(CREDS, logger: Logger.new('/dev/null'))
+  end
+
   def test_list
     assert_gets('/api/v2/user') { wf.list }
   end
@@ -110,6 +118,14 @@ class WavefrontUserTest < WavefrontTestBase
     assert_raises(ArgumentError) { wf.business_functions }
   end
 
+  def test_validate_users
+    assert_posts('/api/v2/user/validateUsers', id_list.to_json) do
+      wf.validate_users(id_list)
+    end
+
+    assert_raises(ArgumentError) { wf.validate_users }
+  end
+
   def test_response_shim
     (RESOURCE_DIR + 'user_responses').each_child do |input|
       # Ugly hack for the 202 in the 'create' file
@@ -156,4 +172,8 @@ class WavefrontUserTest < WavefrontTestBase
   def payload
     { emailAddress: id, groups: %w[browse] }
   end
+
+  def id_list
+    %w[id1 id2 id3]
+  end
 end

data/spec/wavefront-sdk/usergroup_spec.rb

@@ -7,7 +7,8 @@ require_relative '../test_mixins/general'
 # Unit tests for WavefrontUserGroup
 #
 class WavefrontUserGroupTest < WavefrontTestBase
-  attr_reader :users, :groups, :permission, :invalid_groups, :invalid_users
+  attr_reader :users, :groups, :permission, :invalid_groups, :invalid_users,
+              :roles, :invalid_roles
 
   include WavefrontTest::Create
   include WavefrontTest::Delete
@@ -39,22 +40,28 @@ class WavefrontUserGroupTest < WavefrontTestBase
     assert_invalid_id { wf.remove_users_from_group(invalid_id, users) }
   end
 
-  def test_grant
-    assert_posts("/api/v2/usergroup/grant/#{permission}",
-                 groups.to_json) do
-      wf.grant(permission, groups)
+  def test_add_roles_to_group
+    assert_posts("/api/v2/usergroup/#{id}/addRoles", roles.to_json) do
+      wf.add_roles_to_group(id, roles)
     end
 
-    assert_invalid_id { wf.grant(permission, invalid_groups) }
+    assert_raises(Wavefront::Exception::InvalidRoleId) do
+      wf.add_roles_to_group(id, invalid_roles)
+    end
+
+    assert_invalid_id { wf.add_roles_to_group(invalid_id, roles) }
   end
 
-  def test_revoke
-    assert_posts("/api/v2/usergroup/revoke/#{permission}",
-                 groups.to_json) do
-      wf.revoke(permission, groups)
+  def test_remove_roles_from_group
+    assert_posts("/api/v2/usergroup/#{id}/removeRoles", roles.to_json) do
+      wf.remove_roles_from_group(id, roles)
+    end
+
+    assert_raises(Wavefront::Exception::InvalidRoleId) do
+      wf.remove_roles_from_group(id, invalid_roles)
     end
 
-    assert_invalid_id { wf.revoke(permission, invalid_groups) }
+    assert_invalid_id { wf.remove_roles_from_group(invalid_id, roles) }
   end
 
   def setup_fixtures
@@ -63,7 +70,10 @@ class WavefrontUserGroupTest < WavefrontTestBase
     @groups = %w[f8dc0c14-91a0-4ca9-8a2a-7d47f4db4672
                  2659191e-aad4-4302-a94e-9667e1517127]
     @users = %w[someone@somewhere.com other@elsewhere.net]
+    @roles = %w[abcdef14-91a0-4ca9-8a2a-7d47f4db4672
+                fedcba1e-aad4-4302-a94e-9667e1517127]
     @invalid_users = ['bad' * 500, '']
+    @invalid_roles = ['some nonsense']
   end
 
   private

data/spec/wavefront-sdk/validators_spec.rb

@@ -323,19 +323,19 @@ class WavefrontValidatorsTest < MiniTest::Test
     end
   end
 
-  def test_notificant_id
+  def test_wf_notificant_id
     good = %w[CHTo47HvsPzSaGhh]
     bad = ['CTo47HvsPzSaGhh', [], {}, nil, 'bad id']
     good_and_bad('wf_notificant_id?', 'InvalidNotificantId', good, bad)
   end
 
-  def test_integration_id
+  def test_wf_integration_id
     good = %w[aws tutorial elasticsearch cassandra go]
     bad = ['CTo47HvsPzSaGhh', [], {}, nil, 'bad id']
     good_and_bad('wf_integration_id?', 'InvalidIntegrationId', good, bad)
   end
 
-  def test_apitoken_id
+  def test_wf_apitoken_id
     good = %w[2bfdcac7-1c9c-4c4b-9b56-c41c22c586dc
               17db4cc1-65f6-40a8-a1fa-6fcae460c4bd
               fca312fb-5ff4-420d-862d-5d6d99ed6bcf
@@ -345,24 +345,70 @@ class WavefrontValidatorsTest < MiniTest::Test
     good_and_bad('wf_apitoken_id?', 'InvalidApiTokenId', good, bad)
   end
 
-  def test_distribution_interval
+  def test_wf_distribution_interval
     good = %i[m h d]
     bad = ['m', [], {}, nil, 'bad id', :x, 'p']
     good_and_bad('wf_distribution_interval?',
                  'InvalidDistributionInterval', good, bad)
   end
 
-  def test_serviceaccount_id
+  def test_wf_serviceaccount_id
     good = %w[sa::my-id sa::ID]
     bad = %w[sc:id fca312fb-5ff4-420d-862d-5d6d99ed6bcf]
     good_and_bad('wf_serviceaccount_id?',
                  'InvalidServiceAccountId', good, bad)
   end
 
-  def test_permission
+  def test_wf_permission
     good = %w[events_management external_links_management]
     bad = ['events management', 'event_management', 'EVENT_MANAGEMENT']
     good_and_bad('wf_permission?',
                  'InvalidPermission', good, bad)
   end
+
+  def test_wf_ingestionpolicy_id
+    good = %w[another-ingestion-policy-1579538401492
+              testpolicy-1579537565010
+              213452-34-_-0-4-ingestion-policy-1579538556267
+              yet_another-ingestion-policy-1579538414190]
+
+    bad = %w[fa312fb-5ff4-420d-862d-5d6d99ed6bcf thing 123]
+    good_and_bad('wf_ingestionpolicy_id?',
+                 'InvalidIngestionPolicyId',
+                 good,
+                 bad)
+  end
+
+  def test_wf_account_id
+    good = %w[sa::my-id sa::ID Some.User@example.com
+              general99+specific@somewhere.net someone@somewhere.com a user
+              user-name]
+    bad = ['', [], {}, 'a' * 1000]
+    good_and_bad('wf_account_id?', 'InvalidAccountId', good, bad)
+  end
+
+  def test_wf_managedcluster_id
+    good = %w[test-cluster other-cluster cluster]
+    bad = ['', [], {}, 'a' * 1000, '£"^WR"!']
+    good_and_bad('wf_monitoredcluster_id?',
+                 'InvalidMonitoredClusterId',
+                 good,
+                 bad)
+  end
+
+  def test_wf_sampling_value
+    good = [0, 0.01, 0.003, 0.05]
+    bad = ['a', 0.1, 0.99, 1, -1, 1.1]
+    good_and_bad('wf_sampling_value?', 'InvalidSamplingValue', good, bad)
+  end
+
+  def test_wf_role_id
+    good = %w[2bfdcac7-1c9c-4c4b-9b56-c41c22c586dc
+              17db4cc1-65f6-40a8-a1fa-6fcae460c4bd
+              fca312fb-5ff4-420d-862d-5d6d99ed6bcf
+              3a1957e0-459e-49e5-9209-3888a4e8ac5b]
+
+    bad = %w[fa312fb-5ff4-420d-862d-5d6d99ed6bcf thing 123]
+    good_and_bad('wf_role_id?', 'InvalidRoleId', good, bad)
+  end
 end