Source code for freezer.tests.unit.test_job
"""Freezer main.py related tests
(c) Copyright 2014,2015 Hewlett-Packard Development Company, L.P.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
"""
import mock
from mock import patch
from freezer import job as jobs
from freezer.tests import commons
[docs]class TestJob(commons.FreezerBaseTestCase):
[docs] def setUp(self):
super(TestJob, self).setUp()
[docs] def test_execute(self):
opt = commons.BackupOpt1()
job = jobs.InfoJob(opt, opt.storage)
assert job.execute() is None
[docs]class TestInfoJob(TestJob):
[docs] def setUp(self):
super(TestInfoJob, self).setUp()
[docs] def test_execute_nothing_to_do(self):
backup_opt = commons.BackupOpt1()
job = jobs.InfoJob(backup_opt, backup_opt.storage)
job.execute()
[docs] def test_execute_list_containers(self):
backup_opt = commons.BackupOpt1()
job = jobs.InfoJob(backup_opt, backup_opt.storage)
job.execute()
[docs]class TestBackupJob(TestJob):
[docs] def setUp(self):
super(TestBackupJob, self).setUp()
[docs] def test_execute_backup_fs_no_incremental_and_backup_level_raise(self):
backup_opt = commons.BackupOpt1()
backup_opt.mode = 'default'
backup_opt.no_incremental = True
backup_opt.max_level = None
backup_opt.always_level = None
job = jobs.BackupJob(backup_opt, backup_opt.storage)
self.assertRaises(Exception, job.execute) # noqa
[docs] def test_execute_raise(self):
backup_opt = commons.BackupOpt1()
backup_opt.no_incremental = False
backup_opt.mode = None
job = jobs.BackupJob(backup_opt, backup_opt.storage)
self.assertRaises(ValueError, job.execute) # noqa
[docs]class TestAdminJob(TestJob):
[docs] def setUp(self):
super(TestAdminJob, self).setUp()
[docs] def test_execute(self):
backup_opt = commons.BackupOpt1()
jobs.AdminJob(backup_opt, backup_opt.storage).execute()
[docs]class TestExecJob(TestJob):
[docs] def setUp(self):
super(TestExecJob, self).setUp()
# init mock_popen
self.popen = patch('freezer.utils.exec_cmd.subprocess.Popen')
self.mock_popen = self.popen.start()
self.mock_popen.return_value = mock.Mock()
self.mock_popen.return_value.communicate = mock.Mock()
self.mock_popen.return_value.communicate.return_value = ['some stderr']
[docs] def tearDown(self):
super(TestExecJob, self).tearDown()
self.popen.stop()
[docs] def test_execute_nothing_to_do(self):
self.mock_popen.return_value.returncode = 0
backup_opt = commons.BackupOpt1()
backup_opt.command = 'ls '
jobs.ExecJob(backup_opt, backup_opt.storage).execute()
[docs] def test_execute_script(self):
self.mock_popen.return_value.returncode = 0
backup_opt = commons.BackupOpt1()
backup_opt.command = 'echo test'
jobs.ExecJob(backup_opt, backup_opt.storage).execute()
[docs] def test_execute_raise(self):
self.popen = patch('freezer.utils.exec_cmd.subprocess.Popen')
self.mock_popen = self.popen.start()
self.mock_popen.return_value.returncode = 1
backup_opt = commons.BackupOpt1()
backup_opt.command = 'echo test'
job = jobs.ExecJob(backup_opt, backup_opt.storage)
self.assertRaises(Exception, job.execute) # noqa