ArrayUtilsTest.java

package com.upwork.agora.tracker.common.util;

import org.junit.Test;

import java.util.Arrays;
import java.util.List;

import static org.junit.Assert.assertEquals;

public class ArrayUtilsTest {

    @Test
    public void getRightSibling() {
        long[] arr = new long[]{2, 3, 5, 8, 9};
        assertEquals(5, ArrayUtils.getRightSibling(arr, 5));
        assertEquals(8, ArrayUtils.getRightSibling(arr, 6));
        assertEquals(10, ArrayUtils.getRightSibling(arr, 10));
        assertEquals(2, ArrayUtils.getRightSibling(arr, 1));
    }

    @Test
    public void getRightSibling2() {
        long[] arr = new long[]{2, 3, 5, 8, 9};
        assertEquals(5, ArrayUtils.getRightSibling2(arr, 5));
        assertEquals(8, ArrayUtils.getRightSibling2(arr, 6));
        assertEquals(10, ArrayUtils.getRightSibling2(arr, 10));
        assertEquals(2, ArrayUtils.getRightSibling2(arr, 1));
    }

    @Test
    public void getRightSibling2with19() {
        long[] arr = new long[]{2, 3, 5, 8, 9, 10, 11, 15, 18, 19, 21, 24, 28, 29, 27, 29, 31, 33, 36, 38, 39, 41, 45, 47, 49, 51 , 52};
        assertEquals(5, ArrayUtils.getRightSibling2(arr, 5));
        assertEquals(8, ArrayUtils.getRightSibling2(arr, 6));
        assertEquals(10, ArrayUtils.getRightSibling2(arr, 10));
        assertEquals(2, ArrayUtils.getRightSibling2(arr, 1));
    }

    @Test
    public void getRightSibling2with19List() {
        List<Long> arr = Arrays.asList(2L, 3L, 5L, 8L, 9L, 10L, 11L, 15L, 18L, 19L, 21L, 24L, 28L, 29L, 27L, 29L, 31L, 33L, 36L, 38L, 39L, 41L, 45L, 47L, 49L, 51L , 52L);
        assertEquals(5, ArrayUtils.getRightSibling2(arr, 5));
        assertEquals(8, ArrayUtils.getRightSibling2(arr, 6));
        assertEquals(10, ArrayUtils.getRightSibling2(arr, 10));
        assertEquals(2, ArrayUtils.getRightSibling2(arr, 1));
    }

    @Test
    public void getRightSiblingwith19() {
        long[] arr = new long[]{2, 3, 5, 8, 9, 10, 11, 15, 18, 19, 21, 24, 28, 29, 27, 29, 31, 33, 36, 38, 39, 41, 45, 47, 49, 51 , 52};
        assertEquals(5, ArrayUtils.getRightSibling(arr, 5));
        assertEquals(8, ArrayUtils.getRightSibling(arr, 6));
        assertEquals(10, ArrayUtils.getRightSibling(arr, 10));
        assertEquals(2, ArrayUtils.getRightSibling(arr, 1));
    }
}