• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• django-qsessions-master
  • .coveragerc
  • qsessions
    • apps.py
    • middleware.py
    • backends
      • db.py
      • cached_db.py
      • __init__.py
    • models.py
    • geoip.py
    • __init__.py
    • migrations
      • 0001_initial.py
      • 0002_session_created_at.py
      • __init__.py
    • admin.py
    • management
      • commands
        • clearsessions.py
        • download_geoip_db.py
        • __init__.py
      • __init__.py
  • CHANGELOG.md
  • README.rst
  • setup.py
  • setup.cfg
  • .travis.yml
  • tests
    • settings_base.py
    • test_middleware.py
    • geoip
      • GeoIP2-City-Test.mmdb
      • GeoIP2-Country-Test.mmdb
    • test_clearsessions.py
    • urls.py
    • test_model.py
    • test_sessionstore.py
    • __init__.py
    • settings_db.py
    • settings_cached_db.py
    • test_system_checks.py
    • test_admin.py
  • .gitignore
  • LICENSE.txt
  • MANIFEST.in
  • tox.ini

import json

import pytest
from django.conf import settings
from django.contrib.auth import SESSION_KEY
from django.contrib.auth.models import User
from django.urls import reverse

from qsessions import USER_AGENT_SESSION_KEY
from qsessions.models import Session


@pytest.mark.django_db
def test_unmodified_session(client):
    client.get("/", HTTP_USER_AGENT="TestUA/1.1")
    assert settings.SESSION_COOKIE_NAME not in client.cookies


@pytest.mark.django_db
@pytest.mark.parametrize("logged_in", (False, True))
def test_modify_session(client, logged_in):
    if logged_in:
        user = User.objects.create_superuser("user", "", "secret")
        client.force_login(user)
    else:
        user = None

    client.get("/read_session/", HTTP_USER_AGENT="TestUA/1.1")
    client.get("/modify_session/", HTTP_USER_AGENT="TestUA/1.1")
    data = json.loads(client.get("/read_session/", HTTP_USER_AGENT="TestUA/1.1").content.decode("UTF-8"))
    assert data["FOO"] == "BAR"
    assert data[USER_AGENT_SESSION_KEY] == "TestUA/1.1"
    if user:
        assert str(data[SESSION_KEY]) == str(user.id)

    assert settings.SESSION_COOKIE_NAME in client.cookies
    session = Session.objects.get(pk=client.cookies[settings.SESSION_COOKIE_NAME].value)
    assert session.user_agent == "TestUA/1.1"
    assert session.ip == "127.0.0.1"
    assert session.user == user


@pytest.mark.django_db
def test_login(client):
    admin_login_url = reverse("admin:login")
    user = User.objects.create_superuser("user", "", "secret")
    response = client.post(
        admin_login_url,
        data={"username": "user", "password": "secret", "this_is_the_login_form": "1", "next": "/admin/"},
        HTTP_USER_AGENT="TestUA/1.1",
    )
    assert response.url == "/admin/"
    session = Session.objects.get(pk=client.cookies[settings.SESSION_COOKIE_NAME].value)
    assert user == session.user


@pytest.mark.django_db
def test_long_ua(client):
    client.get("/modify_session/", HTTP_USER_AGENT="a" * 500)