FilesExpand file tree

 develop

/

error_test.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

88 lines (77 loc) · 2.07 KB

 develop

/

error_test.go

Copy path

Top

File metadata and controls

• Code
• Blame

88 lines (77 loc) · 2.07 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

85

86

87

88

// Copyright 2016 beego Author. All Rights Reserved.

//

// Licensed under the Apache License, Version 2.0 (the "License");

// you may not use this file except in compliance with the License.

// You may obtain a copy of the License at

//

// http://www.apache.org/licenses/LICENSE-2.0

//

// Unless required by applicable law or agreed to in writing, software

// distributed under the License is distributed on an "AS IS" BASIS,

// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

// See the License for the specific language governing permissions and

// limitations under the License.

package beego

import (

"net/http"

"net/http/httptest"

"strconv"

"strings"

"testing"

)

type errorTestController struct {

Controller

}

const parseCodeError = "parse code error"

func (ec *errorTestController) Get() {

errorCode, err := ec.GetInt("code")

if err != nil {

ec.Abort(parseCodeError)

}

if errorCode != 0 {

ec.CustomAbort(errorCode, ec.GetString("code"))

}

ec.Abort("404")

}

func TestErrorCode_01(t *testing.T) {

registerDefaultErrorHandler()

for k := range ErrorMaps {

r, _ := http.NewRequest("GET", "/error?code="+k, nil)

w := httptest.NewRecorder()

handler := NewControllerRegister()

handler.Add("/error", &errorTestController{})

handler.ServeHTTP(w, r)

code, _ := strconv.Atoi(k)

if w.Code != code {

t.Fail()

}

if !strings.Contains(w.Body.String(), http.StatusText(code)) {

t.Fail()

}

}

}

func TestErrorCode_02(t *testing.T) {

registerDefaultErrorHandler()

r, _ := http.NewRequest("GET", "/error?code=0", nil)

w := httptest.NewRecorder()

handler := NewControllerRegister()

handler.Add("/error", &errorTestController{})

handler.ServeHTTP(w, r)

if w.Code != 404 {

t.Fail()

}

}

func TestErrorCode_03(t *testing.T) {

registerDefaultErrorHandler()

r, _ := http.NewRequest("GET", "/error?code=panic", nil)

w := httptest.NewRecorder()

handler := NewControllerRegister()

handler.Add("/error", &errorTestController{})

handler.ServeHTTP(w, r)

if w.Code != 200 {

t.Fail()

}

if w.Body.String() != parseCodeError {

t.Fail()

}

}