authboss/callbacks_test.go

package authboss

import (
	"bytes"
	"errors"
	"strings"
	"testing"
)

func TestCallbacks(t *testing.T) {
	afterCalled := false
	beforeCalled := false
	c := NewCallbacks()

	c.Before(EventRegister, func(ctx *Context) (Interrupt, error) {
		beforeCalled = true
		return InterruptNone, nil
	})
	c.After(EventRegister, func(ctx *Context) error {
		afterCalled = true
		return nil
	})

	if beforeCalled || afterCalled {
		t.Error("Neither should be called.")
	}

	interrupt, err := c.FireBefore(EventRegister, NewContext())
	if err != nil {
		t.Error("Unexpected error:", err)
	}
	if interrupt != InterruptNone {
		t.Error("It should not have been stopped.")
	}

	if !beforeCalled {
		t.Error("Expected before to have been called.")
	}
	if afterCalled {
		t.Error("Expected after not to be called.")
	}

	c.FireAfter(EventRegister, NewContext())
	if !afterCalled {
		t.Error("Expected after to be called.")
	}
}

func TestCallbacksInterrupt(t *testing.T) {
	before1 := false
	before2 := false
	c := NewCallbacks()

	c.Before(EventRegister, func(ctx *Context) (Interrupt, error) {
		before1 = true
		return InterruptAccountLocked, nil
	})
	c.Before(EventRegister, func(ctx *Context) (Interrupt, error) {
		before2 = true
		return InterruptNone, nil
	})

	interrupt, err := c.FireBefore(EventRegister, NewContext())
	if err != nil {
		t.Error(err)
	}
	if interrupt != InterruptAccountLocked {
		t.Error("The interrupt signal was not account locked:", interrupt)
	}

	if !before1 {
		t.Error("Before1 should have been called.")
	}
	if before2 {
		t.Error("Before2 should not have been called.")
	}
}

func TestCallbacksBeforeErrors(t *testing.T) {
	log := &bytes.Buffer{}
	Cfg = &Config{
		LogWriter: log,
	}
	before1 := false
	before2 := false
	c := NewCallbacks()

	errValue := errors.New("Problem occured")

	c.Before(EventRegister, func(ctx *Context) (Interrupt, error) {
		before1 = true
		return InterruptNone, errValue
	})
	c.Before(EventRegister, func(ctx *Context) (Interrupt, error) {
		before2 = true
		return InterruptNone, nil
	})

	interrupt, err := c.FireBefore(EventRegister, NewContext())
	if err != errValue {
		t.Error("Expected an error to come back.")
	}
	if interrupt != InterruptNone {
		t.Error("It should not have been stopped.")
	}

	if !before1 {
		t.Error("Before1 should have been called.")
	}
	if before2 {
		t.Error("Before2 should not have been called.")
	}

	if estr := log.String(); !strings.Contains(estr, errValue.Error()) {
		t.Error("Error string wrong:", estr)
	}
}

func TestCallbacksAfterErrors(t *testing.T) {
	log := &bytes.Buffer{}
	Cfg = &Config{
		LogWriter: log,
	}
	after1 := false
	after2 := false
	c := NewCallbacks()

	errValue := errors.New("Problem occured")

	c.After(EventRegister, func(ctx *Context) error {
		after1 = true
		return errValue
	})
	c.After(EventRegister, func(ctx *Context) error {
		after2 = true
		return nil
	})

	err := c.FireAfter(EventRegister, NewContext())
	if err != errValue {
		t.Error("Expected an error to come back.")
	}

	if !after1 {
		t.Error("After1 should have been called.")
	}
	if after2 {
		t.Error("After2 should not have been called.")
	}

	if estr := log.String(); !strings.Contains(estr, errValue.Error()) {
		t.Error("Error string wrong:", estr)
	}
}

func TestEventString(t *testing.T) {
	tests := []struct {
		ev  Event
		str string
	}{
		{EventRegister, "EventRegister"},
		{EventAuth, "EventAuth"},
		{EventAuthFail, "EventAuthFail"},
		{EventRecoverStart, "EventRecoverStart"},
		{EventRecoverEnd, "EventRecoverEnd"},
		{EventGet, "EventGet"},
		{EventGetUserSession, "EventGetUserSession"},
	}

	for i, test := range tests {
		if got := test.ev.String(); got != test.str {
			t.Errorf("%d) Wrong string for Event(%d) expected: %v got: %s", i, test.ev, test.str, got)
		}
	}
}

func TestInterruptString(t *testing.T) {
	tests := []struct {
		in  Interrupt
		str string
	}{
		{InterruptNone, "InterruptNone"},
		{InterruptAccountLocked, "InterruptAccountLocked"},
		{InterruptAccountNotConfirmed, "InterruptAccountNotConfirmed"},
		{InterruptSessionExpired, "InterruptSessionExpired"},
	}

	for i, test := range tests {
		if got := test.in.String(); got != test.str {
			t.Errorf("%d) Wrong string for Event(%d) expected: %v got: %s", i, test.in, test.str, got)
		}
	}
}