rollbar_test.go

package rollbar

import (
	"errors"
	"fmt"
	"net/http"
	"os"
	"testing"
)

type CustomError struct {
	s string
}

func (e *CustomError) Error() string {
	return e.s
}

func testErrorStack(s string) {
	testErrorStack2(s)
}

func testErrorStack2(s string) {
	Error("error", errors.New(s))
}

func testErrorStackWithSkip(s string) {
	testErrorStackWithSkip2(s)
}

func testErrorStackWithSkip2(s string) {
	ErrorWithStackSkip("error", errors.New(s), 2)
}

func TestErrorClass(t *testing.T) {
	errors := map[string]error{
		// generic error
		"errors.errorString":          fmt.Errorf("Something is broken!"),
		// custom error
		"rollbar.CustomError": &CustomError{"Terrible mistakes were made."},
	}

	for expected, err := range errors {
		if errorClass(err) != expected {
			t.Error("Got:", errorClass(err), "Expected:", expected)
		}
	}
}

func TestEverything(t *testing.T) {
	Token = os.Getenv("TOKEN")
	Environment = "test"

	Error("critical", errors.New("Normal critical error"))
	Error("error", &CustomError{"This is a custom error"})

	testErrorStack("This error should have a nice stacktrace")
	testErrorStackWithSkip("This error should have a skipped stacktrace")

	done := make(chan bool)
	go func() {
		testErrorStack("I'm in a goroutine")
		done <- true
	}()
	<-done

	Message("error", "This is an error message")
	Message("info", "And this is an info message")

	// If you don't see the message sent on line 65 in Rollbar, that means this
	// is broken:
	Wait()
}

func TestErrorRequest(t *testing.T) {
	r, _ := http.NewRequest("GET", "http://foo.com/somethere?param1=true", nil)
	r.RemoteAddr = "1.1.1.1:123"

	object := errorRequest(r)

	if object["url"] != "http://foo.com/somethere?param1=true" {
		t.Errorf("wrong url, got %v", object["url"])
	}

	if object["method"] != "GET" {
		t.Errorf("wrong method, got %v", object["method"])
	}

	if object["query_string"] != "param1=true" {
		t.Errorf("wrong id, got %v", object["query_string"])
	}
}

func TestFilterParams(t *testing.T) {
	values := map[string][]string{
		"password":     []string{"one"},
		"ok":           []string{"one"},
		"access_token": []string{"one"},
	}

	clean := filterParams(values)
	if clean["password"][0] != FILTERED {
		t.Error("should filter password parameter")
	}

	if clean["ok"][0] == FILTERED {
		t.Error("should keep ok parameter")
	}

	if clean["access_token"][0] != FILTERED {
		t.Error("should filter access_token parameter")
	}
}

func TestFlattenValues(t *testing.T) {
	values := map[string][]string{
		"a": []string{"one"},
		"b": []string{"one", "two"},
	}

	flattened := flattenValues(values)
	if flattened["a"].(string) != "one" {
		t.Error("should flatten single parameter to string")
	}

	if len(flattened["b"].([]string)) != 2 {
		t.Error("should leave multiple parametres as []string")
	}
}

func TestBuildError(t *testing.T) {
	buildError(ERR, nil, BuildStack(0))
	// this should not panic
}

func TestCustomField(t *testing.T) {
	body := buildError(ERR, errors.New("test-custom"), BuildStack(0), &Field{
		Name: "custom",
		Data: map[string]string{
			"NAME1": "VALUE1",
		},
	})

	dataField, ok := body["data"]
	if !ok {
		t.Error("should have field 'data'")
	}

	data, ok := dataField.(map[string]interface{})
	if !ok {
		t.Error("should be of type map[string]interface{}")
	}

	custom, ok := data["custom"]
	if !ok {
		t.Error("should have field 'custom'")
	}

	customMap, ok := custom.(map[string]string)
	if !ok {
		t.Error("should be a map[string]string")
	}

	val, ok := customMap["NAME1"]
	if !ok {
		t.Error("should have a key 'NAME1'")
	}

	if val != "VALUE1" {
		t.Error("should be VALUE1")
	}
}

func TestErrorRead(t *testing.T) {
	Token = os.Getenv("TOKEN")
	Environment = "test"

	bckBuffer, bckEP := Buffer, Endpoint
	defer func() {
		Buffer, Endpoint = bckBuffer, bckEP
	}()

	Buffer = 2
	Endpoint = "https://does.not.exsist/foo/bar"

	go func() {
		errCount := 0
		for err := range PostErrors() {
			t.Log(err)
			errCount++
		}
		if errCount != 2 {
			t.Fatal("didn't receive the right number of errors", errCount)
		}
	}()

	post(nil)
	post(nil)

	Wait()
}