DispatchCallOnTickTest.java

package io.roastedroot.proxywasm;

import static org.junit.jupiter.api.Assertions.assertEquals;

import com.dylibso.chicory.wasm.Parser;
import java.nio.file.Path;
import java.util.Map;
import org.junit.jupiter.api.Test;

/**
 * Java port of https://github.com/proxy-wasm/proxy-wasm-go-sdk/blob/ab4161dcf9246a828008b539a82a1556cf0f2e24/examples/dispatch_call_on_tick/main_test.go
 */
public class DispatchCallOnTickTest {

    static int tickMilliseconds = 100;

    @Test
    public void testOnTick() throws StartException {

        var handler = new MockHandler();
        var module =
                Parser.parse(Path.of("./src/test/go-examples/dispatch_call_on_tick/main.wasm"));
        ProxyWasm.Builder builder = ProxyWasm.builder().withPluginHandler(handler);
        try (var host = builder.build(module)) {
            assertEquals(tickMilliseconds, handler.getTickPeriodMilliseconds());

            for (int i = 1; i <= 10; i++) {
                host.tick(); // call OnTick
            }

            assertEquals(10, handler.getHttpCalls().size());
            handler.getHttpCalls().entrySet().stream()
                    .forEach(
                            entry -> {
                                host.sendHttpCallResponse(
                                        entry.getKey(), Map.of(), Map.of(), new byte[0]);
                            });

            // Check Envoy logs.
            for (int i = 1; i <= 10; i++) {
                handler.assertLogsContain(
                        String.format("called %d for contextID=%d", i, host.contextId()));
            }
        }
    }
}