Examples with TransformRandomLoadBanlance com.alibaba.otter.shared.arbitrate.impl.setl.lb.TransformRandomLoadBanlance used on opensource projects

Example 1 with TransformRandomLoadBanlance

use of com.alibaba.otter.shared.arbitrate.impl.setl.lb.TransformRandomLoadBanlance in project otter by alibaba.

the class RandomLoadBalanceTest method testTransform.

@Test
public void testTransform() {
    // 初始化节点
    Mockit.setUpMock(ArbitrateConfigUtils.class, new Object() {

        @Mock
        public Pipeline getPipeline(Long pipelineId) {
            Pipeline pipeline = new Pipeline();
            pipeline.setId(pipelineId);
            pipeline.setSelectNodes(sourceList);
            pipeline.setExtractNodes(sourceList);
            pipeline.setLoadNodes(targetList);
            return pipeline;
        }
    });
    TransformRandomLoadBanlance transform = new TransformRandomLoadBanlance(pipelineId);
    transform.setNodeMonitor(nodeMonitor);
    sleep(500L);
    try {
        Node n1 = transform.next();
        Node n2 = transform.next();
        Node n3 = transform.next();
        Node n4 = transform.next();
        System.out.printf("n1[%s] n2[%s] n3[%s] n4[%s]", n1.getId(), n2.getId(), n3.getId(), n4.getId());
        want.bool(targetList.contains(n1)).is(true);
        want.bool(targetList.contains(n2)).is(true);
        want.bool(targetList.contains(n3)).is(true);
        want.bool(targetList.contains(n4)).is(true);
    } catch (InterruptedException e) {
        want.fail();
    }
}