1 files changed, 150 insertions, 0 deletions
diff --git a/xpcom/components/test/test_gen_static_components.py b/xpcom/components/test/test_gen_static_components.py
new file mode 100644
index 0000000000..85f731c082
--- /dev/null
+++ b/xpcom/components/test/test_gen_static_components.py
@@ -0,0 +1,150 @@
+# 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 os
+import sys
+import unittest
+
+import mozunit
+
+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"],
+            },
+            "protocol_config": {
+                "scheme": "dummy",
+                "flags": [],
+            },
+        }
+
+        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"],
+                },
+            },
+            "protocol_config": {
+                "scheme": "dummy",
+                "flags": [],
+            },
+        }
+
+        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,
+                },
+            },
+            "protocol_config": {
+                "scheme": "dummy",
+                "flags": [],
+            },
+        }
+
+        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()