123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135 |
- from synapse.rest.client.v2_alpha.register import RegisterRestServlet
- from synapse.api.errors import SynapseError
- from twisted.internet import defer
- from mock import Mock, MagicMock
- from tests import unittest
- import json
- class RegisterRestServletTestCase(unittest.TestCase):
- def setUp(self):
- # do the dance to hook up request data to self.request_data
- self.request_data = ""
- self.request = Mock(
- content=Mock(read=Mock(side_effect=lambda: self.request_data)),
- path='/_matrix/api/v2_alpha/register'
- )
- self.request.args = {}
- self.appservice = None
- self.auth = Mock(get_appservice_by_req=Mock(
- side_effect=lambda x: defer.succeed(self.appservice))
- )
- self.auth_result = (False, None, None)
- self.auth_handler = Mock(
- check_auth=Mock(side_effect=lambda x,y,z: self.auth_result)
- )
- self.registration_handler = Mock()
- self.identity_handler = Mock()
- self.login_handler = Mock()
- # do the dance to hook it up to the hs global
- self.handlers = Mock(
- auth_handler=self.auth_handler,
- registration_handler=self.registration_handler,
- identity_handler=self.identity_handler,
- login_handler=self.login_handler
- )
- self.hs = Mock()
- self.hs.hostname = "superbig~testing~thing.com"
- self.hs.get_auth = Mock(return_value=self.auth)
- self.hs.get_handlers = Mock(return_value=self.handlers)
- self.hs.config.disable_registration = False
- # init the thing we're testing
- self.servlet = RegisterRestServlet(self.hs)
- @defer.inlineCallbacks
- def test_POST_appservice_registration_valid(self):
- user_id = "@kermit:muppet"
- token = "kermits_access_token"
- self.request.args = {
- "access_token": "i_am_an_app_service"
- }
- self.request_data = json.dumps({
- "username": "kermit"
- })
- self.appservice = {
- "id": "1234"
- }
- self.registration_handler.appservice_register = Mock(
- return_value=(user_id, token)
- )
- result = yield self.servlet.on_POST(self.request)
- self.assertEquals(result, (200, {
- "user_id": user_id,
- "access_token": token,
- "home_server": self.hs.hostname
- }))
- @defer.inlineCallbacks
- def test_POST_appservice_registration_invalid(self):
- self.request.args = {
- "access_token": "i_am_an_app_service"
- }
- self.request_data = json.dumps({
- "username": "kermit"
- })
- self.appservice = None # no application service exists
- result = yield self.servlet.on_POST(self.request)
- self.assertEquals(result, (401, None))
- def test_POST_bad_password(self):
- self.request_data = json.dumps({
- "username": "kermit",
- "password": 666
- })
- d = self.servlet.on_POST(self.request)
- return self.assertFailure(d, SynapseError)
- def test_POST_bad_username(self):
- self.request_data = json.dumps({
- "username": 777,
- "password": "monkey"
- })
- d = self.servlet.on_POST(self.request)
- return self.assertFailure(d, SynapseError)
- @defer.inlineCallbacks
- def test_POST_user_valid(self):
- user_id = "@kermit:muppet"
- token = "kermits_access_token"
- self.request_data = json.dumps({
- "username": "kermit",
- "password": "monkey"
- })
- self.registration_handler.check_username = Mock(return_value=True)
- self.auth_result = (True, None, {
- "username": "kermit",
- "password": "monkey"
- })
- self.registration_handler.register = Mock(return_value=(user_id, token))
- result = yield self.servlet.on_POST(self.request)
- self.assertEquals(result, (200, {
- "user_id": user_id,
- "access_token": token,
- "home_server": self.hs.hostname
- }))
- def test_POST_disabled_registration(self):
- self.hs.config.disable_registration = True
- self.request_data = json.dumps({
- "username": "kermit",
- "password": "monkey"
- })
- self.registration_handler.check_username = Mock(return_value=True)
- self.auth_result = (True, None, {
- "username": "kermit",
- "password": "monkey"
- })
- self.registration_handler.register = Mock(return_value=("@user:id", "t"))
- d = self.servlet.on_POST(self.request)
- return self.assertFailure(d, SynapseError)
|