123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125 |
- import json
- from unittest.mock import Mock
- from twisted.test.proto_helpers import MemoryReactorClock
- from synapse.rest.client.v2_alpha.register import register_servlets
- from synapse.util import Clock
- from tests import unittest
- class TermsTestCase(unittest.HomeserverTestCase):
- servlets = [register_servlets]
- def default_config(self):
- config = super().default_config()
- config.update(
- {
- "public_baseurl": "https://example.org/",
- "user_consent": {
- "version": "1.0",
- "policy_name": "My Cool Privacy Policy",
- "template_dir": "/",
- "require_at_registration": True,
- },
- }
- )
- return config
- def prepare(self, reactor, clock, hs):
- self.clock = MemoryReactorClock()
- self.hs_clock = Clock(self.clock)
- self.url = "/_matrix/client/r0/register"
- self.registration_handler = Mock()
- self.auth_handler = Mock()
- self.device_handler = Mock()
- def test_ui_auth(self):
-
- request_data = json.dumps({"username": "kermit", "password": "monkey"})
- channel = self.make_request(b"POST", self.url, request_data)
- self.assertEquals(channel.result["code"], b"401", channel.result)
- self.assertTrue(channel.json_body is not None)
- self.assertIsInstance(channel.json_body["session"], str)
- self.assertIsInstance(channel.json_body["flows"], list)
- for flow in channel.json_body["flows"]:
- self.assertIsInstance(flow["stages"], list)
- self.assertTrue(len(flow["stages"]) > 0)
- self.assertTrue("m.login.terms" in flow["stages"])
- expected_params = {
- "m.login.terms": {
- "policies": {
- "privacy_policy": {
- "en": {
- "name": "My Cool Privacy Policy",
- "url": "https://example.org/_matrix/consent?v=1.0",
- },
- "version": "1.0",
- }
- }
- }
- }
- self.assertIsInstance(channel.json_body["params"], dict)
- self.assertDictContainsSubset(channel.json_body["params"], expected_params)
-
- request_data = json.dumps(
- {
- "username": "kermit",
- "password": "monkey",
- "auth": {
- "session": channel.json_body["session"],
- "type": "m.login.dummy",
- },
- }
- )
- self.registration_handler.check_username = Mock(return_value=True)
- channel = self.make_request(b"POST", self.url, request_data)
-
-
- self.assertEquals(channel.result["code"], b"401", channel.result)
-
- request_data = json.dumps(
- {
- "username": "kermit",
- "password": "monkey",
- "auth": {
- "session": channel.json_body["session"],
- "type": "m.login.terms",
- },
- }
- )
- channel = self.make_request(b"POST", self.url, request_data)
-
-
- self.assertEquals(channel.result["code"], b"200", channel.result)
- self.assertTrue(channel.json_body is not None)
- self.assertIsInstance(channel.json_body["user_id"], str)
- self.assertIsInstance(channel.json_body["access_token"], str)
- self.assertIsInstance(channel.json_body["device_id"], str)
|