def setUp(self):
self.flickr = FlickrClient()
self.token = {'oauth_token_secret': u'secret', 'oauth_token': u'token'}
self.user = {'username': u'palotespaco', 'user_nsid': u'user'}
self.flickr = FlickrClient()
self.flickr.oauth_client = MagicMock()
self.flickr.app = MagicMock()
def setUp(self):
self.flickr = FlickrClient()
self.token = {'oauth_token_secret': u'secret', 'oauth_token': u'token'}
self.user = {'username': u'palotespaco', 'user_nsid': u'user'}
self.flickr = FlickrClient()
self.flickr.client = MagicMock()
self.flickr.app = MagicMock()
# CSRF protection from flask_wtf.csrf import CsrfProtect csrf = CsrfProtect() # Timeouts timeouts = dict() # Ratelimits ratelimits = dict() # Newsletter from newsletter import Newsletter newsletter = Newsletter() # Importer from importers import Importer importer = Importer() # Flickr OAuth integration for importer from pybossa.flickr_client import FlickrClient flickr = FlickrClient() from flask.ext.plugins import PluginManager plugin_manager = PluginManager()
class TestFlickrClient(object):
class Res(object):
def __init__(self, status, data):
self.status_code = status
self.json = lambda: data
def setUp(self):
self.token = {'oauth_token_secret': u'secret', 'oauth_token': u'token'}
self.user = {'username': u'palotespaco', 'user_nsid': u'user'}
self.flickr = FlickrClient('key', MagicMock())
@patch('pybossa.flickr_client.requests')
def test_get_user_albums_calls_flickr_api_endpoint(self, fake_requests):
session = {'flickr_token': self.token, 'flickr_user': self.user}
url = 'https://api.flickr.com/services/rest/'
payload = {'method': 'flickr.photosets.getList',
'api_key': 'key',
'user_id': 'user',
'format': 'json',
'primary_photo_extras':'url_q',
'nojsoncallback': '1'}
self.flickr.get_user_albums(session)
fake_requests.get.assert_called_with(url, params=payload)
@patch('pybossa.flickr_client.requests')
def test_get_user_albums_return_empty_list_on_request_error(self, fake_requests):
response = self.Res(404, 'not found')
fake_requests.get.return_value = response
session = {'flickr_token': self.token, 'flickr_user': self.user}
albums = self.flickr.get_user_albums(session)
assert albums == [], albums
@patch('pybossa.flickr_client.requests')
def test_get_user_albums_return_empty_list_on_request_fail(self, fake_requests):
data = {'stat': 'fail', 'code': 1, 'message': 'User not found'}
response = self.Res(200, data)
fake_requests.get.return_value = response
session = {'flickr_token': self.token, 'flickr_user': self.user}
albums = self.flickr.get_user_albums(session)
assert albums == [], albums
@patch('pybossa.flickr_client.requests')
def test_get_user_albums_log_response_on_request_fail(self, fake_requests):
data = {'stat': 'fail', 'code': 1, 'message': 'User not found'}
response = self.Res(200, data)
fake_requests.get.return_value = response
log_error_msg = ("Bad response from Flickr:\nStatus: %s, Content: %s"
% (response.status_code, response.json()))
session = {'flickr_token': self.token, 'flickr_user': self.user}
albums = self.flickr.get_user_albums(session)
self.flickr.logger.error.assert_called_with(log_error_msg)
@patch('pybossa.flickr_client.requests')
def test_get_user_albums_return_list_with_album_info(self, fake_requests):
data = {
u'stat': u'ok',
u'photosets': {
u'total': 2,
u'perpage': 2,
u'photoset':
[{u'date_update': u'1421313791',
u'visibility_can_see_set': 1,
u'description': {u'_content': u'mis mejores vacaciones'},
u'videos': 0, u'title': {u'_content': u'vacaciones'},
u'farm': 9, u'needs_interstitial': 0,
u'primary': u'16284868505',
u'primary_photo_extras': {
u'height_t': u'63',
u'width_t': u'100',
u'url_q': u'https://farm9.staticflickr.com/8597/16284868505_c4a032a62e_t.jpg'},
u'server': u'8597',
u'date_create': u'1421313790',
u'photos': u'3',
u'secret': u'c4a032a62e',
u'count_comments': u'0',
u'count_views': u'1',
u'can_comment': 0,
u'id': u'72157649886540037'}],
u'page': 1,
u'pages': 1}}
response = self.Res(200, data)
fake_requests.get.return_value = response
session = {'flickr_token': self.token, 'flickr_user': self.user}
albums = self.flickr.get_user_albums(session)
expected_album = response.json()['photosets']['photoset'][0]
expected_album_info = {
'photos': expected_album['photos'],
'thumbnail_url': expected_album['primary_photo_extras']['url_q'],
'id': expected_album['id'],
'title': expected_album['title']['_content']}
assert albums == [expected_album_info], albums
class TestFlickrClient(object):
class Res(object):
def __init__(self, status, data):
self.status = status
self.data = data
def setUp(self):
self.flickr = FlickrClient()
self.token = {'oauth_token_secret': u'secret', 'oauth_token': u'token'}
self.user = {'username': u'palotespaco', 'user_nsid': u'user'}
self.flickr = FlickrClient()
self.flickr.client = MagicMock()
self.flickr.app = MagicMock()
def test_flickr_get_token_returns_None_if_no_token(self):
session = {}
token = self.flickr.get_token(session)
assert token is None, token
def test_flickr_get_token_returns_existing_token_as_a_tuple(self):
session = {'flickr_token': self.token}
token = self.flickr.get_token(session)
assert token == (u'token', u'secret'), token
def test_save_credentials_stores_token_and_user(self):
session = {}
self.flickr.save_credentials(session, self.token, self.user)
assert session.get('flickr_token') is self.token
assert session.get('flickr_user') is self.user
def test_remove_token_deletes_token(self):
session = {'flickr_token': self.token, 'flickr_user': self.user}
self.flickr.remove_credentials(session)
assert session.get('flickr_token') is None
assert session.get('flickr_user') is None
def test_get_user_albums_calls_flickr_api_endpoint(self):
session = {'flickr_token': self.token, 'flickr_user': self.user}
url = ('https://api.flickr.com/services/rest/?'
'method=flickr.photosets.getList&user_id=user'
'&primary_photo_extras=url_q'
'&format=json&nojsoncallback=1')
self.flickr.get_user_albums(session)
# The request MUST NOT include a valid token, to avoid private photos
self.flickr.client.get.assert_called_with(url, token='')
def test_get_user_albums_return_empty_list_on_request_error(self):
response = self.Res(404, 'not found')
self.flickr.client.get.return_value = response
session = {'flickr_token': self.token, 'flickr_user': self.user}
albums = self.flickr.get_user_albums(session)
assert albums == [], albums
def test_get_user_albums_return_empty_list_on_request_fail(self):
data = {'stat': 'fail', 'code': 1, 'message': 'User not found'}
response = self.Res(200, data)
self.flickr.client.get.return_value = response
session = {'flickr_token': self.token, 'flickr_user': self.user}
albums = self.flickr.get_user_albums(session)
assert albums == [], albums
def test_get_user_albums_log_response_on_request_fail(self):
data = {'stat': 'fail', 'code': 1, 'message': 'User not found'}
response = self.Res(200, data)
self.flickr.client.get.return_value = response
log_error_msg = ("Bad response from Flickr:\nStatus: %s, Content: %s" %
(response.status, response.data))
session = {'flickr_token': self.token, 'flickr_user': self.user}
albums = self.flickr.get_user_albums(session)
self.flickr.app.logger.error.assert_called_with(log_error_msg)
def test_get_user_albums_return_list_with_album_info(self):
data = {
u'stat': u'ok',
u'photosets': {
u'total':
2,
u'perpage':
2,
u'photoset': [{
u'date_update': u'1421313791',
u'visibility_can_see_set': 1,
u'description': {
u'_content': u'mis mejores vacaciones'
},
u'videos': 0,
u'title': {
u'_content': u'vacaciones'
},
u'farm': 9,
u'needs_interstitial': 0,
u'primary': u'16284868505',
u'primary_photo_extras': {
u'height_t':
u'63',
u'width_t':
u'100',
u'url_q':
u'https://farm9.staticflickr.com/8597/16284868505_c4a032a62e_t.jpg'
},
u'server': u'8597',
u'date_create': u'1421313790',
u'photos': u'3',
u'secret': u'c4a032a62e',
u'count_comments': u'0',
u'count_views': u'1',
u'can_comment': 0,
u'id': u'72157649886540037'
}],
u'page':
1,
u'pages':
1
}
}
response = self.Res(200, data)
self.flickr.client.get.return_value = response
session = {'flickr_token': self.token, 'flickr_user': self.user}
albums = self.flickr.get_user_albums(session)
expected_album = response.data['photosets']['photoset'][0]
expected_album_info = {
'photos': expected_album['photos'],
'thumbnail_url': expected_album['primary_photo_extras']['url_q'],
'id': expected_album['id'],
'title': expected_album['title']['_content']
}
assert albums == [expected_album_info], albums
def user_albums():
flickr_api_key = current_app.config['FLICKR_API_KEY']
client = FlickrClient(flickr_api_key, current_app.logger)
albums = client.get_user_albums(session)
return Response(json.dumps(albums), mimetype='application/json')
def user_albums():
flickr_api_key = current_app.config['FLICKR_API_KEY']
client = FlickrClient(flickr_api_key, current_app.logger)
albums = client.get_user_albums(session)
return Response(json.dumps(albums), mimetype='application/json')
def setUp(self):
self.token = {'oauth_token_secret': 'secret', 'oauth_token': 'token'}
self.user = {'username': '******', 'user_nsid': 'user'}
self.flickr = FlickrClient('key', MagicMock())
class TestFlickrClient(object):
class Res(object):
def __init__(self, status, data):
self.status = status
self.data = data
def setUp(self):
self.flickr = FlickrClient()
self.token = {'oauth_token_secret': u'secret', 'oauth_token': u'token'}
self.user = {'username': u'palotespaco', 'user_nsid': u'user'}
self.flickr = FlickrClient()
self.flickr.oauth_client = MagicMock()
self.flickr.app = MagicMock()
def test_flickr_get_token_returns_None_if_no_token(self):
session = {}
token = self.flickr.get_token(session)
assert token is None, token
def test_flickr_get_token_returns_existing_token_as_a_tuple(self):
session = {'flickr_token': self.token}
token = self.flickr.get_token(session)
assert token == (u'token', u'secret'), token
def test_save_credentials_stores_token_and_user(self):
session = {}
self.flickr.save_credentials(session, self.token, self.user)
assert session.get('flickr_token') is self.token
assert session.get('flickr_user') is self.user
def test_remove_token_deletes_token(self):
session = {'flickr_token': self.token, 'flickr_user': self.user}
self.flickr.remove_credentials(session)
assert session.get('flickr_token') is None
assert session.get('flickr_user') is None
def test_get_user_albums_calls_flickr_api_endpoint(self):
session = {'flickr_token': self.token, 'flickr_user': self.user}
url = ('https://api.flickr.com/services/rest/?'
'method=flickr.photosets.getList&user_id=user'
'&primary_photo_extras=url_q'
'&format=json&nojsoncallback=1')
self.flickr.get_user_albums(session)
# The request MUST NOT include a valid token, to avoid private photos
self.flickr.oauth_client.get.assert_called_with(url, token='')
def test_get_user_albums_return_empty_list_on_request_error(self):
response = self.Res(404, 'not found')
self.flickr.oauth_client.get.return_value = response
session = {'flickr_token': self.token, 'flickr_user': self.user}
albums = self.flickr.get_user_albums(session)
assert albums == [], albums
def test_get_user_albums_return_empty_list_on_request_fail(self):
data = {'stat': 'fail', 'code': 1, 'message': 'User not found'}
response = self.Res(200, data)
self.flickr.oauth_client.get.return_value = response
session = {'flickr_token': self.token, 'flickr_user': self.user}
albums = self.flickr.get_user_albums(session)
assert albums == [], albums
def test_get_user_albums_log_response_on_request_fail(self):
data = {'stat': 'fail', 'code': 1, 'message': 'User not found'}
response = self.Res(200, data)
self.flickr.oauth_client.get.return_value = response
log_error_msg = ("Bad response from Flickr:\nStatus: %s, Content: %s"
% (response.status, response.data))
session = {'flickr_token': self.token, 'flickr_user': self.user}
albums = self.flickr.get_user_albums(session)
self.flickr.app.logger.error.assert_called_with(log_error_msg)
def test_get_user_albums_return_list_with_album_info(self):
data = {
u'stat': u'ok',
u'photosets': {
u'total': 2,
u'perpage': 2,
u'photoset':
[{u'date_update': u'1421313791',
u'visibility_can_see_set': 1,
u'description': {u'_content': u'mis mejores vacaciones'},
u'videos': 0, u'title': {u'_content': u'vacaciones'},
u'farm': 9, u'needs_interstitial': 0,
u'primary': u'16284868505',
u'primary_photo_extras': {
u'height_t': u'63',
u'width_t': u'100',
u'url_q': u'https://farm9.staticflickr.com/8597/16284868505_c4a032a62e_t.jpg'},
u'server': u'8597',
u'date_create': u'1421313790',
u'photos': u'3',
u'secret': u'c4a032a62e',
u'count_comments': u'0',
u'count_views': u'1',
u'can_comment': 0,
u'id': u'72157649886540037'}],
u'page': 1,
u'pages': 1}}
response = self.Res(200, data)
self.flickr.oauth_client.get.return_value = response
session = {'flickr_token': self.token, 'flickr_user': self.user}
albums = self.flickr.get_user_albums(session)
expected_album = response.data['photosets']['photoset'][0]
expected_album_info = {
'photos': expected_album['photos'],
'thumbnail_url': expected_album['primary_photo_extras']['url_q'],
'id': expected_album['id'],
'title': expected_album['title']['_content']}
assert albums == [expected_album_info], albums
