def test_get_files_in_directory_generator_on_empty_dir(self): temp_dir = tempfile.mkdtemp() try: tsf = TaskSummaryFactory(None) count = 0 for item in tsf._get_files_in_directory_generator(temp_dir): count += 1 self.assertEqual(count, 0) finally: shutil.rmtree(temp_dir)
def test_get_files_in_directory_generator_on_dir_with_file(self): temp_dir = tempfile.mkdtemp() try: tsf = TaskSummaryFactory(None) count = 0 onefile = os.path.join(temp_dir, 'foo.txt') open(onefile, 'a').close() for item in tsf._get_files_in_directory_generator(temp_dir): self.assertEqual(item, onefile) count += 1 self.assertEqual(count, 1) finally: shutil.rmtree(temp_dir)
def test_get_files_in_directory_generator_on_multiple_files(self): temp_dir = tempfile.mkdtemp() try: tsf = TaskSummaryFactory(None) one = os.path.join(temp_dir, '1.txt') two = os.path.join(temp_dir, '2.txt') three = os.path.join(temp_dir, '3.txt') open(one, 'a').close() open(two, 'a').close() open(three, 'a').close() thefiles = [] for item in tsf._get_files_in_directory_generator(temp_dir): thefiles.append(item) self.assertTrue(one in thefiles) self.assertTrue(two in thefiles) self.assertTrue(three in thefiles) self.assertEqual(len(thefiles), 3) finally: shutil.rmtree(temp_dir)
def test_get_files_in_directory_generator_none_passed_in(self): tsf = TaskSummaryFactory(None) count = 0 for item in tsf._get_files_in_directory_generator(None): count += 1 self.assertEqual(count, 0)