hunk.code_window:     # Test the right email is constructed.
       @mock.patch('os.environ.get')
       @mock.patch('airflow.contrib.utils.sendgrid._post_sendgrid_mail')
       def test_send_email_sendgrid_correct_email_extras(self, mock_post, mock_get):
           mock_get.return_value = 'foo@bar.com'
           send_email(self.to, self.subject, self.html_content, cc=self.cc, bcc=self.bcc,
                      personalization_custom_args=self.personalization_custom_args,
                      categories=self.categories)
  
hunk.labels: keep keep keep keep replace keep keep keep
hunk.after_edit:         def get_return(var):
               return {'SENDGRID_MAIL_FROM': 'foo@bar.com',
                       'SENDGRID_MAIL_SENDER': 'Foo'}.get(var)
           mock_get.side_effect = get_return
  
hunk.file_path: tests/contrib/utils/test_sendgrid.py
hunk.type: replace
file: # -*- coding: utf-8 -*-
  #
  # Licensed to the Apache Software Foundation (ASF) under one
  # or more contributor license agreements.  See the NOTICE file
  # distributed with this work for additional information
  # regarding copyright ownership.  The ASF licenses this file
  # to you under the Apache License, Version 2.0 (the
  # "License"); you may not use this file except in compliance
  # with the License.  You may obtain a copy of the License at
  # 
  #   http://www.apache.org/licenses/LICENSE-2.0
  # 
  # Unless required by applicable law or agreed to in writing,
  # software distributed under the License is distributed on an
  # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  # KIND, either express or implied.  See the License for the
  # specific language governing permissions and limitations
  # under the License.
  from datetime import datetime
  
  from airflow.models import DAG
  from airflow.operators.dummy_operator import DummyOperator
  
  DEFAULT_DATE = datetime(2016, 1, 1)
  
  args = {
      'owner': 'airflow',
      'start_date': DEFAULT_DATE,
  }
  
  dag = DAG(dag_id='test_localtaskjob_double_trigger', default_args=args)
  task = DummyOperator(
      task_id='test_localtaskjob_double_trigger_task',
      dag=dag)
  