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