177 lines
7.7 KiB
Python
177 lines
7.7 KiB
Python
# 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.
|
|
|
|
from datetime import datetime
|
|
import mock
|
|
|
|
from oslo_utils import timeutils
|
|
|
|
from karbor.services.protection.bank_plugin import Bank
|
|
from karbor.services.protection.checkpoint import CheckpointCollection
|
|
from karbor.tests import base
|
|
from karbor.tests.unit.protection.fakes import fake_protection_plan
|
|
from karbor.tests.unit.protection.test_bank import _InMemoryBankPlugin
|
|
from karbor.tests.unit.protection.test_bank import _InMemoryLeasePlugin
|
|
|
|
|
|
class CheckpointCollectionTest(base.TestCase):
|
|
def _create_test_collection(self):
|
|
return CheckpointCollection(Bank(_InMemoryBankPlugin()),
|
|
_InMemoryLeasePlugin())
|
|
|
|
def test_create_checkpoint(self):
|
|
collection = self._create_test_collection()
|
|
checkpoint = collection.create(fake_protection_plan())
|
|
checkpoint.status = "finished"
|
|
checkpoint.commit()
|
|
self.assertEqual(
|
|
checkpoint.status,
|
|
collection.get(checkpoint_id=checkpoint.id).status,
|
|
)
|
|
|
|
def test_list_checkpoints(self):
|
|
collection = self._create_test_collection()
|
|
plan = fake_protection_plan()
|
|
provider_id = plan['provider_id']
|
|
project_id = plan['project_id']
|
|
result = {collection.create(plan).id for i in range(10)}
|
|
self.assertEqual(set(collection.list_ids(
|
|
project_id=project_id, provider_id=provider_id)), result)
|
|
|
|
def test_list_checkpoints_by_plan_id(self):
|
|
collection = self._create_test_collection()
|
|
plan_1 = fake_protection_plan()
|
|
plan_1["id"] = "fake_plan_id_1"
|
|
plan_1['provider_id'] = "fake_provider_id_1"
|
|
plan_1["project_id"] = "fake_project_id_1"
|
|
provider_id_1 = plan_1['provider_id']
|
|
checkpoints_plan_1 = {collection.create(plan_1).id for i in range(10)}
|
|
|
|
plan_2 = fake_protection_plan()
|
|
plan_2["id"] = "fake_plan_id_2"
|
|
plan_2['provider_id'] = "fake_provider_id_2"
|
|
plan_2["project_id"] = "fake_project_id_2"
|
|
provider_id_2 = plan_1['provider_id']
|
|
checkpoints_plan_2 = {collection.create(plan_2).id for i in range(10)}
|
|
self.assertEqual(set(collection.list_ids(
|
|
project_id="fake_project_id_1", provider_id=provider_id_1,
|
|
plan_id="fake_plan_id_1")), checkpoints_plan_1)
|
|
self.assertEqual(set(collection.list_ids(
|
|
project_id="fake_project_id_2", provider_id=provider_id_2,
|
|
plan_id="fake_plan_id_2")), checkpoints_plan_2)
|
|
|
|
def test_list_checkpoints_by_plan_id_and_filter_by_start_date(self):
|
|
collection = self._create_test_collection()
|
|
date1 = datetime.strptime("2018-11-12", "%Y-%m-%d")
|
|
date2 = datetime.strptime("2018-11-13", "%Y-%m-%d")
|
|
timeutils.utcnow = mock.MagicMock()
|
|
timeutils.utcnow.return_value = date1
|
|
plan = fake_protection_plan()
|
|
plan["id"] = "fake_plan_id"
|
|
plan['provider_id'] = "fake_provider_id"
|
|
plan["project_id"] = "fake_project_id"
|
|
provider_id = plan['provider_id']
|
|
checkpoints_plan_date1 = {
|
|
collection.create(plan).id for i in range(10)}
|
|
timeutils.utcnow = mock.MagicMock()
|
|
timeutils.utcnow.return_value = date2
|
|
checkpoints_plan_date2 = {
|
|
collection.create(plan).id for i in range(10)}
|
|
self.assertEqual(set(collection.list_ids(
|
|
project_id="fake_project_id", provider_id=provider_id,
|
|
plan_id="fake_plan_id", start_date=date1, end_date=date1)),
|
|
checkpoints_plan_date1)
|
|
self.assertEqual(set(collection.list_ids(
|
|
project_id="fake_project_id", provider_id=provider_id,
|
|
plan_id="fake_plan_id", start_date=date2)),
|
|
checkpoints_plan_date2)
|
|
|
|
def test_list_checkpoints_by_plan_with_marker(self):
|
|
collection = self._create_test_collection()
|
|
plan = fake_protection_plan()
|
|
plan["id"] = "fake_plan_id"
|
|
plan['provider_id'] = "fake_provider_id"
|
|
plan["project_id"] = "fake_project_id"
|
|
provider_id = plan['provider_id']
|
|
checkpoints_plan = {collection.create(plan, {
|
|
'checkpoint_id': i}).id for i in range(10)}
|
|
checkpoints_sorted = sorted(checkpoints_plan)
|
|
self.assertEqual(len(collection.list_ids(
|
|
project_id="fake_project_id", provider_id=provider_id,
|
|
plan_id="fake_plan_id", marker=checkpoints_sorted[0])) < 10, True)
|
|
|
|
def test_list_checkpoints_by_date(self):
|
|
collection = self._create_test_collection()
|
|
date1 = datetime.strptime("2016-06-12", "%Y-%m-%d")
|
|
timeutils.utcnow = mock.MagicMock()
|
|
timeutils.utcnow.return_value = date1
|
|
plan = fake_protection_plan()
|
|
provider_id = plan['provider_id']
|
|
project_id = plan['project_id']
|
|
checkpoints_date_1 = {collection.create(plan).id for i in range(10)}
|
|
date2 = datetime.strptime("2016-06-13", "%Y-%m-%d")
|
|
timeutils.utcnow = mock.MagicMock()
|
|
timeutils.utcnow.return_value = date2
|
|
checkpoints_date_2 = {collection.create(plan).id for i in range(10)}
|
|
self.assertEqual(set(collection.list_ids(
|
|
project_id=project_id,
|
|
provider_id=provider_id,
|
|
start_date=date1,
|
|
end_date=date1)),
|
|
checkpoints_date_1)
|
|
self.assertEqual(set(collection.list_ids(
|
|
project_id=project_id,
|
|
provider_id=provider_id,
|
|
start_date=date2,
|
|
end_date=date2)),
|
|
checkpoints_date_2)
|
|
|
|
def test_list_checkpoints_by_date_with_marker(self):
|
|
collection = self._create_test_collection()
|
|
date = datetime.strptime("2018-11-12", "%Y-%m-%d")
|
|
timeutils.utcnow = mock.MagicMock()
|
|
timeutils.utcnow.return_value = date
|
|
plan = fake_protection_plan()
|
|
plan["id"] = "fake_plan_id"
|
|
plan['provider_id'] = "fake_provider_id"
|
|
plan["project_id"] = "fake_project_id"
|
|
provider_id = plan['provider_id']
|
|
checkpoints_plan = {collection.create(plan, {
|
|
'checkpoint_id': i}).id for i in range(10)}
|
|
checkpoints_sorted = sorted(checkpoints_plan)
|
|
self.assertEqual(len(collection.list_ids(
|
|
project_id="fake_project_id", provider_id=provider_id,
|
|
start_date=date,
|
|
marker=checkpoints_sorted[0])) < 10, True)
|
|
|
|
def test_delete_checkpoint(self):
|
|
collection = self._create_test_collection()
|
|
plan = fake_protection_plan()
|
|
provider_id = plan['provider_id']
|
|
project_id = plan['project_id']
|
|
result = {collection.create(plan).id for i in range(10)}
|
|
checkpoint = collection.get(result.pop())
|
|
checkpoint.purge()
|
|
self.assertEqual(set(collection.list_ids(
|
|
project_id=project_id, provider_id=provider_id)), result)
|
|
|
|
def test_write_checkpoint_with_invalid_lease(self):
|
|
collection = self._create_test_collection()
|
|
checkpoint = collection.create(fake_protection_plan())
|
|
collection._bank_lease.check_lease_validity = mock.MagicMock()
|
|
collection._bank_lease.check_lease_validity.return_value = False
|
|
checkpoint.status = "finished"
|
|
self.assertNotEqual(
|
|
checkpoint.status,
|
|
collection.get(checkpoint_id=checkpoint.id).status,
|
|
)
|