1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
"""
Tests for mediabackup command.
"""
import contextlib
import os
import tempfile
from django.test import TestCase
from dbbackup.management.commands.mediabackup import Command as DbbackupCommand
from dbbackup.storage import get_storage, get_storage_class
from dbbackup.tests.utils import DEV_NULL, HANDLED_FILES, add_public_gpg
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()()
self.command.filename = None
def tearDown(self):
if self.command.path is not None:
with contextlib.suppress(OSError):
os.remove(self.command.path)
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"]))
def test_output_filename(self):
self.command.filename = "my_new_name.tar"
self.command.backup_mediafiles()
self.assertEqual(HANDLED_FILES["written_files"][0][0], self.command.filename)
|
