Create & Init Project...

app/job/main/spy/service/block_test.go

@@ -0,0 +1,78 @@
+package service
+
+import (
+	"context"
+	"fmt"
+	"testing"
+
+	"go-common/app/job/main/spy/model"
+
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+var (
+	testBlockMid int64 = 4780461
+	testLowScore int8  = 7
+)
+
+func Test_BlockReason(t *testing.T) {
+	Convey("Test_BlockReason get block reason", t, WithService(func(s *Service) {
+		reason, remake := s.blockReason(context.TODO(), testBlockMid)
+		fmt.Println("reason remake", reason, remake)
+		So(reason, ShouldNotBeEmpty)
+		So(remake, ShouldNotBeEmpty)
+	}))
+}
+
+func Test_CanBlock(t *testing.T) {
+	Convey("Test_CanBlock can block ", t, WithService(func(s *Service) {
+		tx, err := s.dao.BeginTran(c)
+		So(err, ShouldBeNil)
+		ui := &model.UserInfo{Mid: testBlockMid, State: model.StateNormal}
+		err = s.dao.TxUpdateUserState(c, tx, ui)
+		So(err, ShouldBeNil)
+		err = tx.Commit()
+		So(err, ShouldBeNil)
+
+		ui, ok := s.canBlock(context.TODO(), testBlockMid)
+		fmt.Println("Test_CanBlock ui ", ui, testBlockMid)
+		So(ui, ShouldNotBeNil)
+		So(ok, ShouldBeTrue)
+	}))
+}
+
+func Test_Block(t *testing.T) {
+	Convey("Test_Block block ", t, WithService(func(s *Service) {
+		ui, err := s.dao.UserInfo(context.TODO(), testBlockMid)
+		So(err, ShouldBeNil)
+
+		tx, err := s.dao.BeginTran(context.TODO())
+		So(err, ShouldBeNil)
+		ui.State = model.StateNormal
+		err = s.dao.TxUpdateUserState(c, tx, ui)
+		So(err, ShouldBeNil)
+		err = tx.Commit()
+		So(err, ShouldBeNil)
+
+		ui, err = s.dao.UserInfo(context.TODO(), testBlockMid)
+		So(err, ShouldBeNil)
+		So(ui.State == model.StateNormal, ShouldBeTrue)
+
+		reason, remake := s.blockReason(context.TODO(), testBlockMid)
+		fmt.Println("reason remake", reason, remake)
+		So(reason, ShouldNotBeEmpty)
+		So(remake, ShouldNotBeEmpty)
+
+		Convey("Test_CanBlock do block ", WithService(func(s *Service) {
+			err := s.blockByMid(context.TODO(), testBlockMid)
+			So(err, ShouldBeNil)
+			Convey("Test_CanBlock get block user info ", WithService(func(s *Service) {
+				ui, err := s.dao.UserInfo(context.TODO(), testBlockMid)
+				So(err, ShouldBeNil)
+				So(ui.State == model.StateBlock, ShouldBeTrue)
+			}))
+		}))
+		fmt.Println("Test_Block end ")
+
+	}))
+}