Method of ReportPortalService to get launch UI URL

Author: NikitaFedoraev

reportportal_client/service.py

@@ -249,6 +249,36 @@ def finish_launch(self, end_time, status=None, **kwargs):
         logger.debug("finish_launch - ID: %s", self.launch_id)
         return _get_msg(r)
 
+    def get_launch_info(self):
+        """Get the current launch information.
+
+        :return dict: launch information
+        """
+        url = uri_join(self.base_url_v1, "launch/uuid", self.launch_id)
+        launch_info = _get_json(self.session.get(
+            url=url, verify=self.verify_ssl))
+        logger.debug("get_launch_info - ID: %s", self.launch_id)
+        logger.debug("get_launch_info - Launch info: %s", launch_info)
+        return launch_info
+
+    def get_launch_ui_id(self):
+        """Get UI ID of the current launch.
+
+        :return str: UI ID of the given launch.
+        """
+        return self.get_launch_info()["id"]
+
+    def get_launch_ui_url(self):
+        """Get UI URL of the current launch.
+
+        :return str: launch URL.
+        """
+        ui_id = self.get_launch_ui_id()
+        path = "ui/#{0}/launches/all/{1}".format(self.project, ui_id)
+        url = uri_join(self.endpoint, path)
+        logger.debug("get_launch_ui_url - ID: %s", self.launch_id)
+        return url
+
     def start_test_item(self,
                         name,
                         start_time,

tests/test_service.py

@@ -87,6 +87,53 @@ def test_finish_launch(self, mock_get, rp_service):
             'name', datetime.now().isoformat())
         assert _get_msg == {'id': 111}
 
+    @mock.patch('reportportal_client.service._get_json',
+                mock.Mock(return_value={'id': 112}))
+    def test_get_launch_info(self, rp_service, monkeypatch):
+        """Test get current launch information.
+
+        :param rp_service:  Pytest fixture that represents ReportPortalService
+                            object with mocked session.
+        :param monkeypatch: Pytest fixture to safely set/delete an attribute
+        """
+        mock_get = mock.Mock(return_value={'id': 112})
+        monkeypatch.setattr(rp_service.session, 'get', mock_get)
+
+        launch_id = rp_service.get_launch_info()
+        mock_get.assert_called_once_with(
+            url='{0}/launch/uuid/{1}'.format(rp_service.base_url_v1,
+                                             rp_service.launch_id),
+            verify=rp_service.verify_ssl)
+        assert launch_id == {'id': 112}
+
+    def test_get_launch_ui_id(self, rp_service, monkeypatch):
+        """Test get launch UI ID.
+
+        :param rp_service:  Pytest fixture that represents ReportPortalService
+                            object with mocked session.
+        :param monkeypatch: Pytest fixture to safely set/delete an attribute
+        """
+        mock_get_launch_info = mock.Mock(return_value={'id': 113})
+        monkeypatch.setattr(rp_service,
+                            'get_launch_info',
+                            mock_get_launch_info)
+        assert rp_service.get_launch_ui_id() == 113
+
+    def test_get_launch_ui_url(self, rp_service, monkeypatch):
+        """Test get launch UI URL.
+
+        :param rp_service:  Pytest fixture that represents ReportPortalService
+                            object with mocked session.
+        :param monkeypatch: Pytest fixture to safely set/delete an attribute
+        """
+        mock_get_launch_ui_id = mock.Mock(return_value=1)
+        monkeypatch.setattr(rp_service,
+                            'get_launch_ui_id',
+                            mock_get_launch_ui_id)
+        url = rp_service.get_launch_ui_url()
+        assert url == '{0}/ui/#{1}/launches/all/1'.format(rp_service.endpoint,
+                                                          rp_service.project)
+
     @mock.patch('platform.system', mock.Mock(return_value='linux'))
     @mock.patch('platform.machine', mock.Mock(return_value='Windows-PC'))
     @mock.patch('platform.processor', mock.Mock(return_value='amd'))