forked from mirrors/gecko-dev
3d0d39d82c
For simplicity, this implements just on in `NO_TASKS` (the default) or
on in `ALL_TASKS` (opt-in). This disables all category registrations
when in background task mode; we'll selectively re-enable things as
appropriate.
The flag constants were chosen to smoothly extend to a (16-)bit set in
the future, should we want to add a `JUST_TASKS("task", "other-task")`
option in the future.
This also adds ython tests for gen_static_components.py exercising
categories, simply 'cuz it's easiest to see what this adds in such
tests. Functional tests will follow in patches that actually
implement the new background tasks functionality.
Differential Revision: https://phabricator.services.mozilla.com/D96654
137 lines
4.9 KiB
Python
137 lines
4.9 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/.
|
|
|
|
import mozunit
|
|
import os
|
|
import sys
|
|
import unittest
|
|
|
|
sys.path.append(os.path.join(os.path.dirname(__file__), ".."))
|
|
import gen_static_components
|
|
from gen_static_components import BackgroundTasksSelector
|
|
|
|
|
|
class TestGenStaticComponents(unittest.TestCase):
|
|
def test_string(self):
|
|
# A string: we default to NO_TASKS.
|
|
clas = {
|
|
"cid": "{a8566880-0bc7-4822-adb9-748c9af5cce7}",
|
|
"contract_ids": ["@mozilla.org/dummy-class;1"],
|
|
"jsm": "resource:///modules/DummyClass.jsm",
|
|
"js_name": "dummyClass",
|
|
"constructor": "DummyClassImpl",
|
|
"categories": {
|
|
"dummy1": ["m-dummy1", "m-dummy2"],
|
|
},
|
|
}
|
|
|
|
substs = gen_static_components.gen_substs([{"Classes": [clas]}])
|
|
|
|
self.assertEqual(substs["category_count"], 1)
|
|
self.assertEqual(
|
|
[s.strip() for s in substs["categories"].splitlines()],
|
|
[
|
|
'{ { 0x0 } /* "dummy1" */,',
|
|
"0, 2 },",
|
|
],
|
|
)
|
|
self.assertEqual(
|
|
[s.strip() for s in substs["category_entries"].splitlines()],
|
|
[
|
|
'/* "dummy1" */',
|
|
'{ { 0x7 } /* "m-dummy1" */,',
|
|
'{ 0x10 } /* "@mozilla.org/dummy-class;1" */,',
|
|
"Module::BackgroundTasksSelector::NO_TASKS,",
|
|
"Module::ProcessSelector::ANY_PROCESS },",
|
|
'{ { 0x2b } /* "m-dummy2" */,',
|
|
'{ 0x10 } /* "@mozilla.org/dummy-class;1" */,',
|
|
"Module::BackgroundTasksSelector::NO_TASKS,",
|
|
"Module::ProcessSelector::ANY_PROCESS },",
|
|
],
|
|
)
|
|
|
|
def test_dict(self):
|
|
# A dict, but no backgroundtasks selector: we default to NO_TASKS.
|
|
clas = {
|
|
"cid": "{a8566880-0bc7-4822-adb9-748c9af5cce7}",
|
|
"contract_ids": ["@mozilla.org/dummy-class;1"],
|
|
"jsm": "resource:///modules/DummyClass.jsm",
|
|
"js_name": "dummyClass",
|
|
"constructor": "DummyClassImpl",
|
|
"categories": {
|
|
"dummy1": {
|
|
"name": ["m-dummy1", "m-dummy2"],
|
|
},
|
|
},
|
|
}
|
|
|
|
substs = gen_static_components.gen_substs([{"Classes": [clas]}])
|
|
|
|
self.assertEqual(substs["category_count"], 1)
|
|
self.assertEqual(
|
|
[s.strip() for s in substs["categories"].splitlines()],
|
|
[
|
|
'{ { 0x0 } /* "dummy1" */,',
|
|
"0, 2 },",
|
|
],
|
|
)
|
|
self.assertEqual(
|
|
[s.strip() for s in substs["category_entries"].splitlines()],
|
|
[
|
|
'/* "dummy1" */',
|
|
'{ { 0x7 } /* "m-dummy1" */,',
|
|
'{ 0x10 } /* "@mozilla.org/dummy-class;1" */,',
|
|
"Module::BackgroundTasksSelector::NO_TASKS,",
|
|
"Module::ProcessSelector::ANY_PROCESS },",
|
|
'{ { 0x2b } /* "m-dummy2" */,',
|
|
'{ 0x10 } /* "@mozilla.org/dummy-class;1" */,',
|
|
"Module::BackgroundTasksSelector::NO_TASKS,",
|
|
"Module::ProcessSelector::ANY_PROCESS },",
|
|
],
|
|
)
|
|
|
|
def test_dict_with_selector(self):
|
|
# A dict with a selector.
|
|
clas = {
|
|
"cid": "{a8566880-0bc7-4822-adb9-748c9af5cce7}",
|
|
"contract_ids": ["@mozilla.org/dummy-class;1"],
|
|
"jsm": "resource:///modules/DummyClass.jsm",
|
|
"js_name": "dummyClass",
|
|
"constructor": "DummyClassImpl",
|
|
"categories": {
|
|
"dummy1": {
|
|
"name": ["m-dummy1", "m-dummy2"],
|
|
"backgroundtasks": BackgroundTasksSelector.ALL_TASKS,
|
|
},
|
|
},
|
|
}
|
|
|
|
substs = gen_static_components.gen_substs([{"Classes": [clas]}])
|
|
|
|
self.assertEqual(substs["category_count"], 1)
|
|
self.assertEqual(
|
|
[s.strip() for s in substs["categories"].splitlines()],
|
|
[
|
|
'{ { 0x0 } /* "dummy1" */,',
|
|
"0, 2 },",
|
|
],
|
|
)
|
|
self.assertEqual(
|
|
[s.strip() for s in substs["category_entries"].splitlines()],
|
|
[
|
|
'/* "dummy1" */',
|
|
'{ { 0x7 } /* "m-dummy1" */,',
|
|
'{ 0x10 } /* "@mozilla.org/dummy-class;1" */,',
|
|
"Module::BackgroundTasksSelector::ALL_TASKS,",
|
|
"Module::ProcessSelector::ANY_PROCESS },",
|
|
'{ { 0x2b } /* "m-dummy2" */,',
|
|
'{ 0x10 } /* "@mozilla.org/dummy-class;1" */,',
|
|
"Module::BackgroundTasksSelector::ALL_TASKS,",
|
|
"Module::ProcessSelector::ANY_PROCESS },",
|
|
],
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
mozunit.main()
|