test_user_management.py 36 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853
  1. """
  2. This module tests deSEC's user management.
  3. The tests are separated into two categories, where
  4. (a) the client has an associated user account and
  5. (b) does not have an associated user account.
  6. This involves testing five separate endpoints:
  7. (1) Registration endpoint,
  8. (2) Reset password endpoint,
  9. (3) Change email address endpoint,
  10. (4) delete user endpoint, and
  11. (5) verify endpoint.
  12. """
  13. import random
  14. import re
  15. import time
  16. from unittest import mock
  17. from urllib.parse import urlparse
  18. from django.contrib.auth.hashers import is_password_usable
  19. from django.core import mail
  20. from django.urls import resolve
  21. from rest_framework import status
  22. from rest_framework.reverse import reverse
  23. from rest_framework.test import APIClient
  24. from api import settings
  25. from desecapi.models import Domain, User, Captcha
  26. from desecapi.serializers import AuthenticatedActionSerializer
  27. from desecapi.tests.base import DesecTestCase, PublicSuffixMockMixin
  28. class UserManagementClient(APIClient):
  29. def register(self, email, password, captcha_id, captcha_solution, **kwargs):
  30. return self.post(reverse('v1:register'), {
  31. 'email': email,
  32. 'password': password,
  33. 'captcha': {'id': captcha_id, 'solution': captcha_solution},
  34. **kwargs
  35. })
  36. def login_user(self, email, password):
  37. return self.post(reverse('v1:login'), {
  38. 'email': email,
  39. 'password': password,
  40. })
  41. def logout(self, token):
  42. return self.post(reverse('v1:logout'), HTTP_AUTHORIZATION=f'Token {token}')
  43. def reset_password(self, email, captcha_id, captcha_solution):
  44. return self.post(reverse('v1:account-reset-password'), {
  45. 'email': email,
  46. 'captcha': {'id': captcha_id, 'solution': captcha_solution},
  47. })
  48. def change_email(self, email, password, **payload):
  49. payload['email'] = email
  50. payload['password'] = password
  51. return self.post(reverse('v1:account-change-email'), payload)
  52. def change_email_token_auth(self, token, **payload):
  53. return self.post(reverse('v1:account-change-email'), payload, HTTP_AUTHORIZATION='Token {}'.format(token))
  54. def delete_account(self, email, password):
  55. return self.post(reverse('v1:account-delete'), {
  56. 'email': email,
  57. 'password': password,
  58. })
  59. def view_account(self, token):
  60. return self.get(reverse('v1:account'), HTTP_AUTHORIZATION='Token {}'.format(token))
  61. def verify(self, url, data=None, **kwargs):
  62. return self.post(url, data, **kwargs)
  63. def obtain_captcha(self, **kwargs):
  64. return self.post(reverse('v1:captcha'))
  65. class UserManagementTestCase(DesecTestCase, PublicSuffixMockMixin):
  66. client_class = UserManagementClient
  67. password = None
  68. token = None
  69. def get_captcha(self):
  70. response = self.client.obtain_captcha()
  71. self.assertStatus(response, status.HTTP_201_CREATED)
  72. id = response.data['id']
  73. solution = Captcha.objects.get(id=id).content
  74. return id, solution
  75. def register_user(self, email=None, password=None, **kwargs):
  76. email = email if email is not None else self.random_username()
  77. captcha_id, captcha_solution = self.get_captcha()
  78. return email.strip(), password, self.client.register(email, password, captcha_id, captcha_solution, **kwargs)
  79. def login_user(self, email, password):
  80. response = self.client.login_user(email, password)
  81. token = response.data.get('token')
  82. return token, response
  83. def logout(self, token):
  84. return self.client.logout(token)
  85. def reset_password(self, email):
  86. captcha_id, captcha_solution = self.get_captcha()
  87. return self.client.reset_password(email, captcha_id, captcha_solution)
  88. def change_email(self, new_email):
  89. return self.client.change_email(self.email, self.password, new_email=new_email)
  90. def delete_account(self, email, password):
  91. return self.client.delete_account(email, password)
  92. def assertContains(self, response, text, count=None, status_code=200, msg_prefix='', html=False):
  93. msg_prefix += '\nResponse: %s' % response.data
  94. super().assertContains(response, text, count, status_code, msg_prefix, html)
  95. def assertPassword(self, email, password):
  96. if password is None:
  97. self.assertFalse(is_password_usable(User.objects.get(email=email).password))
  98. return
  99. password = password.strip()
  100. self.assertTrue(User.objects.get(email=email).check_password(password),
  101. 'Expected user password to be "%s" (potentially trimmed), but check failed.' % password)
  102. def assertUserExists(self, email):
  103. try:
  104. User.objects.get(email=email)
  105. except User.DoesNotExist:
  106. self.fail('Expected user %s to exist, but did not.' % email)
  107. def assertUserDoesNotExist(self, email):
  108. # noinspection PyTypeChecker
  109. with self.assertRaises(User.DoesNotExist):
  110. User.objects.get(email=email)
  111. def assertNoEmailSent(self):
  112. self.assertFalse(mail.outbox, "Expected no email to be sent, but %i were sent. First subject line is '%s'." %
  113. (len(mail.outbox), mail.outbox[0].subject if mail.outbox else '<n/a>'))
  114. def assertEmailSent(self, subject_contains='', body_contains='', recipient=None, reset=True, pattern=None):
  115. total = 1
  116. self.assertEqual(len(mail.outbox), total, "Expected %i message in the outbox, but found %i." %
  117. (total, len(mail.outbox)))
  118. email = mail.outbox[-1]
  119. self.assertTrue(subject_contains in email.subject,
  120. "Expected '%s' in the email subject, but found '%s'" %
  121. (subject_contains, email.subject))
  122. self.assertTrue(body_contains in email.body,
  123. "Expected '%s' in the email body, but found '%s'" %
  124. (body_contains, email.body))
  125. if recipient is not None:
  126. if isinstance(recipient, list):
  127. self.assertListEqual(recipient, email.recipients())
  128. else:
  129. self.assertIn(recipient, email.recipients())
  130. body = email.body
  131. self.assertIn('user_id = ', body)
  132. if reset:
  133. mail.outbox = []
  134. return body if not pattern else re.search(pattern, body).group(1)
  135. def assertConfirmationLinkRedirect(self, confirmation_link):
  136. response = self.client.get(confirmation_link)
  137. self.assertResponse(response, status.HTTP_406_NOT_ACCEPTABLE)
  138. response = self.client.get(confirmation_link, HTTP_ACCEPT='text/html')
  139. self.assertResponse(response, status.HTTP_302_FOUND)
  140. self.assertNoEmailSent()
  141. def assertRegistrationEmail(self, recipient, reset=True):
  142. return self.assertEmailSent(
  143. subject_contains='deSEC',
  144. body_contains='Thank you for registering with deSEC!',
  145. recipient=[recipient],
  146. reset=reset,
  147. pattern=r'following link[^:]*:\s+([^\s]*)',
  148. )
  149. def assertResetPasswordEmail(self, recipient, reset=True):
  150. return self.assertEmailSent(
  151. subject_contains='Password reset',
  152. body_contains='We received a request to reset the password for your deSEC account.',
  153. recipient=[recipient],
  154. reset=reset,
  155. pattern=r'following link[^:]*:\s+([^\s]*)',
  156. )
  157. def assertChangeEmailVerificationEmail(self, recipient, reset=True):
  158. return self.assertEmailSent(
  159. subject_contains='Confirmation required: Email address change',
  160. body_contains='You requested to change the email address associated',
  161. recipient=[recipient],
  162. reset=reset,
  163. pattern=r'following link[^:]*:\s+([^\s]*)',
  164. )
  165. def assertChangeEmailNotificationEmail(self, recipient, reset=True):
  166. return self.assertEmailSent(
  167. subject_contains='Account email address changed',
  168. body_contains='We\'re writing to let you know that the email address associated with',
  169. recipient=[recipient],
  170. reset=reset,
  171. )
  172. def assertDeleteAccountEmail(self, recipient, reset=True):
  173. return self.assertEmailSent(
  174. subject_contains='Confirmation required: Delete account',
  175. body_contains='confirm once more',
  176. recipient=[recipient],
  177. reset=reset,
  178. pattern=r'following link[^:]*:\s+([^\s]*)',
  179. )
  180. def assertRegistrationSuccessResponse(self, response):
  181. return self.assertContains(
  182. response=response,
  183. text="Welcome! Please check your mailbox.",
  184. status_code=status.HTTP_202_ACCEPTED
  185. )
  186. def assertLoginSuccessResponse(self, response):
  187. return self.assertContains(
  188. response=response,
  189. text="token",
  190. status_code=status.HTTP_200_OK
  191. )
  192. def assertLogoutSuccessResponse(self, response):
  193. return self.assertStatus(response, status.HTTP_204_NO_CONTENT)
  194. def assertRegistrationFailurePasswordRequiredResponse(self, response):
  195. self.assertContains(
  196. response=response,
  197. text="This field may not be blank",
  198. status_code=status.HTTP_400_BAD_REQUEST
  199. )
  200. self.assertEqual(response.data['password'][0].code, 'blank')
  201. def assertRegistrationFailureDomainUnavailableResponse(self, response, domain):
  202. self.assertContains(
  203. response=response,
  204. text='This domain name is unavailable',
  205. status_code=status.HTTP_400_BAD_REQUEST,
  206. msg_prefix=str(response.data)
  207. )
  208. def assertRegistrationFailureDomainInvalidResponse(self, response, domain):
  209. self.assertContains(
  210. response=response,
  211. text="Invalid value (not a DNS name)",
  212. status_code=status.HTTP_400_BAD_REQUEST,
  213. msg_prefix=str(response.data)
  214. )
  215. def assertRegistrationFailureCaptchaInvalidResponse(self, response):
  216. self.assertContains(
  217. response=response,
  218. text='CAPTCHA could not be validated. Please obtain a new one and try again.',
  219. status_code=status.HTTP_400_BAD_REQUEST,
  220. msg_prefix=str(response.data)
  221. )
  222. def assertRegistrationVerificationSuccessResponse(self, response):
  223. return self.assertContains(
  224. response=response,
  225. text="Success!",
  226. status_code=status.HTTP_200_OK
  227. )
  228. def assertRegistrationWithDomainVerificationSuccessResponse(self, response, domain=None, email=None):
  229. self.assertNoEmailSent() # do not send email in any case
  230. if domain and self.has_local_suffix(domain):
  231. text = 'Success! Here is the password'
  232. else:
  233. text = 'Success! Please check the docs for the next steps'
  234. self.assertContains(response=response, text=text, status_code=status.HTTP_200_OK)
  235. def assertResetPasswordSuccessResponse(self, response):
  236. return self.assertContains(
  237. response=response,
  238. text="Please check your mailbox for further password reset instructions.",
  239. status_code=status.HTTP_202_ACCEPTED
  240. )
  241. def assertResetPasswordVerificationSuccessResponse(self, response):
  242. return self.assertContains(
  243. response=response,
  244. text="Success! Your password has been changed.",
  245. status_code=status.HTTP_200_OK
  246. )
  247. def assertChangeEmailSuccessResponse(self, response):
  248. return self.assertContains(
  249. response=response,
  250. text="Please check your mailbox to confirm email address change.",
  251. status_code=status.HTTP_202_ACCEPTED
  252. )
  253. def assert401InvalidPasswordResponse(self, response):
  254. return self.assertContains(
  255. response=response,
  256. text="Invalid password.",
  257. status_code=status.HTTP_401_UNAUTHORIZED
  258. )
  259. def assertChangeEmailFailureAddressTakenResponse(self, response):
  260. return self.assertContains(
  261. response=response,
  262. text="You already have another account with this email address.",
  263. status_code=status.HTTP_400_BAD_REQUEST
  264. )
  265. def assertChangeEmailFailureSameAddressResponse(self, response):
  266. return self.assertContains(
  267. response=response,
  268. text="Email address unchanged.",
  269. status_code=status.HTTP_400_BAD_REQUEST
  270. )
  271. def assertChangeEmailVerificationSuccessResponse(self, response, new_email):
  272. return self.assertContains(
  273. response=response,
  274. text=f'Success! Your email address has been changed to {new_email}.',
  275. status_code=status.HTTP_200_OK
  276. )
  277. def assertChangeEmailVerificationFailureChangePasswordResponse(self, response):
  278. return self.assertContains(
  279. response=response,
  280. text="This field is not allowed for action ",
  281. status_code=status.HTTP_400_BAD_REQUEST
  282. )
  283. def assertDeleteAccountSuccessResponse(self, response):
  284. return self.assertContains(
  285. response=response,
  286. text="Please check your mailbox for further account deletion instructions.",
  287. status_code=status.HTTP_202_ACCEPTED
  288. )
  289. def assertDeleteAccountFailureStillHasDomainsResponse(self, response):
  290. return self.assertContains(
  291. response=response,
  292. text="To delete your user account, first delete all of your domains.",
  293. status_code=status.HTTP_409_CONFLICT
  294. )
  295. def assertDeleteAccountVerificationSuccessResponse(self, response):
  296. return self.assertContains(
  297. response=response,
  298. text="All your data has been deleted. Bye bye, see you soon! <3",
  299. status_code=status.HTTP_200_OK
  300. )
  301. def assertVerificationFailureInvalidCodeResponse(self, response):
  302. return self.assertContains(
  303. response=response,
  304. text="Invalid code.",
  305. status_code=status.HTTP_400_BAD_REQUEST
  306. )
  307. def assertVerificationFailureExpiredCodeResponse(self, response):
  308. return self.assertContains(
  309. response=response,
  310. text="Invalid code.",
  311. status_code=status.HTTP_400_BAD_REQUEST
  312. )
  313. def assertVerificationFailureUnknownUserResponse(self, response):
  314. return self.assertContains(
  315. response=response,
  316. text="This user does not exist.",
  317. status_code=status.HTTP_400_BAD_REQUEST
  318. )
  319. def _test_registration(self, email=None, password=None, **kwargs):
  320. email, password, response = self.register_user(email, password, **kwargs)
  321. self.assertRegistrationSuccessResponse(response)
  322. self.assertUserExists(email)
  323. self.assertFalse(User.objects.get(email=email).is_active)
  324. self.assertPassword(email, password)
  325. confirmation_link = self.assertRegistrationEmail(email)
  326. self.assertConfirmationLinkRedirect(confirmation_link)
  327. self.assertRegistrationVerificationSuccessResponse(self.client.verify(confirmation_link))
  328. self.assertTrue(User.objects.get(email=email).is_active)
  329. self.assertPassword(email, password)
  330. return email, password
  331. def _test_registration_with_domain(self, email=None, password=None, domain=None, expect_failure_response=None,
  332. tampered_domain=None):
  333. domain = domain or self.random_domain_name()
  334. email, password, response = self.register_user(email, password, domain=domain)
  335. if expect_failure_response:
  336. expect_failure_response(response, domain)
  337. self.assertUserDoesNotExist(email)
  338. return
  339. self.assertRegistrationSuccessResponse(response)
  340. self.assertUserExists(email)
  341. self.assertFalse(User.objects.get(email=email).is_active)
  342. self.assertPassword(email, password)
  343. confirmation_link = self.assertRegistrationEmail(email)
  344. if tampered_domain is not None:
  345. path = urlparse(confirmation_link).path
  346. code = resolve(path).kwargs.get('code')
  347. data = AuthenticatedActionSerializer._unpack_code(code)
  348. data['domain'] = tampered_domain
  349. tampered_code = AuthenticatedActionSerializer._pack_code(data)
  350. confirmation_link = confirmation_link.replace(code, tampered_code)
  351. response = self.client.verify(confirmation_link)
  352. self.assertVerificationFailureInvalidCodeResponse(response)
  353. return
  354. if self.has_local_suffix(domain):
  355. cm = self.requests_desec_domain_creation_auto_delegation(domain)
  356. else:
  357. cm = self.requests_desec_domain_creation(domain)[:-1]
  358. with self.assertPdnsRequests(cm):
  359. response = self.client.verify(confirmation_link)
  360. self.assertRegistrationWithDomainVerificationSuccessResponse(response, domain, email)
  361. self.assertTrue(User.objects.get(email=email).is_active)
  362. self.assertPassword(email, password)
  363. self.assertTrue(Domain.objects.filter(name=domain, owner__email=email).exists())
  364. return email, password, domain
  365. def _test_login(self):
  366. token, response = self.login_user(self.email, self.password)
  367. self.assertLoginSuccessResponse(response)
  368. return token
  369. def _test_logout(self):
  370. response = self.logout(self.token)
  371. self.assertLogoutSuccessResponse(response)
  372. return response
  373. def _test_reset_password(self, email, new_password=None, **kwargs):
  374. new_password = new_password or self.random_password()
  375. self.assertResetPasswordSuccessResponse(self.reset_password(email))
  376. confirmation_link = self.assertResetPasswordEmail(email)
  377. self.assertConfirmationLinkRedirect(confirmation_link)
  378. self.assertResetPasswordVerificationSuccessResponse(
  379. self.client.verify(confirmation_link, data={'new_password': new_password}, **kwargs))
  380. self.assertPassword(email, new_password)
  381. return new_password
  382. def _test_change_email(self):
  383. old_email = self.email
  384. new_email = ' {} '.format(self.random_username()) # test trimming
  385. self.assertChangeEmailSuccessResponse(self.change_email(new_email))
  386. new_email = new_email.strip()
  387. confirmation_link = self.assertChangeEmailVerificationEmail(new_email)
  388. self.assertConfirmationLinkRedirect(confirmation_link)
  389. self.assertChangeEmailVerificationSuccessResponse(self.client.verify(confirmation_link), new_email)
  390. self.assertChangeEmailNotificationEmail(old_email)
  391. self.assertUserExists(new_email)
  392. self.assertUserDoesNotExist(old_email)
  393. self.email = new_email
  394. return self.email
  395. def _test_delete_account(self, email, password):
  396. self.assertDeleteAccountSuccessResponse(self.delete_account(email, password))
  397. confirmation_link = self.assertDeleteAccountEmail(email)
  398. self.assertConfirmationLinkRedirect(confirmation_link)
  399. self.assertDeleteAccountVerificationSuccessResponse(self.client.verify(confirmation_link))
  400. self.assertUserDoesNotExist(email)
  401. class UserLifeCycleTestCase(UserManagementTestCase):
  402. def test_life_cycle(self):
  403. self.email, self.password = self._test_registration(self.random_username(), self.random_password())
  404. self.password = self._test_reset_password(self.email)
  405. mail.outbox = []
  406. self.token = self._test_login()
  407. email = self._test_change_email()
  408. self._test_logout()
  409. self._test_delete_account(email, self.password)
  410. class NoUserAccountTestCase(UserLifeCycleTestCase):
  411. def test_home(self):
  412. self.assertResponse(self.client.get(reverse('v1:root')), status.HTTP_200_OK)
  413. def test_registration(self):
  414. self._test_registration(password=self.random_password())
  415. def test_registration_trim_email(self):
  416. user_email = ' {} '.format(self.random_username())
  417. email, _ = self._test_registration(user_email)
  418. self.assertEqual(email, user_email.strip())
  419. def test_registration_with_domain(self):
  420. PublicSuffixMockMixin.setUpMockPatch(self)
  421. with self.get_psl_context_manager('.'):
  422. _, _, domain = self._test_registration_with_domain()
  423. self._test_registration_with_domain(domain=domain, expect_failure_response=self.assertRegistrationFailureDomainUnavailableResponse)
  424. self._test_registration_with_domain(domain='töö--', expect_failure_response=self.assertRegistrationFailureDomainInvalidResponse)
  425. with self.get_psl_context_manager('co.uk'):
  426. self._test_registration_with_domain(domain='co.uk', expect_failure_response=self.assertRegistrationFailureDomainUnavailableResponse)
  427. local_public_suffix = random.sample(self.AUTO_DELEGATION_DOMAINS, 1)[0]
  428. with self.get_psl_context_manager(local_public_suffix):
  429. self._test_registration_with_domain(domain=self.random_domain_name(suffix=local_public_suffix))
  430. def test_registration_without_domain_and_password(self):
  431. email, password = self._test_registration(self.random_username(), None)
  432. self.assertResetPasswordEmail(email)
  433. def test_registration_with_tampered_domain(self):
  434. PublicSuffixMockMixin.setUpMockPatch(self)
  435. with self.get_psl_context_manager('.'):
  436. self._test_registration_with_domain(tampered_domain='evil.com')
  437. def test_registration_known_account(self):
  438. email, _ = self._test_registration(self.random_username(), self.random_password())
  439. self.assertRegistrationSuccessResponse(self.register_user(email, self.random_password())[2])
  440. self.assertNoEmailSent()
  441. def test_registration_password_required(self):
  442. email = self.random_username()
  443. self.assertRegistrationFailurePasswordRequiredResponse(
  444. response=self.register_user(email=email, password='')[2]
  445. )
  446. self.assertNoEmailSent()
  447. self.assertUserDoesNotExist(email)
  448. def test_no_login_with_unusable_password(self):
  449. email, password = self._test_registration(password=None)
  450. response = self.client.login_user(email, password)
  451. self.assertStatus(response, status.HTTP_400_BAD_REQUEST)
  452. self.assertEqual(response.data['password'][0], 'This field may not be null.')
  453. def test_registration_spam_protection(self):
  454. email = self.random_username()
  455. self.assertRegistrationSuccessResponse(
  456. response=self.register_user(email=email)[2]
  457. )
  458. self.assertRegistrationEmail(email)
  459. for _ in range(5):
  460. self.assertRegistrationSuccessResponse(
  461. response=self.register_user(email=email)[2]
  462. )
  463. self.assertNoEmailSent()
  464. def test_registration_wrong_captcha(self):
  465. email = self.random_username()
  466. password = self.random_password()
  467. captcha_id, _ = self.get_captcha()
  468. self.assertRegistrationFailureCaptchaInvalidResponse(
  469. self.client.register(email, password, captcha_id, 'this is most definitely not a correct CAPTCHA solution')
  470. )
  471. class OtherUserAccountTestCase(UserManagementTestCase):
  472. def setUp(self):
  473. super().setUp()
  474. self.other_email, self.other_password = self._test_registration(password=self.random_password())
  475. def test_reset_password_unknown_user(self):
  476. self.assertResetPasswordSuccessResponse(
  477. response=self.reset_password(self.random_username())
  478. )
  479. self.assertNoEmailSent()
  480. class HasUserAccountTestCase(UserManagementTestCase):
  481. def __init__(self, methodName: str = ...) -> None:
  482. super().__init__(methodName)
  483. self.email = None
  484. self.password = None
  485. def setUp(self):
  486. super().setUp()
  487. self.email, self.password = self._test_registration(password=self.random_password())
  488. self.token = self._test_login()
  489. def _start_reset_password(self):
  490. self.assertResetPasswordSuccessResponse(
  491. response=self.reset_password(self.email)
  492. )
  493. return self.assertResetPasswordEmail(self.email)
  494. def _start_change_email(self):
  495. new_email = self.random_username()
  496. self.assertChangeEmailSuccessResponse(
  497. response=self.change_email(new_email)
  498. )
  499. return self.assertChangeEmailVerificationEmail(new_email), new_email
  500. def _start_delete_account(self):
  501. self.assertDeleteAccountSuccessResponse(self.delete_account(self.email, self.password))
  502. return self.assertDeleteAccountEmail(self.email)
  503. def _finish_reset_password(self, confirmation_link, expect_success=True):
  504. new_password = self.random_password()
  505. response = self.client.verify(confirmation_link, data={'new_password': new_password})
  506. if expect_success:
  507. self.assertResetPasswordVerificationSuccessResponse(response=response)
  508. else:
  509. self.assertVerificationFailureInvalidCodeResponse(response)
  510. return new_password
  511. def _finish_change_email(self, confirmation_link, new_email='', expect_success=True):
  512. response = self.client.verify(confirmation_link)
  513. if expect_success:
  514. self.assertChangeEmailVerificationSuccessResponse(response, new_email)
  515. self.assertChangeEmailNotificationEmail(self.email)
  516. else:
  517. self.assertVerificationFailureInvalidCodeResponse(response)
  518. def _finish_delete_account(self, confirmation_link):
  519. self.assertDeleteAccountVerificationSuccessResponse(self.client.verify(confirmation_link))
  520. self.assertUserDoesNotExist(self.email)
  521. def test_view_account(self):
  522. response = self.client.view_account(self.token)
  523. self.assertEqual(response.status_code, 200)
  524. self.assertTrue('created' in response.data)
  525. self.assertEqual(response.data['email'], self.email)
  526. self.assertEqual(response.data['id'], str(User.objects.get(email=self.email).pk))
  527. self.assertEqual(response.data['limit_domains'], settings.LIMIT_USER_DOMAIN_COUNT_DEFAULT)
  528. def test_view_account_read_only(self):
  529. # Should this test ever be removed (to allow writeable fields), make sure to
  530. # add new tests for each read-only field individually (such as limit_domains)!
  531. for method in [self.client.patch, self.client.put, self.client.post, self.client.delete]:
  532. response = method(
  533. reverse('v1:account'),
  534. {'limit_domains': 99},
  535. HTTP_AUTHORIZATION='Token {}'.format(self.token)
  536. )
  537. self.assertResponse(response, status.HTTP_405_METHOD_NOT_ALLOWED)
  538. def test_reset_password(self):
  539. self._test_reset_password(self.email)
  540. def test_reset_password_inactive_user(self):
  541. user = User.objects.get(email=self.email)
  542. user.is_active = False
  543. user.save()
  544. self.assertResetPasswordSuccessResponse(self.reset_password(self.email))
  545. self.assertNoEmailSent()
  546. def test_reset_password_multiple_times(self):
  547. for _ in range(3):
  548. self._test_reset_password(self.email)
  549. mail.outbox = []
  550. def test_reset_password_during_change_email_interleaved(self):
  551. reset_password_verification_code = self._start_reset_password()
  552. change_email_verification_code, new_email = self._start_change_email()
  553. new_password = self._finish_reset_password(reset_password_verification_code)
  554. self._finish_change_email(change_email_verification_code, expect_success=False)
  555. self.assertUserExists(self.email)
  556. self.assertUserDoesNotExist(new_email)
  557. self.assertPassword(self.email, new_password)
  558. def test_reset_password_during_change_email_nested(self):
  559. change_email_verification_code, new_email = self._start_change_email()
  560. reset_password_verification_code = self._start_reset_password()
  561. new_password = self._finish_reset_password(reset_password_verification_code)
  562. self._finish_change_email(change_email_verification_code, expect_success=False)
  563. self.assertUserExists(self.email)
  564. self.assertUserDoesNotExist(new_email)
  565. self.assertPassword(self.email, new_password)
  566. def test_reset_password_without_new_password(self):
  567. confirmation_link = self._start_reset_password()
  568. response = self.client.verify(confirmation_link)
  569. self.assertResponse(response, status.HTTP_400_BAD_REQUEST)
  570. self.assertEqual(response.data['new_password'][0], 'This field is required.')
  571. self.assertNoEmailSent()
  572. def test_reset_password_validation_unknown_user(self):
  573. confirmation_link = self._start_reset_password()
  574. self._test_delete_account(self.email, self.password)
  575. self.assertVerificationFailureUnknownUserResponse(
  576. response=self.client.verify(confirmation_link, data={'new_password': 'foobar'})
  577. )
  578. self.assertNoEmailSent()
  579. def test_change_email(self):
  580. self._test_change_email()
  581. def test_change_email_requires_password(self):
  582. # Make sure that the account's email address cannot be changed with a token (password required)
  583. new_email = self.random_username()
  584. response = self.client.change_email_token_auth(self.token, new_email=new_email)
  585. self.assertStatus(response, status.HTTP_400_BAD_REQUEST)
  586. self.assertEqual(response.data['email'][0], 'This field is required.')
  587. self.assertEqual(response.data['password'][0], 'This field is required.')
  588. self.assertNoEmailSent()
  589. def test_change_email_multiple_times(self):
  590. for _ in range(3):
  591. self._test_change_email()
  592. def test_change_email_user_exists(self):
  593. known_email, _ = self._test_registration()
  594. # We send a verification link to the new email and check account existence only later, upon verification
  595. self.assertChangeEmailSuccessResponse(
  596. response=self.change_email(known_email)
  597. )
  598. def test_change_email_verification_user_exists(self):
  599. new_email = self.random_username()
  600. self.assertChangeEmailSuccessResponse(self.change_email(new_email))
  601. confirmation_link = self.assertChangeEmailVerificationEmail(new_email)
  602. new_email, new_password = self._test_registration(new_email)
  603. self.assertChangeEmailFailureAddressTakenResponse(
  604. response=self.client.verify(confirmation_link)
  605. )
  606. self.assertUserExists(self.email)
  607. self.assertPassword(self.email, self.password)
  608. self.assertUserExists(new_email)
  609. self.assertPassword(new_email, new_password)
  610. def test_change_email_same_email(self):
  611. self.assertChangeEmailFailureSameAddressResponse(
  612. response=self.change_email(self.email)
  613. )
  614. self.assertUserExists(self.email)
  615. def test_change_email_during_reset_password_interleaved(self):
  616. change_email_verification_code, new_email = self._start_change_email()
  617. reset_password_verification_code = self._start_reset_password()
  618. self._finish_change_email(change_email_verification_code, new_email)
  619. self._finish_reset_password(reset_password_verification_code, expect_success=False)
  620. self.assertUserExists(new_email)
  621. self.assertUserDoesNotExist(self.email)
  622. self.assertPassword(new_email, self.password)
  623. def test_change_email_during_reset_password_nested(self):
  624. reset_password_verification_code = self._start_reset_password()
  625. change_email_verification_code, new_email = self._start_change_email()
  626. self._finish_change_email(change_email_verification_code, new_email)
  627. self._finish_reset_password(reset_password_verification_code, expect_success=False)
  628. self.assertUserExists(new_email)
  629. self.assertUserDoesNotExist(self.email)
  630. self.assertPassword(new_email, self.password)
  631. def test_change_email_nested(self):
  632. verification_code_1, new_email_1 = self._start_change_email()
  633. verification_code_2, new_email_2 = self._start_change_email()
  634. self._finish_change_email(verification_code_2, new_email_2)
  635. self.assertUserDoesNotExist(self.email)
  636. self.assertUserDoesNotExist(new_email_1)
  637. self.assertUserExists(new_email_2)
  638. self._finish_change_email(verification_code_1, expect_success=False)
  639. self.assertUserDoesNotExist(self.email)
  640. self.assertUserDoesNotExist(new_email_1)
  641. self.assertUserExists(new_email_2)
  642. def test_change_email_interleaved(self):
  643. verification_code_1, new_email_1 = self._start_change_email()
  644. verification_code_2, new_email_2 = self._start_change_email()
  645. self._finish_change_email(verification_code_1, new_email_1)
  646. self.assertUserDoesNotExist(self.email)
  647. self.assertUserExists(new_email_1)
  648. self.assertUserDoesNotExist(new_email_2)
  649. self._finish_change_email(verification_code_2, expect_success=False)
  650. self.assertUserDoesNotExist(self.email)
  651. self.assertUserExists(new_email_1)
  652. self.assertUserDoesNotExist(new_email_2)
  653. def test_change_email_validation_unknown_user(self):
  654. confirmation_link, new_email = self._start_change_email()
  655. self._test_delete_account(self.email, self.password)
  656. self.assertVerificationFailureUnknownUserResponse(
  657. response=self.client.verify(confirmation_link)
  658. )
  659. self.assertNoEmailSent()
  660. def test_delete_account_validation_unknown_user(self):
  661. confirmation_link = self._start_delete_account()
  662. self._test_delete_account(self.email, self.password)
  663. self.assertVerificationFailureUnknownUserResponse(
  664. response=self.client.verify(confirmation_link)
  665. )
  666. self.assertNoEmailSent()
  667. def test_delete_account_domains_present(self):
  668. user = User.objects.get(email=self.email)
  669. domain = self.create_domain(owner=user)
  670. self.assertDeleteAccountFailureStillHasDomainsResponse(self.delete_account(self.email, self.password))
  671. domain.delete()
  672. confirmation_link = self._start_delete_account()
  673. domain = self.create_domain(owner=user)
  674. self.assertDeleteAccountFailureStillHasDomainsResponse(response=self.client.verify(confirmation_link))
  675. domain.delete()
  676. self._finish_delete_account(confirmation_link)
  677. def test_reset_password_password_strip(self):
  678. password = ' %s ' % self.random_password()
  679. self._test_reset_password(self.email, password)
  680. self.assertPassword(self.email, password.strip())
  681. self.assertPassword(self.email, password)
  682. def test_reset_password_no_code_override(self):
  683. password = self.random_password()
  684. self._test_reset_password(self.email, password, code='foobar')
  685. self.assertPassword(self.email, password)
  686. def test_action_code_expired(self):
  687. self.assertResetPasswordSuccessResponse(self.reset_password(self.email))
  688. confirmation_link = self.assertResetPasswordEmail(self.email)
  689. with mock.patch('time.time',
  690. return_value=time.time() + settings.VALIDITY_PERIOD_VERIFICATION_SIGNATURE.total_seconds() + 1):
  691. response = self.client.verify(confirmation_link, data={'new_password': self.random_password()})
  692. self.assertVerificationFailureExpiredCodeResponse(response)
  693. def test_action_code_confusion(self):
  694. # Obtain change password code
  695. self.assertResetPasswordSuccessResponse(self.reset_password(self.email))
  696. reset_password_link = self.assertResetPasswordEmail(self.email)
  697. path = urlparse(reset_password_link).path
  698. reset_password_code = resolve(path).kwargs.get('code')
  699. # Obtain deletion code
  700. self.assertDeleteAccountSuccessResponse(self.delete_account(self.email, self.password))
  701. delete_link = self.assertDeleteAccountEmail(self.email)
  702. path = urlparse(delete_link).path
  703. deletion_code = resolve(path).kwargs.get('code')
  704. # Swap codes
  705. self.assertNotEqual(reset_password_code, deletion_code)
  706. delete_link = delete_link.replace(deletion_code, reset_password_code)
  707. reset_password_link = reset_password_link.replace(reset_password_code, deletion_code)
  708. # Make sure links don't work
  709. self.assertVerificationFailureInvalidCodeResponse(self.client.verify(delete_link))
  710. self.assertVerificationFailureInvalidCodeResponse(self.client.verify(reset_password_link,
  711. data={'new_password': 'dummy'}))