LeshanBootstrapServerTest.java

/*******************************************************************************
 * Copyright (c) 2019 Sierra Wireless and others.
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v2.0
 * and Eclipse Distribution License v1.0 which accompany this distribution.
 *
 * The Eclipse Public License is available at
 *    http://www.eclipse.org/legal/epl-v20.html
 * and the Eclipse Distribution License is available at
 *    http://www.eclipse.org/org/documents/edl-v10.html.
 *
 * Contributors:
 *     Sierra Wireless - initial API and implementation
 *******************************************************************************/
package org.eclipse.leshan.server.californium.bootstrap;

import static org.junit.Assert.assertEquals;

import java.net.InetSocketAddress;
import java.net.URI;

import org.eclipse.leshan.core.endpoint.Protocol;
import org.eclipse.leshan.core.request.BootstrapRequest;
import org.eclipse.leshan.core.request.Identity;
import org.eclipse.leshan.core.response.BootstrapResponse;
import org.eclipse.leshan.core.response.SendableResponse;
import org.eclipse.leshan.server.bootstrap.BootstrapConfig;
import org.eclipse.leshan.server.bootstrap.BootstrapConfigStore;
import org.eclipse.leshan.server.bootstrap.BootstrapHandler;
import org.eclipse.leshan.server.bootstrap.BootstrapHandlerFactory;
import org.eclipse.leshan.server.bootstrap.BootstrapSession;
import org.eclipse.leshan.server.bootstrap.BootstrapSessionListener;
import org.eclipse.leshan.server.bootstrap.BootstrapSessionManager;
import org.eclipse.leshan.server.bootstrap.DefaultBootstrapHandler;
import org.eclipse.leshan.server.bootstrap.LeshanBootstrapServer;
import org.eclipse.leshan.server.bootstrap.LeshanBootstrapServerBuilder;
import org.eclipse.leshan.server.bootstrap.request.BootstrapDownlinkRequestSender;
import org.eclipse.leshan.server.californium.bootstrap.endpoint.CaliforniumBootstrapServerEndpointsProvider;
import org.junit.Test;

public class LeshanBootstrapServerTest {

    private BootstrapHandler bsHandler;

    private LeshanBootstrapServer createBootstrapServer() {
        LeshanBootstrapServerBuilder builder = new LeshanBootstrapServerBuilder();
        builder.setBootstrapHandlerFactory(new BootstrapHandlerFactory() {

            @Override
            public BootstrapHandler create(BootstrapDownlinkRequestSender sender,
                    BootstrapSessionManager sessionManager, BootstrapSessionListener listener) {
                bsHandler = new DefaultBootstrapHandler(sender, sessionManager, listener);
                return bsHandler;
            }
        });
        builder.setConfigStore(new BootstrapConfigStore() {

            @Override
            public BootstrapConfig get(String endpoint, Identity deviceIdentity, BootstrapSession session) {
                BootstrapConfig config = new BootstrapConfig();
                config.toDelete.add("/");
                return config;
            }
        });
        builder.setEndpointsProvider(new CaliforniumBootstrapServerEndpointsProvider());
        return builder.build();

    }

    @Test
    public void testStartStopStart() throws InterruptedException {
        LeshanBootstrapServer server = createBootstrapServer();

        server.start();
        Thread.sleep(100);
        server.stop();
        Thread.sleep(100);
        server.start();
    }

    @Test
    public void testStartDestroy() throws InterruptedException {
        // look at nb active thread before.
        int numberOfThreadbefore = Thread.activeCount();

        LeshanBootstrapServer server = createBootstrapServer();
        server.start();
        Thread.sleep(100);
        // HACK force creation thread creation.
        forceThreadsCreation(server.getEndpoint(Protocol.COAP).getURI());
        Thread.sleep(100);
        server.destroy();

        // ensure all thread are destroyed
        Thread.sleep(500);
        assertEquals("All news created threads must be destroyed", numberOfThreadbefore, Thread.activeCount());
    }

    @Test
    public void testStartStopDestroy() throws InterruptedException {
        // look at nb active thread before.
        int numberOfThreadbefore = Thread.activeCount();

        LeshanBootstrapServer server = createBootstrapServer();
        server.start();
        Thread.sleep(100);
        // HACK force creation thread creation.
        forceThreadsCreation(server.getEndpoint(Protocol.COAP).getURI());
        Thread.sleep(100);
        server.stop();
        Thread.sleep(100);
        server.destroy();

        // ensure all thread are destroyed
        Thread.sleep(500);
        assertEquals("All news created threads must be destroyed", numberOfThreadbefore, Thread.activeCount());
    }

    private void forceThreadsCreation(URI endpointURI) {
        SendableResponse<BootstrapResponse> bootstrap = bsHandler
                .bootstrap(Identity.unsecure(new InetSocketAddress(5683)), new BootstrapRequest("test"), endpointURI);
        bootstrap.sent();
    }
}