fix a problem with get avatar

This commit is contained in:
Ivy1996-encode 2021-02-21 12:59:51 +08:00
parent f11eb4cb4e
commit 9bd308579a
2 changed files with 22 additions and 0 deletions

1
bot.go
View File

@ -101,6 +101,7 @@ func (b *Bot) login(data []byte) error {
}
// 设置当前的用户
b.self = &Self{Bot: b, User: &resp.User}
b.self.Self = b.self
b.storage.SetWebInitResponse(*resp)
// 通知手机客户端已经登录

View File

@ -98,3 +98,24 @@ func TestFriends_SearchByRemarkName(t *testing.T) {
fmt.Println(firends2)
fmt.Println(err)
}
func TestUser_GetAvatarResponse(t *testing.T) {
messageHandler := func(message *Message) {
fmt.Println(message)
}
bot := DefaultBot()
bot.RegisterMessageHandler(messageHandler)
bot.UUIDCallback = PrintlnQrcodeUrl
if err := bot.Login(); err != nil {
fmt.Println(err)
return
}
self, _ := bot.GetCurrentUser()
self.SaveAvatar("2.png")
friend, err := self.Friends()
if err != nil {
fmt.Println(err)
return
}
friend[0].SaveAvatar(friend[0].NickName + ".png")
}