forked from mirrors/gecko-dev
53b7bc583c
This requires moving the schema utilities to their own util module. MozReview-Commit-ID: KR5xSJ9ak5Y --HG-- extra : rebase_source : 1c1f6bfb6a08deb8c0be4b2b58db02d85aafeb89
42 lines
978 B
Python
42 lines
978 B
Python
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
from __future__ import absolute_import, print_function, unicode_literals
|
|
|
|
import unittest
|
|
from mozunit import main
|
|
from taskgraph.transforms.base import (
|
|
TransformSequence
|
|
)
|
|
|
|
transforms = TransformSequence()
|
|
|
|
|
|
@transforms.add
|
|
def trans1(config, tests):
|
|
for test in tests:
|
|
test['one'] = 1
|
|
yield test
|
|
|
|
|
|
@transforms.add
|
|
def trans2(config, tests):
|
|
for test in tests:
|
|
test['two'] = 2
|
|
yield test
|
|
|
|
|
|
class TestTransformSequence(unittest.TestCase):
|
|
|
|
def test_sequence(self):
|
|
tests = [{}, {'two': 1, 'second': True}]
|
|
res = list(transforms({}, tests))
|
|
self.assertEqual(res, [
|
|
{u'two': 2, u'one': 1},
|
|
{u'second': True, u'two': 2, u'one': 1},
|
|
])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|