def test_issue_label_set(self, jira_mock):
jira_mock.return_value.issue.return_value = minimal_test_ticket
ticket_label_sensor = JiraTicketSensor(
task_id='search-ticket-test',
ticket_id='TEST-1226',
field_checker_func=TestJiraSensor.field_checker_func,
timeout=518400,
poke_interval=10,
dag=self.dag)
ticket_label_sensor.run(start_date=DEFAULT_DATE,
end_date=DEFAULT_DATE, ignore_ti_state=True)
self.assertTrue(jira_mock.called)
self.assertTrue(jira_mock.return_value.issue.called)
def test_issue_label_set(self, jira_mock):
jira_mock.return_value.issue.return_value = minimal_test_ticket
ticket_label_sensor = JiraTicketSensor(task_id='search-ticket-test',
ticket_id='TEST-1226',
field_checker_func=
TestJiraSensor.field_checker_func,
timeout=518400,
poke_interval=10,
dag=self.dag)
ticket_label_sensor.run(start_date=DEFAULT_DATE,
end_date=DEFAULT_DATE, ignore_ti_state=True)
self.assertTrue(jira_mock.called)
self.assertTrue(jira_mock.return_value.issue.called)