add unit test

Signed-off-by: bufferflies <1045931706@qq.com>

Author: bufferflies

server/cluster/cluster.go

@@ -100,7 +100,6 @@ type RaftCluster struct {
 	core    *core.BasicCluster
 	meta    *metapb.Cluster
 	opt     *config.PersistOptions
-	cfg     *config.Config
 	storage storage.Storage
 	id      id.Allocator
 	limiter *StoreLimiter

server/config/config_test.go

@@ -91,6 +91,7 @@ func (s *testConfigSuite) TestReloadConfig(c *C) {
 	}
 	c.Assert(newOpt.GetMaxReplicas(), Equals, 5)
 	c.Assert(newOpt.GetMaxSnapshotCount(), Equals, uint64(10))
+	c.Assert(newOpt.GetCfg().MaxRegionSize, Equals, uint64(defaultMaxRegionSize))
 }
 
 func (s *testConfigSuite) TestReloadUpgrade(c *C) {
@@ -166,6 +167,7 @@ func (s *testConfigSuite) TestAdjust(c *C) {
 name = ""
 lease = 0
 max-request-bytes = 20000000
+max-region-size = 10000
 
 [pd-server]
 metric-storage = "http://127.0.0.1:9090"
@@ -187,6 +189,7 @@ leader-schedule-limit = 0
 	c.Assert(cfg.Name, Equals, fmt.Sprintf("%s-%s", defaultName, host))
 	c.Assert(cfg.LeaderLease, Equals, defaultLeaderLease)
 	c.Assert(cfg.MaxRequestBytes, Equals, uint(20000000))
+	c.Assert(cfg.MaxRegionSize, Equals, uint64(10000))
 	// When defined, use values from config file.
 	c.Assert(cfg.Schedule.MaxMergeRegionSize, Equals, uint64(0))
 	c.Assert(cfg.Schedule.EnableOneWayMerge, IsTrue)

server/schedule/checker/merge_checker_test.go

@@ -113,6 +113,12 @@ func (s *testMergeCheckerSuite) TestBasic(c *C) {
 	c.Assert(ops[0].RegionID(), Equals, s.regions[2].GetID())
 	c.Assert(ops[1].RegionID(), Equals, s.regions[1].GetID())
 
+	// change the max region size
+	s.cluster.GetCfg().MaxRegionSize = 200
+	s.cluster.PutRegion(s.regions[1].Clone(core.SetApproximateSize(200)))
+	ops = s.mc.Check(s.regions[2])
+	c.Assert(ops, NotNil)
+	s.cluster.GetCfg().MaxRegionSize = 96
 	// Test the peer store check.
 	store := s.cluster.GetStore(1)
 	c.Assert(store, NotNil)