Prometheus client test refactor

closes influxdata#318

CHANGELOG.md

@@ -8,6 +8,7 @@ changed to just run docker commands in the Makefile. See `make docker-run` and
 
 ### Features
 - [#325](https://github.com/influxdb/telegraf/pull/325): NSQ output. Thanks @jrxFive!
+- [#318](https://github.com/influxdb/telegraf/pull/318): Prometheus output. Thanks @oldmantaiter!
 
 ### Bugfixes
 

README.md

@@ -221,6 +221,7 @@ found by running `telegraf -sample-config`.
 * amqp (rabbitmq)
 * mqtt
 * librato
+* prometheus
 * riemann
 
 ## Contributing

outputs/prometheus_client/prometheus_client.go

@@ -2,15 +2,15 @@ package prometheus_client
 
 import (
 	"fmt"
+	"net/http"
+
 	"github.com/influxdb/influxdb/client/v2"
 	"github.com/influxdb/telegraf/outputs"
 	"github.com/prometheus/client_golang/prometheus"
-	"net/http"
 )
 
 type PrometheusClient struct {
 	Listen  string
-	server  *http.Server
 	metrics map[string]*prometheus.UntypedVec
 }
 
@@ -21,16 +21,16 @@ var sampleConfig = `
 
 func (p *PrometheusClient) Start() error {
 	if p.Listen == "" {
-		p.Listen = ":9126"
+		p.Listen = "localhost:9126"
 	}
+
 	http.Handle("/metrics", prometheus.Handler())
 	server := &http.Server{
-		Addr:    p.Listen,
-		Handler: prometheus.Handler(),
+		Addr: p.Listen,
 	}
-	p.server = server
+
 	p.metrics = make(map[string]*prometheus.UntypedVec)
-	go p.server.ListenAndServe()
+	go server.ListenAndServe()
 	return nil
 }
 

outputs/prometheus_client/prometheus_client_test.go

@@ -1,26 +1,21 @@
 package prometheus_client
 
 import (
+	"testing"
+
 	"github.com/influxdb/influxdb/client/v2"
 	"github.com/influxdb/telegraf/plugins/prometheus"
 	"github.com/influxdb/telegraf/testutil"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
-	"testing"
 )
 
-var pTesting *PrometheusClient = &PrometheusClient{}
-
-func TestPrometheusStart(t *testing.T) {
-	require.NoError(t, pTesting.Start())
-}
+var pTesting *PrometheusClient
 
 func TestPrometheusWritePointEmptyTag(t *testing.T) {
-	if testing.Short() {
-		t.Skip("Skipping integration test in short mode")
-	}
+
 	p := &prometheus.Prometheus{
-		Urls: []string{"http://" + testutil.GetLocalHost() + ":9126/metrics"},
+		Urls: []string{"http://localhost:9126/metrics"},
 	}
 	tags := make(map[string]string)
 	var points = []*client.Point{
@@ -53,11 +48,9 @@ func TestPrometheusWritePointEmptyTag(t *testing.T) {
 }
 
 func TestPrometheusWritePointTag(t *testing.T) {
-	if testing.Short() {
-		t.Skip("Skipping integration test in short mode")
-	}
+
 	p := &prometheus.Prometheus{
-		Urls: []string{"http://" + testutil.GetLocalHost() + ":9126/metrics"},
+		Urls: []string{"http://localhost:9126/metrics"},
 	}
 	tags := make(map[string]string)
 	tags["testtag"] = "testvalue"
@@ -88,3 +81,8 @@ func TestPrometheusWritePointTag(t *testing.T) {
 		assert.True(t, acc.CheckTaggedValue(e.name, e.value, tags))
 	}
 }
+
+func init() {
+	pTesting = &PrometheusClient{Listen: "localhost:9126"}
+	pTesting.Start()
+}