ro.contezi.websockets.main.SampleWebSocketsApplicationTest.java Source code

Introduction

Here is the source code for ro.contezi.websockets.main.SampleWebSocketsApplicationTest.java

Source

/*
 * Copyright 2012-2015 the original author or authors.
 *
 * 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 ro.contezi.websockets.main;

import static org.junit.Assert.assertEquals;

import java.util.concurrent.atomic.AtomicReference;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.autoconfigure.PropertyPlaceholderAutoConfiguration;
import org.springframework.boot.builder.SpringApplicationBuilder;
import org.springframework.boot.test.SpringApplicationConfiguration;
import org.springframework.boot.test.WebIntegrationTest;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import ro.contezi.websockets.ClientConfiguration;
import ro.contezi.websockets.SampleWebSocketApplication;

@RunWith(SpringJUnit4ClassRunner.class)
@SpringApplicationConfiguration(SampleWebSocketApplication.class)
@WebIntegrationTest
public class SampleWebSocketsApplicationTest {

    @Value("${local.server.port}")
    private int port;

    @Test
    public void reverseEndpoint() throws Exception {
        ConfigurableApplicationContext context = new SpringApplicationBuilder(ClientConfiguration.class,
                PropertyPlaceholderAutoConfiguration.class)
                        .properties("websocket.uri:ws://localhost:" + this.port + "/reverse")
                        .run("--spring.main.web_environment=false");
        long count = context.getBean(ClientConfiguration.class).getLatch().getCount();
        AtomicReference<String> messagePayloadReference = context.getBean(ClientConfiguration.class)
                .getMessagePayload();
        context.close();
        assertEquals(0, count);
        assertEquals("Reversed: !olleH", messagePayloadReference.get());
    }

}