def test_get_image_from_src_url(self):
uri = 'http://httpbin.org/image/png'
mock_request(uri, method='GET', status=200, fixture='image1.png', include_location=False)
web_data = image.get_image_from_src(uri)
local_data = get_fixture('image1.png', as_binary=True)
self.assertEqual(web_data, local_data)
def test_get_image_from_src_invalid_data(self):
img_data = 'test data'
result = image.get_image_from_src(img_data)
self.assertEqual('test data', result)
def test_get_image_from_src_data(self):
img_data = get_fixture('image1.data', as_binary=True)
result = image.get_image_from_src(img_data)
self.assertEqual(img_data, result)