HBASE-24894 [Flakey Test] TestStochasticLoadBalancer.testMoveCostMultiplier

Closes #2270

Signed-off-by: Viraj Jasani <vjasani@apache.org>
This commit is contained in:
Zheng Wang 2020-08-30 20:08:28 +05:30 committed by Viraj Jasani
parent 2c4204ad75
commit 2eb2c96960
No known key found for this signature in database
GPG Key ID: B3D6C0B41C8ADFD5
1 changed files with 8 additions and 1 deletions

View File

@ -32,6 +32,7 @@ import java.util.List;
import java.util.Map;
import java.util.Queue;
import java.util.Set;
import java.util.TimeZone;
import java.util.TreeMap;
import org.apache.hadoop.conf.Configuration;
@ -52,6 +53,7 @@ import org.apache.hadoop.hbase.master.balancer.StochasticLoadBalancer.ServerLoca
import org.apache.hadoop.hbase.testclassification.MasterTests;
import org.apache.hadoop.hbase.testclassification.MediumTests;
import org.apache.hadoop.hbase.util.Bytes;
import org.apache.hadoop.hbase.util.EnvironmentEdgeManager;
import org.junit.ClassRule;
import org.junit.Test;
import org.junit.experimental.categories.Category;
@ -295,9 +297,14 @@ public class TestStochasticLoadBalancer extends BalancerTestBase {
assertEquals(StochasticLoadBalancer.MoveCostFunction.DEFAULT_MOVE_COST,
costFunction.getMultiplier(), 0.01);
//In offpeak hours, the multiplier of move cost should be lower
// In offpeak hours, the multiplier of move cost should be lower
conf.setInt("hbase.offpeak.start.hour",0);
conf.setInt("hbase.offpeak.end.hour",23);
// Set a fixed time which hour is 15, so it will always in offpeak
// See HBASE-24898 for more info of the calculation here
long deltaFor15 = TimeZone.getDefault().getRawOffset() - 28800000;
long timeFor15 = 1597907081000L - deltaFor15;
EnvironmentEdgeManager.injectEdge(() -> timeFor15);
costFunction = new StochasticLoadBalancer.MoveCostFunction(conf);
costFunction.init(cluster);
costFunction.cost();