gecko-dev/taskcluster/taskgraph/test/test_transforms_base.py
Dustin J. Mitchell ffca84ff7d Bug 1281004: Specify test tasks more flexibly; r=gps; r=gbrown
This introduces a completely new way of specifying test task in-tree,
completely replacing the old spider-web of YAML files.

The high-level view is this:

 - some configuration files are used to determine which test suites to run
   for each test platform, and against which build platforms

 - each test suite is then represented by a dictionary, and modified by a
   sequence of transforms, duplicating as necessary (e.g., chunks), until
   it becomes a task definition

The transforms allow sufficient generality to support just about any desired
configuration, with the advantage that common configurations are "easy" while
unusual configurations are supported but notable for their oddness (they
require a custom transform).

As of this commit, this system produces the same set of test graphs as the
existing YAML, modulo:

  - extra.treeherder.groupName -- this was not consistent in the YAML
  - extra.treeherder.build -- this is ignored by taskcluster-treeherder anyway
  - mozharness command argument order
  - boolean True values for environment variables are now the string "true"
  - metadata -- this is now much more consistent, with task name being the label

Testing of this commit demonstrates that it produces the same set of test tasks for
the following projects (those which had special cases defined in the YAML):

  - autoland
  - ash (*)
  - willow
  - mozilla-inbound
  - mozilla-central
  - try:
    -b do -p all -t all -u all
    -b d -p linux64,linux64-asan -u reftest -t none
    -b d -p linux64,linux64-asan -u reftest[x64] -t none[x64]

(*) this patch omits the linux64/debug tc-M-e10s(dt) test, which is enabled on
ash; ash will require a small changeset to re-enable this test.

IGNORE BAD COMMIT MESSAGES (because the hook flags try syntax!)

MozReview-Commit-ID: G34dg9f17Hq

--HG--
rename : taskcluster/taskgraph/kind/base.py => taskcluster/taskgraph/task/base.py
rename : taskcluster/taskgraph/kind/docker_image.py => taskcluster/taskgraph/task/docker_image.py
rename : taskcluster/taskgraph/kind/legacy.py => taskcluster/taskgraph/task/legacy.py
extra : rebase_source : 03e70902c2d3a297eb9e3ce852f8737c2550d5a6
extra : histedit_source : d4d9f4b192605af21f41d83495fc3c923759c3cb
2016-07-11 23:27:14 +00:00

129 lines
3.2 KiB
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 (
validate_schema,
get_keyed_by,
TransformSequence
)
from voluptuous import Schema
schema = Schema({
'x': int,
'y': basestring,
})
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},
])
class TestValidateSchema(unittest.TestCase):
def test_valid(self):
validate_schema(schema, {'x': 10, 'y': 'foo'}, "pfx")
def test_invalid(self):
try:
validate_schema(schema, {'x': 'not-int'}, "pfx")
self.fail("no exception raised")
except Exception, e:
self.failUnless(str(e).startswith("pfx\n"))
class TestKeyedBy(unittest.TestCase):
def test_simple_value(self):
test = {
'test-name': 'tname',
'option': 10,
}
self.assertEqual(get_keyed_by(test, 'option', 'x'), 10)
def test_by_value(self):
test = {
'test-name': 'tname',
'option': {
'by-other-value': {
'a': 10,
'b': 20,
},
},
'other-value': 'b',
}
self.assertEqual(get_keyed_by(test, 'option', 'x'), 20)
def test_by_value_default(self):
test = {
'test-name': 'tname',
'option': {
'by-other-value': {
'a': 10,
'default': 30,
},
},
'other-value': 'xxx',
}
self.assertEqual(get_keyed_by(test, 'option', 'x'), 30)
def test_by_value_invalid_dict(self):
test = {
'test-name': 'tname',
'option': {
'by-something-else': {},
'by-other-value': {},
},
}
self.assertRaises(Exception, get_keyed_by, test, 'option', 'x')
def test_by_value_invalid_no_default(self):
test = {
'test-name': 'tname',
'option': {
'by-other-value': {
'a': 10,
},
},
'other-value': 'b',
}
self.assertRaises(Exception, get_keyed_by, test, 'option', 'x')
def test_by_value_invalid_no_by(self):
test = {
'test-name': 'tname',
'option': {
'other-value': {},
},
}
self.assertRaises(Exception, get_keyed_by, test, 'option', 'x')
if __name__ == '__main__':
main()