vendor/github.com/go-kit/kit/metrics/statsd/statsd_test.go

   1 package statsd
   2
   3 import (
   4         "testing"
   5
   6         "github.com/go-kit/kit/log"
   7         "github.com/go-kit/kit/metrics/teststat"
   8 )
   9
  10 func TestCounter(t *testing.T) {
  11         prefix, name := "abc.", "def"
  12         label, value := "label", "value" // ignored
  13         regex := `^` + prefix + name + `:([0-9\.]+)\|c$`
  14         s := New(prefix, log.NewNopLogger())
  15         counter := s.NewCounter(name, 1.0).With(label, value)
  16         valuef := teststat.SumLines(s, regex)
  17         if err := teststat.TestCounter(counter, valuef); err != nil {
  18                 t.Fatal(err)
  19         }
  20 }
  21
  22 func TestCounterSampled(t *testing.T) {
  23         // This will involve multiplying the observed sum by the inverse of the
  24         // sample rate and checking against the expected value within some
  25         // tolerance.
  26         t.Skip("TODO")
  27 }
  28
  29 func TestGauge(t *testing.T) {
  30         prefix, name := "ghi.", "jkl"
  31         label, value := "xyz", "abc" // ignored
  32         regex := `^` + prefix + name + `:([0-9\.]+)\|g$`
  33         s := New(prefix, log.NewNopLogger())
  34         gauge := s.NewGauge(name).With(label, value)
  35         valuef := teststat.LastLine(s, regex)
  36         if err := teststat.TestGauge(gauge, valuef); err != nil {
  37                 t.Fatal(err)
  38         }
  39 }
  40
  41 // StatsD timings just emit all observations. So, we collect them into a generic
  42 // histogram, and run the statistics test on that.
  43
  44 func TestTiming(t *testing.T) {
  45         prefix, name := "statsd.", "timing_test"
  46         label, value := "abc", "def" // ignored
  47         regex := `^` + prefix + name + `:([0-9\.]+)\|ms$`
  48         s := New(prefix, log.NewNopLogger())
  49         timing := s.NewTiming(name, 1.0).With(label, value)
  50         quantiles := teststat.Quantiles(s, regex, 50) // no |@0.X
  51         if err := teststat.TestHistogram(timing, quantiles, 0.01); err != nil {
  52                 t.Fatal(err)
  53         }
  54 }
  55
  56 func TestTimingSampled(t *testing.T) {
  57         prefix, name := "statsd.", "sampled_timing_test"
  58         label, value := "foo", "bar" // ignored
  59         regex := `^` + prefix + name + `:([0-9\.]+)\|ms\|@0\.01[0]*$`
  60         s := New(prefix, log.NewNopLogger())
  61         timing := s.NewTiming(name, 0.01).With(label, value)
  62         quantiles := teststat.Quantiles(s, regex, 50)
  63         if err := teststat.TestHistogram(timing, quantiles, 0.02); err != nil {
  64                 t.Fatal(err)
  65         }
  66 }