|
| 1 | +import os |
| 2 | +import pytest |
| 3 | + |
| 4 | +from playwright.sync_api import sync_playwright |
| 5 | + |
| 6 | +from django.conf import settings |
| 7 | +from django.contrib.admin.sites import site as admin_site |
| 8 | +from django.core.management import call_command |
| 9 | +from django.urls import reverse |
| 10 | + |
| 11 | +from finder.models.folder import FolderModel |
| 12 | +from finder.models.realm import RealmModel |
| 13 | + |
| 14 | +from .utils import create_random_image |
| 15 | + |
| 16 | +os.environ.setdefault('DJANGO_ALLOW_ASYNC_UNSAFE', 'true') |
| 17 | + |
| 18 | + |
| 19 | +@pytest.fixture(autouse=True, scope='session') |
| 20 | +def create_assets(): |
| 21 | + os.makedirs(settings.BASE_DIR / 'workdir/assets', exist_ok=True) |
| 22 | + for counter in range(10): |
| 23 | + image = create_random_image() |
| 24 | + image.save(settings.BASE_DIR / 'workdir/assets' / f'image_{counter:01d}.png') |
| 25 | + |
| 26 | + |
| 27 | +@pytest.fixture(scope='session') |
| 28 | +def django_db_setup(django_db_blocker): |
| 29 | + database_file = settings.BASE_DIR / 'workdir/test_db.sqlite3' |
| 30 | + settings.DATABASES['default']['NAME'] = database_file |
| 31 | + with django_db_blocker.unblock(): |
| 32 | + call_command('migrate', verbosity=0) |
| 33 | + yield |
| 34 | + os.remove(database_file) |
| 35 | + |
| 36 | + |
| 37 | +@pytest.fixture |
| 38 | +def realm(admin_client): |
| 39 | + if realm := RealmModel.objects.first(): |
| 40 | + return realm |
| 41 | + response = admin_client.get(reverse('admin:finder_foldermodel_changelist')) |
| 42 | + assert response.status_code == 302 |
| 43 | + realm = RealmModel.objects.first() |
| 44 | + assert realm is not None |
| 45 | + redirected = reverse('admin:finder_inodemodel_change', kwargs={'inode_id': realm.root_folder.id}) |
| 46 | + assert response.url == redirected |
| 47 | + assert realm.root_folder.is_folder is True |
| 48 | + assert realm.root_folder.is_trash is False |
| 49 | + assert realm.root_folder.owner == response.wsgi_request.user |
| 50 | + assert realm.root_folder.name == '__root__' |
| 51 | + assert realm.root_folder.parent is None |
| 52 | + assert realm.root_folder.is_root |
| 53 | + assert realm.trash_folders.count() == 0 |
| 54 | + return realm |
| 55 | + |
| 56 | + |
| 57 | +class Connector: |
| 58 | + def __init__(self, live_server): |
| 59 | + print(f"\nStarting end-to-end test server at {live_server}\n") |
| 60 | + self.live_server = live_server |
| 61 | + |
| 62 | + def __enter__(self): |
| 63 | + def print_args(msg): |
| 64 | + if msg.type in ['info', 'debug']: |
| 65 | + return |
| 66 | + for arg in msg.args: |
| 67 | + print(arg.json_value()) |
| 68 | + |
| 69 | + self.playwright = sync_playwright().start() |
| 70 | + self.browser = self.playwright.chromium.launch() |
| 71 | + return self |
| 72 | + |
| 73 | + def __exit__(self, exc_type, exc_val, exc_tb): |
| 74 | + self.browser.close() |
| 75 | + self.playwright.stop() |
| 76 | + |
| 77 | + |
| 78 | +@pytest.fixture(scope='session') |
| 79 | +def connector(live_server): |
| 80 | + with Connector(live_server) as connector: |
| 81 | + yield connector |
| 82 | + |
| 83 | + |
| 84 | +@pytest.fixture |
| 85 | +def locale(): |
| 86 | + return 'en-US' |
| 87 | + |
| 88 | + |
| 89 | +@pytest.fixture |
| 90 | +def language(): |
| 91 | + return 'en' |
| 92 | + |
| 93 | + |
| 94 | +def print_args(msg): |
| 95 | + """ |
| 96 | + Print messages from the browser console. |
| 97 | + """ |
| 98 | + for arg in msg.args: |
| 99 | + print(arg.json_value()) |
| 100 | + |
| 101 | + |
| 102 | +@pytest.fixture() |
| 103 | +def page(connector, viewname, locale, language): |
| 104 | + context = connector.browser.new_context(locale=locale) |
| 105 | + context.add_cookies([{'name': 'django_language', 'value': language, 'domain': 'localhost', 'path': '/'}]) |
| 106 | + page = context.new_page() |
| 107 | + # page.on('console', print_args) |
| 108 | + page.goto(connector.live_server.url + reverse(viewname)) |
| 109 | + # django_formset = page.locator('django-formset:defined') |
| 110 | + # django_formset.wait_for() |
| 111 | + return page |
0 commit comments