FilesExpand file tree

 develop

/

flash_test.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

54 lines (47 loc) · 1.52 KB

 develop

/

flash_test.go

Copy path

Top

File metadata and controls

• Code
• Blame

54 lines (47 loc) · 1.52 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

// Copyright 2014 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"

"strings"

"testing"

)

type TestFlashController struct {

Controller

}

func (t *TestFlashController) TestWriteFlash() {

flash := NewFlash()

flash.Notice("TestFlashString")

flash.Store(&t.Controller)

// we choose to serve json because we don't want to load a template html file

t.ServeJSON(true)

}

func TestFlashHeader(t *testing.T) {

// create fake GET request

r, _ := http.NewRequest("GET", "/", nil)

w := httptest.NewRecorder()

// setup the handler

handler := NewControllerRegister()

handler.Add("/", &TestFlashController{}, "get:TestWriteFlash")

handler.ServeHTTP(w, r)

// get the Set-Cookie value

sc := w.Header().Get("Set-Cookie")

// match for the expected header

res := strings.Contains(sc, "BEEGO_FLASH=%00notice%23BEEGOFLASH%23TestFlashString%00")

// validate the assertion

if !res {

t.Errorf("TestFlashHeader() unable to validate flash message")

}

}