1 files changed, 99 insertions, 0 deletions
diff --git a/tests/api/generate_test_eventauth.py b/tests/api/generate_test_eventauth.py
new file mode 100644
index 0000000000..8181f674d2
--- /dev/null
+++ b/tests/api/generate_test_eventauth.py
@@ -0,0 +1,99 @@
+# -*- coding: utf-8 -*-
+# Copyright 2015 - 2016 OpenMarket Ltd
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+import sys
+import json
+import os
+import pprint
+
+testcases = json.load(sys.stdin)
+
+w = sys.stdout.write
+
+w("""
+# -*- coding: utf-8 -*-
+# Generated by """ + os.path.basename(__file__) + """
+
+# Copyright 2015 - 2016 OpenMarket Ltd
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from synapse.api.auth import Auth
+from synapse.api.errors import SynapseError
+from tests import unittest
+from tests.utils import setup_test_homeserver
+from twisted.internet import defer
+from synapse.events import FrozenEvent
+
+
+class EventAuthTestCase(unittest.TestCase):
+
+    @defer.inlineCallbacks
+    def setUp(self):
+        self.hs = yield setup_test_homeserver(handlers=None)
+        self.auth = Auth(self.hs)
+""")
+
+indent = " " * 4
+for name, case in sorted(testcases.items()):
+    w("""
+    def test_""" + name[4:] + """(self):
+""")
+    auth_data = case.get("auth_events", {})
+    auth_events = {}
+    if "create" in auth_data:
+        auth_events[("m.room.create", "")] = auth_data["create"]
+    for user_id, member in auth_data.get("member", {}).items():
+        auth_events[("m.room.member", user_id)] = member
+
+    w(indent * 2 + "auth_events = (\n" + indent * 3)
+    data = pprint.pformat(auth_events, width=80 - len(indent * 3))
+    w(data.replace("\n", "\n" + indent * 3))
+    w("\n" + indent * 2 + ")\n")
+    w("""
+        auth_events = {k: FrozenEvent(v) for k, v in auth_events.items()}
+"""[1:])
+
+    w("\n" + indent * 2 + "# Allowed events\n")
+    for allowed in case.get("allowed", ()):
+        reason = allowed.pop("unsigned", {}).pop("allowed", None)
+        if reason is not None:
+            w(indent * 2 + "#Allowed: " + reason)
+        w(indent * 2 + "self.auth.check(FrozenEvent(\n" + indent * 3)
+        data = pprint.pformat(allowed, width=80 - len(indent * 3))
+        w(data.replace("\n", "\n" + indent * 3))
+        w("\n" + indent * 2 + "), auth_events=auth_events, do_sig_check=False)\n")
+
+    w("\n" + indent * 2 + "# Disallowed events\n")
+    for not_allowed in case.get("not_allowed", ()):
+        reason = not_allowed.pop("unsigned", {}).pop("not_allowed", None)
+        if reason is not None:
+            w(indent * 2 + "# " + reason + "\n")
+        w(indent * 2 + "self.assertRaises(SynapseError,")
+        w(" self.auth.check, FrozenEvent(\n" + indent * 3)
+        data = pprint.pformat(not_allowed, width=80 - len(indent * 3))
+        w(data.replace("\n", "\n" + indent * 3))
+        w("\n" + indent * 2 + "), auth_events=auth_events, do_sig_check=False)\n")