SpectoLabs/hoverfly

package io.specto.hoverfly.junit5;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import io.specto.hoverfly.junit.core.Hoverfly;
import io.specto.hoverfly.junit.core.SimulationPreprocessor;
import io.specto.hoverfly.junit.core.model.*;
import io.specto.hoverfly.junit5.api.HoverflyConfig;
import io.specto.hoverfly.junit5.api.HoverflySimulate;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import java.time.LocalDateTime;
import java.util.HashSet;
import java.util.Set;

@HoverflySimulate(
config = @HoverflyConfig(
simulationPreprocessor = PreProcessResponseTest.TimeSimulationPreprocessor.class
), source = @HoverflySimulate.Source(value = "hoverfly/test-service.json",
type = HoverflySimulate.SourceType.CLASSPATH)
)
@ExtendWith({HoverflyExtension.class})
public class PreProcessResponseTest {
static class TimeSimulationPreprocessor implements SimulationPreprocessor {
@Override
public void accept(Simulation simulation) {
HoverflyData originData= simulation.getHoverflyData();
HoverflyMetaData originMetaData=simulation.getHoverflyMetaData();
Set<RequestResponsePair> pairs= new HashSet<>();
originData.getPairs().stream()
.forEach(p ->
{

                 Response origin= p.getResponse();
                    String body =origin.getBody();
                    String replaced=    body.replace("NOW", LocalDateTime.now().toString());
                    Response rsp= new Response(origin.getStatus(),replaced,origin.isEncodedBody(),origin.isTemplated(),origin.getHeaders()
                    ,origin.getTransitionsState(),origin.getRemovesState(),origin.getFixedDelay(),origin.getLogNormalDelay());

                    origin=rsp;

                    ObjectMapper mapper = new ObjectMapper();
                    try {
                        System.out.println(mapper.writeValueAsString(origin));
                        System.out.println(mapper.writeValueAsString(rsp));
                    } catch (JsonProcessingException e) {
                        e.printStackTrace();
                    }
                    RequestResponsePair pair =new RequestResponsePair(p.getRequest(),rsp);
                    pairs.add(pair);
                });
      //  HoverflyData  hoverflyData=new HoverflyData(pairs,simulation.getHoverflyData().getGlobalActions());
        simulation=new Simulation(new HoverflyData(pairs,simulation.getHoverflyData().getGlobalActions()), simulation.getHoverflyMetaData());
      //  simulation.setHoverflyData(hoverflyData);
        ObjectMapper mapper = new ObjectMapper();
        try {
            System.out.println("class::\n"+mapper.writeValueAsString(simulation));
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
    }

}
@Test
void shouldUseCustomPorts(Hoverfly hoverfly) throws JsonProcessingException {
    ObjectMapper mapper = new ObjectMapper();
    System.out.println("test::\n"+mapper.writeValueAsString(hoverfly.getSimulation()));
}

}