com.capitalone.dashboard.testutil.TestRestOperations Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of hygieia-common Show documentation

Core package shared by API layer and Microservices

The newest version!

package com.capitalone.dashboard.testutil;

import com.capitalone.dashboard.client.RestOperationsSupplier;
import org.springframework.http.HttpStatus;
import org.springframework.web.client.RestOperations;

import java.util.HashMap;
import java.util.Map;

public class TestRestOperations implements RestOperationsSupplier {
    TestRestTemplate template;

    private Map> response;

    public TestRestOperations(Map> response) {
        this.response = response;
    }

    public Map> getResponse() {
        return response;
    }

    public void addResponse(String key, TestResponse testResponse) {
        if (response == null) {
            response = new HashMap<>();
        }
        this.response.put(key, testResponse);
    }

    public void addResponse(String key, T body, HttpStatus httpStatus) {
        if (response == null) {
            response = new HashMap<>();
        }
        this.response.put(key, new TestResponse<>(body, httpStatus));
        if (template != null) {
            template.addResponse(key, new TestResponse<>(body, httpStatus));
        }
    }

    public TestRestTemplate getTemplate() {
        return template;
    }

    public void setTemplate(TestRestTemplate template) {
        this.template = template;
    }

    public RestOperations get() {
        if (template == null) {
            template = new TestRestTemplate<>(response);
        }
        return template;
    }
}