Commit b6e94bde authored by Dylan Baker's avatar Dylan Baker

framework/profile: Allow a group manager class to be overwritten

This is useful for classes like the coming BuiltInConstantsTest class,
where just one or two tests need to change the test class.
Tested-by: Rafael Antognolli's avatarRafael Antognolli <rafael.antognolli@intel.com>
parent 30909302
......@@ -205,7 +205,7 @@ class TestDict(collections.MutableMapping):
"""
assert isinstance(group, six.string_types), type(group)
def adder(args, name=None, **kwargs):
def adder(args, name=None, override_class=None, **kwargs):
"""Helper function that actually adds the tests.
Arguments:
......@@ -228,7 +228,9 @@ class TestDict(collections.MutableMapping):
assert isinstance(name, six.string_types)
lgroup = grouptools.join(group, name)
self[lgroup] = test_class(
class_ = override_class or test_class
self[lgroup] = class_(
args,
**dict(itertools.chain(six.iteritems(default_args),
six.iteritems(kwargs))))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment