|
1 | 1 | package org.baeldung.guava.collections; |
2 | 2 |
|
3 | | -import static org.hamcrest.Matchers.equalTo; |
4 | 3 | import static org.hamcrest.Matchers.nullValue; |
5 | 4 | import static org.junit.Assert.assertThat; |
| 5 | +import static org.junit.Assert.assertTrue; |
6 | 6 |
|
7 | 7 | import java.util.Arrays; |
8 | 8 | import java.util.Collections; |
9 | 9 | import java.util.List; |
10 | 10 |
|
11 | 11 | import org.junit.Test; |
12 | 12 |
|
| 13 | +import com.google.common.collect.Lists; |
13 | 14 | import com.google.common.collect.Ordering; |
14 | 15 | import com.google.common.primitives.Ints; |
15 | 16 |
|
16 | 17 | public class GuavaOrderingExamplesTest { |
17 | 18 |
|
| 19 | + private final class OrderingByLenght extends Ordering<String> { |
| 20 | + @Override |
| 21 | + public final int compare(final String s1, final String s2) { |
| 22 | + return Ints.compare(s1.length(), s2.length()); |
| 23 | + } |
| 24 | + } |
| 25 | + |
18 | 26 | // tests |
19 | 27 |
|
20 | 28 | @Test |
21 | 29 | public final void givenCollectionWithNulls_whenSortingWithNullsLast_thenNullsAreLast() { |
22 | | - final List<Integer> nums = Arrays.asList(3, 5, 4, null, 1, 2); |
23 | | - Collections.sort(nums, Ordering.natural().nullsLast()); |
24 | | - assertThat(nums.get(nums.size() - 1), nullValue()); |
| 30 | + final List<Integer> toSort = Arrays.asList(3, 5, 4, null, 1, 2); |
| 31 | + Collections.sort(toSort, Ordering.natural().nullsLast()); |
| 32 | + assertThat(toSort.get(toSort.size() - 1), nullValue()); |
25 | 33 | } |
26 | 34 |
|
27 | 35 | @Test |
28 | 36 | public final void givenCollectionWithNulls_whenSortingWithNullsFirst_thenNullsAreFirst() { |
29 | | - final List<Integer> nums = Arrays.asList(3, 5, 4, null, 1, 2); |
30 | | - Collections.sort(nums, Ordering.natural().nullsFirst()); |
31 | | - assertThat(nums.get(0), nullValue()); |
| 37 | + final List<Integer> toSort = Arrays.asList(3, 5, 4, null, 1, 2); |
| 38 | + Collections.sort(toSort, Ordering.natural().nullsFirst()); |
| 39 | + assertThat(toSort.get(0), nullValue()); |
32 | 40 | } |
33 | 41 |
|
34 | 42 | @Test |
35 | 43 | public final void whenCollectionIsSortedNullsLastReversed_thenNullAreFirst() { |
36 | | - final List<Integer> nums = Arrays.asList(3, 5, 4, null, 1, 2); |
37 | | - Collections.sort(nums, Ordering.natural().nullsLast().reverse()); |
38 | | - assertThat(nums.get(0), nullValue()); |
| 44 | + final List<Integer> toSort = Arrays.asList(3, 5, 4, null, 1, 2); |
| 45 | + Collections.sort(toSort, Ordering.natural().nullsLast().reverse()); |
| 46 | + assertThat(toSort.get(0), nullValue()); |
| 47 | + } |
| 48 | + |
| 49 | + @Test |
| 50 | + public final void givenCollectionIsSorted_whenUsingOrderingApiToCheckOrder_thenCheckCanBePerformed() { |
| 51 | + final List<String> toSort = Arrays.asList("zz", "aa", "b", "ccc"); |
| 52 | + final Ordering<String> byLength = new OrderingByLenght(); |
| 53 | + Collections.sort(toSort, byLength); |
| 54 | + |
| 55 | + final Ordering<String> expectedOrder = Ordering.explicit(Lists.newArrayList("b", "zz", "aa", "ccc")); |
| 56 | + assertTrue(expectedOrder.isOrdered(toSort)); |
39 | 57 | } |
40 | 58 |
|
41 | 59 | @Test |
42 | 60 | public final void whenSortingCollectionsOfStringsByLenght_thenCorrectlySorted() { |
43 | | - final List<String> toSort = Arrays.asList("aa", "b", null, "ccc"); |
44 | | - final Ordering<String> byLength = new Ordering<String>() { |
45 | | - @Override |
46 | | - public int compare(final String s1, final String s2) { |
47 | | - return Ints.compare(s1.length(), s2.length()); |
48 | | - } |
49 | | - }; |
50 | | - |
51 | | - Collections.sort(toSort, byLength.nullsFirst()); |
52 | | - assertThat(toSort.get(0), nullValue()); |
53 | | - assertThat(toSort.get(1), equalTo("b")); |
| 61 | + final List<String> toSort = Arrays.asList("zz", "aa", "b", "ccc"); |
| 62 | + final Ordering<String> byLength = new OrderingByLenght(); |
| 63 | + |
| 64 | + Collections.sort(toSort, byLength); |
| 65 | + |
| 66 | + final Ordering<String> expectedOrder = Ordering.explicit(Lists.newArrayList("b", "zz", "aa", "ccc")); |
| 67 | + assertTrue(expectedOrder.isOrdered(toSort)); |
| 68 | + } |
| 69 | + |
| 70 | + @Test |
| 71 | + public final void whenSortingCollectionsOfStringsByLenghtWithSecondaryNaturalOrdering_thenCorrectlySorted() { |
| 72 | + final List<String> toSort = Arrays.asList("zz", "aa", "b", "ccc"); |
| 73 | + final Ordering<String> byLength = new OrderingByLenght(); |
| 74 | + |
| 75 | + Collections.sort(toSort, byLength.compound(Ordering.natural())); |
| 76 | + |
| 77 | + final Ordering<String> expectedOrder = Ordering.explicit(Lists.newArrayList("b", "aa", "zz", "ccc")); |
| 78 | + assertTrue(expectedOrder.isOrdered(toSort)); |
54 | 79 | } |
55 | 80 |
|
56 | 81 | } |
0 commit comments