Create & Init Project...

2019-04-22 18:49:16 +08:00
commit fc4fa37393
25440 changed files with 4054998 additions and 0 deletions
--- a/app/admin/ep/saga/service/wechat/wechat_test.go
+++ b/app/admin/ep/saga/service/wechat/wechat_test.go
@ -0,0 +1,143 @@
+package wechat
+
+import (
+	"context"
+	"flag"
+	"os"
+	"testing"
+
+	"go-common/app/admin/ep/saga/conf"
+	"go-common/app/admin/ep/saga/dao"
+	"go-common/app/admin/ep/saga/model"
+
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+var (
+	mydao  *dao.Dao
+	wechat *Wechat
+	ctx    = context.Background()
+)
+
+func TestMain(m *testing.M) {
+	var err error
+	flag.Set("conf", "../../cmd/saga-admin-test.toml")
+	if err = conf.Init(); err != nil {
+		panic(err)
+	}
+
+	mydao = dao.New()
+	defer mydao.Close()
+	wechat = New(mydao)
+
+	os.Exit(m.Run())
+}
+
+func TestAddRequireVisible(t *testing.T) {
+	var (
+		err     error
+		userMap = make(map[string]model.RequireVisibleUser)
+	)
+
+	Convey("TEST addRequireVisible", t, func() {
+		err = wechat.addRequireVisible(ctx, "000000")
+		So(err, ShouldNotBeNil)
+		err = wechat.addRequireVisible(ctx, "001134")
+		So(err, ShouldBeNil)
+		err = mydao.RequireVisibleUsersRedis(ctx, &userMap)
+		So(err, ShouldBeNil)
+		So(userMap, ShouldContainKey, "001134")
+	})
+}
+
+func TestAlreadyInCache(t *testing.T) {
+	var (
+		err         error
+		result      bool
+		contactInfo model.ContactInfo
+	)
+
+	contactInfo = model.ContactInfo{
+		ID:          "111",
+		UserName:    "zhangsan",
+		UserID:      "222",
+		NickName:    "xiaolizi",
+		VisibleSaga: true,
+	}
+
+	Convey("TEST alreadyInCache", t, func() {
+		result, err = wechat.alreadyInCache(ctx, "000")
+		So(err, ShouldBeNil)
+		So(result, ShouldEqual, false)
+		So(mydao.SetRequireVisibleUsersRedis(ctx, &contactInfo), ShouldBeNil)
+		result, err = wechat.alreadyInCache(ctx, "222")
+		So(err, ShouldBeNil)
+		So(result, ShouldEqual, true)
+	})
+}
+
+func TestSyncContacts(t *testing.T) {
+	var (
+		err         error
+		contactInfo = &model.ContactInfo{
+			UserID:   "E10021",
+			UserName: "eyotang",
+			NickName: "ben大神点C",
+		}
+		modify = &model.ContactInfo{
+			UserID:      "000328",
+			UserName:    "eyotang",
+			NickName:    "ben大神点C",
+			VisibleSaga: false,
+		}
+		target      *model.ContactInfo
+		almostEqual bool
+	)
+
+	Convey("TEST sync after add incorrect", t, func() {
+		err = wechat.dao.CreateContact(contactInfo)
+		So(err, ShouldBeNil)
+		target, err = wechat.dao.QueryUserByID(contactInfo.UserID)
+		So(err, ShouldBeNil)
+		almostEqual = contactInfo.AlmostEqual(target)
+		So(almostEqual, ShouldBeTrue)
+
+		err = wechat.SyncContacts(ctx)
+		So(err, ShouldBeNil)
+		target, err = wechat.dao.QueryUserByID(contactInfo.UserID)
+		So(err, ShouldNotBeNil)
+	})
+
+	Convey("TEST aync after change", t, func() {
+		contactInfo, err = wechat.dao.QueryUserByID(modify.UserID)
+		So(err, ShouldBeNil)
+		modify.ID = contactInfo.ID
+		err = wechat.dao.UptContact(contactInfo)
+		So(err, ShouldBeNil)
+
+		err = wechat.SyncContacts(ctx)
+		So(err, ShouldBeNil)
+		target, err = wechat.dao.QueryUserByID(modify.UserID)
+		So(err, ShouldBeNil)
+		So(target.VisibleSaga, ShouldBeTrue)
+		So(target.UserName, ShouldNotEqual, "eyotang")
+	})
+}
+
+func TestPushMsg(t *testing.T) {
+	var err error
+	userName := []string{"wuwei"}
+	content := "测试发送企业微信"
+	Convey("TEST push message", t, func() {
+		err = wechat.PushMsg(ctx, userName, content)
+		So(err, ShouldBeNil)
+	})
+}
+
+func TestAnalysisContacts(t *testing.T) {
+	var err error
+	Convey("TEST analysis contacts", t, func() {
+		err = wechat.AnalysisContacts(ctx)
+		So(err, ShouldBeNil)
+	})
+}