class MediabackupBackupMediafilesTest(TestCase): def setUp(self): HANDLED_FILES.clean() self.command = DbbackupCommand() self.command.servername = 'foo-server' self.command.storage = get_storage() self.command.stdout = DEV_NULL self.command.compress = False self.command.encrypt = False self.command.path = None self.command.media_storage = get_storage_class()() def tearDown(self): if self.command.path is not None: try: os.remove(self.command.path) except OSError: pass def test_func(self): self.command.backup_mediafiles() self.assertEqual(1, len(HANDLED_FILES['written_files'])) def test_compress(self): self.command.compress = True self.command.backup_mediafiles() self.assertEqual(1, len(HANDLED_FILES['written_files'])) self.assertTrue(HANDLED_FILES['written_files'][0][0].endswith('.gz')) def test_encrypt(self): self.command.encrypt = True add_public_gpg() self.command.backup_mediafiles() self.assertEqual(1, len(HANDLED_FILES['written_files'])) outputfile = HANDLED_FILES['written_files'][0][1] outputfile.seek(0) self.assertTrue(outputfile.read().startswith(b'-----BEGIN PGP MESSAGE-----')) def test_compress_and_encrypt(self): self.command.compress = True self.command.encrypt = True add_public_gpg() self.command.backup_mediafiles() self.assertEqual(1, len(HANDLED_FILES['written_files'])) outputfile = HANDLED_FILES['written_files'][0][1] outputfile.seek(0) self.assertTrue(outputfile.read().startswith(b'-----BEGIN PGP MESSAGE-----')) def test_write_local_file(self): self.command.path = tempfile.mktemp() self.command.backup_mediafiles() self.assertTrue(os.path.exists(self.command.path)) self.assertEqual(0, len(HANDLED_FILES['written_files']))
class MediabackupBackupMediafilesTest(TestCase): def setUp(self): HANDLED_FILES.clean() self.command = DbbackupCommand() self.command.servername = 'foo-server' self.command.storage = FakeStorage() self.command.stdout = DEV_NULL def test_func(self): self.command.backup_mediafiles(encrypt=False, compress=False) self.assertEqual(1, len(HANDLED_FILES['written_files'])) def test_compress(self): self.command.backup_mediafiles(encrypt=False, compress=True) self.assertEqual(1, len(HANDLED_FILES['written_files'])) self.assertTrue(HANDLED_FILES['written_files'][0][0].endswith('.gz')) def test_encrypt(self): add_public_gpg() self.command.backup_mediafiles(encrypt=True, compress=False) self.assertEqual(1, len(HANDLED_FILES['written_files'])) outputfile = HANDLED_FILES['written_files'][0][1] outputfile.seek(0) self.assertTrue( outputfile.read().startswith(b'-----BEGIN PGP MESSAGE-----')) def test_compress_and_encrypt(self): add_public_gpg() self.command.backup_mediafiles(encrypt=True, compress=True) self.assertEqual(1, len(HANDLED_FILES['written_files'])) outputfile = HANDLED_FILES['written_files'][0][1] outputfile.seek(0) self.assertTrue( outputfile.read().startswith(b'-----BEGIN PGP MESSAGE-----'))
class MediabackupBackupMediafilesTest(TestCase): def setUp(self): HANDLED_FILES.clean() self.command = DbbackupCommand() self.command.servername = 'foo-server' self.command.storage = FakeStorage() self.command.stdout = DEV_NULL def test_func(self): self.command.backup_mediafiles(encrypt=False, compress=False) self.assertEqual(1, len(HANDLED_FILES['written_files'])) def test_compress(self): self.command.backup_mediafiles(encrypt=False, compress=True) self.assertEqual(1, len(HANDLED_FILES['written_files'])) self.assertTrue(HANDLED_FILES['written_files'][0][0].endswith('.gz')) def test_encrypt(self): cmd = ('gpg --import %s' % GPG_PUBLIC_PATH).split() subprocess.call(cmd, stdout=DEV_NULL, stderr=DEV_NULL) self.command.backup_mediafiles(encrypt=True, compress=False) self.assertEqual(1, len(HANDLED_FILES['written_files'])) outputfile = HANDLED_FILES['written_files'][0][1] outputfile.seek(0) self.assertTrue(outputfile.read().startswith(b'-----BEGIN PGP MESSAGE-----')) def test_compress_and_encrypt(self): cmd = ('gpg --import %s' % GPG_PUBLIC_PATH).split() subprocess.call(cmd, stdout=DEV_NULL, stderr=DEV_NULL) self.command.backup_mediafiles(encrypt=True, compress=True) self.assertEqual(1, len(HANDLED_FILES['written_files'])) outputfile = HANDLED_FILES['written_files'][0][1] outputfile.seek(0) self.assertTrue(outputfile.read().startswith(b'-----BEGIN PGP MESSAGE-----'))
class MediabackupBackupMediafilesTest(TestCase): def setUp(self): HANDLED_FILES.clean() self.command = DbbackupCommand() self.command.servername = "foo-server" self.command.storage = FakeStorage() self.command.stdout = DEV_NULL def test_func(self): self.command.backup_mediafiles(encrypt=False, compress=False) self.assertEqual(1, len(HANDLED_FILES["written_files"])) def test_compress(self): self.command.backup_mediafiles(encrypt=False, compress=True) self.assertEqual(1, len(HANDLED_FILES["written_files"])) self.assertTrue(HANDLED_FILES["written_files"][0][0].endswith(".gz")) def test_encrypt(self): cmd = ("gpg --import %s" % GPG_PUBLIC_PATH).split() subprocess.call(cmd, stdout=DEV_NULL, stderr=DEV_NULL) self.command.backup_mediafiles(encrypt=True, compress=False) self.assertEqual(1, len(HANDLED_FILES["written_files"])) outputfile = HANDLED_FILES["written_files"][0][1] outputfile.seek(0) self.assertTrue(outputfile.read().startswith(b"-----BEGIN PGP MESSAGE-----")) def test_compress_and_encrypt(self): cmd = ("gpg --import %s" % GPG_PUBLIC_PATH).split() subprocess.call(cmd, stdout=DEV_NULL, stderr=DEV_NULL) self.command.backup_mediafiles(encrypt=True, compress=True) self.assertEqual(1, len(HANDLED_FILES["written_files"])) outputfile = HANDLED_FILES["written_files"][0][1] outputfile.seek(0) self.assertTrue(outputfile.read().startswith(b"-----BEGIN PGP MESSAGE-----"))
class MediabackupBackupMediafilesTest(TestCase): def setUp(self): HANDLED_FILES.clean() self.command = DbbackupCommand() self.command.servername = 'foo-server' self.command.storage = FakeStorage() self.command.stdout = DEV_NULL def test_func(self): self.command.backup_mediafiles(encrypt=False, compress=False) self.assertEqual(1, len(HANDLED_FILES['written_files'])) def test_compress(self): self.command.backup_mediafiles(encrypt=False, compress=True) self.assertEqual(1, len(HANDLED_FILES['written_files'])) self.assertTrue(HANDLED_FILES['written_files'][0][0].endswith('.gz')) def test_encrypt(self): add_public_gpg() self.command.backup_mediafiles(encrypt=True, compress=False) self.assertEqual(1, len(HANDLED_FILES['written_files'])) outputfile = HANDLED_FILES['written_files'][0][1] outputfile.seek(0) self.assertTrue(outputfile.read().startswith(b'-----BEGIN PGP MESSAGE-----')) def test_compress_and_encrypt(self): add_public_gpg() self.command.backup_mediafiles(encrypt=True, compress=True) self.assertEqual(1, len(HANDLED_FILES['written_files'])) outputfile = HANDLED_FILES['written_files'][0][1] outputfile.seek(0) self.assertTrue(outputfile.read().startswith(b'-----BEGIN PGP MESSAGE-----'))