RingTest.java

package com.workshare.msnos.core;

import static com.workshare.msnos.core.CoreHelper.asNetwork;
import static com.workshare.msnos.core.CoreHelper.asPublicNetwork;
import static com.workshare.msnos.core.CoreHelper.asSet;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import java.util.UUID;

import org.junit.Test;

import com.workshare.msnos.core.geo.Location;
import com.workshare.msnos.core.protocols.ip.BaseEndpoint;
import com.workshare.msnos.core.protocols.ip.Endpoint;
import com.workshare.msnos.core.protocols.ip.Endpoint.Type;
import com.workshare.msnos.core.protocols.ip.HttpEndpoint;
import com.workshare.msnos.usvc.IMicroservice;

public class RingTest {

    @Test
    public void shouldCalculateRingBasedOnUDPEndpoint() {
        Endpoint ssh = new BaseEndpoint(Type.SSH, asPublicNetwork("25.25.25.25"));
        Endpoint htp = new HttpEndpoint(asPublicNetwork("25.25.25.25"), "http://25.25.25.25");
        Endpoint udp = new BaseEndpoint(Type.UDP, asNetwork("192.168.0.199",(short)16));

        Ring ring = Ring.make(asSet(udp, ssh, htp));
        
        long v1 = toLong(udp.getNetwork().getAddress());
        long v2 = udp.getNetwork().getPrefix();
        assertEquals(new UUID(v1,v2), ring.uuid());
    }

    @Test
    public void shouldReturnRandomRingWhenNoValidEndpoints() {
        Ring ring = Ring.make(CoreHelper.<Endpoint>asSet());
        assertNotNull(ring.uuid());
    }

    @Test
    public void shouldLocationBeUnknownInitially() {
        Ring ring = Ring.make(CoreHelper.<Endpoint>asSet());
        assertEquals(Location.UNKNOWN, ring.location());
    }

    @Test
    public void shouldLocationBeUpdatedWhenMicroserviceJoinWithLocation() {
        Ring ring = Ring.random();

        Location location = mock(Location.class);
        when(location.getPrecision()).thenReturn(100);
        IMicroservice uservice = mock(IMicroservice.class);
        when(uservice.getLocation()).thenReturn(location);
        ring.onMicroserviceJoin(uservice);
        
        assertEquals(location, ring.location());

        IMicroservice nowhereman = mock(IMicroservice.class);
        when(nowhereman.getLocation()).thenReturn(Location.UNKNOWN);
        ring.onMicroserviceJoin(uservice);
        
        assertEquals(location, ring.location());
    }

    private long toLong(byte[] values) {
        long shift = 0;
        long total = 0;
        for (byte value : values) {
            final long lval = (int)(value&0xff);
            total = total + lval << shift;
            shift += 8;
        }
        
        return total;
    }
}