def test_update_issue(self, jira_mock):
jira_mock.return_value.add_comment.return_value = True
add_comment_operator = JiraOperator(
task_id='add_comment_test',
jira_method="add_comment",
jira_method_args={'issue': minimal_test_ticket.get("key"), 'body': 'this is test comment'},
dag=self.dag,
)
add_comment_operator.run(start_date=DEFAULT_DATE, end_date=DEFAULT_DATE, ignore_ti_state=True)
self.assertTrue(jira_mock.called)
self.assertTrue(jira_mock.return_value.add_comment.called)
def test_issue_search(self, jira_mock):
jql_str = 'issuekey=TEST-1226'
jira_mock.return_value.search_issues.return_value = minimal_test_ticket
jira_ticket_search_operator = JiraOperator(
task_id='search-ticket-test',
jira_method="search_issues",
jira_method_args={'jql_str': jql_str, 'maxResults': '1'},
dag=self.dag,
)
jira_ticket_search_operator.run(start_date=DEFAULT_DATE, end_date=DEFAULT_DATE, ignore_ti_state=True)
self.assertTrue(jira_mock.called)
self.assertTrue(jira_mock.return_value.search_issues.called)