serverspec-opentsdb 0.1.2 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: eb7d750ca51fdc6882af8898d1c61dce352d9dd1
-  data.tar.gz: 4a71312b90f2ea0da2240eb1745652352fb021bc
+  metadata.gz: 4e14aecc85327b60ff742690af0d5c6741ed8b57
+  data.tar.gz: bf15e8763fd0d42daf345ccbe5762d6839a6b661
 SHA512:
-  metadata.gz: a5de7f226e225a8fadf8b5a48f4c0fe77227b8d3a083c62d44e0e0db66f1cb30341520beb09e5666ee38db15ef45d5b4291d472f01be10adb95e6957b3613bd2
-  data.tar.gz: eb2df067423f0e909aa1f4d7dea37d2fd0d912a23c8dd249cedf8582a0f70c7fc1035d9e127306c71d3ddc8b6710a5251ab88623dc47455e4307fab0acfc6cdf
+  metadata.gz: ee0f32ff64f71b2aed419b4831e2c227370af3375a0a113833b3c206011593210bfabe95e84f2eee5cf9474586195fa50499e58fe289835b8b0f6b071e99967a
+  data.tar.gz: c6fd058611f5da321652b36625628b0dbc15c97f2b3863410712d85ef2111e182b05de016454508a5bc72e46351c6586f98b7dfc650492e51b62467b72e6066a

data/lib/serverspec/matcher/have_data_points.rb

@@ -0,0 +1,22 @@
+RSpec::Matchers.define :have_data_points do |metric|
+  match do |subject|
+    if subject.class.name == 'Serverspec::Type::OpenTSDBQuery'
+      subject.has_data_points?(metric, @start, @aggregator, @tags)
+    elsif subject.respond_to?(:has_data_points?)
+      subject.has_data_points?(metric)
+    end
+  end
+
+  # For opentsdb
+  chain :with_start do |start|
+    @start = start
+  end
+
+  chain :with_aggregator do |aggregator|
+    @aggregator = aggregator
+  end
+
+  chain :with_tags do |tags|
+    @tags = tags
+  end
+end

data/lib/serverspec/opentsdb.rb

@@ -1 +1,2 @@
+require_relative "matcher/have_data_points.rb"
 require_relative "type/opentsdb_query.rb"

data/lib/serverspec/type/opentsdb_query.rb

@@ -6,22 +6,29 @@ require "uri"
 
 module Serverspec
   module Type
-    class OpentsdbQuery < Base
-      def initialize(metric, aggregate, tags)
+    class OpenTSDBQuery < Base
+      def initialize(metric)
+        @metric = metric
         @host = "opentsdb"
         @port = 4242
         @start = "2m-ago"
-        @metric = metric
-        @aggregate = aggregate
-        @tags = tags
+        @aggregator = "sum"
+        @tags = {}
       end
 
       def response
         JSON.parse(Net::HTTP.get_response(generate_uri).body)
       end
 
-      def data_points
-        response.first['dps']
+      def has_data_points?(metric, start, aggregator, tags)
+        @start = start if start
+        @aggregator = aggregator if aggregator
+        @tags = tags if tags
+        response.first['dps'].any?
+      end
+
+      def to_s
+        "OpenTSDB Query"
       end
 
       private
@@ -31,14 +38,14 @@ module Serverspec
         uri = URI.parse("http://#{@host}:#{@port}/api/query")
         uri.query = URI.encode_www_form(
           "start" => @start,
-          "m" => "#{@aggregate}:#{@metric}{#{joined_tags}}"
+          "m" => "#{@aggregator}:#{@metric}{#{joined_tags}}"
         )
         uri
       end
     end
 
-    def opentsdb_query(metric, aggregate, tags)
-      OpentsdbQuery.new(metric, aggregate, tags)
+    def opentsdb_query(metric)
+      OpenTSDBQuery.new(metric)
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: serverspec-opentsdb
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.4
 platform: ruby
 authors:
 - Mike
@@ -31,6 +31,7 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- lib/serverspec/matcher/have_data_points.rb
 - lib/serverspec/opentsdb.rb
 - lib/serverspec/type/opentsdb_query.rb
 homepage: https://rubygems.org/gems/serverspec-opentsdb