package merr

import (
	"context"
	"errors"
	. "testing"

	"github.com/mediocregopher/mediocre-go-lib/mctx"
	"github.com/mediocregopher/mediocre-go-lib/mtest/massert"
)

func TestError(t *T) {
	massert.Require(t, massert.Nil(Wrap(nil)))

	ctx := mctx.Annotate(context.Background(),
		"a", "aaa aaa\n",
		"c", "ccc\nccc\n",
		"d\t", "weird key but ok")

	{
		e := New("foo", ctx)
		exp := `foo
	* a: aaa aaa
	* c: 
		ccc
		ccc
	* d: weird key but ok
	* errLoc: merr/merr_test.go:21`
		massert.Require(t, massert.Equal(exp, e.Error()))
	}

	{
		e := Wrap(errors.New("foo"), ctx)
		exp := `foo
	* a: aaa aaa
	* c: 
		ccc
		ccc
	* d: weird key but ok
	* errLoc: merr/merr_test.go:33`
		massert.Require(t, massert.Equal(exp, e.Error()))
	}
}

func TestBase(t *T) {
	errFoo, errBar := errors.New("foo"), errors.New("bar")
	erFoo := Wrap(errFoo)
	massert.Require(t,
		massert.Nil(Base(nil)),
		massert.Equal(errFoo, Base(erFoo)),
		massert.Equal(errBar, Base(errBar)),
		massert.Not(massert.Equal(errFoo, erFoo)),
		massert.Not(massert.Equal(errBar, Base(erFoo))),
		massert.Equal(true, Equal(errFoo, erFoo)),
		massert.Equal(false, Equal(errBar, erFoo)),
	)
}

func TestValue(t *T) {
	massert.Require(t,
		massert.Nil(WithValue(nil, "foo", "bar")),
		massert.Nil(Value(nil, "foo")),
	)

	e1 := New("foo")
	e1 = WithValue(e1, "a", "A")
	e2 := WithValue(errors.New("bar"), "a", "A")
	massert.Require(t,
		massert.Equal("A", Value(e1, "a")),
		massert.Equal("A", Value(e2, "a")),
	)

	e3 := WithValue(e2, "a", "AAA")
	massert.Require(t,
		massert.Equal("A", Value(e1, "a")),
		massert.Equal("A", Value(e2, "a")),
		massert.Equal("AAA", Value(e3, "a")),
	)
}

func mkErr(ctx context.Context, err error) error {
	return Wrap(err, ctx)
}

func TestCtx(t *T) {
	ctxA := mctx.Annotate(context.Background(), "0", "ZERO", "1", "one")
	ctxB := mctx.Annotate(context.Background(), "1", "ONE", "2", "TWO")

	// use mkErr so that it's easy to test that the stack info isn't overwritten
	// when Wrap is called with ctxB.
	e := mkErr(ctxA, errors.New("hello"))
	e = Wrap(e, ctxB)

	err := massert.Equal(map[string]string{
		"0":      "ZERO",
		"1":      "ONE",
		"2":      "TWO",
		"err":    "hello",
		"errLoc": "merr/merr_test.go:82",
	}, mctx.Annotations(Context(e)).StringMap()).Assert()
	if err != nil {
		t.Fatal(err)
	}
}