core/net/src/test/java/org/onosproject/cluster/impl/MastershipProxyManagerTest.java

/*
 * Copyright 2018-present Open Networking Foundation
 *
 * 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 org.onosproject.cluster.impl;

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.atomic.AtomicInteger;

import org.junit.Test;
import org.onlab.packet.IpAddress;
import org.onosproject.cluster.ClusterServiceAdapter;
import org.onosproject.cluster.ControllerNode;
import org.onosproject.cluster.DefaultControllerNode;
import org.onosproject.cluster.NodeId;
import org.onosproject.mastership.MastershipProxyFactory;
import org.onosproject.mastership.MastershipServiceAdapter;
import org.onosproject.net.DeviceId;
import org.onosproject.store.serializers.KryoNamespaces;
import org.onosproject.store.service.Serializer;

import static org.junit.Assert.assertEquals;

/**
 * Mastership proxy manager test.
 */
public class MastershipProxyManagerTest {
    @Test
    public void testProxyManager() throws Exception {
        TestClusterCommunicationServiceFactory clusterCommunicatorFactory =
            new TestClusterCommunicationServiceFactory();

        NodeId a = NodeId.nodeId("a");
        NodeId b = NodeId.nodeId("b");

        DeviceId deviceId = DeviceId.deviceId("a");

        Serializer serializer = Serializer.using(KryoNamespaces.BASIC);

        ProxyInterfaceImpl proxyInterface1 = new ProxyInterfaceImpl();
        MastershipProxyManager proxyManager1 = new MastershipProxyManager();
        proxyManager1.clusterService = new ClusterServiceAdapter() {
            @Override
            public ControllerNode getLocalNode() {
                return new DefaultControllerNode(a, IpAddress.valueOf(0));
            }
        };
        proxyManager1.clusterCommunicator = clusterCommunicatorFactory.newCommunicationService(a);
        proxyManager1.mastershipService = new MastershipServiceAdapter() {
            @Override
            public NodeId getMasterFor(DeviceId deviceId) {
                return b;
            }
        };
        proxyManager1.activate();
        proxyManager1.registerProxyService(ProxyInterface.class, proxyInterface1, serializer);

        ProxyInterfaceImpl proxyInterface2 = new ProxyInterfaceImpl();
        MastershipProxyManager proxyManager2 = new MastershipProxyManager();
        proxyManager2.clusterService = new ClusterServiceAdapter() {
            @Override
            public ControllerNode getLocalNode() {
                return new DefaultControllerNode(b, IpAddress.valueOf(0));
            }
        };
        proxyManager2.clusterCommunicator = clusterCommunicatorFactory.newCommunicationService(b);
        proxyManager2.mastershipService = new MastershipServiceAdapter() {
            @Override
            public NodeId getMasterFor(DeviceId deviceId) {
                return b;
            }
        };
        proxyManager2.activate();
        proxyManager2.registerProxyService(ProxyInterface.class, proxyInterface2, serializer);

        MastershipProxyFactory<ProxyInterface> proxyFactory1 =
            proxyManager1.getProxyFactory(ProxyInterface.class, serializer);
        assertEquals("Hello world!", proxyFactory1.getProxyFor(deviceId).sync("Hello world!"));
        assertEquals(1, proxyInterface2.syncCalls.get());
        assertEquals("Hello world!", proxyFactory1.getProxyFor(deviceId).async("Hello world!").join());
        assertEquals(1, proxyInterface2.asyncCalls.get());

        MastershipProxyFactory<ProxyInterface> proxyFactory2 =
            proxyManager2.getProxyFactory(ProxyInterface.class, serializer);
        assertEquals("Hello world!", proxyFactory2.getProxyFor(deviceId).sync("Hello world!"));
        assertEquals(2, proxyInterface2.syncCalls.get());
        assertEquals("Hello world!", proxyFactory2.getProxyFor(deviceId).async("Hello world!").join());
        assertEquals(2, proxyInterface2.asyncCalls.get());

        proxyManager1.deactivate();
        proxyManager2.deactivate();
    }

    interface ProxyInterface {
        String sync(String arg);
        CompletableFuture<String> async(String arg);
    }

    class ProxyInterfaceImpl implements ProxyInterface {
        private final AtomicInteger syncCalls = new AtomicInteger();
        private final AtomicInteger asyncCalls = new AtomicInteger();

        @Override
        public String sync(String arg) {
            syncCalls.incrementAndGet();
            return arg;
        }

        @Override
        public CompletableFuture<String> async(String arg) {
            asyncCalls.incrementAndGet();
            return CompletableFuture.completedFuture(arg);
        }
    }
}