From 0b5359589bec09399d90071db3e47160a94059ea Mon Sep 17 00:00:00 2001
From: meixiongfeng <766881921@qq.com>
Date: Wed, 8 Jun 2022 19:36:28 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BB=A3=E7=A0=81=E6=B3=A8=E9=87=8A=20?=
 =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E6=B5=8B=E8=AF=95=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 modules/mail/api_comp.go |  4 ++++
 sys/db/mail.go           |  2 ++
 sys/db/mail_test.go      | 37 +++++++++++++++++++++++++++++++++++++
 3 files changed, 43 insertions(+)
 create mode 100644 sys/db/mail_test.go

diff --git a/modules/mail/api_comp.go b/modules/mail/api_comp.go
index 703b44e56..332e25a8b 100644
--- a/modules/mail/api_comp.go
+++ b/modules/mail/api_comp.go
@@ -50,6 +50,7 @@ func (this *Api_Comp) Start() (err error) {
 	return
 }
 
+// 查看所有邮件信息
 func (this *Api_Comp) QueryUserMailReq(ctx context.Context, session comm.IUserSession, req *pb.QueryUserMailReq) (err error) {
 
 	code := pb.ErrorCode_Success
@@ -70,6 +71,7 @@ func (this *Api_Comp) QueryUserMailReq(ctx context.Context, session comm.IUserSe
 	return
 }
 
+// 查看某一封邮件
 func (this *Api_Comp) ReadUserMailReq(ctx context.Context, session comm.IUserSession, req *pb.ReadUserMailReq) (err error) {
 	var (
 		code pb.ErrorCode
@@ -91,6 +93,7 @@ func (this *Api_Comp) ReadUserMailReq(ctx context.Context, session comm.IUserSes
 	return
 }
 
+// 领取附件
 func (this *Api_Comp) GetUserMailAttachmentReq(ctx context.Context, session comm.IUserSession, req *pb.GetUserMailAttachmentReq) (err error) {
 
 	var (
@@ -124,6 +127,7 @@ func (this *Api_Comp) GetUserMailAttachmentReq(ctx context.Context, session comm
 	return
 }
 
+// 删除邮件
 func (this *Api_Comp) DelUserMailReq(ctx context.Context, session comm.IUserSession, req *pb.DelUserMailReq) (err error) {
 
 	code := pb.ErrorCode_Success
diff --git a/sys/db/mail.go b/sys/db/mail.go
index 562d93206..322f16dcc 100644
--- a/sys/db/mail.go
+++ b/sys/db/mail.go
@@ -38,6 +38,7 @@ func (this *DB) QueryUserMail(uId string) (mail []*pb.DB_MailData, err error) {
 	return
 }
 
+// 插入一封新的邮件
 func (this *DB) InsertUserMail(mail *pb.DB_MailData) (err error) {
 
 	mail.ObjId = primitive.NewObjectID().Hex()
@@ -101,6 +102,7 @@ func (this *DB) UpdateMailAttachmentState(objId string) bool {
 	return true
 }
 
+// 删除一封邮件
 func (this *DB) DelUserMail(objId string) bool {
 	var obj *pb.DB_MailData
 	err := this.mgo.FindOne(DB_MailTable, bson.M{"_id": objId}).Decode(obj)
diff --git a/sys/db/mail_test.go b/sys/db/mail_test.go
new file mode 100644
index 000000000..c167aa08e
--- /dev/null
+++ b/sys/db/mail_test.go
@@ -0,0 +1,37 @@
+package db
+
+import (
+	"go_dreamfactory/pb"
+	"log"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestCreateEmail(t *testing.T) {
+	err := db.InsertUserMail(&pb.DB_MailData{
+
+		UserId:     "uid123",
+		Title:      "系统邮件",
+		Contex:     "恭喜获得专属礼包一份",
+		CreateTime: uint64(time.Now().Unix()),
+		DueTime:    uint64(time.Now().Unix()) + 30*24*3600,
+		Check:      false,
+		Reward:     false,
+	})
+	_data, err := db.QueryUserMail("uid123")
+
+	for _, v := range _data {
+		log.Printf("userid = %s", v.UserId)
+	}
+	require.Nil(t, err, nil)
+}
+
+func TestReadEmail(t *testing.T) {
+	data, err := db.ReadOneMail("62a078c0726ea54890c34937")
+	if err != nil {
+		log.Printf("%v", data.Reward)
+	}
+	require.Nil(t, err, data)
+}