test: extended util_schema_test to include test_optionally_keyed_by_mulitple_keys

Signed-off-by: AnsahMohammad <mohdansah10@gmail.com>

Author: AnsahMohammad

test/test_util_schema.py

@@ -233,3 +233,20 @@ def test_optionally_keyed_by():
 
     with pytest.raises(MultipleInvalid):
         validator({"by-bar": {"a": "b"}})
+
+
+def test_optionally_keyed_by_mulitple_keys():
+    validator = optionally_keyed_by("foo", "bar", str)
+    assert validator("baz") == "baz"
+    assert validator({"by-foo": {"a": "b", "c": "d"}}) == {"a": "b", "c": "d"}
+    assert validator({"by-bar": {"x": "y"}}) == {"x": "y"}
+    assert validator({"by-foo": {"a": {"by-bar": {"x": "y"}}}}) == {"a": {"x": "y"}}
+
+    with pytest.raises(Invalid):
+        validator({"by-foo": {"a": 123, "c": "d"}})
+
+    with pytest.raises(MultipleInvalid):
+        validator({"by-bar": {"a": 1}})
+
+    with pytest.raises(MultipleInvalid):
+        validator({"by-unknown": {"a": "b"}})