Page MenuHome
Differential D8254 Diff 26646 films/tests/test_previous_next_asset.py

Changeset View

Standalone View

View Options

films/tests/test_previous_next_asset.py

import datetime as dt import datetime as dt
from unittest.mock import patch from unittest.mock import patch
from django.test import TestCase from django.test import TestCase
from django.urls.base import reverse from django.urls.base import reverse
from common.factories.assets import StaticAssetFactory from common.factories.static_assets import StaticAssetFactory
from common.factories.films import ( from common.factories.films import (
FilmFactory, FilmFactory,
CollectionFactory, CollectionFactory,
AssetFactory, AssetFactory,
ProductionLogFactory, ProductionLogFactory,
ProductionLogEntryFactory, ProductionLogEntryFactory,
ProductionLogEntryAssetFactory, ProductionLogEntryAssetFactory,
) )
from common.factories.user import UserFactory from common.factories.user import UserFactory
from films.views.api.assets import SiteContexts from films.views.api.assets import SiteContexts
@patch('sorl.thumbnail.base.ThumbnailBackend.get_thumbnail', return_value=None)
class TestSiteContextResolution(TestCase): class TestSiteContextResolution(TestCase):
@classmethod @classmethod
def setUpTestData(cls) -> None: def setUpTestData(cls) -> None:
cls.asset = AssetFactory() cls.asset = AssetFactory()
cls.other_asset = AssetFactory(film=cls.asset.film, collection=cls.asset.collection) cls.other_asset = AssetFactory(film=cls.asset.film, collection=cls.asset.collection)
@patch('films.views.api.assets.get_next_asset_in_gallery', return_value=None) @patch('films.views.api.assets.get_next_asset_in_gallery', return_value=None)
@patch('films.views.api.assets.get_previous_asset_in_gallery', return_value=None) @patch('films.views.api.assets.get_previous_asset_in_gallery', return_value=None)
def test_gallery_site_context(self, get_previous_asset_mock, get_next_asset_mock): def test_gallery_site_context(
self, get_previous_asset_mock, get_next_asset_mock, get_thumbnail_mock
):
query_string = f'site_context={SiteContexts.GALLERY.value}' query_string = f'site_context={SiteContexts.GALLERY.value}'
response = self.client.get(f'{reverse("api-asset", args=(self.asset.pk,))}?{query_string}') response = self.client.get(f'{reverse("api-asset", args=(self.asset.pk,))}?{query_string}')
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
get_previous_asset_mock.assert_called_once_with(self.asset) get_previous_asset_mock.assert_called_once_with(self.asset)
get_next_asset_mock.assert_called_once_with(self.asset) get_next_asset_mock.assert_called_once_with(self.asset)
@patch('films.views.api.assets.get_next_asset_in_featured_artwork', return_value=None) @patch('films.views.api.assets.get_next_asset_in_featured_artwork', return_value=None)
@patch('films.views.api.assets.get_previous_asset_in_featured_artwork', return_value=None) @patch('films.views.api.assets.get_previous_asset_in_featured_artwork', return_value=None)
def test_featured_artwork_site_context(self, get_previous_asset_mock, get_next_asset_mock): def test_featured_artwork_site_context(
self, get_previous_asset_mock, get_next_asset_mock, get_thumbnail_mock
):
query_string = f'site_context={SiteContexts.FEATURED_ARTWORK.value}' query_string = f'site_context={SiteContexts.FEATURED_ARTWORK.value}'
response = self.client.get(f'{reverse("api-asset", args=(self.asset.pk,))}?{query_string}') response = self.client.get(f'{reverse("api-asset", args=(self.asset.pk,))}?{query_string}')
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
get_previous_asset_mock.assert_called_once_with(self.asset) get_previous_asset_mock.assert_called_once_with(self.asset)
get_next_asset_mock.assert_called_once_with(self.asset) get_next_asset_mock.assert_called_once_with(self.asset)
@patch('films.views.api.assets.get_next_asset_in_production_logs', return_value=None) @patch('films.views.api.assets.get_next_asset_in_production_logs', return_value=None)
@patch('films.views.api.assets.get_previous_asset_in_production_logs', return_value=None) @patch('films.views.api.assets.get_previous_asset_in_production_logs', return_value=None)
def test_production_logs_site_context(self, get_previous_asset_mock, get_next_asset_mock): def test_production_logs_site_context(
self, get_previous_asset_mock, get_next_asset_mock, get_thumbnail_mock
):
query_string = f'site_context={SiteContexts.PRODUCTION_LOGS.value}' query_string = f'site_context={SiteContexts.PRODUCTION_LOGS.value}'
response = self.client.get(f'{reverse("api-asset", args=(self.asset.pk,))}?{query_string}') response = self.client.get(f'{reverse("api-asset", args=(self.asset.pk,))}?{query_string}')
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
get_previous_asset_mock.assert_called_once_with(self.asset) get_previous_asset_mock.assert_called_once_with(self.asset)
get_next_asset_mock.assert_called_once_with(self.asset) get_next_asset_mock.assert_called_once_with(self.asset)
def test_wrong_site_context(self): def test_wrong_site_context(self, get_thumbnail_mock):
query_string = 'site_context=definitely-incorrect' query_string = 'site_context=definitely-incorrect'
response = self.client.get(f'{reverse("api-asset", args=(self.asset.pk,))}?{query_string}') response = self.client.get(f'{reverse("api-asset", args=(self.asset.pk,))}?{query_string}')
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
self.assertEqual(response.context.get('asset'), self.asset) self.assertEqual(response.context.get('asset'), self.asset)
self.assertEqual(response.context.get('previous_asset'), None) self.assertEqual(response.context.get('previous_asset'), None)
self.assertEqual(response.context.get('next_asset'), None) self.assertEqual(response.context.get('next_asset'), None)
def test_no_site_context_in_query_string(self): def test_no_site_context_in_query_string(self, get_thumbnail_mock):
response = self.client.get(reverse("api-asset", args=(self.asset.pk,))) response = self.client.get(reverse("api-asset", args=(self.asset.pk,)))
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
self.assertEqual(response.context.get('asset'), self.asset) self.assertEqual(response.context.get('asset'), self.asset)
self.assertEqual(response.context.get('previous_asset'), None) self.assertEqual(response.context.get('previous_asset'), None)
self.assertEqual(response.context.get('next_asset'), None) self.assertEqual(response.context.get('next_asset'), None)
@patch('sorl.thumbnail.base.ThumbnailBackend.get_thumbnail', return_value=None)
class TestAssetOrderingInGallery(TestCase): class TestAssetOrderingInGallery(TestCase):
@classmethod @classmethod
def setUpTestData(cls) -> None: def setUpTestData(cls) -> None:
cls.site_context = SiteContexts.GALLERY.value cls.site_context = SiteContexts.GALLERY.value
film = FilmFactory() film = FilmFactory()
cls.collection_a = CollectionFactory(film=film) cls.collection_a = CollectionFactory(film=film)
collection_b = CollectionFactory(film=film, order=1) collection_b = CollectionFactory(film=film, order=1)
collection_c = CollectionFactory(film=film, order=2) collection_c = CollectionFactory(film=film, order=2)
other_film = FilmFactory() other_film = FilmFactory()
other_collection = CollectionFactory(film=other_film) other_collection = CollectionFactory(film=other_film)
# assets from collection A: should be sorted by (order, name) # assets from collection A: should be sorted by (order, name)
cls.asset_a_2 = AssetFactory( cls.asset_a_2 = AssetFactory(
film=film, collection=cls.collection_a, order=2, name='Aa', is_featured=True film=film, collection=cls.collection_a, order=2, name='Aa', is_featured=True
) )
cls.asset_a_3 = AssetFactory(film=film, collection=cls.collection_a, name='Aa') cls.asset_a_3 = AssetFactory(film=film, collection=cls.collection_a, name='Aa')
cls.asset_a_0 = AssetFactory(film=film, collection=cls.collection_a, order=1, name='Bb') cls.asset_a_0 = AssetFactory(film=film, collection=cls.collection_a, order=1, name='Bb')
cls.asset_a_1 = AssetFactory(film=film, collection=cls.collection_a, order=1, name='Cc') cls.asset_a_1 = AssetFactory(film=film, collection=cls.collection_a, order=1, name='Cc')
# assets from other films and collections, should not be included in context: # assets from other films and collections, should not be included in context:
fsiddiUnsubmitted
Done
Inline Actions
cls.asset_a_1 = AssetFactory(film=film, collection=cls.collection_a, order=1, name='Cc')
- # assets from other films and collections, should not be included in context:
+ # Assets from other films and collections, should not be included in context:
asset_b_0 = AssetFactory(film=film, collection=collection_b, is_featured=True)

