/
githubmirror
/
beego
Обзор
Документация
Войти
/
githubmirror
/
beego
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
develop
server/web/flash_test.go
54 строки
2 KB
Ming Deng
Make compatible
03 янв 2021, 17:00
03 янв 2021, 17:00
babf0df
Код
Авторство
О чём код?
// 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 web 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{}, WithRouterMethods(&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") } }