From dd43f001bf0bc90045c260f272d47676950cc040 Mon Sep 17 00:00:00 2001 From: Han Xiao Date: Thu, 9 Apr 2020 13:10:54 +0200 Subject: [PATCH] refactor: log server config Signed-off-by: Han Xiao --- tests/test_flow.py | 4 ++-- tests/yaml/test-server-config.yml | 10 ++++++++++ 2 files changed, 12 insertions(+), 2 deletions(-) create mode 100644 tests/yaml/test-server-config.yml diff --git a/tests/test_flow.py b/tests/test_flow.py index fa36c5b7e24a4..b22846dc7acb0 100644 --- a/tests/test_flow.py +++ b/tests/test_flow.py @@ -113,13 +113,13 @@ def test_flow_no_container(self): fl.index(raw_bytes=random_docs(10), in_proto=True) def test_flow_yaml_dump(self): - f = Flow(port_sse=12345, + f = Flow(logserver_config='yaml/test-server-config.yml', optimize_level=FlowOptimizeLevel.IGNORE_GATEWAY, no_gateway=True) f.save_config('test1.yml') fl = Flow.load_config('test1.yml') - self.assertEqual(f.args.port_sse, fl.args.port_sse) + self.assertEqual(f.args.logserver_config, fl.args.logserver_config) self.assertEqual(f.args.no_gateway, fl.args.no_gateway) self.assertEqual(f.args.optimize_level, fl.args.optimize_level) self.add_tmpfile('test1.yml') diff --git a/tests/yaml/test-server-config.yml b/tests/yaml/test-server-config.yml new file mode 100644 index 0000000000000..a99f7e95e4aeb --- /dev/null +++ b/tests/yaml/test-server-config.yml @@ -0,0 +1,10 @@ +host: 0.0.0.0 +port: 5000 +endpoints: + log: /log/stream # fetching log in SSE stream + profile: /stream/profile # fetching profiling log in SSE stream + yaml: /yaml # get the YAML spec of a flow + shutdown: /action/shutdown # shutdown the log server + ready: /action/isready # tell if the log server is ready, return 200 if yes + +