Add assertions for histograms, sets, and key values

Shopify · May 7, 2014 · ae348dd · ae348dd
1 parent 5865dbb
commit ae348dd
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/lib/statsd/instrument/assertions.rb b/lib/statsd/instrument/assertions.rb
@@ -27,6 +27,18 @@ def assert_statsd_gauge(metric_name, options = {}, &block)
     assert_statsd_call(:g, metric_name, options, &block)
   end
 
+  def assert_statsd_histogram(metric_name, options = {}, &block)
+    assert_statsd_call(:h, metric_name, options, &block)
+  end
+
+  def assert_statsd_set(metric_name, options = {}, &block)
+    assert_statsd_call(:s, metric_name, options, &block)
+  end
+
+  def assert_statsd_key_value(metric_name, options = {}, &block)
+    assert_statsd_call(:kv, metric_name, options, &block)
+  end
+
   private
 
   def assert_statsd_call(metric_type, metric_name, options = {}, &block)