def test_execute(self, request_mock):
resp = requests.Response()
resp.status_code = 200
request_mock.return_value = resp
operator = HipChatAPISendRoomNotificationOperator(
task_id='test_hipchat_success',
owner='airflow',
token='abc123',
room_id='room_id',
message='hello world!')
operator.execute(None)
def test_execute(self, request_mock):
resp = requests.Response()
resp.status_code = 200
request_mock.return_value = resp
operator = HipChatAPISendRoomNotificationOperator(
task_id='test_hipchat_success',
owner='airflow',
token='abc123',
room_id='room_id',
message='hello world!'
)
operator.execute(None)
def test_execute_error_response(self, request_mock):
resp = requests.Response()
resp.status_code = 404
resp.reason = 'Not Found'
request_mock.return_value = resp
operator = HipChatAPISendRoomNotificationOperator(
task_id='test_hipchat_failure',
owner='airflow',
token='abc123',
room_id='room_id',
message='hello world!')
with self.assertRaises(AirflowException):
operator.execute(None)
def test_execute_error_response(self, request_mock):
resp = requests.Response()
resp.status_code = 404
resp.reason = 'Not Found'
request_mock.return_value = resp
operator = HipChatAPISendRoomNotificationOperator(
task_id='test_hipchat_failure',
owner='airflow',
token='abc123',
room_id='room_id',
message='hello world!'
)
with self.assertRaises(AirflowException):
operator.execute(None)
