當前位置: 首頁>>代碼示例>>Python>>正文

Python dbrestore.Command類代碼示例

本文整理匯總了Python中dbbackup.management.commands.dbrestore.Command類的典型用法代碼示例。如果您正苦於以下問題：Python Command類的具體用法？Python Command怎麽用？Python Command使用的例子？那麽, 這裏精選的類代碼示例或許可以為您提供幫助。


在下文中一共展示了Command類的15個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Python代碼示例。

示例1: DbMongoRestoreCommandRestoreBackupTest

class DbMongoRestoreCommandRestoreBackupTest(TestCase):
    def setUp(self):
        self.command = DbrestoreCommand()
        self.command.stdout = DEV_NULL
        self.command.uncompress = False
        self.command.decrypt = False
        self.command.backup_extension = 'bak'
        self.command.path = None
        self.command.filename = 'foofile'
        self.command.database = TEST_MONGODB
        self.command.passphrase = None
        self.command.interactive = True
        self.command.storage = get_storage()
        self.command.connector = MongoDumpConnector()
        self.command.database_name = 'mongo'
        self.command.servername = HOSTNAME
        HANDLED_FILES.clean()
        add_private_gpg()

    def test_mongo_settings_backup_command(self, mock_runcommands, *args):
        self.command.storage.file_read = TARED_FILE
        self.command.filename = TARED_FILE
        HANDLED_FILES['written_files'].append((TARED_FILE, open(TARED_FILE, 'rb')))
        self.command._restore_backup()
        self.assertTrue(mock_runcommands.called)
開發者ID:ZuluPro，項目名稱:django-dbbackup，代碼行數:25，代碼來源:test_dbrestore.py

示例2: DbrestoreCommandRestoreBackupTest

class DbrestoreCommandRestoreBackupTest(TestCase):
    def setUp(self):
        self.command = DbrestoreCommand()
        self.command.stdout = DEV_NULL
        self.command.uncompress = False
        self.command.decrypt = False
        self.command.backup_extension = 'bak'
        self.command.filename = 'foofile'
        self.command.database = TEST_DATABASE
        self.command.dbcommands = DBCommands(TEST_DATABASE)
        self.command.passphrase = None
        self.command.interactive = True
        self.command.storage = FakeStorage()
        HANDLED_FILES.clean()
        add_private_gpg()

    def tearDown(self):
        clean_gpg_keys()

    def test_no_filename(self, *args):
        # Prepare backup
        HANDLED_FILES['written_files'].append(
            (utils.filename_generate('foo'), BytesIO(b'bar')))
        # Check
        self.command.path = None
        self.command.filename = None
        self.command._restore_backup()

    def test_no_backup_found(self, *args):
        self.command.path = None
        self.command.filename = None
        with self.assertRaises(CommandError):
            self.command._restore_backup()

    def test_uncompress(self, *args):
        self.command.storage.file_read = COMPRESSED_FILE
        self.command.path = None
        self.command.filename = COMPRESSED_FILE
        HANDLED_FILES['written_files'].append((COMPRESSED_FILE, open(COMPRESSED_FILE, 'rb')))
        self.command.uncompress = True
        self.command._restore_backup()

    @patch('dbbackup.utils.getpass', return_value=None)
    def test_decrypt(self, *args):
        self.command.path = None
        self.command.decrypt = True
        self.command.filename = ENCRYPTED_FILE
        HANDLED_FILES['written_files'].append((ENCRYPTED_FILE, open(ENCRYPTED_FILE, 'rb')))
        self.command._restore_backup()

    def test_path(self, *args):
        self.command.path = COMPRESSED_FILE
        self.command._restore_backup()
開發者ID:CalPolyResDev，項目名稱:django-dbbackup，代碼行數:53，代碼來源:test_dbrestore.py

示例3: DbrestoreCommandGetExtensionTest

