Выборка данных в представление Django работает вручную, но не работает с Selenium. Почему?

Я использую Selenium для тестирования созданного мной приложения Django.
Следующий код отлично работает "вручную", но не работает с Selenium.
Почему?


## dummy.html ## 
{% extends 'main.html' %}
{% load static %}

{% block title %}
Dummy page
{% endblock title %}

{% block meta %}
<meta name="robots" content="noindex">
{% endblock meta %}

{% block content %}

<div class="container">
    <div class="row">
        <div class="col d-flex justify-content-center align-items-center">
            <h1 class="text-primary font-handwriting">Dummy page</h1>
        </div>
    </div>
    <br>
    <div class="row">
        <div class="col d-flex justify-content-center align-items-center">
            <h2 class="text-primary font-heading">THIS IS A DUMMY PAGE</h2>
        </div>
    </div>
    <br>
    <div class="row">
        <button id="trigger-dummy-button" type="button"
            class="btn btn-primary text-white font-base fw-bold text-uppercase"
            title="Trigger dummy button">Trigger dummy button</button>
    </div>
</div>

<script> 
    var someDummyURL = "{% url 'home_page:another_dummy_view' %}";
</script>

<script type="module" src="{% static '/js/dummy.js' %}">  </script>

{% endblock content %}

## dummy.js ## 
'use strict';

var newValue;

async function postDummyFunction(request) {
    var response = await fetch(request);
    if (response.ok) {
        var updatedReturnedValue = await response.json();
        newValue = updatedReturnedValue.returnedValue;
        alert('-- postDummyFunction --  |  newValue = ' + newValue);
        // When the script is triggered manually by clicking, response.ok = true and newValue = "Hello World OK !"
        return newValue;
    } else {
        // When the script is triggered by Selenium, response.ok = false
        alert('-- postDummyFunction --  |  response.ok ? ' + response.ok);
        return false;
    }
};


document.addEventListener("DOMContentLoaded", function () {

    var triggerDummyButton = document.getElementById("trigger-dummy-button");
    var someDummyVariable = "Hello World ";
    var toPostURL = someDummyURL;
    
    triggerDummyButton.onclick = (e) => {
        var data = {"dummyValue": someDummyVariable};
        var request = new Request(toPostURL, {
            method: "POST",
            mode: "same-origin",
            headers: {
                "Content-Type": "application/json",
                'X-Requested-With':'XMLHttpRequest',
                "Accept": "application/json",
                'X-CSRFToken': csrftoken
            },
            body: JSON.stringify(data),
        });
        // Works with Selenium
        // if alert(data.dummyValue), browser correctly displays "Hello World "
        postDummyFunction(request);
    }
});

## views.py ## 
import json

from django.shortcuts import render, redirect
from django.http import JsonResponse


def dev_dummy_script(request):
    context = {}
    return render(request, "dummy.html", context)


def another_dummy_view(request):
    if request.method == "POST":
        post_data = json.loads(request.body.decode("utf-8"))
        posted_value = post_data["dummyValue"]
        value_to_return = posted_value + " " + "OK !"
        response = JsonResponse({"returnedValue": value_to_return})
        return response

## test_selenium.py ## 

import time
import os
from decimal import Decimal
from selenium import webdriver
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

from django.contrib.staticfiles.testing import StaticLiveServerTestCase

from core.settings import BASE_DIR

PATH_TO_WEBDRIVER_CHROME = os.path.join(
    BASE_DIR, "test", "webdrivers", "chromedriver.exe"
)

PATH_TO_WEBDRIVER_FIREFOX = os.path.join(BASE_DIR, "test", "webdrivers", "geckodriver")

SHORT_TIME = 2.5
LONG_TIME = 6
DEFAULT_WAIT_TIME = 10


class FooTest(StaticLiveServerTestCase):
    def setUp(self):
        self.browser = webdriver.Firefox(executable_path=PATH_TO_WEBDRIVER_FIREFOX)

    def tearDown(self):
        self.browser.close()

    def testfoo(self):

        self.browser.get(("%s%s" % (self.live_server_url, "/dev/dummy_script/")))

        elem_trigger_btn = WebDriverWait(self.browser, DEFAULT_WAIT_TIME).until(
            EC.element_to_be_clickable((By.ID, "trigger-dummy-button"))
        )

        elem_trigger_btn.click()

        time.sleep(LONG_TIME)

        self.browser.close()

версии

  • Django==3.2
  • selenium==3.141.0
  • Firefox 96.0.3

Если я запускаю python manage.py runserver, вручную перехожу к dummy_script и нажимаю на кнопку, то все работает нормально, т.е. всплывает окно предупреждения и отображается "Hello World OK !".

Если я запускаю скрипт с помощью pytest и selenium, код выполняется, но ответ, возвращенный из fetch, - ko.

Я понял, что когда я использую Selenium, функция getCookie возвращает csrftoken = null. В результате, невозможно получить данные в django view.

Чтобы заставить его работать с Selenium только в тестовом режиме, я добавил следующий условный декоратор перед 'another_dummy_view'.

views.py


def conditional_decorator(dec, condition):
    def decorator(func):
        if not condition:
            # Return the function unchanged, not decorated.
            return func
        return dec(func)
    return decorator


def dev_dummy_script(request):
    context = {}
    return render(request, "dummy_2.html", context)


@conditional_decorator(csrf_exempt, DEBUG)
def another_dummy_view(request):
    if request.method == "POST":
        post_data = json.loads(request.body.decode("utf-8"))
        posted_value = post_data["dummyValue"]
        value_to_return = posted_value + " " + "OK !"
        response = JsonResponse({"returnedValue": value_to_return})
        return response
Вернуться на верх