Dockery/Pinry
1
0
Fork 0
mirror of https://github.com/pinry/pinry.git synced 2025-11-15 09:25:50 +01:00
Code Issues Packages Projects Releases Activity

Fix test by mocking requests.get function

Browse Source
This commit is contained in:
Krzysztof Klimonda
2013-03-19 15:41:55 +01:00
parent 98876bb0bd
commit 494f7fbd8c
2 changed files with 13 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
pinry/users/tests.py
View File

@@ -9,8 +9,9 @@ from ..core.models import Image, Pin
from .models import User
def mock_urlopen(url):
return open('logo.png')
def mock_requests_get(url):
response = mock.Mock(content=open('logo.png').read())
return response
class CombinedAuthBackendTest(TestCase):
@@ -33,14 +34,14 @@ class CombinedAuthBackendTest(TestCase):
def test_authenticate_unknown_user(self):
self.assertIsNone(self.backend.authenticate(username='wrong-username', password='wrong-password'))
@mock.patch('urllib2.urlopen', mock_urlopen)
@mock.patch('requests.get', mock_requests_get)
def test_has_perm_on_pin(self):
image = Image.objects.create_for_url('http://testserver/mocked/screenshot.png')
user = User.objects.get(pk=1)
pin = Pin.objects.create(submitter=user, image=image)
self.assertTrue(self.backend.has_perm(user, 'add_pin', pin))
@mock.patch('urllib2.urlopen', mock_urlopen)
@mock.patch('requests.get', mock_requests_get)
def test_has_perm_on_pin_unauthorized(self):
image = Image.objects.create_for_url('http://testserver/mocked/screenshot.png')
user = User.objects.get(pk=1)
@@ -73,4 +74,4 @@ class LogoutViewTest(TestCase):
def test_logout_view(self):
response = self.client.get(reverse('users:logout'))
self.assertRedirects(response, reverse('core:recent-pins'))
self.assertRedirects(response, reverse('core:recent-pins'))