gerrors/gerrors_test.go at master · ngopher/gerrors · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
package gerrors_test

import (
	"encoding/json"
	"errors"
	"testing"

	"github.com/ghoroubi/gerrors"
)

var (
	ErrValidationError = errors.New("invalid data provided")
	wrongJSONStr       = `
{
	"id":"1",
	"username":"admin"
}
`
	trueJSONStr = `
{
	"id":1,
	"username":"admin"
}
`
)

// User
type user struct {
	ID       int64  `json:"id"`
	Username string `json:"username"`
}

func TestWrapError(t *testing.T) {

	u := new(user)

	// False
	err := json.Unmarshal([]byte(wrongJSONStr), &u)
	err = gerrors.WrapError(ErrValidationError, err)
	if !errors.Is(err, ErrValidationError) {
		t.FailNow()
	}

	// True
	err = json.Unmarshal([]byte(trueJSONStr), &u)
	err = gerrors.WrapError(ErrValidationError, err)
	if errors.Is(err, ErrValidationError) {
		t.FailNow()
	}
}

func TestHandleGracefully(t *testing.T) {
	u := new(user)

	// False
	err := json.Unmarshal([]byte(wrongJSONStr), &u)

	err = gerrors.WrapError(ErrValidationError, err)
	if !errors.Is(err, ErrValidationError) {
		t.FailNow()
	}

	// Error in production mode
	productionErr := gerrors.HandleGracefully(err, false)

	// Error in development/debug mode
	developErr := gerrors.HandleGracefully(err, true)

	if !errors.As(ErrValidationError, &productionErr) {
		t.FailNow()
	}

	if !errors.As(ErrValidationError, &developErr) {
		t.FailNow()
	}

	if !errors.Is(ErrValidationError, productionErr) {
		t.FailNow()
	}

	if !errors.Is(ErrValidationError, developErr) {
		t.FailNow()
	}
}