Add 'mongo_collection' to template_fields in MongoToS3Operator (#13361)

This commit is contained in:
Adrián Matellanes 2021-01-02 11:32:07 +01:00 коммит произвёл GitHub
Родитель e436883583
Коммит f7a1334abe
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
2 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

@ -38,7 +38,7 @@ class MongoToS3Operator(BaseOperator):
to perform transformations unique to those operators needs
"""
template_fields = ['s3_key', 'mongo_query']
template_fields = ['s3_key', 'mongo_query', 'mongo_collection']
# pylint: disable=too-many-instance-attributes
@apply_defaults

Просмотреть файл

@ -65,7 +65,7 @@ class TestMongoToS3Operator(unittest.TestCase):
self.assertEqual(self.mock_operator.s3_key, S3_KEY)
def test_template_field_overrides(self):
self.assertEqual(self.mock_operator.template_fields, ['s3_key', 'mongo_query'])
self.assertEqual(self.mock_operator.template_fields, ['s3_key', 'mongo_query', 'mongo_collection'])
def test_render_template(self):
ti = TaskInstance(self.mock_operator, DEFAULT_DATE)