launch_uuid_print and print_output argument tests

Author: HardNorth

reportportal_client/client.py

@@ -91,7 +91,7 @@ def __init__(
             log_batch_payload_size: int = MAX_LOG_BATCH_PAYLOAD_SIZE,
             mode: str = 'DEFAULT',
             launch_uuid_print: bool = False,
-            print_output: TextIO = sys.stdout,
+            print_output: Optional[TextIO] = None,
             **kwargs: Any
     ) -> None:
         """Initialize required attributes.
@@ -137,7 +137,7 @@ def __init__(
         self.mode = mode
         self._skip_analytics = getenv('AGENT_NO_ANALYTICS')
         self.launch_uuid_print = launch_uuid_print
-        self.print_output = print_output
+        self.print_output = print_output or sys.stdout
 
         self.api_key = api_key
         if not self.api_key:

tests/test_client.py

@@ -10,6 +10,7 @@
 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 #  See the License for the specific language governing permissions and
 #  limitations under the License
+import sys
 from io import StringIO
 
 import pytest
@@ -217,3 +218,25 @@ def test_no_launch_uuid_print():
     client._skip_analytics = True
     client.start_launch('Test Launch', timestamp())
     assert 'Report Portal Launch UUID: ' not in str_io.getvalue()
+
+
+@mock.patch('reportportal_client.client.sys.stdout', new_callable=StringIO)
+def test_launch_uuid_print_default_io(mock_stdout):
+    client = RPClient(endpoint='http://endpoint', project='project',
+                      api_key='test', launch_uuid_print=True)
+    client.session = mock.Mock()
+    client._skip_analytics = True
+    client.start_launch('Test Launch', timestamp())
+
+    assert 'Report Portal Launch UUID: ' in mock_stdout.getvalue()
+
+
+@mock.patch('reportportal_client.client.sys.stdout', new_callable=StringIO)
+def test_launch_uuid_print_default_print(mock_stdout):
+    client = RPClient(endpoint='http://endpoint', project='project',
+                      api_key='test')
+    client.session = mock.Mock()
+    client._skip_analytics = True
+    client.start_launch('Test Launch', timestamp())
+
+    assert 'Report Portal Launch UUID: ' not in mock_stdout.getvalue()