django/tests/admin_views/test_related_object_lookups.py

123 lines
4.7 KiB
Python

from django.contrib.admin.tests import AdminSeleniumTestCase
from django.contrib.auth.models import User
from django.test import override_settings
from django.urls import reverse
@override_settings(ROOT_URLCONF="admin_views.urls")
class SeleniumTests(AdminSeleniumTestCase):
available_apps = ["admin_views"] + AdminSeleniumTestCase.available_apps
def setUp(self):
self.superuser = User.objects.create_superuser(
username="super", password="secret", email="super@example.com"
)
self.admin_login(
username="super", password="secret", login_url=reverse("admin:index")
)
def test_related_object_link_images_attributes(self):
from selenium.webdriver.common.by import By
album_add_url = reverse("admin:admin_views_album_add")
self.selenium.get(self.live_server_url + album_add_url)
tests = [
"add_id_owner",
"change_id_owner",
"delete_id_owner",
"view_id_owner",
]
for link_id in tests:
with self.subTest(link_id):
link_image = self.selenium.find_element(
By.XPATH, f'//*[@id="{link_id}"]/img'
)
self.assertEqual(link_image.get_attribute("alt"), "")
self.assertEqual(link_image.get_attribute("width"), "20")
self.assertEqual(link_image.get_attribute("height"), "20")
def test_related_object_lookup_link_initial_state(self):
from selenium.webdriver.common.by import By
album_add_url = reverse("admin:admin_views_album_add")
self.selenium.get(self.live_server_url + album_add_url)
tests = [
"change_id_owner",
"delete_id_owner",
"view_id_owner",
]
for link_id in tests:
with self.subTest(link_id):
link = self.selenium.find_element(By.XPATH, f'//*[@id="{link_id}"]')
self.assertEqual(link.get_attribute("aria-disabled"), "true")
def test_related_object_lookup_link_enabled(self):
from selenium.webdriver.common.by import By
from selenium.webdriver.support.select import Select
album_add_url = reverse("admin:admin_views_album_add")
self.selenium.get(self.live_server_url + album_add_url)
select_element = self.selenium.find_element(By.XPATH, '//*[@id="id_owner"]')
option = Select(select_element).options[-1]
self.assertEqual(option.text, "super")
select_element.click()
option.click()
tests = [
"add_id_owner",
"change_id_owner",
"delete_id_owner",
"view_id_owner",
]
for link_id in tests:
with self.subTest(link_id):
link = self.selenium.find_element(By.XPATH, f'//*[@id="{link_id}"]')
self.assertIsNone(link.get_attribute("aria-disabled"))
def test_related_object_update_with_camel_casing(self):
from selenium.webdriver.common.by import By
add_url = reverse("admin:admin_views_camelcaserelatedmodel_add")
self.selenium.get(self.live_server_url + add_url)
interesting_name = "A test name"
# Add a new CamelCaseModel using the "+" icon next to the "fk" field.
self.selenium.find_element(By.ID, "add_id_fk").click()
# Switch to the add popup window.
self.wait_for_and_switch_to_popup()
# Find the "interesting_name" field and enter a value, then save it.
self.selenium.find_element(By.ID, "id_interesting_name").send_keys(
interesting_name
)
self.selenium.find_element(By.NAME, "_save").click()
# Return to the main window.
self.wait_until(lambda d: len(d.window_handles) == 1, 1)
self.selenium.switch_to.window(self.selenium.window_handles[0])
# Check that both the "Available" m2m box and the "Fk" dropdown now
# include the newly added CamelCaseModel instance.
fk_dropdown = self.selenium.find_element(By.ID, "id_fk")
self.assertHTMLEqual(
fk_dropdown.get_attribute("innerHTML"),
f"""
<option value="" selected="">---------</option>
<option value="1" selected>{interesting_name}</option>
""",
)
# Check the newly added instance is not also added in the "to" box.
m2m_to = self.selenium.find_element(By.ID, "id_m2m_to")
self.assertHTMLEqual(m2m_to.get_attribute("innerHTML"), "")
m2m_box = self.selenium.find_element(By.ID, "id_m2m_from")
self.assertHTMLEqual(
m2m_box.get_attribute("innerHTML"),
f"""
<option value="1">{interesting_name}</option>
""",
)