def test_iterate_on_dir(): work_dir = os.getcwd() test_dir_path = os.path.join(work_dir, 'dir') mkdir(test_dir_path) result = set() def fun(name, ntfile_instance): del ntfile_instance result.add(name) return True, False try: ntfile = NTFile(test_dir_path) status = ntfile.iterate_on_dir(fun, default_result=False) assert not result assert not status finally: ntfile.close() for n in range(0, 40): touch(os.path.join(test_dir_path, '%s.txt' % n)) try: ntfile = NTFile(test_dir_path) status = ntfile.iterate_on_dir(fun, default_result=False) assert status assert len(result) == 40 finally: ntfile.close()
def test_iterate_on_dir(): work_dir = os.getcwd() test_dir_path = os.path.join(work_dir, 'dir') mkdir(test_dir_path) result = set() def fun(name, ntfile_instance): result.add(name) return True, False try: ntfile = NTFile(test_dir_path) status = ntfile.iterate_on_dir(fun, default_result=False) assert not result assert not status finally: ntfile.close() for n in range(0, 40): touch(os.path.join(test_dir_path, '%s.txt' % n)) try: ntfile = NTFile(test_dir_path) status = ntfile.iterate_on_dir(fun, default_result=False) assert status assert len(result) == 40 finally: ntfile.close()
def test_iterate_on_dir(): work_dir = os.getcwd() test_dir_path = os.path.join(work_dir, "dir") mkdir(test_dir_path) result = set() def fun(name, ntfile_instance): del ntfile_instance result.add(name) return True, False try: ntfile = NTFile(test_dir_path) status = ntfile.iterate_on_dir(fun, default_result=False) assert not result assert not status finally: ntfile.close() for n in range(0, 40): touch(os.path.join(test_dir_path, "%s.txt" % n)) try: ntfile = NTFile(test_dir_path) status = ntfile.iterate_on_dir(fun, default_result=False) assert status assert len(result) == 40 finally: ntfile.close() test_file = os.path.join(test_dir_path, "not-a-directory.txt") touch(test_file) ntfile = NTFile(test_file) try: with pytest.raises(NTException): status = ntfile.iterate_on_dir(fun, default_result=False) finally: ntfile.close()