class BaseDbBackupCommandSetLoggerLevelTest(TestCase):
def setUp(self):
self.command = BaseDbBackupCommand()
def test_0_level(self):
self.command.verbosity = 0
self.command._set_logger_level()
self.assertEqual(self.command.logger.level, logging.WARNING)
def test_1_level(self):
self.command.verbosity = 1
self.command._set_logger_level()
self.assertEqual(self.command.logger.level, logging.INFO)
def test_2_level(self):
self.command.verbosity = 2
self.command._set_logger_level()
self.assertEqual(self.command.logger.level, logging.DEBUG)
def test_3_level(self):
self.command.verbosity = 3
self.command._set_logger_level()
self.assertEqual(self.command.logger.level, logging.DEBUG)
def test_quiet(self):
self.command.quiet = True
self.command._set_logger_level()
self.assertGreater(self.command.logger.level, logging.ERROR)
class BaseDbBackupCommandCleanupOldBackupsTest(TestCase):
def setUp(self):
HANDLED_FILES.clean()
self.command = BaseDbBackupCommand()
self.command.stdout = DEV_NULL
self.command.encrypt = False
self.command.compress = False
self.command.servername = 'foo-server'
self.command.storage = get_storage()
HANDLED_FILES['written_files'] = [(f, None) for f in [
'2015-02-06-042810.tar',
'2015-02-07-042810.tar',
'2015-02-08-042810.tar',
'foodb-2015-02-06-042810.dump',
'foodb-2015-02-07-042810.dump',
'foodb-2015-02-08-042810.dump',
]]
@patch('dbbackup.settings.CLEANUP_KEEP', 1)
def test_clean_db(self):
self.command.content_type = 'db'
self.command.database = 'foodb'
self.command._cleanup_old_backups()
self.assertEqual(2, len(HANDLED_FILES['deleted_files']))
self.assertNotIn('foodb-2015-02-08-042810.dump',
HANDLED_FILES['deleted_files'])
@patch('dbbackup.settings.CLEANUP_KEEP_MEDIA', 1)
def test_clean_media(self):
self.command.content_type = 'media'
self.command._cleanup_old_backups()
self.assertEqual(2, len(HANDLED_FILES['deleted_files']))
self.assertNotIn('2015-02-08-042810.tar',
HANDLED_FILES['deleted_files'])
class BaseDbBackupCommandCleanupOldBackupsTest(TestCase):
def setUp(self):
HANDLED_FILES.clean()
self.command = BaseDbBackupCommand()
self.command.stdout = DEV_NULL
self.command.encrypt = False
self.command.compress = False
self.command.servername = 'foo-server'
self.command.storage = get_storage()
HANDLED_FILES['written_files'] = [(f, None) for f in [
'2015-02-06-042810.tar',
'2015-02-07-042810.tar',
'2015-02-08-042810.tar',
'foodb-2015-02-06-042810.dump',
'foodb-2015-02-07-042810.dump',
'foodb-2015-02-08-042810.dump',
]]
@patch('dbbackup.settings.CLEANUP_KEEP', 1)
def test_clean_db(self):
self.command.content_type = 'db'
self.command.database = 'foodb'
self.command._cleanup_old_backups()
self.assertEqual(2, len(HANDLED_FILES['deleted_files']))
self.assertNotIn('foodb-2015-02-08-042810.dump', HANDLED_FILES['deleted_files'])
@patch('dbbackup.settings.CLEANUP_KEEP_MEDIA', 1)
def test_clean_media(self):
self.command.content_type = 'media'
self.command._cleanup_old_backups()
self.assertEqual(2, len(HANDLED_FILES['deleted_files']))
self.assertNotIn('2015-02-08-042810.tar', HANDLED_FILES['deleted_files'])
class BaseDbBackupCommandSetLoggerLevelTest(TestCase):
def setUp(self):
self.command = BaseDbBackupCommand()
def test_0_level(self):
self.command.verbosity = 0
self.command._set_logger_level()
self.assertEqual(self.command.logger.level, logging.WARNING)
def test_1_level(self):
self.command.verbosity = 1
self.command._set_logger_level()
self.assertEqual(self.command.logger.level, logging.INFO)
def test_2_level(self):
self.command.verbosity = 2
self.command._set_logger_level()
self.assertEqual(self.command.logger.level, logging.DEBUG)
def test_3_level(self):
self.command.verbosity = 3
self.command._set_logger_level()
self.assertEqual(self.command.logger.level, logging.DEBUG)
def test_quiet(self):
self.command.quiet = True
self.command._set_logger_level()
self.assertGreater(self.command.logger.level, logging.ERROR)
def setUp(self):
HANDLED_FILES.clean()
self.command = BaseDbBackupCommand()
self.command.stdout = DEV_NULL
self.command.encrypt = False
self.command.compress = False
self.command.servername = 'foo-server'
self.command.storage = get_storage()
HANDLED_FILES['written_files'] = [(f, None) for f in [
'2015-02-06-042810.tar',
'2015-02-07-042810.tar',
'2015-02-08-042810.tar',
'foodb-2015-02-06-042810.dump',
'foodb-2015-02-07-042810.dump',
'foodb-2015-02-08-042810.dump',
]]
def setUp(self):
HANDLED_FILES.clean()
self.command = BaseDbBackupCommand()
self.command.stdout = DEV_NULL
self.command.encrypt = False
self.command.compress = False
self.command.servername = 'foo-server'
self.command.storage = get_storage()
HANDLED_FILES['written_files'] = [(f, None) for f in [
'2015-02-06-042810.tar',
'2015-02-07-042810.tar',
'2015-02-08-042810.tar',
'foodb-2015-02-06-042810.dump',
'foodb-2015-02-07-042810.dump',
'foodb-2015-02-08-042810.dump',
]]
class BaseDbBackupCommandSetLoggerLevelTest(TestCase):
def setUp(self):
self.command = BaseDbBackupCommand()
def test_debug_level(self):
self.command.verbosity = 0
self.command._set_logger_level()
self.assertEqual(self.command.logger.level, 10)
def test_info_level(self):
self.command.verbosity = 1
self.command._set_logger_level()
self.assertEqual(self.command.logger.level, 20)
def test_quiet(self):
self.command.quiet = True
self.command._set_logger_level()
self.assertGreater(self.command.logger.level, 50)
class BaseDbBackupCommandLogTest(TestCase):
def setUp(self):
self.command = BaseDbBackupCommand()
self.command.stdout = StringIO()
def test_less_level(self):
self.command.verbosity = 1
self.command.log("foo", 2)
self.command.stdout.seek(0)
self.assertFalse(self.command.stdout.read())
def test_more_level(self):
self.command.verbosity = 1
self.command.log("foo", 0)
self.command.stdout.seek(0)
self.assertEqual('foo', self.command.stdout.read())
def test_quiet(self):
self.command.quiet = True
self.command.verbosity = 1
self.command.log("foo", 0)
self.command.stdout.seek(0)
self.assertFalse(self.command.stdout.read())
def setUp(self):
self.command = BaseDbBackupCommand()
self.command.stdout = StringIO()
def setUp(self):
HANDLED_FILES.clean()
self.command = BaseDbBackupCommand()
self.command.storage = get_storage()
class BaseDbBackupCommandMethodsTest(TestCase):
def setUp(self):
HANDLED_FILES.clean()
self.command = BaseDbBackupCommand()
self.command.storage = get_storage()
def test_read_from_storage(self):
HANDLED_FILES['written_files'].append(['foo', File(six.BytesIO(b'bar'))])
file_ = self.command.read_from_storage('foo')
self.assertEqual(file_.read(), b'bar')
def test_write_to_storage(self):
self.command.write_to_storage(six.BytesIO(b'foo'), 'bar')
self.assertEqual(HANDLED_FILES['written_files'][0][0], 'bar')
def test_read_local_file(self):
# setUp
self.command.path = '/tmp/foo.bak'
open(self.command.path, 'w').close()
# Test
output_file = self.command.read_local_file(self.command.path)
# tearDown
os.remove(self.command.path)
def test_write_local_file(self):
fd, path = File(six.BytesIO(b"foo")), '/tmp/foo.bak'
self.command.write_local_file(fd, path)
self.assertTrue(os.path.exists(path))
# tearDown
os.remove(path)
def test_ask_confirmation(self):
# Yes
with patch('dbbackup.management.commands._base.input', return_value='y'):
self.command._ask_confirmation()
with patch('dbbackup.management.commands._base.input', return_value='Y'):
self.command._ask_confirmation()
with patch('dbbackup.management.commands._base.input', return_value=''):
self.command._ask_confirmation()
with patch('dbbackup.management.commands._base.input', return_value='foo'):
self.command._ask_confirmation()
# No
with patch('dbbackup.management.commands._base.input', return_value='n'):
with self.assertRaises(SystemExit):
self.command._ask_confirmation()
with patch('dbbackup.management.commands._base.input', return_value='N'):
with self.assertRaises(SystemExit):
self.command._ask_confirmation()
with patch('dbbackup.management.commands._base.input', return_value='No'):
with self.assertRaises(SystemExit):
self.command._ask_confirmation()
def setUp(self):
self.command = BaseDbBackupCommand()
