Add more tests

tests/aio/test_threaded_client.py

@@ -11,8 +11,13 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License
 import pickle
+import sys
+from unittest import mock
+
+import pytest
 
 from reportportal_client.aio import ThreadedRPClient
+from reportportal_client.helpers import timestamp
 
 
 def test_threaded_rp_client_pickling():
@@ -67,3 +72,64 @@ def test_clone():
     )
     assert cloned._item_stack.qsize() == 1 \
            and async_client.current_item() == cloned.current_item()
+
+
+@pytest.mark.skipif(sys.version_info < (3, 8),
+                    reason='the test requires AsyncMock which was introduced in Python 3.8')
+@pytest.mark.parametrize(
+    'launch_uuid, method, params',
+    [
+        ('test_launch_uuid', 'start_test_item', ['Test Item', timestamp(), 'STEP']),
+        ('test_launch_uuid', 'finish_test_item', ['test_item_id', timestamp()]),
+        ('test_launch_uuid', 'get_launch_info', []),
+        ('test_launch_uuid', 'get_launch_ui_id', []),
+        ('test_launch_uuid', 'get_launch_ui_url', []),
+        ('test_launch_uuid', 'log', [timestamp(), 'Test message']),
+        (None, 'start_test_item', ['Test Item', timestamp(), 'STEP']),
+        (None, 'finish_test_item', ['test_item_id', timestamp()]),
+        (None, 'get_launch_info', []),
+        (None, 'get_launch_ui_id', []),
+        (None, 'get_launch_ui_url', []),
+        (None, 'log', [timestamp(), 'Test message']),
+    ]
+)
+def test_launch_uuid_usage(launch_uuid, method, params):
+    started_launch_uuid = 'new_test_launch_uuid'
+    aio_client = mock.AsyncMock()
+    aio_client.start_launch.return_value = started_launch_uuid
+    client = ThreadedRPClient('http://endpoint', 'project', api_key='api_key',
+                              client=aio_client, launch_uuid=launch_uuid, log_batch_size=1)
+    actual_launch_uuid = (client.start_launch('Test Launch', timestamp())).blocking_result()
+    getattr(client, method)(*params).blocking_result()
+    finish_launch_message = (client.finish_launch(timestamp())).blocking_result()
+
+    if launch_uuid is None:
+        aio_client.start_launch.assert_called_once()
+        assert actual_launch_uuid == started_launch_uuid
+        assert client.launch_uuid.blocking_result() == started_launch_uuid
+        aio_client.finish_launch.assert_called_once()
+        assert finish_launch_message
+    else:
+        aio_client.start_launch.assert_not_called()
+        assert actual_launch_uuid == launch_uuid
+        assert client.launch_uuid.blocking_result() == launch_uuid
+        aio_client.finish_launch.assert_not_called()
+        assert finish_launch_message == ''
+    assert client.launch_uuid.blocking_result() == actual_launch_uuid
+
+    if method == 'log':
+        assert len(getattr(aio_client, 'log_batch').call_args_list) == 2
+        args, kwargs = getattr(aio_client, 'log_batch').call_args_list[0]
+        batch = args[0]
+        assert isinstance(batch, list)
+        assert len(batch) == 1
+        log = batch[0]
+        assert log.launch_uuid.blocking_result() == actual_launch_uuid
+        assert log.time == params[0]
+        assert log.message == params[1]
+    else:
+        getattr(aio_client, method).assert_called_once()
+        args, kwargs = getattr(aio_client, method).call_args_list[0]
+        assert args[0].blocking_result() == actual_launch_uuid
+        for i, param in enumerate(params):
+            assert args[i + 1] == param