Comments start with capital letter :)

fsiddi: Comments start with capital letter :)
asset_b_0 = AssetFactory(film=film, collection=collection_b, is_featured=True) asset_b_0 = AssetFactory(film=film, collection=collection_b, is_featured=True)
asset_c_0 = AssetFactory(film=film, collection=collection_c, is_featured=True) asset_c_0 = AssetFactory(film=film, collection=collection_c, is_featured=True)
other_asset_0 = AssetFactory(film=other_film, collection=other_collection) other_asset_0 = AssetFactory(film=other_film, collection=other_collection)
other_asset_1 = AssetFactory(film=other_film, collection=other_collection, is_featured=True) other_asset_1 = AssetFactory(film=other_film, collection=other_collection, is_featured=True)
def test_previous_asset_for_first_asset_is_none(self): def test_previous_asset_for_first_asset_is_none(self, get_thumbnail_mock):
first_asset = self.asset_a_0 first_asset = self.asset_a_0
response = self.client.get( response = self.client.get(
f'{reverse("api-asset", args=(first_asset.pk,))}?site_context={self.site_context}' f'{reverse("api-asset", args=(first_asset.pk,))}?site_context={self.site_context}'
) )
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
self.assertEqual(response.context.get('asset'), first_asset) self.assertEqual(response.context.get('asset'), first_asset)
self.assertEqual(response.context.get('previous_asset'), None) self.assertEqual(response.context.get('previous_asset'), None)
def test_next_asset_for_last_asset_is_none(self): def test_next_asset_for_last_asset_is_none(self, get_thumbnail_mock):
last_asset = self.asset_a_3 last_asset = self.asset_a_3
response = self.client.get( response = self.client.get(
f'{reverse("api-asset", args=(last_asset.pk,))}?site_context={self.site_context}' f'{reverse("api-asset", args=(last_asset.pk,))}?site_context={self.site_context}'
) )
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
self.assertEqual(response.context.get('asset'), last_asset) self.assertEqual(response.context.get('asset'), last_asset)
self.assertEqual(response.context.get('next_asset'), None) self.assertEqual(response.context.get('next_asset'), None)
def test_assets_from_collection_sorted_by_order_and_name(self): def test_assets_from_collection_sorted_by_order_and_name(self, get_thumbnail_mock):
assets = [self.asset_a_0, self.asset_a_1, self.asset_a_2, self.asset_a_3] assets = [self.asset_a_0, self.asset_a_1, self.asset_a_2, self.asset_a_3]
for previous_asset, asset, next_asset in zip(assets, assets[1:], assets[2:]): for previous_asset, asset, next_asset in zip(assets, assets[1:], assets[2:]):
response = self.client.get( response = self.client.get(
f'{reverse("api-asset", args=(asset.pk,))}?site_context={self.site_context}' f'{reverse("api-asset", args=(asset.pk,))}?site_context={self.site_context}'
) )
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
self.assertEqual(response.context.get('previous_asset'), previous_asset) self.assertEqual(response.context.get('previous_asset'), previous_asset)
self.assertEqual(response.context.get('asset'), asset) self.assertEqual(response.context.get('asset'), asset)
self.assertEqual(response.context.get('next_asset'), next_asset) self.assertEqual(response.context.get('next_asset'), next_asset)
@patch('sorl.thumbnail.base.ThumbnailBackend.get_thumbnail', return_value=None)
class TestAssetOrderingInFeaturedArtwork(TestCase): class TestAssetOrderingInFeaturedArtwork(TestCase):
@classmethod @classmethod
def setUpTestData(cls) -> None: def setUpTestData(cls) -> None:
cls.site_context = SiteContexts.FEATURED_ARTWORK.value cls.site_context = SiteContexts.FEATURED_ARTWORK.value
film = FilmFactory() film = FilmFactory()
collection_a = CollectionFactory(film=film) collection_a = CollectionFactory(film=film)
collection_b = CollectionFactory(film=film, order=1) collection_b = CollectionFactory(film=film, order=1)
Show All 12 Lines def setUpTestData(cls) -> None:
cls.featured_asset_2 = AssetFactory(film=film, collection=collection_c, is_featured=True) cls.featured_asset_2 = AssetFactory(film=film, collection=collection_c, is_featured=True)
asset_a_2 = AssetFactory(film=film, collection=collection_a, order=3) asset_a_2 = AssetFactory(film=film, collection=collection_a, order=3)
cls.featured_asset_3 = AssetFactory(film=film, collection=collection_a, is_featured=True) cls.featured_asset_3 = AssetFactory(film=film, collection=collection_a, is_featured=True)
# assets from the other film should not be included in context: # assets from the other film should not be included in context:
other_asset_0 = AssetFactory(film=other_film, collection=other_collection) other_asset_0 = AssetFactory(film=other_film, collection=other_collection)
other_asset_1 = AssetFactory(film=other_film, collection=other_collection, is_featured=True) other_asset_1 = AssetFactory(film=other_film, collection=other_collection, is_featured=True)
def test_previous_asset_for_first_asset_is_none(self): def test_previous_asset_for_first_asset_is_none(self, get_thumbnail_mock):
first_asset = self.featured_asset_0 first_asset = self.featured_asset_0
response = self.client.get( response = self.client.get(
f'{reverse("api-asset", args=(first_asset.pk,))}?site_context={self.site_context}' f'{reverse("api-asset", args=(first_asset.pk,))}?site_context={self.site_context}'
) )
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
self.assertEqual(response.context.get('asset'), first_asset) self.assertEqual(response.context.get('asset'), first_asset)
self.assertEqual(response.context.get('previous_asset'), None) self.assertEqual(response.context.get('previous_asset'), None)
def test_next_asset_for_last_asset_is_none(self): def test_next_asset_for_last_asset_is_none(self, get_thumbnail_mock):
last_asset = self.featured_asset_3 last_asset = self.featured_asset_3
response = self.client.get( response = self.client.get(
f'{reverse("api-asset", args=(last_asset.pk,))}?site_context={self.site_context}' f'{reverse("api-asset", args=(last_asset.pk,))}?site_context={self.site_context}'
) )
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
self.assertEqual(response.context.get('asset'), last_asset) self.assertEqual(response.context.get('asset'), last_asset)
self.assertEqual(response.context.get('next_asset'), None) self.assertEqual(response.context.get('next_asset'), None)
def test_featured_assets_sorted_by_date_created(self): def test_featured_assets_sorted_by_date_created(self, get_thumbnail_mock):
assets = [ assets = [
self.featured_asset_0, self.featured_asset_0,
self.featured_asset_1, self.featured_asset_1,
self.featured_asset_2, self.featured_asset_2,
self.featured_asset_3, self.featured_asset_3,
] ]
for previous_asset, asset, next_asset in zip(assets, assets[1:], assets[2:]): for previous_asset, asset, next_asset in zip(assets, assets[1:], assets[2:]):
response = self.client.get( response = self.client.get(
f'{reverse("api-asset", args=(asset.pk,))}?site_context={self.site_context}' f'{reverse("api-asset", args=(asset.pk,))}?site_context={self.site_context}'
) )
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
self.assertEqual(response.context.get('previous_asset'), previous_asset) self.assertEqual(response.context.get('previous_asset'), previous_asset)
self.assertEqual(response.context.get('asset'), asset) self.assertEqual(response.context.get('asset'), asset)
self.assertEqual(response.context.get('next_asset'), next_asset) self.assertEqual(response.context.get('next_asset'), next_asset)
@patch('sorl.thumbnail.base.ThumbnailBackend.get_thumbnail', return_value=None)
class TestAssetOrderingInProductionLogs(TestCase): class TestAssetOrderingInProductionLogs(TestCase):
@classmethod @classmethod
def setUpTestData(cls) -> None: def setUpTestData(cls) -> None:
cls.site_context = SiteContexts.PRODUCTION_LOGS.value cls.site_context = SiteContexts.PRODUCTION_LOGS.value
author_a = UserFactory() author_a = UserFactory()
author_b = UserFactory() author_b = UserFactory()
Show All 18 Lines def setUpTestData(cls) -> None:
other_entry = ProductionLogEntryFactory( other_entry = ProductionLogEntryFactory(
user=author_a, production_log=ProductionLogFactory(start_date=dt.date(1997, 12, 1)) user=author_a, production_log=ProductionLogFactory(start_date=dt.date(1997, 12, 1))
) )
ProductionLogEntryAssetFactory(production_log_entry=other_entry, asset=other_asset_a) ProductionLogEntryAssetFactory(production_log_entry=other_entry, asset=other_asset_a)
asset_b = AssetFactory(film=film, static_asset=StaticAssetFactory(user=author_b)) asset_b = AssetFactory(film=film, static_asset=StaticAssetFactory(user=author_b))
entry_b = ProductionLogEntryFactory(production_log=prod_log, author=author_b) entry_b = ProductionLogEntryFactory(production_log=prod_log, author=author_b)
ProductionLogEntryAssetFactory(production_log_entry=entry_b, asset=asset_b) ProductionLogEntryAssetFactory(production_log_entry=entry_b, asset=asset_b)
def test_previous_asset_for_first_asset_is_none(self): def test_previous_asset_for_first_asset_is_none(self, get_thumbnail_mock):
first_asset = self.asset_a_0 first_asset = self.asset_a_0
response = self.client.get( response = self.client.get(
f'{reverse("api-asset", args=(first_asset.pk,))}?site_context={self.site_context}' f'{reverse("api-asset", args=(first_asset.pk,))}?site_context={self.site_context}'
) )
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
self.assertEqual(response.context.get('asset'), first_asset) self.assertEqual(response.context.get('asset'), first_asset)
self.assertEqual(response.context.get('previous_asset'), None) self.assertEqual(response.context.get('previous_asset'), None)
def test_next_asset_for_last_asset_is_none(self): def test_next_asset_for_last_asset_is_none(self, get_thumbnail_mock):
last_asset = self.asset_a_3 last_asset = self.asset_a_3
response = self.client.get( response = self.client.get(
f'{reverse("api-asset", args=(last_asset.pk,))}?site_context={self.site_context}' f'{reverse("api-asset", args=(last_asset.pk,))}?site_context={self.site_context}'
) )
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
self.assertEqual(response.context.get('asset'), last_asset) self.assertEqual(response.context.get('asset'), last_asset)
self.assertEqual(response.context.get('next_asset'), None) self.assertEqual(response.context.get('next_asset'), None)
def test_assets_in_production_log_entry_sorted_by_date_created(self): def test_assets_in_production_log_entry_sorted_by_date_created(self, get_thumbnail_mock):
assets = [ assets = [
self.asset_a_0, self.asset_a_0,
self.asset_a_1, self.asset_a_1,
self.asset_a_2, self.asset_a_2,
self.asset_a_3, self.asset_a_3,
] ]
for previous_asset, asset, next_asset in zip(assets, assets[1:], assets[2:]): for previous_asset, asset, next_asset in zip(assets, assets[1:], assets[2:]):
response = self.client.get( response = self.client.get(
f'{reverse("api-asset", args=(asset.pk,))}?site_context={self.site_context}' f'{reverse("api-asset", args=(asset.pk,))}?site_context={self.site_context}'
) )
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
self.assertEqual(response.context.get('previous_asset'), previous_asset) self.assertEqual(response.context.get('previous_asset'), previous_asset)
self.assertEqual(response.context.get('asset'), asset) self.assertEqual(response.context.get('asset'), asset)
self.assertEqual(response.context.get('next_asset'), next_asset) self.assertEqual(response.context.get('next_asset'), next_asset)