class DbrestoreCommandGetExtensionTest(TestCase):
    def setUp(self):
        self.command = DbrestoreCommand()

    def test_tar(self):
        ext = self.command.get_extension("foo.tar")
        self.assertEqual(ext, ".tar")

    def test_tar_gz(self):
        ext = self.command.get_extension("foo.tar.gz")
        self.assertEqual(ext, ".gz")

    def test_no_extension(self):
        ext = self.command.get_extension("foo")
        self.assertEqual(ext, "")
開發者ID:RoPP，項目名稱:django-dbbackup，代碼行數:15，代碼來源:test_dbrestore.py

示例4: DbrestoreCommandGetExtensionTest

class DbrestoreCommandGetExtensionTest(TestCase):
    def setUp(self):
        self.command = DbrestoreCommand()

    def test_tar(self):
        ext = self.command.get_extension('foo.tar')
        self.assertEqual(ext, '.tar')

    def test_tar_gz(self):
        ext = self.command.get_extension('foo.tar.gz')
        self.assertEqual(ext, '.gz')

    def test_no_extension(self):
        ext = self.command.get_extension('foo')
        self.assertEqual(ext, '')
開發者ID:gardenia，項目名稱:django-dbbackup，代碼行數:15，代碼來源:test_dbrestore.py

示例5: DbrestoreCommandGetDatabaseTest

class DbrestoreCommandGetDatabaseTest(TestCase):
    def setUp(self):
        self.command = DbrestoreCommand()

    def test_give_db_name(self):
        db = self.command._get_database({"database": "default"})
        self.assertEqual(db, settings.DATABASES["default"])

    def test_no_given_db(self):
        db = self.command._get_database({})
        self.assertEqual(db, settings.DATABASES["default"])

    @patch("django.conf.settings.DATABASES", {"db1": {}, "db2": {}})
    def test_no_given_db_multidb(self):
        with self.assertRaises(CommandError):
            self.command._get_database({})
開發者ID:RoPP，項目名稱:django-dbbackup，代碼行數:16，代碼來源:test_dbrestore.py

示例6: DbrestoreCommandUncompressTest

class DbrestoreCommandUncompressTest(TestCase):
    def setUp(self):
        self.command = DbrestoreCommand()

    def test_uncompress(self):
        inputfile = open(COMPRESSED_FILE, 'rb')
        fd = self.command.uncompress_file(inputfile)
        fd.seek(0)
        self.assertEqual(fd.read(), b'foo\n')
開發者ID:balavec，項目名稱:django-dbbackup，代碼行數:9，代碼來源:test_restore.py

示例7: setUp

 def setUp(self):
     self.command = DbrestoreCommand()
     self.command.stdout = DEV_NULL
     self.command.uncompress = False
     self.command.decrypt = False
     self.command.backup_extension = 'bak'
     self.command.filepath = 'foofile'
     self.command.database = TEST_DATABASE
     self.command.dbcommands = DBCommands(TEST_DATABASE)
     self.command.passphrase = None
     self.command.storage = FakeStorage()
開發者ID:balavec，項目名稱:django-dbbackup，代碼行數:11，代碼來源:test_restore.py

示例8: DbbackupReadLocalFileTest

class DbbackupReadLocalFileTest(TestCase):
    def setUp(self):
        self.command = DbrestoreCommand()
        self.command.path = '/tmp/foo.bak'

    def test_read(self):
        # setUp
        open(self.command.path, 'w').close()
        # Test
        output_file = self.command.read_local_file(self.command.path)
        # tearDown
        os.remove(self.command.path)
開發者ID:Fisiu，項目名稱:django-dbbackup，代碼行數:12，代碼來源:test_dbrestore.py

示例9: setUp

 def setUp(self):
     self.command = DbrestoreCommand()
     self.command.stdout = DEV_NULL
     self.command.uncompress = False
     self.command.decrypt = False
     self.command.backup_extension = 'bak'
     self.command.filename = 'foofile'
     self.command.database = TEST_DATABASE
     self.command.passphrase = None
     self.command.interactive = True
     self.command.storage = FakeStorage()
     self.command.connector = get_connector()
     HANDLED_FILES.clean()
