def setUp(self): HANDLED_FILES.clean() self.storage = get_storage() # foodb files HANDLED_FILES['written_files'] += [ (utils.filename_generate(ext, 'foodb'), None) for ext in ('db', 'db.gz', 'db.gpg', 'db.gz.gpg') ] HANDLED_FILES['written_files'] += [ (utils.filename_generate(ext, 'hamdb', 'fooserver'), None) for ext in ('db', 'db.gz', 'db.gpg', 'db.gz.gpg') ] # Media file HANDLED_FILES['written_files'] += [ (utils.filename_generate(ext, None, None, 'media'), None) for ext in ('tar', 'tar.gz', 'tar.gpg', 'tar.gz.gpg') ] HANDLED_FILES['written_files'] += [ (utils.filename_generate(ext, 'bardb', 'barserver'), None) for ext in ('db', 'db.gz', 'db.gpg', 'db.gz.gpg') ] # barserver files HANDLED_FILES['written_files'] += [ ('file_without_date', None) ]
def setUp(self): self.storage = FakeStorage() HANDLED_FILES['written_files'] += [ (utils.filename_generate(ext, 'foo'), None) for ext in ('db', 'db.gz', 'db.gpg', 'db.gz.gpg') ] HANDLED_FILES['written_files'] += [ (utils.filename_generate(ext, 'foo', None, 'media'), None) for ext in ('media', 'media.gz', 'media.gpg', 'media.gz.gpg') ] HANDLED_FILES['written_files'] += [('file_without_date', None)]
def setUp(self): HANDLED_FILES.clean() self.storage = get_storage() HANDLED_FILES['written_files'] += [ (utils.filename_generate(ext, 'foo'), None) for ext in ('db', 'db.gz', 'db.gpg', 'db.gz.gpg') ] HANDLED_FILES['written_files'] += [ (utils.filename_generate(ext, 'foo', None, 'media'), None) for ext in ('tar', 'tar.gz', 'tar.gpg', 'tar.gz.gpg') ] HANDLED_FILES['written_files'] += [('file_without_date', None)]
def setUp(self): self.storage = FakeStorage() HANDLED_FILES['written_files'] += [ (utils.filename_generate(ext, 'foo'), None) for ext in ('db', 'db.gz', 'db.gpg', 'db.gz.gpg') ] HANDLED_FILES['written_files'] += [ (utils.filename_generate(ext, 'foo', None, 'media'), None) for ext in ('media', 'media.gz', 'media.gpg', 'media.gz.gpg') ] HANDLED_FILES['written_files'] += [ ('file_without_date', None) ]
def setUp(self): HANDLED_FILES.clean() self.storage = get_storage() HANDLED_FILES['written_files'] += [ (utils.filename_generate(ext, 'foo'), None) for ext in ('db', 'db.gz', 'db.gpg', 'db.gz.gpg') ] HANDLED_FILES['written_files'] += [ (utils.filename_generate(ext, 'foo', None, 'media'), None) for ext in ('tar', 'tar.gz', 'tar.gpg', 'tar.gz.gpg') ] HANDLED_FILES['written_files'] += [ ('file_without_date', None) ]
def test_no_filepath(self, *args): # Create backup HANDLED_FILES['written_files'].append( (utils.filename_generate('foo'), BytesIO(b'bar'))) # Check self.command.filepath = None self.command.restore_backup()
def test_no_filepath(self, *args): # Create backup HANDLED_FILES['written_files'].append( (utils.filename_generate('foo'), BytesIO(b'bar'))) # Check self.command.filepath = None self.command._restore_backup()
def test_no_filename(self, *args): # Prepare backup HANDLED_FILES['written_files'].append( (utils.filename_generate('foo'), get_dump())) # Check self.command.path = None self.command.filename = None self.command._restore_backup()
def run(self): self.directory = tempfile.mkdtemp() self.write_migration_level() self.backup_db() self.backup_git_sources() self.backup_ruleset_middleware() # create tar archive of dir call_dir = os.getcwd() os.chdir(self.directory) filename = filename_generate('tar.bz2', self.dbcommands.settings.database['NAME'], self.servername) outputfile = tempfile.SpooledTemporaryFile() ts = tarfile.open(filename, 'w:bz2', fileobj=outputfile) for dfile in os.listdir('.'): ts.add(dfile) ts.close() self.storage.write_file(outputfile, filename) os.chdir(call_dir)
def backup_media(): extension = "tar.gz" filename = utils.filename_generate(extension, content_type='media') # Create tarball media_storage = get_storage_class()() outputfile = utils.create_spooled_temporary_file() tar_file = tarfile.open(name=filename, fileobj=outputfile, mode='w:gz') for media_filename in explore_storage(media_storage): tarinfo = tarfile.TarInfo(media_filename) media_file = media_storage.open(media_filename) tarinfo.size = len(media_file) tar_file.addfile(tarinfo, media_file) # Close the TAR for writing tar_file.close() # Store backup outputfile.seek(0) return outputfile, filename
def test_generated_filename(self): filename = utils.filename_generate('bak', 'default') datestring = utils.filename_to_date(filename)
def generate_filename(self, server_name=None): return utils.filename_generate(self.extension, self.database_name, server_name)
def test_no_filepath(self, *args): # Create backup HANDLED_FILES["written_files"].append((utils.filename_generate("foo"), BytesIO(b"bar"))) # Check self.command.filepath = None self.command.restore_backup()
def test_generated_filename(self): filename = utils.filename_generate("bak", "default") datestring = utils.filename_to_date(filename)
def generate_filename(self, server_name=None): return utils.filename_generate(self.extension, self.settings['NAME'], server_name)
def test_template_is_callable(self, *args): extension = 'foo' generated_name = utils.filename_generate(extension) self.assertTrue(generated_name.endswith('foo'))
def test_tz_true(self): filename = utils.filename_generate('bak', 'default') datestring = utils.filename_to_datestring(filename) self.assertIn(datestring, filename)
def test_media(self, *args): extension = 'foo' generated_name = utils.filename_generate(extension, content_type='media') self.assertTrue(generated_name.startswith(settings.HOSTNAME)) self.assertTrue(generated_name.endswith(extension))
def test_func(self, *args): extension = "foo" generated_name = utils.filename_generate(extension) self.assertTrue("--" not in generated_name) self.assertFalse(generated_name.startswith("-"))
def test_generated_filename(self): filename = utils.filename_generate("bak", "default") datestring = utils.filename_to_datestring(filename) self.assertIn(datestring, filename)
def test_db(self, *args): extension = 'foo' generated_name = utils.filename_generate(extension) self.assertTrue(generated_name.startswith(settings.HOSTNAME)) self.assertTrue(generated_name.endswith(extension))
def test_media(self, *args): extension = 'foo' generated_name = utils.filename_generate(extension, content_type='media') self.assertTrue(generated_name.startswith(dbbackup_settings.HOSTNAME)) self.assertTrue(generated_name.endswith(extension))
def test_generated_filename(self): filename = utils.filename_generate('bak', 'default') datestring = utils.filename_to_datestring(filename) self.assertIn(datestring, filename)
def test_func(self, *args): extension = 'foo' generated_name = utils.filename_generate(extension) self.assertTrue('--' not in generated_name) self.assertFalse(generated_name.startswith('-'))