def test_list_files_should_list_files_when_root_folder_is_passed(
self, files_fixture):
self.user.getNotInSetPhotos.return_value = files_fixture
storage = FlickrStorage(self.config, Resiliently(self.config))
folder = RootFolder()
files = list(storage.list_files(folder))
assert len(files) == 2
def test_list_files_should_return_files_given_there_are_files(
self, folders_fixture, files_fixture):
self.user.getPhotosets.return_value = folders_fixture
folders_fixture[0].getPhotos.return_value = files_fixture
storage = FlickrStorage(self.config, Resiliently(self.config))
folders = list(storage.list_folders())
files = list(storage.list_files(folders[0]))
assert len(files) == 2
def test_list_files_should_not_list_file_given_its_not_included(
self, folders_fixture, files_fixture):
self.config.include = 'image1'
self.user.getPhotosets.return_value = folders_fixture
folders_fixture[0].getPhotos.return_value = files_fixture
storage = FlickrStorage(self.config, Resiliently(self.config))
folders = list(storage.list_folders())
files = list(storage.list_files(folders[0]))
assert len(files) == 1
assert files[0].name == 'image1.jpg'