Merge pull request #77 from jiaozifs/test/auth_test

test: add test for cookie and jwt

Author: TsumikiQAQ

controller/user_ctl.go

@@ -33,7 +33,6 @@ type UserController struct {
 }
 
 func (userCtl UserController) Login(ctx context.Context, w *api.JiaozifsResponse, r *http.Request, body api.LoginJSONRequestBody) {
-
 	// get user encryptedPassword by username
 	ep, err := userCtl.Repo.UserRepo().GetEPByName(ctx, body.Name)
 	if err != nil {

integrationtest/branch_test.go

@@ -23,7 +23,7 @@ func BranchSpec(ctx context.Context, urlStr string) func(c convey.C) {
 		prefix := "feat/"
 
 		createUser(ctx, c, client, userName)
-		loginAndSwitch(ctx, c, client, userName)
+		loginAndSwitch(ctx, c, client, "mike login", userName, false)
 		createRepo(ctx, c, client, repoName)
 
 		c.Convey("create branch", func(c convey.C) {

integrationtest/commit_test.go

@@ -19,7 +19,7 @@ func GetEntriesInRefSpec(ctx context.Context, urlStr string) func(c convey.C) {
 		branchName := "feat/get_entries_test"
 
 		createUser(ctx, c, client, userName)
-		loginAndSwitch(ctx, c, client, userName)
+		loginAndSwitch(ctx, c, client, "kitty login", userName, false)
 		createRepo(ctx, c, client, repoName)
 		createBranch(ctx, c, client, userName, repoName, "main", branchName)
 		createWip(ctx, c, client, "feat get entries test0", userName, repoName, branchName)

integrationtest/helper_test.go

@@ -114,19 +114,25 @@ func createUser(ctx context.Context, c convey.C, client *api.Client, userName st
 	})
 }
 
-func loginAndSwitch(ctx context.Context, c convey.C, client *api.Client, userName string) {
-	c.Convey("login "+userName, func() {
+func loginAndSwitch(ctx context.Context, c convey.C, client *api.Client, title, userName string, useCookie bool) {
+	c.Convey("login "+title, func() {
 		resp, err := client.Login(ctx, api.LoginJSONRequestBody{
 			Name:     userName,
 			Password: "12345678",
 		})
 		convey.So(err, convey.ShouldBeNil)
 		convey.So(resp.StatusCode, convey.ShouldEqual, http.StatusOK)
+		loginResult, err := api.ParseLoginResponse(resp)
+		convey.So(err, convey.ShouldBeNil)
 
 		client.RequestEditors = nil
 		client.RequestEditors = append(client.RequestEditors, func(ctx context.Context, req *http.Request) error {
-			for _, cookie := range resp.Cookies() {
-				req.AddCookie(cookie)
+			if useCookie {
+				for _, cookie := range resp.Cookies() {
+					req.AddCookie(cookie)
+				}
+			} else {
+				req.Header.Add("Authorization", "Bearer "+loginResult.JSON200.Token)
 			}
 			return nil
 		})

integrationtest/objects_test.go

@@ -22,7 +22,7 @@ func ObjectSpec(ctx context.Context, urlStr string) func(c convey.C) {
 		branchName := "feat/obj_test"
 
 		createUser(ctx, c, client, userName)
-		loginAndSwitch(ctx, c, client, userName)
+		loginAndSwitch(ctx, c, client, "molly login", userName, false)
 		createRepo(ctx, c, client, repoName)
 		createBranch(ctx, c, client, userName, repoName, "main", branchName)
 		createWip(ctx, c, client, "feat get obj test", userName, repoName, branchName)

integrationtest/repo_test.go

@@ -19,7 +19,7 @@ func RepoSpec(ctx context.Context, urlStr string) func(c convey.C) {
 		userName := "jimmy"
 		repoName := "happyrun"
 		createUser(ctx, c, client, userName)
-		loginAndSwitch(ctx, c, client, userName)
+		loginAndSwitch(ctx, c, client, "jimmy login", userName, false)
 
 		c.Convey("create repo", func(c convey.C) {
 			c.Convey("forbidden create repo name", func() {

integrationtest/user_test.go

@@ -30,12 +30,20 @@ func UserSpec(ctx context.Context, urlStr string) func(c convey.C) {
 			convey.So(resp.StatusCode, convey.ShouldEqual, http.StatusUnauthorized)
 		})
 
-		loginAndSwitch(ctx, c, client, userName)
+		loginAndSwitch(ctx, c, client, "admin login", userName, false)
 
 		c.Convey("usr profile", func() {
 			resp, err := client.GetUserInfo(ctx)
 			convey.So(err, convey.ShouldBeNil)
 			convey.So(resp.StatusCode, convey.ShouldEqual, http.StatusOK)
 		})
+
+		loginAndSwitch(ctx, c, client, "admin login again", userName, true)
+
+		c.Convey("usr profile with cookie", func() {
+			resp, err := client.GetUserInfo(ctx)
+			convey.So(err, convey.ShouldBeNil)
+			convey.So(resp.StatusCode, convey.ShouldEqual, http.StatusOK)
+		})
 	}
 }

integrationtest/wip_object_test.go

@@ -18,7 +18,7 @@ func WipObjectSpec(ctx context.Context, urlStr string) func(c convey.C) {
 		branchName := "feat/wip_obj_test"
 
 		createUser(ctx, c, client, userName)
-		loginAndSwitch(ctx, c, client, userName)
+		loginAndSwitch(ctx, c, client, "jude login", userName, false)
 		createRepo(ctx, c, client, repoName)
 		createBranch(ctx, c, client, userName, repoName, "main", branchName)
 		createWip(ctx, c, client, "get wip obj test", userName, repoName, branchName)

integrationtest/wip_test.go

@@ -18,7 +18,7 @@ func WipSpec(ctx context.Context, urlStr string) func(c convey.C) {
 		branchNameForDelete := "feat/wip_test2"
 
 		createUser(ctx, c, client, userName)
-		loginAndSwitch(ctx, c, client, userName)
+		loginAndSwitch(ctx, c, client, "july login", userName, false)
 		createRepo(ctx, c, client, repoName)
 		createBranch(ctx, c, client, userName, repoName, "main", branchName)
 		createBranch(ctx, c, client, userName, repoName, "main", branchNameForDelete)