def send_logs( self, tracker_issue: TrackerIssue, logs: List[Log], ) -> SendLogsResult: """ Send logs to the tracker. Args: tracker_issue: information about the tracker issue logs: a list of logs Returns: information about the sent logs """ jira_issue = self._get_issue(issue_id=tracker_issue.issue_id, ) tracker_comments = SendLogsResult( tracker_issue=tracker_issue, added_comments=[], ) for log in logs: jira_comment = self._add_comment( issue=jira_issue, log=log, ) tracker_comments.added_comments.append( TrackerIssueComment( created_at=self._parse_date(date=jira_comment.created, ), author=jira_comment.author.displayName, comment_id=str(jira_comment.id), body=jira2markdown(src=jira_comment.body, ), attachments={}, ), ) return tracker_comments
def send_logs( self, tracker_issue: TrackerIssue, logs: Iterable[Log], ) -> SendLogsResult: """ Send logs to the tracker. Args: tracker_issue: information about the tracker issue logs: a list of comments Raises: GitHubTrackerClientError: if an error occurs Returns: information about the sent comments """ self._ensure_auth() tracker_comments = SendLogsResult( tracker_issue=tracker_issue, added_comments=[], ) github_issue = self._get_github_issue( issue_id=tracker_issue.issue_id, ) if not github_issue: raise GitHubTrackerClientError( f'GitHub issue {tracker_issue.issue_id} not found') for log in logs: github_comment = self._add_comment( github_issue=github_issue, log=log, ) self._upload_comment_attachments( issue=github_issue, comment=github_comment, attachments=log.attachments, ) tracker_comments.added_comments.append( TrackerIssueComment( author=github_comment.user.name or github_comment.user.login, created_at=self._ensure_timezone( dt=github_comment.created_at, tz=self._default_timezone, ), comment_id=str(github_comment.id), body=github_comment.body, attachments={}, ), ) return tracker_comments
def send_logs( self, tracker_issue: TrackerIssue, logs: Iterable[Log], ) -> SendLogsResult: """ Send logs to the tracker. Args: tracker_issue: information about the tracker issue logs: a list of comments Raises: GitLabTrackerClientError: if an error occurs Returns: information about the sent comments """ self._ensure_auth() tracker_comments = SendLogsResult( tracker_issue=tracker_issue, added_comments=[], ) gitlab_project = self._get_gitlab_project() gitlab_issue = self._get_gitlab_issue( issue_id=tracker_issue.issue_id, ) if not gitlab_issue: raise GitLabTrackerClientError( f'GitLab issue {tracker_issue.issue_id} not found in project {self.configuration.project}', ) for log in logs: gitlab_comment = self._add_comment( gitlab_project=gitlab_project, gitlab_issue=gitlab_issue, log=log, ) tracker_comments.added_comments.append( TrackerIssueComment( created_at=self._parse_date( date=gitlab_comment.created_at, ), author=gitlab_comment.author.get( 'name', self._default_author_name), comment_id=str(gitlab_comment.id), body=gitlab_comment.body, attachments={}, ), ) return tracker_comments
def _extract_comment( self, gitlab_note: ProjectIssueNote, ) -> TrackerIssueComment: gitlab_body = gitlab_note.body inline_images = _RE_IMAGE.findall(gitlab_body) comment_attachments: Dict[str, TrackerAttachment] = {} for _, inline_path in inline_images: attachment = self._download_attachment(path=inline_path, ) if attachment: comment_attachments[inline_path] = attachment return TrackerIssueComment( created_at=self._parse_date(date=gitlab_note.created_at, ), author=gitlab_note.author.get('name', self._default_author_name), comment_id=str(gitlab_note.id), body=gitlab_body, attachments=comment_attachments, )
def _extract_comment( self, github_comment: IssueComment, ) -> TrackerIssueComment: github_body = github_comment.body inline_images = _RE_IMAGE.findall(github_body) comment_attachments: Dict[str, TrackerAttachment] = {} for _, inline_url in inline_images: attachment = self._download_attachment(url=inline_url, ) if attachment: comment_attachments[inline_url] = attachment return TrackerIssueComment( created_at=self._ensure_timezone( dt=github_comment.created_at, tz=self._default_timezone, ), author=github_comment.user.name or github_comment.user.login, comment_id=str(github_comment.id), body=github_body, attachments=comment_attachments, )
def _extract_comment( self, jira_issue: JIRAIssue, jira_comment: JIRAComment, ) -> TrackerIssueComment: jira_body = jira_comment.body inline_images = _RE_IMAGE.findall(jira_body) comment_attachments: Dict[str, TrackerAttachment] = {} for jira_attachment in jira_issue.fields.attachment: if jira_attachment.filename in inline_images: comment_attachments[ jira_attachment.filename] = TrackerAttachment( filename=jira_attachment.filename, mime_type=jira_attachment.mimeType, content=jira_attachment.get(), ) return TrackerIssueComment( created_at=self._parse_date(date=jira_comment.created, ), author=jira_comment.author.displayName, comment_id=str(jira_comment.id), body=jira2markdown(src=jira_body, ), attachments=comment_attachments, )
def test_partially_synced(self, ) -> None: comment_log1 = CommentLog( created_at='2021-01-01T00:00:00+00:00', log_id=1, log_type='comment', private=True, author=Author(username='******', ), message_html='This is a comment', attachments=[], ) tracking_status = TrackingStatusLog( created_at='2021-01-01T00:30:00+00:00', log_id=1, log_type='tracking-status', private=True, author=Author(username='******', ), message_html='Tracked', attachments=[], tracker_name='my-tracker', tracker_url='http://tracker/issue/1', tracker_id='1', ) tracker_update_log1 = TrackerUpdateLog( created_at='2021-01-01T01:00:00+00:00', log_id=2, log_type='tracker-update', private=True, author=Author(username='******', ), message_html='This is a a tracker update', attachments=[], tracker_name='my-tracker', tracker_id='1', tracker_url='http://tracker/issue/1', tracker_token=StateEncryptor.encrypt( key='123', state=TrackerIssueState( closed=False, bugtracker_name='my-tracker', ), ), ) comment_log2 = CommentLog( created_at='2021-01-01T02:00:00+00:00', log_id=3, log_type='comment', private=True, author=Author(username='******', ), message_html='This is another comment', attachments=[], ) report = self._build_report( report_id=123, tracking_status='T', logs=[ comment_log1, tracking_status, tracker_update_log1, comment_log2, ], ) ywh_api_client_mock = create_autospec(YesWeHackApiClient, spec_set=True) tracker_client_mock = create_autospec(TrackerClient, spec_set=True) tracker_client_mock.tracker_type = 'MyTracker' issue = TrackerIssue( tracker_url='http://tracker/issue/1', project='my-project', issue_id='1', issue_url='http://tracker/issue/1', closed=False, ) tracker_client_mock.get_tracker_issue.return_value = issue tracker_client_mock.send_logs.return_value = SendLogsResult( tracker_issue=issue, added_comments=[ TrackerIssueComment( created_at=datetime.datetime( year=2020, month=1, day=1, hour=15, minute=17, second=23, microsecond=420000, tzinfo=datetime.timezone.utc, ), author='tracker-user', comment_id='456', body='', attachments={}, ), ], ) Given( case=self, report=report, yeswehack_client=ywh_api_client_mock, tracker_name='my-tracker', tracker_client=tracker_client_mock, synchronize_options=SynchronizeOptions( upload_private_comments=True, upload_public_comments=True, upload_details_updates=True, upload_rewards=True, upload_status_updates=True, ), feedback_options=FeedbackOptions(), message_formatter=SimpleMessageFormatter( tracking_status_update_format= 'issue url: {tracker_issue.issue_url}', synchronization_done_format= 'issue url: {send_logs_result.tracker_issue.issue_url}', download_comment_format='comment: {comment}', status_update_comment_format='comment: {comment}', ), ).when_synchronize_report().then_assert_no_error( ).then_assert_has_result().then_assert_is_existing_issue( ).then_assert_tracker_client_send_report_not_called( ).then_assert_tracker_client_send_logs_called_once_with( tracker_issue=issue, logs=[ comment_log2, ], ).then_assert_yeswehack_client_post_report_tracker_update_called_once_with( report=report, tracker_name='my-tracker', issue_id='1', issue_url='http://tracker/issue/1', comment='issue url: http://tracker/issue/1', token=StateEncryptor.encrypt( key='123', state=TrackerIssueState(closed=False, bugtracker_name='my-tracker', downloaded_comments=['456']), ), )