diff --git a/bot.go b/bot.go index 5c7b0d4..4c79c62 100644 --- a/bot.go +++ b/bot.go @@ -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) // 通知手机客户端已经登录 diff --git a/bot_test.go b/bot_test.go index bb97f18..72ca882 100644 --- a/bot_test.go +++ b/bot_test.go @@ -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") +}