def test_cached_snapshot_images_gets_plans(self, get_plans):
"""More or less a metatest that verifies that the mocking
is actually working.
"""
project = self.create_project()
plan = self.create_plan(project)
snapshot = self.create_snapshot(project)
snapshot_image = self.create_snapshot_image(snapshot, plan)
cached_snapshot_image = self.create_cached_snapshot_image(snapshot_image)
get_plans.return_value = set()
gc.get_cached_snapshot_images('cluster')
get_plans.assert_called_with('cluster')
def test_cached_snapshot_images_get_current_datetime(self, get_plans, get_current_datetime):
"""More or less a metatest that verifies that the mocking
is actually working.
"""
project = self.create_project()
plan = self.create_plan(project)
snapshot = self.create_snapshot(project)
snapshot_image = self.create_snapshot_image(snapshot, plan)
cached_snapshot_image = self.create_cached_snapshot_image(snapshot_image)
get_plans.return_value = set()
get_current_datetime.return_value = self.mock_datetime
gc.get_cached_snapshot_images('cluster')
get_current_datetime.assert_any_call()
def test_cached_snapshot_images_gets_plans(self, get_plans):
"""More or less a metatest that verifies that the mocking
is actually working.
"""
project = self.create_project()
plan = self.create_plan(project)
snapshot = self.create_snapshot(project)
snapshot_image = self.create_snapshot_image(snapshot, plan)
cached_snapshot_image = self.create_cached_snapshot_image(
snapshot_image)
get_plans.return_value = set()
gc.get_cached_snapshot_images('cluster')
get_plans.assert_called_with('cluster')
def test_cached_snapshot_images_all(self, get_plans, get_current_datetime):
"""
Tests the four fundamental cases for an image:
- No expiration date
- Not yet expired
- Expired
- Not part of the plans associated with the cluster
"""
project = self.create_project()
plan1 = self.create_plan(project)
plan2 = self.create_plan(project)
plan3 = self.create_plan(project)
plan4 = self.create_plan(project)
get_current_datetime.return_value = self.mock_datetime
get_plans.return_value = {plan1, plan2, plan3}
snapshot = self.create_snapshot(project)
snapshot_image1 = self.create_snapshot_image(snapshot, plan1)
snapshot_image2 = self.create_snapshot_image(snapshot, plan2)
snapshot_image3 = self.create_snapshot_image(snapshot, plan3)
snapshot_image4 = self.create_snapshot_image(snapshot, plan4)
self.create_cached_snapshot_image(snapshot_image1, expiration_date=None)
self.create_cached_snapshot_image(snapshot_image2,
expiration_date=self.mock_datetime + datetime.timedelta(0, 1))
self.create_cached_snapshot_image(snapshot_image3,
expiration_date=self.mock_datetime - datetime.timedelta(0, 1))
self.create_cached_snapshot_image(snapshot_image4, expiration_date=None)
cached_snapshot_ids = [s.id for s in gc.get_cached_snapshot_images('cluster')]
assert len(cached_snapshot_ids) == 2
assert snapshot_image1.id in cached_snapshot_ids
assert snapshot_image2.id in cached_snapshot_ids
def test_cached_snapshot_images_get_current_datetime(
self, get_plans, get_current_datetime):
"""More or less a metatest that verifies that the mocking
is actually working.
"""
project = self.create_project()
plan = self.create_plan(project)
snapshot = self.create_snapshot(project)
snapshot_image = self.create_snapshot_image(snapshot, plan)
cached_snapshot_image = self.create_cached_snapshot_image(
snapshot_image)
get_plans.return_value = set()
get_current_datetime.return_value = self.mock_datetime
gc.get_cached_snapshot_images('cluster')
get_current_datetime.assert_any_call()
def test_cached_snapshot_images_no_plans(self, get_plans):
project = self.create_project()
plan = self.create_plan(project)
snapshot = self.create_snapshot(project)
snapshot_image = self.create_snapshot_image(snapshot, plan)
cached_snapshot_image = self.create_cached_snapshot_image(snapshot_image)
get_plans.return_value = set()
assert gc.get_cached_snapshot_images('cluster') == []
def test_cached_snapshot_images_no_plans(self, get_plans):
project = self.create_project()
plan = self.create_plan(project)
snapshot = self.create_snapshot(project)
snapshot_image = self.create_snapshot_image(snapshot, plan)
cached_snapshot_image = self.create_cached_snapshot_image(
snapshot_image)
get_plans.return_value = set()
assert gc.get_cached_snapshot_images('cluster') == []
def test_cached_snapshot_images_all(self, get_plans, get_current_datetime):
"""
Tests the four fundamental cases for an image:
- No expiration date
- Not yet expired
- Expired
- Not part of the plans associated with the cluster
"""
project = self.create_project()
plan1 = self.create_plan(project)
plan2 = self.create_plan(project)
plan3 = self.create_plan(project)
plan4 = self.create_plan(project)
get_current_datetime.return_value = self.mock_datetime
get_plans.return_value = {plan1, plan2, plan3}
snapshot = self.create_snapshot(project)
snapshot_image1 = self.create_snapshot_image(snapshot, plan1)
snapshot_image2 = self.create_snapshot_image(snapshot, plan2)
snapshot_image3 = self.create_snapshot_image(snapshot, plan3)
snapshot_image4 = self.create_snapshot_image(snapshot, plan4)
self.create_cached_snapshot_image(snapshot_image1,
expiration_date=None)
self.create_cached_snapshot_image(snapshot_image2,
expiration_date=self.mock_datetime +
datetime.timedelta(0, 1))
self.create_cached_snapshot_image(snapshot_image3,
expiration_date=self.mock_datetime -
datetime.timedelta(0, 1))
self.create_cached_snapshot_image(snapshot_image4,
expiration_date=None)
cached_snapshot_ids = [
s.id for s in gc.get_cached_snapshot_images('cluster')
]
assert len(cached_snapshot_ids) == 2
assert snapshot_image1.id in cached_snapshot_ids
assert snapshot_image2.id in cached_snapshot_ids
def get(self, cluster):
images = gc.get_cached_snapshot_images(cluster)
return self.respond([image.id for image in images])
