mygpo/data/tests.py

   1 import re
   2 import json
   3
   4 from django.test import TestCase
   5
   6 from . import flickr
   7
   8 import responses
   9
  10
  11 MEDIUM_URL = "https://farm6.staticflickr.com/5001/1246644888_36863b0856.jpg"
  12
  13 API_RESPONSE = {
  14     "stat": "ok",
  15     "sizes": {
  16         "canblog": 0,
  17         "size": [
  18             {
  19                 "source": "https://farm6.staticflickr.com/5001/1234533888_45673b0856_s.jpg",
  20                 "url": "https://www.flickr.com/photos/someuser/135643888/sizes/sq/",
  21                 "media": "photo",
  22                 "height": 75,
  23                 "width": 75,
  24                 "label": "Square",
  25             },
  26             {
  27                 "source": MEDIUM_URL,
  28                 "url": "https://www.flickr.com/photos/someuser/3465234888/sizes/m/",
  29                 "media": "photo",
  30                 "height": "500",
  31                 "width": "333",
  32                 "label": "Medium",
  33             },
  34         ],
  35         "candownload": 1,
  36         "canprint": 0,
  37     },
  38 }
  39
  40 FLICKR_URL = re.compile(
  41     r"https://api.flickr.com/services/rest/\?method=flickr.photos.getSizes&api_key=.*photo_id=.*&format=json&nojsoncallback=1"
  42 )
  43
  44
  45 class FlickrTests(TestCase):
  46     def test_get_sizes(self):
  47         with responses.RequestsMock() as rsps:
  48             rsps.add(
  49                 responses.GET, FLICKR_URL, status=200, body=json.dumps(API_RESPONSE)
  50             )
  51
  52             sizes = flickr.get_photo_sizes("1235123123")
  53
  54         self.assertEqual(sizes, API_RESPONSE["sizes"]["size"])
  55
  56     def test_display_image(self):
  57         with responses.RequestsMock() as rsps:
  58             rsps.add(
  59                 responses.GET, FLICKR_URL, status=200, body=json.dumps(API_RESPONSE)
  60             )
  61
  62             disp_photo = flickr.get_display_photo(
  63                 "https://farm9.staticflickr.com/8747/12346789012_bf1e234567_b.jpg"
  64             )
  65
  66         self.assertEqual(disp_photo, MEDIUM_URL)