openbilibili/app/job/main/answer/service/service_test.go

package service

import (
	"context"
	"flag"
	"fmt"
	"os"
	"testing"

	"go-common/app/job/main/answer/conf"
	"go-common/app/job/main/answer/model"

	. "github.com/smartystreets/goconvey/convey"
)

var (
	s *Service
)

func TestMain(m *testing.M) {
	if os.Getenv("DEPLOY_ENV") != "" {
		flag.Set("app_id", "")
		flag.Set("conf_appid", "")
		flag.Set("conf_token", "")
		flag.Set("tree_id", "")
		flag.Set("conf_version", "server-1")
		flag.Set("deploy_env", "dev")
		flag.Set("conf_env", "10")
		flag.Set("conf_host", "config.bilibili.co")
		flag.Set("conf_path", "/tmp")
		flag.Set("region", "sh")
		flag.Set("zone", "sh001")
	} else {
		flag.Set("conf", "../cmd/answer-job-test.toml")
	}
	flag.Parse()
	if err := conf.Init(); err != nil {
		panic(err)
	}
	s = New(conf.Conf)
	os.Exit(m.Run())
}

func TestUpdateBfs(t *testing.T) {
	Convey("create img", t, func(s *Service) {
		err := s.createBFSImg(context.TODO(), &model.LabourQs{
			Question: "测试附加题3",
			ID:       3,
		})
		fmt.Println("err:", err)
		So(err, ShouldBeNil)
	})
}

func TestPing(t *testing.T) {
	Convey("Test_Ping", t, func(s *Service) {
		err := s.Ping(context.TODO())
		So(err, ShouldBeNil)
	})
}

// go test  -test.v -test.run TestAddQue
func TestCreateBFSImg(t *testing.T) {
	Convey("TestAddQue add question data", t, func(s *Service) {
		for i := 90; i < 100; i++ {
			que := &model.LabourQs{
				Question: fmt.Sprintf("测试d=====(￣▽￣*)b厉害 %d", i),
				Ans:      int8(i%2 + 1),
				AvID:     int64(i),
				Status:   int8(i%2 + 1),
				Source:   int8(1),
				State:    model.HadCreateImg,
				ID:       int64(i),
			}
			err := s.createBFSImg(context.TODO(), que)
			So(err, ShouldBeNil)
			err = s.dao.AddQs(context.TODO(), que)
			So(err, ShouldBeNil)
		}
	})
}