41e5403f4e
* pyupgrade on contentstore/views * Apply suggestions from code review Co-authored-by: Usama Sadiq <usama.sadiq@arbisoft.com> Co-authored-by: Usama Sadiq <usama.sadiq@arbisoft.com>
91 lines
3.1 KiB
Python
91 lines
3.1 KiB
Python
"""
|
|
Utilities for view tests.
|
|
"""
|
|
|
|
|
|
import json
|
|
|
|
from cms.djangoapps.contentstore.tests.utils import CourseTestCase
|
|
from xmodule.modulestore.tests.factories import ItemFactory
|
|
|
|
from ..helpers import xblock_studio_url
|
|
|
|
|
|
class StudioPageTestCase(CourseTestCase):
|
|
"""
|
|
Base class for all tests of Studio pages.
|
|
"""
|
|
|
|
def setUp(self):
|
|
super().setUp()
|
|
self.chapter = ItemFactory.create(parent_location=self.course.location,
|
|
category='chapter', display_name="Week 1")
|
|
self.sequential = ItemFactory.create(parent_location=self.chapter.location,
|
|
category='sequential', display_name="Lesson 1")
|
|
|
|
def get_page_html(self, xblock):
|
|
"""
|
|
Returns the HTML for the page representing the xblock.
|
|
"""
|
|
url = xblock_studio_url(xblock)
|
|
self.assertIsNotNone(url)
|
|
resp = self.client.get_html(url)
|
|
self.assertEqual(resp.status_code, 200)
|
|
return resp.content.decode(resp.charset)
|
|
|
|
def get_preview_html(self, xblock, view_name):
|
|
"""
|
|
Returns the HTML for the xblock when shown within a unit or container page.
|
|
"""
|
|
preview_url = f'/xblock/{xblock.location}/{view_name}'
|
|
resp = self.client.get_json(preview_url)
|
|
self.assertEqual(resp.status_code, 200)
|
|
resp_content = json.loads(resp.content.decode('utf-8'))
|
|
return resp_content['html']
|
|
|
|
def validate_preview_html(self, xblock, view_name, can_add=True, can_reorder=True, can_move=True,
|
|
can_edit=True, can_duplicate=True, can_delete=True):
|
|
"""
|
|
Verify that the specified xblock's preview has the expected HTML elements.
|
|
"""
|
|
html = self.get_preview_html(xblock, view_name)
|
|
self.validate_html_for_action_button(
|
|
html,
|
|
'<div class="add-xblock-component new-component-item adding"></div>',
|
|
can_add
|
|
)
|
|
self.validate_html_for_action_button(
|
|
html,
|
|
'<span data-tooltip="Drag to reorder" class="drag-handle action"></span>',
|
|
can_reorder
|
|
)
|
|
self.validate_html_for_action_button(
|
|
html,
|
|
'<button data-tooltip="Move" class="btn-default move-button action-button">',
|
|
can_move
|
|
)
|
|
self.validate_html_for_action_button(
|
|
html,
|
|
'button class="btn-default edit-button action-button">',
|
|
can_edit
|
|
)
|
|
self.validate_html_for_action_button(
|
|
html,
|
|
'<button data-tooltip="Delete" class="btn-default delete-button action-button">',
|
|
can_duplicate
|
|
)
|
|
self.validate_html_for_action_button(
|
|
html,
|
|
'<button data-tooltip="Duplicate" class="btn-default duplicate-button action-button">',
|
|
can_delete
|
|
)
|
|
|
|
def validate_html_for_action_button(self, html, expected_html, can_action=True):
|
|
"""
|
|
Validate that the specified HTML has specific action..
|
|
"""
|
|
if can_action:
|
|
self.assertIn(expected_html, html)
|
|
else:
|
|
self.assertNotIn(expected_html, html)
|