53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
import pytest
|
|
from django.contrib.auth import get_user_model
|
|
from django.utils import timezone
|
|
from scrobbles.models import Scrobble
|
|
from tasks.models import Task
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_standard_toolbar_on_base_list_page(client):
|
|
user = get_user_model().objects.create_user(
|
|
username="toolbaruser", password="testpass"
|
|
)
|
|
task = Task.objects.create(title="Ship the toolbar")
|
|
Scrobble.objects.create(
|
|
user=user,
|
|
task=task,
|
|
media_type="Task",
|
|
timestamp=timezone.now(),
|
|
)
|
|
client.force_login(user)
|
|
response = client.get(f"/tasks/{task.uuid}/")
|
|
assert response.status_code == 200
|
|
content = response.content.decode()
|
|
# Standard toolbar present on a base_list-based detail page
|
|
assert "Search..." in content
|
|
assert "Calendar" in content
|
|
assert "Admin" in content
|
|
assert "Charts" in content
|
|
assert "Live" in content
|
|
assert "Browse" in content
|
|
assert "media-browser-menu" in content
|
|
# Grid view toggle kept
|
|
assert "Grid View" in content
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_standard_toolbar_on_home(client):
|
|
user = get_user_model().objects.create_user(
|
|
username="toolbarhome", password="testpass"
|
|
)
|
|
task = Task.objects.create(title="Task")
|
|
Scrobble.objects.create(
|
|
user=user, task=task, media_type="Task", timestamp=timezone.now()
|
|
)
|
|
client.force_login(user)
|
|
response = client.get("/")
|
|
assert response.status_code == 200
|
|
content = response.content.decode()
|
|
assert "Search..." in content
|
|
assert "Calendar" in content
|
|
assert "Live" in content
|
|
assert "media-browser-menu" in content
|