def test_date_setting(self):
datefile = Dated_file("/path/2003_10_02_test_file.txt")
datefile.set_date("2005_11_04")
self.assertEqual(
datefile.get_date(),
"2005_11_04"
)
def test_tuncate_file_name(self):
datefile = Dated_file("/path/2003_10_02_test_file.txt")
self.assertEqual(
datefile.get_file_name(),
"test_file.txt"
)
self.assertEqual(
datefile.get_full_file_name(),
"2003_10_02_test_file.txt"
)
def test_test_date_found(self):
'''
if there is no date and older file don't exist we should set the \
current date.
'''
datefile = Dated_file("/path/2003_10_02_test_file.txt")
self.assertEqual(
datefile.get_date(),
"2003_10_02"
)
def test_set_date_existed_file(self):
with open(os.path.abspath("./data/examples/test_file2.txt"), 'w'):
datefile = Dated_file(
"./data/examples/test_file2.txt",
"2008_12_02")
datefile = Dated_file("./data/examples/2008_12_02_test_file2.txt")
self.assertEqual(
datefile.get_full_file_name(),
"2008_12_02_test_file2.txt"
)
os.remove(os.path.abspath(
"./data/examples/2008_12_02_test_file2.txt"))
def test_set_date_existed_folder(self):
newpath = "./data/examples/test_folder2"
if not os.path.exists(newpath):
os.makedirs(newpath)
datefile = Dated_file(
"./data/examples/test_folder2",
"2008_12_02")
datefile = Dated_file("./data/examples/2008_12_02_test_folder2")
self.assertEqual(
datefile.get_full_file_name(),
"2008_12_02_test_folder2"
)
os.rmdir(os.path.abspath(
"./data/examples/2008_12_02_test_folder2"))
def test_set_to_dated_existing_folder(self):
datefile = Dated_file("./data/examples/2004_12_02_test_folder")
self.assertEqual(
datefile.get_date(),
"2004_12_02"
)
self.assertEqual(
datefile.get_file_name(),
"test_folder"
)
self.assertEqual(
datefile.get_full_file_name(),
"2004_12_02_test_folder"
)
def test_set_to_last_existing_file(self):
datefile = Dated_file("./data/examples/test_file.txt")
self.assertEqual(
datefile.get_date(),
"2006_02_08"
)
self.assertEqual(
datefile.get_file_name(),
"test_file.txt"
)
self.assertEqual(
datefile.get_full_file_name(),
"2006_02_08_test_file.txt"
)
def test_test_no_date_found(self):
'''
if there is no date and older file don't exist we should set the \
current date.
'''
datefile = Dated_file("/path/test_file.txt")
current_date = datetime.date.today()
self.assertEqual(
datefile.get_date(),
current_date.strftime("%Y_%m_%d")
)
self.assertEqual(
datefile.get_full_file_name(),
current_date.strftime("%Y_%m_%d_") + "test_file.txt"
)
def test_date_existed_file(self):
with open(os.path.abspath("./data/examples/test_file2.txt"), 'w'):
datefile = Dated_file("./data/examples/test_file2.txt")
current_date = datetime.date.today()
self.assertTrue(
os.path.isfile(
"./data/examples/" +
current_date.strftime("%Y_%m_%d_") +
"test_file2.txt"))
self.assertEqual(
datefile.get_full_file_name(),
current_date.strftime("%Y_%m_%d_") + "test_file2.txt"
)
os.remove(os.path.abspath(
"./data/examples/" +
current_date.strftime("%Y_%m_%d_") +
"test_file2.txt"))
def test_date_existed_folder(self):
newpath = "./data/examples/test_folder"
if not os.path.exists(newpath):
os.makedirs(newpath)
datefile = Dated_file("./data/examples/test_folder",
date=None,
redate=True)
current_date = datetime.date.today()
self.assertTrue(
os.path.exists(
"./data/examples/" +
current_date.strftime("%Y_%m_%d_") +
"test_folder"))
self.assertEqual(
datefile.get_full_file_name(),
current_date.strftime("%Y_%m_%d_") + "test_folder"
)
os.rmdir(os.path.abspath(
"./data/examples/" +
current_date.strftime("%Y_%m_%d_") +
"test_folder"))
def test_list_files_empty(self):
datefile = Dated_file("./data/examples/2004_10_02_test_file.txt")
self.assertEqual(
datefile[2],
"2004_10_02_test_file.txt"
)
self.assertEqual(
datefile[1],
"2004_12_02_test_file.txt"
)
self.assertEqual(
datefile[0],
"2006_02_08_test_file.txt"
)
def test_abs_path(self):
datefile = Dated_file("/path/2003_10_02_test_file.txt")
self.assertEqual(
datefile.get_file_path(),
os.path.abspath("/path/")
)