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'