1 files changed, 132 insertions, 0 deletions
diff --git a/dom/bindings/parser/tests/test_union_callback_dict.py b/dom/bindings/parser/tests/test_union_callback_dict.py
new file mode 100644
index 0000000000..3e87e16ad4
--- /dev/null
+++ b/dom/bindings/parser/tests/test_union_callback_dict.py
@@ -0,0 +1,132 @@
+import WebIDL
+
+
+def WebIDLTest(parser, harness):
+    parser = parser.reset()
+    threw = False
+    try:
+        parser.parse(
+            """
+            dictionary TestDict {
+            DOMString member;
+            };
+            [LegacyTreatNonObjectAsNull] callback TestCallback = undefined ();
+            typedef (TestCallback or TestDict) TestUnionCallbackDict;
+            """
+        )
+        results = parser.finish()
+    except WebIDL.WebIDLError:
+        threw = True
+    harness.ok(
+        threw,
+        "Should not allow Dict/Callback union where callback is [LegacyTreatNonObjectAsNull]",
+    )
+
+    parser = parser.reset()
+
+    threw = False
+    try:
+        parser.parse(
+            """
+            dictionary TestDict {
+            DOMString member;
+            };
+            [LegacyTreatNonObjectAsNull] callback TestCallback = undefined ();
+            typedef (TestDict or TestCallback) TestUnionCallbackDict;
+            """
+        )
+        results = parser.finish()
+    except WebIDL.WebIDLError:
+        threw = True
+    harness.ok(
+        threw,
+        "Should not allow Dict/Callback union where callback is [LegacyTreatNonObjectAsNull]",
+    )
+
+    parser = parser.reset()
+
+    parser.parse(
+        """
+        dictionary TestDict {
+          DOMString member;
+        };
+        callback TestCallback = undefined ();
+        typedef (TestCallback or TestDict) TestUnionCallbackDict;
+        """
+    )
+    results = parser.finish()
+
+    harness.ok(True, "TestUnionCallbackDict interface parsed without error")
+    harness.check(len(results), 3, "Document should have 3 types")
+
+    myDict = results[0]
+    harness.ok(isinstance(myDict, WebIDL.IDLDictionary), "Expect an IDLDictionary")
+
+    myCallback = results[1]
+    harness.ok(isinstance(myCallback, WebIDL.IDLCallback), "Expect an IDLCallback")
+
+    myUnion = results[2]
+    harness.ok(isinstance(myUnion, WebIDL.IDLTypedef), "Expect a IDLTypedef")
+    harness.ok(
+        isinstance(myUnion.innerType, WebIDL.IDLUnionType), "Expect a IDLUnionType"
+    )
+    harness.ok(
+        isinstance(myUnion.innerType.memberTypes[0], WebIDL.IDLCallbackType),
+        "Expect a IDLCallbackType",
+    )
+    harness.ok(
+        isinstance(myUnion.innerType.memberTypes[1], WebIDL.IDLWrapperType),
+        "Expect a IDLDictionary",
+    )
+    harness.ok(
+        (myUnion.innerType.memberTypes[0].callback == myCallback),
+        "Expect left Union member to be MyCallback",
+    )
+    harness.ok(
+        (myUnion.innerType.memberTypes[1].inner == myDict),
+        "Expect right Union member to be MyDict",
+    )
+
+    parser = parser.reset()
+
+    parser.parse(
+        """
+        dictionary TestDict {
+          DOMString member;
+        };
+        callback TestCallback = undefined ();
+        typedef (TestDict or TestCallback) TestUnionCallbackDict;
+        """
+    )
+    results = parser.finish()
+
+    harness.ok(True, "TestUnionCallbackDict interface parsed without error")
+    harness.check(len(results), 3, "Document should have 3 types")
+
+    myDict = results[0]
+    harness.ok(isinstance(myDict, WebIDL.IDLDictionary), "Expect an IDLDictionary")
+
+    myCallback = results[1]
+    harness.ok(isinstance(myCallback, WebIDL.IDLCallback), "Expect an IDLCallback")
+
+    myUnion = results[2]
+    harness.ok(isinstance(myUnion, WebIDL.IDLTypedef), "Expect a IDLTypedef")
+    harness.ok(
+        isinstance(myUnion.innerType, WebIDL.IDLUnionType), "Expect a IDLUnionType"
+    )
+    harness.ok(
+        isinstance(myUnion.innerType.memberTypes[0], WebIDL.IDLWrapperType),
+        "Expect a IDLDictionary",
+    )
+    harness.ok(
+        isinstance(myUnion.innerType.memberTypes[1], WebIDL.IDLCallbackType),
+        "Expect a IDLCallbackType",
+    )
+    harness.ok(
+        (myUnion.innerType.memberTypes[0].inner == myDict),
+        "Expect right Union member to be MyDict",
+    )
+    harness.ok(
+        (myUnion.innerType.memberTypes[1].callback == myCallback),
+        "Expect left Union member to be MyCallback",
+    )