knowledgeBase/rpc/knowledge/test/knowledge_test.go

package test

import (
	cm "app.yhyue.com/moapp/jybase/common"
	"app.yhyue.com/moapp/jybase/mysql"
	"bp.jydev.jianyu360.cn/SocialPlatform/knowledgeBase/entity"
	"bp.jydev.jianyu360.cn/SocialPlatform/knowledgeBase/rpc/knowledge/internal/config"
	"bp.jydev.jianyu360.cn/SocialPlatform/knowledgeBase/rpc/knowledge/knowledge"
	"context"
	"flag"
	"github.com/zeromicro/go-zero/core/conf"
	"github.com/zeromicro/go-zero/zrpc"
	"log"
	"testing"
	"time"
)

var configFile = flag.String("f", "../etc/knowledge.yaml", "the config file")
var c config.Config

func init() {
	conf.MustLoad(*configFile, &c)
	mm := c.MysqlMain
	if mm.Address != "" {
		log.Println("--初始化 mysql--")
		entity.Mysql = &mysql.Mysql{
			Address:      mm.Address,
			UserName:     mm.UserName,
			PassWord:     mm.PassWord,
			DBName:       mm.DbName,
			MaxOpenConns: mm.MaxOpenConns,
			MaxIdleConns: mm.MaxIdleConns,
		}
		entity.Mysql.Init()
	}
}

func Test_KnowledgeAdd(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
	TestSystem := knowledge.NewKnowledge(zrpc.MustNewClient(c.TestConf))
	req := &knowledge.AddRequest{}
	req.AppId = "10000"
	req.EntId = 14929
	req.Question = "超级订阅为什么购买不了1111"
	req.Answer = "超级订阅按购买区域、时间计价，您看下购买页，是否有选择购买的省份，选择购买的区域后，系统会自动计价，您下单购买就可以了。"
	req.EntUserId = 4315
	res, err := TestSystem.KnowledgeAdd(ctx, req)
	log.Println("res ", res)
	log.Println("err ", err)
}

func Test_KnowledgeList(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
	TestSystem := knowledge.NewKnowledge(zrpc.MustNewClient(c.TestConf))
	req := &knowledge.ListRequest{}
	req.EntId = 14929
	req.PageSize = 20
	req.PageIndex = 3
	res, err := TestSystem.KnowledgeList(ctx, req)
	log.Println("res ", res)
	log.Println("err ", err)
}

func Test_KnowledgeInfo(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
	TestSystem := knowledge.NewKnowledge(zrpc.MustNewClient(c.TestConf))
	req := &knowledge.KnowledgeEntity{}
	req.AnswerId = 30238
	res, err := TestSystem.KnowledgeInfo(ctx, req)
	log.Println("res ", res)
	log.Println("err ", err)
}

func Test_KnowledgeEdit(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
	TestSystem := knowledge.NewKnowledge(zrpc.MustNewClient(c.TestConf))
	req := &knowledge.KnowledgeEditReq{}
	req.AnswerId = 23
	req.KnowledgeId = 1
	req.EntId = 10000
	req.Question = "大会员权益有哪些，如何购买?"
	req.Answer = "大会员有超级多的权益，好处多多，在剑鱼网站直接购买。"
	res, err := TestSystem.KnowledgeEdit(ctx, req)
	log.Println("res ", res)
	log.Println("err ", err)
}

func Test_KnowledgeDel(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
	TestSystem := knowledge.NewKnowledge(zrpc.MustNewClient(c.TestConf))
	req := &knowledge.KnowledgeDelReq{}
	req.AnswerId = 32
	res, err := TestSystem.KnowledgeDel(ctx, req)
	log.Println("res ", res)
	log.Println("err ", err)
}

func Test_FindAnswer(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
	TestSystem := knowledge.NewKnowledge(zrpc.MustNewClient(c.TestConf))
	req := &knowledge.FindAnswerReq{}
	req.Question = "在"
	req.RobotEntId = "455b495c5c"
	req.Type = 1
	res, err := TestSystem.FindAnswer(ctx, req)
	log.Println("res ", res)
	log.Println("err ", err)
}

func Test_RecommendAnswer(t *testing.T) {
	ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
	TestSystem := knowledge.NewKnowledge(zrpc.MustNewClient(c.TestConf))
	req := &knowledge.FindAnswerReq{}
	req.Question = "订阅购买"
	req.RobotEntId = "455b495c5c"
	req.Type = 1
	res, err := TestSystem.RecommendAnswer(ctx, req)
	log.Println("res ", res)
	log.Println("err ", err)
}

func Test_DataSync(t *testing.T) {
	//查询mysql
	data := entity.Mysql.SelectBySql("SELECT q.content as question,k.ent_id,a.content as answer,a.create_person FROM socialize_question q LEFT JOIN socialize_answer a ON q.answer_id = a.id LEFT JOIN socialize_knowledge k ON a.knowledge_id = k.id")
	log.Println(data)
	if data != nil && len(*data) > 0 {
		ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
		TestSystem := knowledge.NewKnowledge(zrpc.MustNewClient(c.TestConf))
		for _, val := range *data {

			req := &knowledge.AddRequest{}
			req.AppId = "10000"
			req.EntId = cm.Int64All(val["ent_id"])
			req.Question = cm.ObjToString(val["question"])
			req.Answer = cm.ObjToString(val["answer"])
			req.EntUserId = cm.Int64All(val["create_person"])
			res, err := TestSystem.KnowledgeAdd(ctx, req)
			log.Println("res ", res, err)
		}
	}
}