test: simplify post test (#201)

Joan Fontanals · hanxiao · web-flow · commit 1b5f536ddb5f · 2022-03-15T08:47:09.000+01:00
* test: simplify post test

* style: fix black

Co-authored-by: Han Xiao &lt;han.xiao@jina.ai&gt;
diff --git a/tests/unit/array/mixins/test_post.py b/tests/unit/array/mixins/test_post.py
@@ -21,42 +21,18 @@
 def test_post_to_a_flow(show_pbar, conn_config, batch_size):
     from jina import Flow
 
-    def start_flow(stop_event, **kwargs):
-        """start a blocking Flow."""
-        with Flow(**kwargs) as f:
-            f.block(stop_event=stop_event)
-            print('bye')
-
-    e = multiprocessing.Event()  # create new Event
-
     p = random_port()
-    t = multiprocessing.Process(
-        name='Blocked-Flow',
-        target=start_flow,
-        args=(e,),
-        kwargs={**conn_config[0], 'port': p},
-    )
-    t.start()
-
-    time.sleep(3)
-
     da = DocumentArray.empty(100)
-    try:
+    with Flow(**{**conn_config[0], 'port': p}):
         da.post(conn_config[1].replace('$port', str(p)), batch_size=batch_size)
-    except:
-        raise
-    finally:
-        e.set()
-        t.join()
-        time.sleep(1)
 
 
 @pytest.mark.parametrize(
     'hub_uri',
     [
         'jinahub://Hello',
         'jinahub+sandbox://Hello',
-        #'jinahub+docker://Hello',  this somehow does not work on GH workflow
+        # 'jinahub+docker://Hello',  this somehow does not work on GH workflow
     ],
 )
 def test_post_with_jinahub(hub_uri):
@@ -95,7 +71,7 @@ def start_flow(stop_event, **kwargs):
     )
     t.start()
 
-    time.sleep(1)
+    time.sleep(5)
     N = 100
     da = DocumentArray.empty(N)
     try:
