com.vsct.dt.strowgr.admin.nsq.payload.PayloadTest.java Source code

Introduction

Here is the source code for com.vsct.dt.strowgr.admin.nsq.payload.PayloadTest.java

Source

/*
 * Copyright (C) 2016 VSCT
 *
 *  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 com.vsct.dt.strowgr.admin.nsq.payload;

import com.fasterxml.jackson.core.util.DefaultPrettyPrinter;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.vsct.dt.strowgr.admin.nsq.payload.fragment.Conf;
import com.vsct.dt.strowgr.admin.nsq.payload.fragment.Header;
import com.vsct.dt.strowgr.admin.nsq.payload.fragment.Server;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

import java.io.IOException;
import java.net.URISyntaxException;
import java.net.URL;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.HashMap;
import java.util.Map;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;

public class PayloadTest {
    private ObjectMapper mapper;

    @Before
    public void setUp() throws Exception {
        mapper = new ObjectMapper();
    }

    private void assertSerialization(Object message, String file) throws IOException, URISyntaxException {
        String result = mapper
                .writer(new DefaultPrettyPrinter().withArrayIndenter(new DefaultPrettyPrinter.FixedSpaceIndenter()))
                .writeValueAsString(message);

        URL resource = getClass().getClassLoader().getResource(file);
        String expected = new String(Files.readAllBytes(Paths.get(resource.toURI())));
        assertEquals(expected.replaceAll("\r\n", "\n"), result.replaceAll("\r\n", "\n"));
    }

    @Test
    public void testCommitRequested() throws IOException, URISyntaxException {
        CommitRequested message = new CommitRequested(new Header("test-id", "TST", "REL1", 1L, "test"),
                new Conf("abcde", "fghij", "hapversion", "127.0.0.1"));
        assertSerialization(message, "commitRequested.expected.json");
    }

    @Test
    public void testCommitCompleted() throws IOException, URISyntaxException {
        CommitCompleted message = new CommitCompleted(new Header("test-id", "TST", "REL1", 1L, "test"));
        assertSerialization(message, "commitCompleted.expected.json");
    }

    @Test
    public void testCommitFailed() throws IOException, URISyntaxException {
        CommitFailed message = new CommitFailed(new Header("test-id", "TST", "REL1", 1L, "test"));
        assertSerialization(message, "commitFailed.expected.json");
    }

    @Test
    public void testDeleteRequested() throws IOException, URISyntaxException {
        DeleteRequested message = new DeleteRequested(new Header("test-id", "TST", "REL1", 1L, "test"));
        assertSerialization(message, "deleteRequested.expected.json");
    }

    @Test
    public void should_serialize_register_server_with_expected_format() throws IOException, URISyntaxException {
        // given
        Map<String, String> context = new HashMap<>();
        context.put("key1", "val1");
        context.put("key2", "val2");
        Server server = new Server("server-id", "backend-id", "1.2.3.4", "1234", context);

        RegisterServer message = new RegisterServer(new Header("test-id", "TST", "REL1", 1L, "test"), server);

        // test and check
        assertSerialization(message, "registerServer.expected.json");
    }

    @Test
    public void should_deserialize_commit_completed_message() throws IOException {
        // given
        String payload = "{\"header\":{\"correlationId\":\"16f484b9-3935-415b-bd8e-aaeaaf1020ac\",\"application\":\"STR\",\"platform\":\"REL1\",\"timestamp\":1467824168749,\"source\":\"sidekick-default-name-master\"}}";

        // test
        CommitCompleted commitCompleted = new ObjectMapper().readValue(payload.getBytes(), CommitCompleted.class);

        // check
        assertNotNull(commitCompleted);
        assertNotNull(commitCompleted.getHeader());
        Assert.assertEquals("16f484b9-3935-415b-bd8e-aaeaaf1020ac", commitCompleted.getHeader().getCorrelationId());
    }

    @Test
    public void should_deserialize_header_with_corelationId() throws IOException {
        String payload = "{\"application\":\"DEM\",\"platform\": \"REL1\",\"correlationId\":\"cid\",\"timestamp\":\"1\",\"source\":\"source\"}";

        Header header = new ObjectMapper().readValue(payload.getBytes(), Header.class);

        Header expected = new Header("cid", "DEM", "REL1", 1L, "source");
        assertEquals(expected, header);
    }

    @Test
    public void should_deserialize_header_without_corelationId_timestamp_source() throws IOException {

        String payload = "{\"application\":\"DEM\",\"platform\": \"REL1\"}";

        Header header = new ObjectMapper().readValue(payload.getBytes(), Header.class);

        assertEquals("DEM", header.getApplication());
        assertEquals("REL1", header.getPlatform());
        assertNotNull(header.getCorrelationId());
        assertNotNull(header.getTimestamp());
        assertEquals("unknown", header.getSource());
    }
}