def test_collect_file_resources___different_types():
from StringIO import StringIO as SIO
from cStringIO import StringIO as cSIO
file_obj = Mock(spec=file)
sio_obj = SIO('GIF89a')
csio_obj = cSIO('GIF89a')
request_obj = dict(
folder=dict(
name='mordor',
pic=file_obj,
hd=sio_obj,
ld=csio_obj
)
)
file_cid_map = _collect_file_resources(request_obj)
assert type(request_obj['folder']['pic']) == str
assert type(request_obj['folder']['hd']) == str
assert type(request_obj['folder']['ld']) == str
assert file_cid_map[extract_cid(request_obj['folder']['pic'])] == file_obj
assert file_cid_map[extract_cid(request_obj['folder']['hd'])] == sio_obj
assert file_cid_map[extract_cid(request_obj['folder']['ld'])] == csio_obj
def test_collect_file_resources___nested():
pic_obj1 = Mock(spec=file)
pic_obj2 = Mock(spec=file)
request_obj = dict(
folder=dict(
name='mordor',
pic = pic_obj1,
child=dict(
name='gondor',
pic = pic_obj2
)
)
)
file_cid_map = _collect_file_resources(request_obj)
assert type(request_obj['folder']['pic']) == str
assert type(request_obj['folder']['child']['pic']) == str
assert file_cid_map[extract_cid(request_obj['folder']['pic'])] == pic_obj1
assert file_cid_map[extract_cid(request_obj['folder']['child']['pic'])] == pic_obj2
