From e803eeed62f80d2e5128155a510e4dc955a376e4 Mon Sep 17 00:00:00 2001
From: Matt Bostock <matt@mattbostock.com>
Date: Fri, 6 Mar 2015 16:15:30 +0000
Subject: [PATCH] Add integration test to Airbrake hook

Add a test for the Airbrake hook to:

a) document how the hook is intended to work

b) test that an XML payload is received with the expected message
---
 hooks/airbrake/airbrake_test.go | 57 +++++++++++++++++++++++++++++++++
 1 file changed, 57 insertions(+)
 create mode 100644 hooks/airbrake/airbrake_test.go

diff --git a/hooks/airbrake/airbrake_test.go b/hooks/airbrake/airbrake_test.go
new file mode 100644
index 0000000..d2fd61d
--- /dev/null
+++ b/hooks/airbrake/airbrake_test.go
@@ -0,0 +1,57 @@
+package logrus_airbrake
+
+import (
+	"encoding/xml"
+	"errors"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+	"time"
+
+	"github.com/Sirupsen/logrus"
+	"github.com/tobi/airbrake-go"
+)
+
+type notice struct {
+	Error struct {
+		Message string `xml:"message"`
+	} `xml:"error"`
+}
+
+func TestNoticeReceived(t *testing.T) {
+	msg := make(chan string, 1)
+	expectedMsg := "foo"
+
+	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		var notice notice
+		if err := xml.NewDecoder(r.Body).Decode(&notice); err != nil {
+			t.Error(err)
+		}
+		r.Body.Close()
+
+		msg <- notice.Error.Message
+	}))
+	defer ts.Close()
+
+	hook := &AirbrakeHook{}
+
+	airbrake.Environment = "production"
+	airbrake.Endpoint = ts.URL
+	airbrake.ApiKey = "foo"
+
+	log := logrus.New()
+	log.Hooks.Add(hook)
+
+	log.WithFields(logrus.Fields{
+		"error": errors.New(expectedMsg),
+	}).Error("Airbrake will not see this string")
+
+	select {
+	case received := <-msg:
+		if received != expectedMsg {
+			t.Errorf("Unexpected message received: %s", received)
+		}
+	case <-time.After(time.Second):
+		t.Error("Timed out; no notice received by Airbrake API")
+	}
+}