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)