開發者ID:Fisiu，項目名稱:django-dbbackup，代碼行數:13，代碼來源:test_dbrestore.py

示例10: DbMongoRestoreCommandRestoreBackupTest

class DbMongoRestoreCommandRestoreBackupTest(TestCase):
    def setUp(self):
        self.command = DbrestoreCommand()
        self.command.stdout = DEV_NULL
        self.command.uncompress = False
        self.command.decrypt = False
        self.command.backup_extension = "bak"
        self.command.filepath = "foofile"
        self.command.database = TEST_MONGODB
        self.command.dbcommands = MongoDBCommands(TEST_MONGODB)
        self.command.passphrase = None
        self.command.interactive = True
        self.command.storage = FakeStorage()
        HANDLED_FILES.clean()
        add_private_gpg()

    def test_mongo_settings_backup_command(self, mock_runcommands, *args):
        self.command.storage.file_read = TARED_FILE
        self.command.filepath = TARED_FILE
        HANDLED_FILES["written_files"].append((TARED_FILE, open(TARED_FILE, "rb")))
        self.command.restore_backup()
        self.assertTrue(mock_runcommands.called)
開發者ID:RoPP，項目名稱:django-dbbackup，代碼行數:22，代碼來源:test_dbrestore.py

示例11: setUp

 def setUp(self):
     self.command = DbrestoreCommand()
     self.command.stdout = DEV_NULL
     self.command.uncompress = False
     self.command.decrypt = False
     self.command.backup_extension = 'bak'
     self.command.filepath = 'foofile'
     self.command.database = TEST_DATABASE
     self.command.dbcommands = DBCommands(TEST_DATABASE)
     self.command.passphrase = None
     self.command.storage = FakeStorage()
     cmd = ('gpg --import %s' % GPG_PRIVATE_PATH).split()
     subprocess.call(cmd, stdout=DEV_NULL, stderr=DEV_NULL)
開發者ID:goodtune，項目名稱:django-dbbackup，代碼行數:13，代碼來源:test_dbrestore.py

示例12: setUp

 def setUp(self):
     self.command = DbrestoreCommand()
     self.command.stdout = DEV_NULL
     self.command.uncompress = False
     self.command.decrypt = False
     self.command.backup_extension = "bak"
     self.command.filepath = "foofile"
     self.command.database = TEST_MONGODB
     self.command.dbcommands = MongoDBCommands(TEST_MONGODB)
     self.command.passphrase = None
     self.command.interactive = True
     self.command.storage = FakeStorage()
     HANDLED_FILES.clean()
     add_private_gpg()
開發者ID:RoPP，項目名稱:django-dbbackup，代碼行數:14，代碼來源:test_dbrestore.py

示例13: DbrestoreCommandRestoreBackupTest

class DbrestoreCommandRestoreBackupTest(TestCase):
    def setUp(self):
        self.command = DbrestoreCommand()
        self.command.stdout = DEV_NULL
        self.command.uncompress = False
        self.command.decrypt = False
        self.command.backup_extension = 'bak'
        self.command.filepath = 'foofile'
        self.command.database = TEST_DATABASE
        self.command.dbcommands = DBCommands(TEST_DATABASE)
        self.command.passphrase = None
        self.command.storage = FakeStorage()
        HANDLED_FILES.clean()
        cmd = ('gpg --import %s' % GPG_PRIVATE_PATH).split()
        subprocess.call(cmd, stdout=DEV_NULL, stderr=DEV_NULL)

    def tearDown(self):
        clean_gpg_keys()

    def test_no_filepath(self, *args):
        # Create backup
        HANDLED_FILES['written_files'].append((':memory:.bak', BytesIO(b'bar')))
        # Check
        self.command.filepath = None
        self.command.restore_backup()

    def test_no_backup_found(self, *args):
        self.command.filepath = None
        with self.assertRaises(CommandError):
            self.command.restore_backup()

    def test_uncompress(self, *args):
        self.command.storage.file_read = COMPRESSED_FILE
        self.command.filepath = COMPRESSED_FILE
        HANDLED_FILES['written_files'].append((COMPRESSED_FILE, open(COMPRESSED_FILE, 'rb')))
        self.command.uncompress = True
        self.command.restore_backup()

    @patch('dbbackup.utils.getpass', return_value=None)
    def test_decrypt(self, *args):
        if six.PY3:
            self.skipTest("Decryption isn't implemented in Python3")
        self.command.decrypt = True
        self.command.filepath = ENCRYPTED_FILE
        HANDLED_FILES['written_files'].append((ENCRYPTED_FILE, open(ENCRYPTED_FILE)))
        self.command.restore_backup()
