telegraf/plugins/inputs/webhooks/particle/particle_webhooks_test.go

package particle

import (
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/influxdata/telegraf/testutil"
)

func postWebhooks(t *testing.T, rb *Webhook, eventBody string) *httptest.ResponseRecorder {
	req, err := http.NewRequest("POST", "/", strings.NewReader(eventBody))
	require.NoError(t, err)
	w := httptest.NewRecorder()
	w.Code = 500

	rb.eventHandler(w, req)

	return w
}

func TestNewItem(t *testing.T) {
	t.Parallel()
	var acc testutil.Accumulator
	rb := &Webhook{Path: "/particle", acc: &acc}
	resp := postWebhooks(t, rb, newItemJSON())
	if resp.Code != http.StatusOK {
		t.Errorf("POST new_item returned HTTP status code %v.\nExpected %v", resp.Code, http.StatusOK)
	}

	fields := map[string]interface{}{
		"temp_c":    26.680000,
		"temp_f":    80.024001,
		"infrared":  528.0,
		"lux":       0.0,
		"humidity":  44.937500,
		"pressure":  998.998901,
		"altitude":  119.331436,
		"broadband": 1266.0,
	}

	tags := map[string]string{
		"id":       "230035001147343438323536",
		"location": "TravelingWilbury",
	}

	acc.AssertContainsTaggedFields(t, "mydata", fields, tags)
}

func TestUnknowItem(t *testing.T) {
	t.Parallel()
	var acc testutil.Accumulator
	rb := &Webhook{Path: "/particle", acc: &acc}
	resp := postWebhooks(t, rb, unknownJSON())
	if resp.Code != http.StatusOK {
		t.Errorf("POST unknown returned HTTP status code %v.\nExpected %v", resp.Code, http.StatusOK)
	}
}

func TestDefaultMeasurementName(t *testing.T) {
	t.Parallel()
	var acc testutil.Accumulator
	rb := &Webhook{Path: "/particle", acc: &acc}
	resp := postWebhooks(t, rb, blankMeasurementJSON())
	if resp.Code != http.StatusOK {
		t.Errorf("POST new_item returned HTTP status code %v.\nExpected %v", resp.Code, http.StatusOK)
	}

	fields := map[string]interface{}{
		"temp_c": 26.680000,
	}

	tags := map[string]string{
		"id": "230035001147343438323536",
	}

	acc.AssertContainsTaggedFields(t, "eventName", fields, tags)
}

func blankMeasurementJSON() string {
	return `
	{
	  "event": "eventName",
	  "data": {
		  "tags": {
			  "id": "230035001147343438323536"
		  },
		  "values": {
			  "temp_c": 26.680000
		  }
	  },
	  "ttl": 60,
	  "published_at": "2017-09-28T21:54:10.897Z",
	  "coreid": "123456789938323536",
	  "userid": "1234ee123ac8e5ec1231a123d",
	  "version": 10,
	  "public": false,
	  "productID": 1234,
	  "name": "sensor",
	  "measurement": ""
  }`
}

func newItemJSON() string {
	return `
	{
	  "event": "temperature",
	  "data": {
		  "tags": {
			  "id": "230035001147343438323536",
			  "location": "TravelingWilbury"
		  },
		  "values": {
			  "temp_c": 26.680000,
			  "temp_f": 80.024001,
			  "humidity": 44.937500,
			  "pressure": 998.998901,
			  "altitude": 119.331436,
			  "broadband": 1266.0,
			  "infrared": 528.0,
			  "lux": 0.0
		  }
	  },
	  "ttl": 60,
	  "published_at": "2017-09-28T21:54:10.897Z",
	  "coreid": "123456789938323536",
	  "userid": "1234ee123ac8e5ec1231a123d",
	  "version": 10,
	  "public": false,
	  "productID": 1234,
	  "name": "sensor",
	  "measurement": "mydata"
  }`
}

func unknownJSON() string {
	return `
    {
      "event": "roger"
    }`
}