class ClsTest(unittest.TestCase): def setUp(self): self.cwd = os.getcwd() os.chdir(TEST_DIR) os.mkdir("Class") self.cls = Class() def tearDown(self): os.rmdir("Class") os.chdir(self.cwd) del self.cls def test__fill_template(self): self.assertEqual(self.cls._fill_template("Hello", "World"), "Foo|Bar|Hello|World") def test__create_class1(self): class_string = "" name = "Baz" target = os.path.join("Class", name + ".php") self.cls._create_class(name, "World") with open(target) as class_file: for line in class_file: class_string += line self.assertEqual(class_string, "Foo|Bar|" + name + "|World") os.remove(target) def test__create_class2(self): class_string = "" name = "Quux_Qux" target = os.path.join("Class", "Quux", "Qux.php") self.cls._create_class(name, "World") with open(target) as class_file: for line in class_file: class_string += line self.assertEqual(class_string, "Foo|Bar|Quux_Qux|World") os.remove(target) os.rmdir(target[:target.rfind(os.sep)]) def test__words_to_dirs1(self): self.assertEqual(self.cls._words_to_dirs(TEST_DIR, "Product"), TEST_DIR) def test__words_to_dirs2(self): self.assertEqual(self.cls._words_to_dirs(TEST_DIR, "Abc_Def"), os.path.join(TEST_DIR, "Abc")) def test__words_to_dirs3(self): self.assertEqual(self.cls._words_to_dirs(TEST_DIR, "Abc_Def_Ghi"), os.path.join(TEST_DIR, "Abc", "Def")) def test__create_missing_dirs1(self): self.cls._create_missing_dirs(os.path.join("abc", "def", "ghi")) os.rmdir(os.path.join("abc", "def", "ghi")) os.rmdir(os.path.join("abc", "def")) os.rmdir("abc") def test__create_missing_dirs2(self): self.cls._create_missing_dirs(os.path.join(os.path.abspath(os.getcwd()), "abc", "def", "ghi")) os.rmdir(os.path.join("abc", "def", "ghi")) os.rmdir(os.path.join("abc", "def")) os.rmdir("abc") def test__prepare_path_to(self): self.assertEqual(self.cls._prepare_path_to("Product"), os.path.join(TEST_DIR, "Class", "Product.php"))