109 lines
4.3 KiB
Python
109 lines
4.3 KiB
Python
from bson import ObjectId
|
|
|
|
from pillar.tests import AbstractPillarTest
|
|
from pillar.tests import common_test_data as ctd
|
|
from pillar.web import system_util
|
|
from pillar.web.projects.routes import project_navigation_links, find_project_or_404
|
|
|
|
|
|
class ProjectTest(AbstractPillarTest):
|
|
def setUp(self, **kwargs):
|
|
super().setUp(**kwargs)
|
|
|
|
self.pid, self.project = self.ensure_project_exists()
|
|
self.owner_uid = self.create_user(24 * 'a',
|
|
groups=[ctd.EXAMPLE_ADMIN_GROUP_ID],
|
|
token='admin-token')
|
|
|
|
# Create a Node of type page.
|
|
self.node_id = self.create_node({
|
|
'_id': ObjectId('572761099837730efe8e120d'),
|
|
'description': 'This the about page',
|
|
'node_type': 'page',
|
|
'user': self.owner_uid,
|
|
'properties': {
|
|
'status': 'published',
|
|
'url': 'about',
|
|
},
|
|
'name': 'About',
|
|
'project': self.pid,
|
|
})
|
|
|
|
self.user_uid = self.create_user(24 * 'b', groups=[ctd.EXAMPLE_ADMIN_GROUP_ID],
|
|
token='user-token')
|
|
|
|
def test_project_navigation_links_one_page(self):
|
|
"""Test link generation for project navigation."""
|
|
with self.app.test_request_context():
|
|
api = system_util.pillar_api()
|
|
project = find_project_or_404(self.project['url'], api=api)
|
|
navigation_links = project_navigation_links(project, api=api)
|
|
|
|
# We expect only one link to be in the list
|
|
links = [{'url': '/p/default-project/about', 'slug': 'about', 'label': 'About'}]
|
|
self.assertListEqual(links, navigation_links)
|
|
|
|
def test_project_navigation_links_pages_and_blog(self):
|
|
"""Test link generation for a project with two Pages and one Blog."""
|
|
# Add one more page to the project
|
|
self.create_node({
|
|
'_id': ObjectId(),
|
|
'description': 'This the awards page',
|
|
'node_type': 'page',
|
|
'user': self.owner_uid,
|
|
'properties': {
|
|
'status': 'published',
|
|
'url': 'awards',
|
|
},
|
|
'name': 'Awards',
|
|
'project': self.pid,
|
|
})
|
|
# Create a Node of type blog.
|
|
self.create_node({
|
|
'_id': ObjectId(),
|
|
'description': 'This the blog page',
|
|
'node_type': 'blog',
|
|
'user': self.owner_uid,
|
|
'properties': {
|
|
'status': 'published',
|
|
},
|
|
'name': 'Blog',
|
|
'project': self.pid,
|
|
})
|
|
# Create a Node of type asset
|
|
self.create_node({
|
|
'_id': ObjectId(),
|
|
'description': 'This is an asset without file',
|
|
'node_type': 'asset',
|
|
'user': self.owner_uid,
|
|
'properties': {
|
|
'status': 'published',
|
|
'content_type': 'image',
|
|
},
|
|
'name': 'Image test',
|
|
'project': self.pid,
|
|
})
|
|
|
|
with self.app.test_request_context():
|
|
api = system_util.pillar_api()
|
|
project = find_project_or_404(self.project['url'], api=api)
|
|
navigation_links = project_navigation_links(project, api=api)
|
|
expected_list = [
|
|
# Blog is the first element of the list (since it's added first)
|
|
{'url': '/blog/', 'slug': 'blog', 'label': 'Blog'},
|
|
{'url': '/p/default-project/about', 'slug': 'about', 'label': 'About'},
|
|
{'url': '/p/default-project/awards', 'slug': 'awards', 'label': 'Awards'}]
|
|
|
|
self.assertListEqual(expected_list, navigation_links)
|
|
|
|
def test_project_no_navigation_links(self):
|
|
"""Test link generation in a project with no Pages or Blog."""
|
|
with self.app.test_request_context():
|
|
# Delete the existing page from the database
|
|
self.app.db('nodes').delete_one({'_id': ObjectId('572761099837730efe8e120d')})
|
|
api = system_util.pillar_api()
|
|
project = find_project_or_404(self.project['url'], api=api)
|
|
navigation_links = project_navigation_links(project, api=api)
|
|
# This should result in an empty list
|
|
self.assertListEqual([], navigation_links)
|