sonny
177755e302
commit 99fd94580f95dcbfb77b73e2de846f76a5709ef9
Author: Sonny <sonnyba871@gmail.com>
Date: Sat Feb 15 21:45:16 2020 +0100
Use postgres password
As of https://gitlab.com/gitlab-com/support-forum/issues/5199
161 lines
5.1 KiB
Python
161 lines
5.1 KiB
Python
from unittest.mock import patch
|
|
from urllib.parse import urlencode
|
|
from uuid import uuid4
|
|
|
|
from django.core.cache import cache
|
|
from django.test import TestCase
|
|
from django.urls import reverse
|
|
|
|
from newsreader.accounts.models import User
|
|
from newsreader.accounts.tests.factories import UserFactory
|
|
from newsreader.news.collection.exceptions import StreamTooManyException
|
|
|
|
|
|
class SettingsViewTestCase(TestCase):
|
|
def setUp(self):
|
|
self.user = UserFactory(email="test@test.nl", password="test")
|
|
self.client.force_login(self.user)
|
|
|
|
self.url = reverse("accounts:settings")
|
|
|
|
def test_simple(self):
|
|
response = self.client.get(self.url)
|
|
|
|
self.assertEquals(response.status_code, 200)
|
|
self.assertContains(response, "Authorize Reddit account")
|
|
|
|
def test_user_credential_change(self):
|
|
response = self.client.post(
|
|
reverse("accounts:settings"),
|
|
{"first_name": "First name", "last_name": "Last name"},
|
|
)
|
|
|
|
user = User.objects.get()
|
|
|
|
self.assertRedirects(response, reverse("accounts:settings"))
|
|
|
|
self.assertEquals(user.first_name, "First name")
|
|
self.assertEquals(user.last_name, "Last name")
|
|
|
|
def test_linked_reddit_account(self):
|
|
self.user.reddit_refresh_token = "test"
|
|
self.user.save()
|
|
|
|
response = self.client.get(self.url)
|
|
|
|
self.assertEquals(response.status_code, 200)
|
|
self.assertNotContains(response, "Authorize Reddit account")
|
|
|
|
|
|
class RedditTemplateViewTestCase(TestCase):
|
|
def setUp(self):
|
|
self.user = UserFactory(email="test@test.nl", password="test")
|
|
self.client.force_login(self.user)
|
|
|
|
self.base_url = reverse("accounts:reddit-template")
|
|
self.state = str(uuid4())
|
|
|
|
self.patch = patch("newsreader.news.collection.reddit.post")
|
|
self.mocked_post = self.patch.start()
|
|
|
|
def tearDown(self):
|
|
patch.stopall()
|
|
|
|
def test_simple(self):
|
|
response = self.client.get(self.base_url)
|
|
|
|
self.assertEquals(response.status_code, 200)
|
|
self.assertContains(response, "Return to settings page")
|
|
|
|
def test_successful_authorization(self):
|
|
self.mocked_post.return_value.json.return_value = {
|
|
"access_token": "1001010412",
|
|
"refresh_token": "134510143",
|
|
}
|
|
|
|
cache.set(f"{self.user.email}-reddit-auth", self.state)
|
|
|
|
params = {"state": self.state, "code": "Valid code"}
|
|
url = f"{self.base_url}?{urlencode(params)}"
|
|
|
|
response = self.client.get(url)
|
|
|
|
self.mocked_post.assert_called_once()
|
|
|
|
self.assertEquals(response.status_code, 200)
|
|
self.assertContains(response, "Your reddit account was successfully linked.")
|
|
|
|
self.user.refresh_from_db()
|
|
|
|
self.assertEquals(self.user.reddit_access_token, "1001010412")
|
|
self.assertEquals(self.user.reddit_refresh_token, "134510143")
|
|
|
|
self.assertEquals(cache.get(f"{self.user.email}-reddit-auth"), None)
|
|
|
|
def test_error(self):
|
|
params = {"error": "Denied authorization"}
|
|
|
|
url = f"{self.base_url}?{urlencode(params)}"
|
|
|
|
response = self.client.get(url)
|
|
|
|
self.assertEquals(response.status_code, 200)
|
|
self.assertContains(response, "Denied authorization")
|
|
|
|
def test_invalid_state(self):
|
|
cache.set(f"{self.user.email}-reddit-auth", str(uuid4()))
|
|
|
|
params = {"code": "Valid code", "state": "Invalid state"}
|
|
|
|
url = f"{self.base_url}?{urlencode(params)}"
|
|
|
|
response = self.client.get(url)
|
|
|
|
self.assertEquals(response.status_code, 200)
|
|
self.assertContains(
|
|
response, "The saved state for Reddit authorization did not match"
|
|
)
|
|
|
|
def test_stream_error(self):
|
|
self.mocked_post.side_effect = StreamTooManyException
|
|
|
|
cache.set(f"{self.user.email}-reddit-auth", self.state)
|
|
|
|
params = {"state": self.state, "code": "Valid code"}
|
|
url = f"{self.base_url}?{urlencode(params)}"
|
|
|
|
response = self.client.get(url)
|
|
|
|
self.mocked_post.assert_called_once()
|
|
|
|
self.assertEquals(response.status_code, 200)
|
|
self.assertContains(response, "Too many requests")
|
|
|
|
self.user.refresh_from_db()
|
|
|
|
self.assertEquals(self.user.reddit_access_token, None)
|
|
self.assertEquals(self.user.reddit_refresh_token, None)
|
|
|
|
self.assertEquals(cache.get(f"{self.user.email}-reddit-auth"), self.state)
|
|
|
|
def test_unexpected_json(self):
|
|
self.mocked_post.return_value.json.return_value = {"message": "Happy eastern"}
|
|
|
|
cache.set(f"{self.user.email}-reddit-auth", self.state)
|
|
|
|
params = {"state": self.state, "code": "Valid code"}
|
|
url = f"{self.base_url}?{urlencode(params)}"
|
|
|
|
response = self.client.get(url)
|
|
|
|
self.mocked_post.assert_called_once()
|
|
|
|
self.assertEquals(response.status_code, 200)
|
|
self.assertContains(response, "Access and refresh token not found in response")
|
|
|
|
self.user.refresh_from_db()
|
|
|
|
self.assertEquals(self.user.reddit_access_token, None)
|
|
self.assertEquals(self.user.reddit_refresh_token, None)
|
|
|
|
self.assertEquals(cache.get(f"{self.user.email}-reddit-auth"), self.state)
|