func TestUpdateUser(t *testing.T) { phone := "18801234567" secret := "8428d916f8cca9ba5971bf58b34d38da20bc3dff" phone2 := "18801234568" // insert two user, err := models.CreateUserByPhone(&phone, secret) assert.Zero(t, err) user2, err := models.CreateUserByPhone(&phone2, secret) assert.Zero(t, err) weixin := "wx:piasy_umumu" user.WeiXin = &weixin user.Nickname = "Piasy" user.Gender = 1 err = models.UpdateUser(user) assert.Zero(t, err) got, err := models.GetUserByToken(user.Token) assert.Zero(t, err) assertUserEquals(t, user, got) user.Phone = &phone2 err = models.UpdateUser(user) assert.Equal(t, utils.ERROR_CODE_SYSTEM_ERROR, err) deleteUser(t, user.Id) deleteUser(t, user2.Id) }
// @Title CreateUserWithPhone // @Description 通过手机号注册, 返回所有字段 // @Param phone query string true "用户手机号" // @Param code query string true "手机验证码" // @Param secret query string true "加密处理后的密码" // @Param Authorization header string true "Basic auth的授权码, 计算方式见wiki" // @Success 201 {object} models.User // @Failure 401 basic auth失败 // @Failure 403 参数错误:缺失或格式错误 // @Failure 422 手机号已注册 // @Failure 500 系统错误 // @router / [post] func (this *UsersController) Post() { phone := this.GetString("phone") code := this.GetString("code") secret := this.GetString("secret") authorization := this.Ctx.Request.Header.Get("Authorization") if authorization != BASIC_AUTH_AUTHORIZATION { this.Ctx.ResponseWriter.WriteHeader(401) this.Data["json"] = utils.Issue(utils.ERROR_CODE_BASIC_AUTH_FAIL, this.Ctx.Request.URL.String()) } else if !utils.IsValidPhone(phone) || code == "" || len(secret) != 40 { this.Ctx.ResponseWriter.WriteHeader(403) this.Data["json"] = utils.Issue(utils.ERROR_CODE_PARAM_ERROR, this.Ctx.Request.URL.String()) } else if exists := models.UserPhoneExists(&phone); exists { this.Ctx.ResponseWriter.WriteHeader(422) this.Data["json"] = utils.Issue(utils.ERROR_CODE_USERS_PHONE_REGISTERED, this.Ctx.Request.URL.String()) } else if err := models.CheckVerifyCode(phone, code); err > 0 { this.Ctx.ResponseWriter.WriteHeader(422) this.Data["json"] = utils.Issue(err, this.Ctx.Request.URL.String()) } else if user, err := models.CreateUserByPhone(&phone, secret); err > 0 { this.Ctx.ResponseWriter.WriteHeader(500) this.Data["json"] = utils.Issue(err, this.Ctx.Request.URL.String()) } else { this.Ctx.ResponseWriter.WriteHeader(201) this.Data["json"] = user } this.ServeJSON() }
func TestCreateUserByPhone(t *testing.T) { initORM() phone := "18801234567" secret := "8428d916f8cca9ba5971bf58b34d38da20bc3dff" password := "******" // insert one user, err := models.CreateUserByPhone(&phone, secret) assert.NotNil(t, user) assert.Zero(t, err) assert.Empty(t, user.Nickname) assert.True(t, user.Uid >= utils.USER_MIN_UID) assert.Equal(t, password, user.Password) assert.True(t, len(user.Token) == 40) assert.True(t, utils.AreStringEquals(user.Phone, &phone)) now := utils.GetTimeMillis() assert.True(t, now-1000 < user.CreateAt) assert.True(t, user.CreateAt < now+1000) // get it by phone getByPhone, err := models.GetUserByUid(user.Uid) assert.Zero(t, err) assertUserEquals(t, user, getByPhone) // get it by uid getByUid, err := models.GetUserByUid(user.Uid) assert.Zero(t, err) assertUserEquals(t, user, getByUid) // get it by token getByToken, err := models.GetUserByToken(user.Token) assert.Zero(t, err) assertUserEquals(t, user, getByToken) // clean up deleteUser(t, user.Id) // no such user after delete user, err = models.GetUserByUid(user.Uid) assert.Nil(t, user) assert.Equal(t, utils.ERROR_CODE_USERS_USER_NOT_EXISTS, err) }