test/rocketmq-eventbridge-e2etest/src/test/java/org/apache/rocketmq/eventbridge/e2etest/BaseTest.java

package org.apache.rocketmq.eventbridge.e2etest;

import com.fasterxml.jackson.core.JsonParseException;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.Version;
import com.fasterxml.jackson.databind.DeserializationFeature;
import com.fasterxml.jackson.databind.JsonMappingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.module.SimpleModule;
import org.junit.Assert;
import org.junit.Before;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.web.client.TestRestTemplate;
import org.springframework.boot.web.server.LocalServerPort;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;

import java.io.IOException;
import java.net.URL;
import java.util.Map;

public class BaseTest {

    protected final static ObjectMapper mapper = new ObjectMapper();

    static {
        SimpleModule m = new SimpleModule("WorkflowJob", Version.unknownVersion());
        mapper.registerModule(m);

        mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
    }

    protected URL baseUrl;

    @LocalServerPort
    protected int port;

    @Autowired
    protected TestRestTemplate template;

    @Before
    public void setUp() throws Exception {
        this.baseUrl = new URL("http://localhost:" + port);
    }

    protected <T> T parseDataFromResponse(ResponseEntity<Object> response, Class<T> dataType, Map<Class, Class> modelMapping) {
        Assert.assertEquals(HttpStatus.OK, response.getStatusCode());
        ObjectMapper mapper = new ObjectMapper();
        mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
        try {
            for (Map.Entry<Class, Class> e : modelMapping.entrySet()) {
                SimpleModule model = new SimpleModule(e.getKey().getSimpleName(), Version.unknownVersion());
                model.addAbstractTypeMapping(e.getKey(), e.getValue());
                mapper.registerModule(model);
            }
            return mapper.readValue(mapper.writeValueAsString(response.getBody()), dataType);
        } catch (JsonParseException e) {
            e.printStackTrace();
            Assert.fail(e.toString());
        } catch (JsonMappingException e) {
            e.printStackTrace();
            Assert.fail(e.toString());
        } catch (JsonProcessingException e) {
            e.printStackTrace();
            Assert.fail(e.toString());
        } catch (IOException e) {
            e.printStackTrace();
            Assert.fail(e.toString());
        }

        return null;
    }

    protected <T> T parseDataFromResponse(ResponseEntity<Object> response, Class<T> dataType) {
        Assert.assertEquals(HttpStatus.OK, response.getStatusCode());

        try {
            return mapper.readValue(mapper.writeValueAsString(response.getBody()), dataType);
        } catch (JsonParseException e) {
            e.printStackTrace();
            Assert.fail(e.toString());
        } catch (JsonMappingException e) {
            e.printStackTrace();
            Assert.fail(e.toString());
        } catch (JsonProcessingException e) {
            e.printStackTrace();
            Assert.fail(e.toString());
        } catch (IOException e) {
            e.printStackTrace();
            Assert.fail(e.toString());
        }

        return null;
    }
}