Mirrors/sftpgo
1
0
Fork 0
mirror of https://github.com/drakkan/sftpgo.git synced 2025-12-07 23:00:55 +03:00
Code Issues Packages Projects Releases Wiki Activity

WebClient: fix test cases

Browse Source 
Signed-off-by: Nicola Murino <nicola.murino@gmail.com>
This commit is contained in:
Nicola Murino
2023-12-12 19:04:32 +01:00
parent ff2eed8ee9
commit f721cf5c40
3 changed files with 20 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
internal/httpd/flash_test.go
View File

@@ -16,6 +16,7 @@ package httpd
import (
"encoding/base64"
"encoding/json"
"fmt"
"net/http"
"net/http/httptest"
@@ -23,18 +24,29 @@ import (
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/drakkan/sftpgo/v2/internal/util"
)
func TestFlashMessages(t *testing.T) {
rr := httptest.NewRecorder()
req, err := http.NewRequest(http.MethodGet, "/url", nil)
require.NoError(t, err)
message := "test message"
setFlashMessage(rr, req, flashMessage{ErrorString: message})
req.Header.Set("Cookie", fmt.Sprintf("%v=%v", flashCookieName, base64.URLEncoding.EncodeToString([]byte(message))))
message := flashMessage{
ErrorString: "error",
I18nMessage: util.I18nChangePwdTitle,
}
setFlashMessage(rr, req, message)
value, err := json.Marshal(message)
assert.NoError(t, err)
req.Header.Set("Cookie", fmt.Sprintf("%v=%v", flashCookieName, base64.URLEncoding.EncodeToString(value)))
msg := getFlashMessage(rr, req)
assert.Equal(t, message, msg.ErrorString)
assert.Equal(t, message, msg)
assert.Equal(t, util.I18nChangePwdTitle, msg.getI18nError().Message)
req.Header.Set("Cookie", fmt.Sprintf("%v=%v", flashCookieName, "a"))
msg = getFlashMessage(rr, req)
assert.Empty(t, msg)
req.Header.Set("Cookie", fmt.Sprintf("%v=%v", flashCookieName, "YQ=="))
msg = getFlashMessage(rr, req)
assert.Empty(t, msg)
}