77e69b9cf3
Signed-off-by: Olivier Gambier <olivier@docker.com>
88 lines
1.6 KiB
Go
88 lines
1.6 KiB
Go
package bugsnag
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"log"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func TestMiddlewareOrder(t *testing.T) {
|
|
|
|
result := make([]int, 0, 7)
|
|
stack := middlewareStack{}
|
|
stack.OnBeforeNotify(func(e *Event, c *Configuration) error {
|
|
result = append(result, 2)
|
|
return nil
|
|
})
|
|
stack.OnBeforeNotify(func(e *Event, c *Configuration) error {
|
|
result = append(result, 1)
|
|
return nil
|
|
})
|
|
stack.OnBeforeNotify(func(e *Event, c *Configuration) error {
|
|
result = append(result, 0)
|
|
return nil
|
|
})
|
|
|
|
stack.Run(nil, nil, func() error {
|
|
result = append(result, 3)
|
|
return nil
|
|
})
|
|
|
|
if !reflect.DeepEqual(result, []int{0, 1, 2, 3}) {
|
|
t.Errorf("unexpected middleware order %v", result)
|
|
}
|
|
}
|
|
|
|
func TestBeforeNotifyReturnErr(t *testing.T) {
|
|
|
|
stack := middlewareStack{}
|
|
err := fmt.Errorf("test")
|
|
|
|
stack.OnBeforeNotify(func(e *Event, c *Configuration) error {
|
|
return err
|
|
})
|
|
|
|
called := false
|
|
|
|
e := stack.Run(nil, nil, func() error {
|
|
called = true
|
|
return nil
|
|
})
|
|
|
|
if e != err {
|
|
t.Errorf("Middleware didn't return the error")
|
|
}
|
|
|
|
if called == true {
|
|
t.Errorf("Notify was called when BeforeNotify returned False")
|
|
}
|
|
}
|
|
|
|
func TestBeforeNotifyPanic(t *testing.T) {
|
|
|
|
stack := middlewareStack{}
|
|
|
|
stack.OnBeforeNotify(func(e *Event, c *Configuration) error {
|
|
panic("oops")
|
|
})
|
|
|
|
called := false
|
|
b := &bytes.Buffer{}
|
|
|
|
stack.Run(nil, &Configuration{Logger: log.New(b, log.Prefix(), 0)}, func() error {
|
|
called = true
|
|
return nil
|
|
})
|
|
|
|
logged := b.String()
|
|
|
|
if logged != "bugsnag/middleware: unexpected panic: oops\n" {
|
|
t.Errorf("Logged: %s", logged)
|
|
}
|
|
|
|
if called == false {
|
|
t.Errorf("Notify was not called when BeforeNotify panicked")
|
|
}
|
|
}
|