@Test @SuppressWarnings("unchecked") public void testGroupSet() { String[] ids = new String[]{ UUID.randomUUID().toString(), UUID.randomUUID().toString()}; FieldPartitioner hashFunc = Accessor.getDefault().getFieldPartitioners(hash).get(0); Constraints c = emptyConstraints.partitionedBy(hash).with("id", ids[0]); Marker marker0 = new Marker.Builder() .add("id_hash", hashFunc.apply(ids[0])).add("id_copy", ids[0]).build(); MarkerRange actual = Iterables.getOnlyElement(c.toKeyRanges()); Assert.assertEquals(marker0, actual.getStart().getBound()); Assert.assertEquals(marker0, actual.getEnd().getBound()); c = emptyConstraints.partitionedBy(hash).with("id", (Object[]) ids); Marker marker1 = new Marker.Builder() .add("id_hash", hashFunc.apply(ids[1])).add("id_copy", ids[1]).build(); assertIterableEquals( Sets.newHashSet( new MarkerRange(hashCmp).of(marker0), new MarkerRange(hashCmp).of(marker1)), c.toKeyRanges()); }
@Test public void testOneSidedRange() { // Thu Sep 12 14:15:47 PDT 2013 Constraints c = emptyConstraints.partitionedBy(time) .from("timestamp", 1379020547042L); Marker sep = new Marker.Builder() .add("year", 2013).add("month", 9).add("day", 12).build(); assertIterableEquals(Sets.newHashSet(new MarkerRange(timeCmp).from(sep)), c.toKeyRanges()); }
@Test public void testGroupRange() { Constraints c = emptyConstraints.partitionedBy(num) .from("number", 5).toBefore("number", 18); Marker start = new Marker.Builder("id_number", 5).build(); Marker stop = new Marker.Builder("id_number", 18).build(); Assert.assertEquals( new MarkerRange(numCmp).from(start).to(stop), Iterables.getOnlyElement(c.toKeyRanges())); }
.add("year", 2013).add("month", 9).add("day", 12) .add("id_hash", hashFunc.apply(ids[0])).add("id_copy", ids[0]) .build(); Marker nov0 = new Marker.Builder() .add("year", 2013).add("month", 11).add("day", 11) .add("id_hash", hashFunc.apply(ids[0])).add("id_copy", ids[0]) .build(); .build(); Marker nov1 = new Marker.Builder() .add("year", 2013).add("month", 11).add("day", 11) .add("id_hash", hashFunc.apply(ids[1])).add("id_copy", ids[1]) .build(); assertIterableEquals(Sets.newHashSet( new MarkerRange(cmp).from(sep0).to(nov0), .add("id_hash", hashFunc.apply(ids[0])).add("id_copy", ids[0]) .add("id_component", "com.company.Main") .build(); Marker nov2 = new Marker.Builder() .add("year", 2013).add("month", 11).add("day", 11) .add("id_hash", hashFunc.apply(ids[0])).add("id_copy", ids[0]) .add("id_component", "com.company.Main") .build(); Marker sep3 = new Marker.Builder() .add("year", 2013).add("month", 9).add("day", 12) .add("id_hash", hashFunc.apply(ids[1])).add("id_copy", ids[1]) .add("id_component", "com.company.Main") .build();
@BeforeClass public static void setup() { YEAR = new Marker.Builder("year", 2013).build(); OCT = new Marker.Builder() .add("year", 2013) .add("month", 10) .build(); OCT_12 = new Marker.Builder() .add("year", 2013) .add("month", 10) .add("day", 12) .build(); SEPT = new Marker.Builder() .add("year", 2013) .add("month", 9) .build(); SEPT_30 = new Marker.Builder() .add("year", 2013) .add("month", 9) .add("day", 30) .build(); NOV = new Marker.Builder() .add("year", 2013) .add("month", 11) .build(); NOV_1 = new Marker.Builder() .add("year", 2013) .add("month", 11) .add("day", 1) .build();
.add("year", 2013) .add("month", 10) .build(); OCT_12 = new Marker.Builder() .add("year", 2013) .add("month", 10) .add("day", 12) .build(); SEPT = new Marker.Builder() .add("year", 2013) .add("month", 9) .build(); SEPT_30 = new Marker.Builder() .add("year", 2013) .add("month", 9) .add("day", 30) .build(); NOV = new Marker.Builder() .add("year", 2013) .add("month", 11) .build(); NOV_1 = new Marker.Builder() .add("year", 2013) .add("month", 11) .add("day", 1) .build(); PartitionStrategy strategy = new PartitionStrategy.Builder() .year("timestamp")
@BeforeClass public static void setup() { OCT_12 = new Marker.Builder() .add("year", 2013) .add("month", 10) .add("day", 12) .build(); OCT_15 = new Marker.Builder() .add("year", 2013) .add("month", 10) .add("day", 15) .build(); SEPT_30 = new Marker.Builder() .add("year", 2013) .add("month", 9) .add("day", 30) .build(); NOV_1 = new Marker.Builder() .add("year", 2013) .add("month", 11) .add("day", 1) .build(); PartitionStrategy strategy = new PartitionStrategy.Builder() .year("timestamp") .month("timestamp") .day("timestamp") .build(); comparator = new MarkerComparator(strategy); }
@Test public void testTimeSet() { Long[] timestamps = new Long[] { 1379020547042L, // Thu Sep 12 14:15:47 PDT 2013 1381612547042L, // Sat Oct 12 14:15:47 PDT 2013 1384204547042L // Mon Nov 11 13:15:47 PST 2013 }; Constraints c = emptyConstraints.partitionedBy(time) .with("timestamp", (Object[]) timestamps); Marker sep = new Marker.Builder().add("year", 2013).add("month", 9).add("day", 12).build(); Marker oct = new Marker.Builder().add("year", 2013).add("month", 10).add("day", 12).build(); Marker nov = new Marker.Builder().add("year", 2013).add("month", 11).add("day", 11).build(); assertIterableEquals(Sets.newHashSet( new MarkerRange(timeCmp).of(sep), new MarkerRange(timeCmp).of(oct), new MarkerRange(timeCmp).of(nov)), c.toKeyRanges()); }
@Test public void testSingleSet() { Constraints c = emptyConstraints.partitionedBy(id) .with("component", "com.company.Main"); Marker main = new Marker.Builder("id_component", "com.company.Main").build(); MarkerRange actual = Iterables.getOnlyElement(c.toKeyRanges()); Assert.assertEquals(main, actual.getStart().getBound()); Assert.assertEquals(main, actual.getEnd().getBound()); Assert.assertEquals(new MarkerRange(idCmp).of(main), actual); c = emptyConstraints.partitionedBy(id).with("component", "com.company.Main", "com.company.SomeClass"); Marker sc = new Marker.Builder("id_component", "com.company.SomeClass").build(); assertIterableEquals( Sets.newHashSet( new MarkerRange(idCmp).of(main), new MarkerRange(idCmp).of(sc)), c.toKeyRanges()); }
@Test public void testTimeRange() { Long[] timestamps = new Long[] { 1379020547042L, // Thu Sep 12 14:15:47 PDT 2013 1381612547042L, // Sat Oct 12 14:15:47 PDT 2013 1384204547042L // Mon Nov 11 13:15:47 PST 2013 }; Constraints c = emptyConstraints.partitionedBy(time) .from("timestamp", timestamps[0]).to("timestamp", timestamps[2]); Marker sep = new Marker.Builder().add("year", 2013).add("month", 9).add("day", 12).build(); Marker nov = new Marker.Builder().add("year", 2013).add("month", 11).add("day", 11).build(); Assert.assertEquals( new MarkerRange(timeCmp).from(sep).to(nov), Iterables.getOnlyElement(c.toKeyRanges())); }
@Test public void testSingleRange() { Constraints c = emptyConstraints.partitionedBy(num2) .from("number", 5).toBefore("number", 18) .to("number2", 9); Marker start = new Marker.Builder("id_number", 5).build(); Marker stop = new Marker.Builder().add("id_number", 18).add("id_number2", 9).build(); Assert.assertEquals( new MarkerRange(num2Cmp).from(start).to(stop), Iterables.getOnlyElement(c.toKeyRanges())); }
@Test public void testHashRange() { Constraints c = emptyConstraints.partitionedBy(hash) .from("id", "0000").toBefore("id", "0001"); // note the lack of a hash field -- ranges cannot be projected through hash Marker start = new Marker.Builder("id_copy", "0000").build(); Marker stop = new Marker.Builder("id_copy", "0001").build(); Assert.assertEquals( new MarkerRange(hashCmp).from(start).to(stop), Iterables.getOnlyElement(c.toKeyRanges())); }