class TestTXTHandler(unittest.TestCase):
def setUp(self):
self.txt_handler = TXTHandler()
# get the pathdir where the application is running
file = sys.argv[0]
self.path_name = os.path.dirname(file) + "/"
# create an file to test the existing file case
my_file = open(self.path_name + 'test_exist_file.txt', 'w')
my_file.write("te")
my_file.close()
def tearDown(self):
os.remove(self.path_name + 'test_exist_file.txt')
def test_the_name_file_increments_when_the_same_file_exist(self):
file_name = self.txt_handler.get_file_name(self.path_name, 'test_exist_file.txt', 0)
self.assertEquals('test_exist_file1.txt', file_name)
def test_the_name_file_is_the_same_when_the_file_not_exist(self):
file_name = self.txt_handler.get_file_name(self.path_name, 'test_not_exist_file.txt', 0)
self.assertEquals('test_not_exist_file.txt', file_name)
def test_write_sudoku_returns_true_file_has_been_write(self):
sudoku_was_written = self.txt_handler.write_sudoku(self.path_name, 'test_not_exist_file.txt', "test")
os.remove(self.path_name + 'test_not_exist_file.txt')
self.assertTrue(sudoku_was_written)
def test_write_sudoku_writes_the_specified_line(self):
line_to_write = "test\n" + \
"this is a test"
self.txt_handler.write_sudoku(self.path_name ,
'test_write_specified_line.txt', line_to_write)
file_written = open(self.path_name + 'test_write_specified_line.txt', "r")
content = file_written.readlines()
line_written = ""
for line in content :
line_written = line_written + line
file_written.close()
os.remove(self.path_name + 'test_write_specified_line.txt')
self.assertEquals(line_written,line_to_write)
