io.neow3j.test.WireMockTestHelper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of test-tools Show documentation
neow3j: test tools
The newest version!
package io.neow3j.test;

import com.github.tomakehurst.wiremock.client.WireMock;
import com.github.tomakehurst.wiremock.matching.ContainsPattern;
import com.github.tomakehurst.wiremock.matching.RegexPattern;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;

public class WireMockTestHelper {

    public static void setUpWireMockForCall(String call, String responseFile, String... params) throws IOException {

        String responseBody = loadFile("/responses/" + responseFile);

        StringBuilder regexPattern = new StringBuilder()
                .append(".*\"method\":\"").append(call).append("\".*")
                .append(".*\"params\":.*");
        for (String param : params) {
            regexPattern.append(".*").append(param).append(".*");
        }
        WireMock.stubFor(WireMock.post(WireMock.urlEqualTo("/"))
                .withRequestBody(new RegexPattern(regexPattern.toString()))
                .willReturn(WireMock.aResponse()
                        .withStatus(200)
                        .withBody(responseBody)));
    }

    public static void setUpWireMockForGetBlockCount(long blockCount) throws IOException {
        String responseBody = loadFile("/responses/getblockcount_" + blockCount + ".json");

        WireMock.stubFor(WireMock.post(WireMock.urlEqualTo("/"))
                .withRequestBody(new ContainsPattern("\"method\":\"getblockcount\""))
                .willReturn(WireMock.aResponse()
                        .withStatus(200)
                        .withBody(responseBody)));
    }

    public static void setUpWireMockForInvokeFunction(String contractFunction, String responseFile)
            throws IOException {

        String responseBody = loadFile("/responses/" + responseFile);

        WireMock.stubFor(WireMock.post(WireMock.urlEqualTo("/"))
                .withRequestBody(new RegexPattern(""
                        + ".*\"method\":\"invokefunction\""
                        + ".*\"params\":.*\"" + contractFunction +
                        "\".*"))
                .willReturn(WireMock.aResponse()
                        .withStatus(200)
                        .withBody(responseBody)));
    }

    public static void setUpWireMockForBalanceOf(String accountScriptHash, String responseFile)
            throws IOException {

        String responseBody = loadFile("/responses/" + responseFile);

        WireMock.stubFor(WireMock.post(WireMock.urlEqualTo("/"))
                .withRequestBody(new RegexPattern(""
                        + ".*\"method\":\"invokefunction\""
                        + ".*\"params\":.*\"balanceOf\".*"
                        + ".*\"" + accountScriptHash + "\".*"))
                .willReturn(WireMock.aResponse()
                        .withStatus(200)
                        .withBody(responseBody)));
    }

    public static String loadFile(String fileName) throws IOException {
        String absFileName = WireMockTestHelper.class.getResource(fileName).getFile();
        FileInputStream inStream = new FileInputStream(new File(absFileName));
        return Files.lines(new File(absFileName).toPath(), StandardCharsets.UTF_8).reduce((a, b) -> a + b).get();
    }

}