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)