add new function to change passsword with test
This commit is contained in:
40
db_test.go
40
db_test.go
@@ -185,11 +185,15 @@ func TestLoginAndAuthorization(t *testing.T) {
|
||||
auth.GET("", func(ctx *gin.Context) {
|
||||
ctx.JSON(http.StatusOK, "ok")
|
||||
})
|
||||
auth2 := middleware.Group("", aH.AuthorizeRole("/login/change", "password"))
|
||||
|
||||
auth2.POST("/login/change/password", aH.ChangePassword)
|
||||
|
||||
// ---- Step 1: Perform login ----
|
||||
user := models.User{
|
||||
Name: "guest",
|
||||
Password: "passwordd1",
|
||||
Name: "guest",
|
||||
Password: "passwordd1",
|
||||
NewPassword: "Newpasswordd1",
|
||||
}
|
||||
jsonBody, _ := json.Marshal(user)
|
||||
|
||||
@@ -218,22 +222,40 @@ func TestLoginAndAuthorization(t *testing.T) {
|
||||
}
|
||||
|
||||
type request struct {
|
||||
Name string
|
||||
Method string
|
||||
Path string
|
||||
Cookie *http.Cookie
|
||||
Name string
|
||||
Method string
|
||||
Path string
|
||||
Payload any
|
||||
Cookie *http.Cookie
|
||||
ignoreError bool
|
||||
}
|
||||
var requests []request
|
||||
|
||||
user.Id = 2
|
||||
|
||||
correctUser := user
|
||||
correctUser.Password = user.NewPassword
|
||||
|
||||
requests = append(requests,
|
||||
request{Name: "Refresh", Method: "POST", Path: "/login/refresh", Cookie: refreshCookie},
|
||||
request{Name: "Me", Method: "GET", Path: "/login/me", Cookie: accessCookie},
|
||||
request{Name: "Authorization", Method: "GET", Path: "/members", Cookie: accessCookie},
|
||||
request{Name: "Change Password", Method: "POST", Path: "/login/change/password", Cookie: accessCookie, Payload: user},
|
||||
request{Name: "Logout", Method: "GET", Path: "/logout", Cookie: refreshCookie},
|
||||
request{Name: "New wrong login", Method: "POST", Path: "/login", Payload: user, ignoreError: true},
|
||||
request{Name: "New login", Method: "POST", Path: "/login", Payload: correctUser},
|
||||
)
|
||||
|
||||
for _, request := range requests {
|
||||
req, _ := http.NewRequest(request.Method, request.Path, nil)
|
||||
var body io.Reader
|
||||
if request.Payload != nil {
|
||||
jsonBytes, err := json.Marshal(request.Payload)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
body = bytes.NewBuffer(jsonBytes)
|
||||
}
|
||||
req, _ := http.NewRequest(request.Method, request.Path, body)
|
||||
if request.Cookie != nil {
|
||||
req.AddCookie(request.Cookie) // attach refresh_token cookie
|
||||
}
|
||||
@@ -242,6 +264,8 @@ func TestLoginAndAuthorization(t *testing.T) {
|
||||
r.ServeHTTP(w, req)
|
||||
|
||||
t.Log(request.Name+" response:", w.Body.String())
|
||||
assert.Equal(t, http.StatusOK, w.Code)
|
||||
if !request.ignoreError {
|
||||
assert.Equal(t, http.StatusOK, w.Code)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user