add set environment variable for log

python/paddle_serving_server/env_check/run.py

@@ -31,6 +31,11 @@ inference_test_cases = ["test_fit_a_line.py::TestFitALine::test_inference"]
 cpp_test_cases = ["test_fit_a_line.py::TestFitALine::test_cpu", "test_fit_a_line.py::TestFitALine::test_gpu"]
 pipeline_test_cases = ["test_uci_pipeline.py::TestUCIPipeline::test_cpu", "test_uci_pipeline.py::TestUCIPipeline::test_gpu"]
 
+def set_serving_log_path():
+    if 'SERVING_LOG_PATH' not in os.environ:
+        serving_log_path = os.path.expanduser(os.getcwd())
+        os.environ['SERVING_LOG_PATH']=serving_log_path
+
 def run_test_cases(cases_list, case_type, is_open_std):
     old_stdout, old_stderr = sys.stdout, sys.stderr
     real_path = os.path.dirname(os.path.realpath(__file__))
@@ -56,7 +61,7 @@ def unset_env(key):
     del os.environ[key]
 
 def check_env(mode):
-    
+    set_serving_log_path()
     if 'https_proxy' in os.environ or 'http_proxy' in os.environ:
         unset_env("https_proxy") 
         unset_env("http_proxy")     

python/paddle_serving_server/env_check/test_fit_a_line.py

@@ -90,6 +90,7 @@ class TestFitALine(object):
         return output_dict
 
     def test_inference(self):
+        self.serving_util.start_server_by_shell(cmd="", sleep=1)
         assert self.truth_val['price'].size != 0, "The result of inference is empty"
 
 

python/paddle_serving_server/env_check/test_uci_pipeline.py

@@ -105,9 +105,6 @@ class TestUCIPipeline(object):
         assert count_process_num_on_port(9998) == 1, "Error occured when Paddle Server started"  # gRPC Server
         assert count_process_num_on_port(18082) == 1, "Error occured when Paddle Server started"  # gRPC gateway
 
-        # 3.keywords check
-        check_keywords_in_server_log("MKLDNN is enabled", filename="stderr.log")
-
         # 4.predict by rpc
         result = self.predict_pipeline_rpc(batch_size=1)
         self.serving_util.check_result(result_data=result, truth_data=self.truth_val, batch_size=1)

python/pipeline/logger.py

@@ -24,8 +24,11 @@ class SectionLevelFilter(object):
     def filter(self, logRecord):
         return logRecord.levelno in self._levels
 
-
 log_dir = "PipelineServingLogs"
+if 'SERVING_LOG_PATH' in os.environ:
+    serving_log_path = os.environ['SERVING_LOG_PATH']
+    log_dir = os.path.join(serving_log_path, log_dir)
+
 if not os.path.exists(log_dir):
     os.makedirs(log_dir)