add example for using server (#345)

LazyAGI · Nov 14, 2024 · 7ec1ca2 · 7ec1ca2
1 parent 89eaf71
commit 7ec1ca2
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/basic_tests/test_engine.py b/tests/basic_tests/test_engine.py
@@ -392,6 +392,12 @@ def test_engine_server(self):
         assert engine.status(gid) == {'1': 'running', '2': lazyllm.launcher.Status.Running, '3': 'running'}
         assert engine.run(gid, 1) == 2
         time.sleep(3)
+
+        server = engine.build_node('graph-1').func._g
+        assert isinstance(server, lazyllm.ServerModule)
+        m = lazyllm.UrlModule(url=server._url)
+        assert m(2) == 4
+
         web = engine.build_node('graph-1').func._web
         assert engine.build_node('graph-1').func.api_url is not None
         assert engine.build_node('graph-1').func.web_url == web.url