def testAddImportToInit(self): """Tests adding a line to an init file.""" test_path = os.path.join(os.path.dirname( os.path.dirname(os.path.dirname(__file__))), 'test_data') test_file = os.path.join(test_path, 'test_init.py') with open(test_file, 'r') as fh: test_file_content = fh.read() new_import = 'from secret.project.parsers import foobar\n' temp_file = tempfile.NamedTemporaryFile() with open(temp_file.name, 'w') as fh: fh.write(test_file_content) handler = file_handler.FileHandler() handler.AddImportToInit(temp_file.name, new_import) with open(temp_file.name, 'r') as fh: content = fh.read() os.remove(temp_file.name) expected_file_path = os.path.join(test_path, 'test_init_fixed.py') with open(expected_file_path, 'r') as fh: expected_content = fh.read() self.assertEqual(content, expected_content)
def testCreateFileFromPath(self): """Tests if the creation of a file none existing beforehand works.""" handler = file_handler.FileHandler() with tempfile.TemporaryDirectory() as directory: source = os.path.join(directory, self.file) self.assertFalse(os.path.exists(source)) handler.CreateFileFromPath(source) self.assertTrue(os.path.exists(source))
def testCreateFile(self): """Tests if the creation of a file none existing beforehand works.""" with tempfile.TemporaryDirectory() as directory: handler = file_handler.FileHandler() file_path = os.path.join(directory, self.file) self.assertFalse(os.path.exists(file_path)) handler.CreateFile(directory, self.name, self.suffix) self.assertTrue(os.path.exists(file_path))
def testCreateFileIfPathNotExisting(self): """Tests creation of non-existing file.""" with tempfile.TemporaryDirectory() as directory: new_path = os.path.join(directory, "temp") handler = file_handler.FileHandler() file_path = os.path.join(new_path, self.file) self.assertFalse(os.path.exists(file_path)) handler.CreateFile(new_path, self.name, self.suffix) self.assertTrue(os.path.exists(file_path))
def testCreateFolder(self): """Tests if the creation of a folder works.""" with tempfile.TemporaryDirectory() as directory: new_path = os.path.join(directory, "newfolder") self.assertFalse(os.path.exists(new_path)) handler = file_handler.FileHandler() handler._CreateFolder(new_path) # pylint: disable=protected-access actual = os.path.exists(new_path) self.assertTrue(actual)
def __init__(self): """Initializes the engine.""" super(ScaffolderEngine, self).__init__() self._attributes = {} self._definition = '' self._definition_root_path = '' self._file_handler = file_handler.FileHandler() self._file_name_prefix = '' self._scaffolder = None self.module_name = ''
def testCopyFile(self): """Tests if the copying of a file none existing beforehand works.""" expected_content = "this is test content." with tempfile.TemporaryDirectory() as directory: source = os.path.join(directory, self.file) destination = os.path.join(directory, "copy", self.file) with open(source, "a") as f: f.write(expected_content) handler = file_handler.FileHandler() self.assertFalse(os.path.exists(destination)) handler.CopyFile(source, destination) self.assertTrue(os.path.exists(destination)) self.assertTrue(filecmp.cmp(destination, source))
def testAddContentIfFileDoesNotExist(self): """Tests if the editing of a file not existing works.""" content = "this is test content. " expected = content with tempfile.TemporaryDirectory() as directory: source = os.path.join(directory, self.file) handler = file_handler.FileHandler() self.assertFalse(os.path.exists(source)) handler.AddContent(source, content) self.assertTrue(os.path.exists(source)) with open(source, "r") as f: actual = f.read() self.assertEqual(expected, actual)
def testAddContentIfFileAndFolderDoesNotExist(self): """Tests creation or modification of non-existing file with content.""" content = "this is test content. " expected = content with tempfile.TemporaryDirectory() as directory: new_path = os.path.join(directory, "newfolder") source = os.path.join(new_path, self.file) handler = file_handler.FileHandler() self.assertFalse(os.path.exists(source)) handler.CreateOrModifyFileWithContent(source, content) self.assertTrue(os.path.exists(source)) with open(source, "r") as f: actual = f.read() self.assertEqual(expected, actual)
def testCreateOrModifyFileWithContentIfFileExists(self): """Tests creation or modification of existing file with content.""" content = "this is test content. " expected = content + content with tempfile.TemporaryDirectory() as directory: source = os.path.join(directory, self.file) with open(source, "a") as f: f.write(content) handler = file_handler.FileHandler() self.assertTrue(os.path.exists(source)) handler.CreateOrModifyFileWithContent(source, content) self.assertTrue(os.path.exists(source)) with open(source, "r") as f: actual = f.read() self.assertEqual(expected, actual)