Class RPCTest

jvm/core/src/test/java/org/apache/tvm/rpc/RPCTest.java:31–120

import org.slf4j.LoggerFactory;

public class RPCTest {
  private final Logger logger = LoggerFactory.getLogger(RPCTest.class);

  @Ignore("RPC test is not enabled")
  @Test
  public void test_addone() {
    if (!Module.enabled("rpc")) {
      logger.warn("RPC is not enabled. Skip.");
      return;
    }
    Function.register("test.rpc.addone", new Function.Callback() {
      @Override
      public Object invoke(TVMValue... args) {
        return args[0].asLong() + 1L;
      }
    });

    TestUtils.RefInt port = new TestUtils.RefInt();
    Server server = null;
    try {
      server = TestUtils.startServer(port);
      RPCSession client = Client.connect("127.0.0.1", port.value);
      Function func = client.getFunction("test.rpc.addone");
      assertEquals(11L, func.call(10).asLong());
    } finally {
      if (server != null) {
        server.terminate();
      }
    }
  }

  @Ignore("RPC test is not enabled")
  @Test
  public void test_strcat() {
    if (!Module.enabled("rpc")) {
      logger.warn("RPC is not enabled. Skip.");
      return;
    }
    Function.register("test.rpc.strcat", new Function.Callback() {
      @Override
      public Object invoke(TVMValue... args) {
        return args[0].asString() + ":" + args[1].asLong();
      }
    });

    TestUtils.RefInt port = new TestUtils.RefInt();
    Server server = null;
    try {
      server = TestUtils.startServer(port);
      RPCSession client = Client.connect("127.0.0.1", port.value);
      Function func = client.getFunction("test.rpc.strcat");
      assertEquals("abc:11", func.call("abc", 11L).asString());
    } finally {
      if (server != null) {
        server.terminate();
      }
    }
  }