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)
     ]
示例#2
0
 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)
     ]
示例#3
0
 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)]
示例#5
0
 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)
     ]
示例#7
0
 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()
示例#9
0
 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()
示例#10
0
 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()
示例#11
0
 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)
示例#12
0
 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)
示例#13
0
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
示例#14
0
 def test_generated_filename(self):
     filename = utils.filename_generate('bak', 'default')
     datestring = utils.filename_to_date(filename)
示例#15
0
 def generate_filename(self, server_name=None):
     return utils.filename_generate(self.extension, self.database_name,
                                    server_name)
示例#16
0
 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()
示例#17
0
 def test_generated_filename(self):
     filename = utils.filename_generate("bak", "default")
     datestring = utils.filename_to_date(filename)
示例#18
0
 def generate_filename(self, server_name=None):
     return utils.filename_generate(self.extension, self.settings['NAME'],
                                    server_name)
示例#19
0
 def test_template_is_callable(self, *args):
     extension = 'foo'
     generated_name = utils.filename_generate(extension)
     self.assertTrue(generated_name.endswith('foo'))
示例#20
0
 def test_tz_true(self):
     filename = utils.filename_generate('bak', 'default')
     datestring = utils.filename_to_datestring(filename)
     self.assertIn(datestring, filename)
示例#21
0
 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))
示例#22
0
 def test_func(self, *args):
     extension = "foo"
     generated_name = utils.filename_generate(extension)
     self.assertTrue("--" not in generated_name)
     self.assertFalse(generated_name.startswith("-"))
示例#23
0
 def test_generated_filename(self):
     filename = utils.filename_generate("bak", "default")
     datestring = utils.filename_to_datestring(filename)
     self.assertIn(datestring, filename)
示例#24
0
 def generate_filename(self, server_name=None):
     return utils.filename_generate(self.extension, self.database_name,
                                    server_name)
示例#25
0
 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))
示例#26
0
 def generate_filename(self, server_name=None):
     return utils.filename_generate(self.extension, self.settings['NAME'],
                                    server_name)
示例#27
0
 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))
示例#28
0
 def test_generated_filename(self):
     filename = utils.filename_generate('bak', 'default')
     datestring = utils.filename_to_datestring(filename)
     self.assertIn(datestring, filename)
示例#29
0
 def test_generated_filename(self):
     filename = utils.filename_generate('bak', 'default')
     datestring = utils.filename_to_date(filename)
示例#30
0
 def test_func(self, *args):
     extension = 'foo'
     generated_name = utils.filename_generate(extension)
     self.assertTrue('--' not in generated_name)
     self.assertFalse(generated_name.startswith('-'))
示例#31
0
 def test_func(self, *args):
     extension = 'foo'
     generated_name = utils.filename_generate(extension)
     self.assertTrue('--' not in generated_name)
     self.assertFalse(generated_name.startswith('-'))
示例#32
0
 def test_template_is_callable(self, *args):
     extension = 'foo'
     generated_name = utils.filename_generate(extension)
     self.assertTrue(generated_name.endswith('foo'))
示例#33
0
 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))