metrics_test.go

package goa_test

import (
	"time"

	"github.com/goadesign/goa"

	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"

	"github.com/armon/go-metrics"
)

// mock metrics
type mockSink struct{}

func (m *mockSink) SetGauge(key []string, val float32)         {}
func (m *mockSink) EmitKey(key []string, val float32)          {}
func (m *mockSink) IncrCounter(key []string, val float32)      {}
func (m *mockSink) AddSample(key []string, val float32)        {}
func (m *mockSink) MeasureSince(key []string, start time.Time) {}

var _ = Describe("Metrics", func() {
	var keys = [6]string{}
	var metriks *metrics.Metrics
	var sink *mockSink

	BeforeEach(func() {
		sink = &mockSink{}

		var err error
		metriks, err = metrics.New(metrics.DefaultConfig("UnitTest Service"), sink)

		if err != nil {
			panic("Unable to create test instance of metrics")
		}

		keys = [6]string{
			"foo_bar_*/*",
			"foo_*_baz",
			"foo/baz",
			"foo/bar/baz",
			"foo/bar*_*/*",
			"//foo/bar*",
		}
	})

	Describe("Add sample", func() {
		Context("With invalid characters in key", func() {
			It("should replace invalid characters with normalized characters", func() {
				goa.SetMetrics(metriks)
				goa.AddSample(keys[:], 3.14)
				Ω(keys).Should(ConsistOf([]string{
					"foo_bar_all",
					"foo___baz",
					"foo_baz",
					"foo_bar_baz",
					"foo_bar__all",
					"__foo_bar_",
				}))
			})
		})
	})

	Describe("Emit key", func() {
		Context("With invalid characters in key", func() {
			It("should replace invalid characters with normalized characters", func() {
				goa.SetMetrics(metriks)
				goa.EmitKey(keys[:], 3.14)
				Ω(keys).Should(ConsistOf([]string{
					"foo_bar_all",
					"foo___baz",
					"foo_baz",
					"foo_bar_baz",
					"foo_bar__all",
					"__foo_bar_",
				}))
			})
		})
	})

	Describe("Increment Counter", func() {
		Context("With invalid characters in key", func() {
			It("should replace invalid characters with normalized characters", func() {
				goa.SetMetrics(metriks)
				goa.IncrCounter(keys[:], 3.14)
				Ω(keys).Should(ConsistOf([]string{
					"foo_bar_all",
					"foo___baz",
					"foo_baz",
					"foo_bar_baz",
					"foo_bar__all",
					"__foo_bar_",
				}))
			})
		})
	})

	Describe("Measure since", func() {
		Context("With invalid characters in key", func() {
			It("should replace invalid characters with normalized characters", func() {
				goa.SetMetrics(metriks)
				goa.MeasureSince(keys[:], time.Time{})
				Ω(keys).Should(ConsistOf([]string{
					"foo_bar_all",
					"foo___baz",
					"foo_baz",
					"foo_bar_baz",
					"foo_bar__all",
					"__foo_bar_",
				}))
			})
		})
	})

	Describe("Set gauge", func() {
		Context("With invalid characters in key", func() {
			It("should replace invalid characters with normalized characters", func() {
				goa.SetMetrics(metriks)
				goa.SetGauge(keys[:], 3.14)
				Ω(keys).Should(ConsistOf([]string{
					"foo_bar_all",
					"foo___baz",
					"foo_baz",
					"foo_bar_baz",
					"foo_bar__all",
					"__foo_bar_",
				}))
			})
		})
	})
})