/*
* Copyright (C) 2010-2101 Alibaba Group Holding Limited.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.alibaba.otter.node.etl.load;
import java.util.List;
import org.testng.annotations.Test;
import com.alibaba.otter.node.etl.load.loader.weight.WeightBuckets;
import com.alibaba.otter.node.etl.BaseOtterTest;
public class WeightBucketTest extends BaseOtterTest {
@Test
public void test_simple() {
WeightBuckets<String> bucket = new WeightBuckets();
int iCount = 10;
int jCount = 100;
for (int i = 0; i < iCount; i++) {
for (int j = 0; j < jCount; j++) {
bucket.addItem(i, "name:" + i + "_" + j);
}
}
for (int i = 0; i < iCount; i++) {
List<String> items = bucket.getItems(i);
want.collection(items).sizeEq(jCount);
for (int j = 0; j < jCount; j++) {
want.string(items.get(j)).isEqualTo("name:" + i + "_" + j);
}
}
}
@Test
public void test_random() {
WeightBuckets<String> bucket = new WeightBuckets();
int iCount = 10;
int jCount = 100;
for (int i = 0; i < iCount; i++) {
for (int j = 0; j < jCount; j++) {
bucket.addItem(j, "name:" + i + "_" + j);
}
}
for (int j = 0; j < jCount; j++) {
List<String> items = bucket.getItems(j);
want.collection(items).sizeEq(iCount);
for (int i = 0; i < iCount; i++) {
want.string(items.get(i)).isEqualTo("name:" + i + "_" + j);
}
}
}
@Test
public void test_custom() {
WeightBuckets<String> bucket = new WeightBuckets();
bucket.addItem(6, "6");
bucket.addItem(1, "1");
bucket.addItem(5, "5");
bucket.addItem(3, "3");
bucket.addItem(5, "51");
bucket.addItem(2, "2");
bucket.addItem(4, "4");
bucket.addItem(6, "61");
List<Long> weights = bucket.weights();
want.number(weights.get(0)).isEqualTo(1);
want.number(weights.get(1)).isEqualTo(2);
want.number(weights.get(2)).isEqualTo(3);
want.number(weights.get(3)).isEqualTo(4);
want.number(weights.get(4)).isEqualTo(5);
want.number(weights.get(5)).isEqualTo(6);
}
}