1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
# -*- coding: utf-8 -*-
# thumbor imaging service
# https://github.com/thumbor/thumbor/wiki
# Licensed under the MIT license:
# http://www.opensource.org/licenses/mit-license
# Copyright (c) 2011 globo.com thumbor@googlegroups.com
from os.path import abspath, dirname, join
from preggy import expect
from tornado.testing import gen_test
from tests.base import TestCase
from thumbor.config import Config
from thumbor.context import Context
from thumbor.importer import Importer
from thumbor.loaders import LoaderResult
STORAGE_PATH = abspath(join(dirname(__file__), "../fixtures/images/"))
class CompatibilityLoaderTestCase(TestCase):
def get_context(self):
config = Config(
FILE_LOADER_ROOT_PATH=STORAGE_PATH,
COMPATIBILITY_LEGACY_LOADER="tests.compatibility.legacy_file_loader",
)
importer = Importer(config)
importer.import_modules()
return Context(config=config, importer=importer)
async def load_file(self, context, file_name):
Importer.deprecated_monkey_patch_tornado_return_future()
from thumbor.compatibility.loader import ( # pylint: disable=import-outside-toplevel
load,
)
return await load(context, file_name)
@gen_test
async def test_should_raise_for_invalid_compatibility_loader(self):
config = Config(
FILE_LOADER_ROOT_PATH=STORAGE_PATH,
)
importer = Importer(config)
importer.import_modules()
ctx = Context(config=config, importer=importer)
with expect.error_to_happen(
RuntimeError,
message=(
"The 'COMPATIBILITY_LEGACY_LOADER' configuration should point "
"to a valid loader when using compatibility loader."
),
):
await self.load_file(ctx, "image.jpg")
@gen_test
async def test_should_load_file(self):
result = await self.load_file(self.context, "image.jpg")
expect(result).to_be_instance_of(LoaderResult)
with open(join(STORAGE_PATH, "image.jpg"), "rb") as img:
expect(result.buffer).to_equal(img.read())
expect(result.successful).to_be_true()
|
