From 9bd308579a5f327cf9630e7336b954bfe9718e71 Mon Sep 17 00:00:00 2001
From: Ivy1996-encode <2682904957@qq.com>
Date: Sun, 21 Feb 2021 12:59:51 +0800
Subject: [PATCH] fix a problem with get avatar

---
 bot.go      |  1 +
 bot_test.go | 21 +++++++++++++++++++++
 2 files changed, 22 insertions(+)

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")
+}