__author__ = 'stubbfel' import unittest from utilities.textfileutility import TextFile import os.path class TextFileUtilitiesTest(unittest.TestCase): def setUp(self): self.__tf = TextFile("test.txt") self.__tf.check_and_remove_file() def test_write_file(self): self.__tf.write_text_file("this is a test") self.__tf.check_and_remove_file() def test_exists_file_true_case(self): self.assertEqual(os.path.isfile("test.txt"), self.__tf.exists_file) self.__tf.check_and_remove_file() def test_exists_file_false_case(self): self.__tf.check_and_remove_file() self.assertEqual(os.path.isfile("test.txt"), self.__tf.exists_file) def test_check_and_remove_file(self): self.__tf.write_text_file("this is a test") self.__tf.check_and_remove_file() self.assertFalse(self.__tf.exists_file) def test_get_filename_true(self): self.assertEqual(self.__tf.get_filename, "test.txt") def test_get_filename_false(self): self.assertNotEqual(self.__tf.get_filename, "test1.txt") if __name__ == '__main__': unittest.main()