開發者ID:gardenia，項目名稱:django-dbbackup，代碼行數:46，代碼來源:test_dbrestore.py

示例14: DbrestoreCommandDecryptTest

class DbrestoreCommandDecryptTest(TestCase):
    def setUp(self):
        self.command = DbrestoreCommand()
        self.command.passphrase = None
        cmd = ('gpg --import %s' % GPG_PRIVATE_PATH).split()
        subprocess.call(cmd, stdout=DEV_NULL, stderr=DEV_NULL)

    @patch('dbbackup.management.commands.dbrestore.input', return_value=None)
    @patch('dbbackup.management.commands.dbrestore.getpass', return_value=None)
    def test_decrypt(self, *args):
        if six.PY3:
            self.skipTest("Decryption isn't implemented in Python3")
        inputfile = open(ENCRYPTED_FILE, 'r+b')
        uncryptfile = self.command.unencrypt_file(inputfile)
        uncryptfile.seek(0)
        self.assertEqual('foo\n', uncryptfile.read())
開發者ID:balavec，項目名稱:django-dbbackup，代碼行數:16，代碼來源:test_restore.py

示例15: DbrestoreCommandRestoreBackupTest

class DbrestoreCommandRestoreBackupTest(TestCase):
    def setUp(self):
        self.command = DbrestoreCommand()
        self.command.stdout = DEV_NULL
        self.command.uncompress = False
        self.command.decrypt = False
        self.command.backup_extension = 'bak'
        self.command.filepath = 'foofile'
        self.command.database = TEST_DATABASE
        self.command.dbcommands = DBCommands(TEST_DATABASE)
        self.command.passphrase = None
        self.command.storage = FakeStorage()
        cmd = ('gpg --import %s' % GPG_PRIVATE_PATH).split()
        subprocess.call(cmd, stdout=DEV_NULL, stderr=DEV_NULL)

    def tearDown(self):
        clean_gpg_keys()

    def test_no_filepath(self, *args):
        self.command.storage.list_files = ['foo.bak']
        self.command.filepath = None
        self.command.restore_backup()

    def test_no_backup_found(self, *args):
        self.command.filepath = None
        with self.assertRaises(CommandError):
            self.command.restore_backup()

    def test_uncompress(self, *args):
        self.command.storage.file_read = COMPRESSED_FILE
        self.command.filepath = COMPRESSED_FILE
        self.command.uncompress = True
        self.command.restore_backup()

    @patch('dbbackup.management.commands.dbrestore.getpass', return_value=None)
    def test_decrypt(self, *args):
        if six.PY3:
            self.skipTest("Decryption isn't implemented in Python3")
        self.command.decrypt = True
        self.command.filepath = ENCRYPTED_FILE
        self.command.restore_backup()
開發者ID:goodtune，項目名稱:django-dbbackup，代碼行數:41，代碼來源:test_dbrestore.py


注：本文中的dbbackup.management.commands.dbrestore.Command類示例由純淨天空整理自Github/MSDocs等開源代碼及文檔管理平台，相關代碼片段篩選自各路編程大神貢獻的開源項目，源碼版權歸原作者所有，傳播和使用請參考對應項目的License；未經允許，請勿轉載。