mirror of
https://github.com/peeringdb/peeringdb.git
synced 2024-05-11 05:55:09 +00:00
85 lines
2.5 KiB
Python
85 lines
2.5 KiB
Python
import pytest
|
|
import json
|
|
|
|
from django.test import Client, TestCase, RequestFactory
|
|
from django.contrib.auth.models import Group
|
|
|
|
import peeringdb_server.models as models
|
|
|
|
#from django.template import Context, Template
|
|
#from django.utils import translation
|
|
|
|
|
|
class UserLocaleTests(TestCase):
|
|
"""
|
|
Test peeringdb_server.models.User functions
|
|
"""
|
|
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
user_group = Group.objects.create(name="user")
|
|
for name in ["user_undef", "user_en", "user_pt"]:
|
|
setattr(cls, name,
|
|
models.User.objects.create_user(
|
|
name, "%s@localhost" % name, first_name=name,
|
|
last_name=name, password=name))
|
|
|
|
cls.user_en.set_locale('en')
|
|
cls.user_pt.set_locale('pt')
|
|
|
|
user_group.user_set.add(cls.user_en)
|
|
user_group.user_set.add(cls.user_pt)
|
|
user_group.user_set.add(cls.user_undef)
|
|
|
|
cls.user_undef.save()
|
|
cls.user_en.save()
|
|
cls.user_pt.save()
|
|
|
|
def setUp(self):
|
|
self.factory = RequestFactory()
|
|
|
|
def test_user_locale(self):
|
|
"""
|
|
Tests if user profile page has the right language
|
|
|
|
Note: Don't use Client.login(...) since it will miss language setting in the session
|
|
"""
|
|
|
|
#t = Template("{% load i18n %}{% get_current_language as LANGUAGE_CODE %}{{ LANGUAGE_CODE }}")
|
|
#print(t.render(Context({})))
|
|
#translation.activate('pt')
|
|
#print(t.render(Context({})))
|
|
#u_pt = models.User.objects.get(username="user_pt")
|
|
#print(u_pt.get_locale())
|
|
|
|
c = Client()
|
|
resp = c.get("/profile", follow=True)
|
|
data = {
|
|
"next": "/profile",
|
|
"username": "user_en",
|
|
"password": "user_en"
|
|
}
|
|
resp = c.post("/auth", data, follow=True)
|
|
self.assertGreater(
|
|
resp.content.find('<!-- Current language: en -->'), -1)
|
|
|
|
c.logout()
|
|
data = {
|
|
"next": "/profile",
|
|
"username": "user_pt",
|
|
"password": "user_pt"
|
|
}
|
|
resp = c.post("/auth", data, follow=True)
|
|
self.assertGreater(
|
|
resp.content.find('<!-- Current language: pt -->'), -1)
|
|
|
|
c.logout()
|
|
data = {
|
|
"next": "/profile",
|
|
"username": "user_undef",
|
|
"password": "user_undef"
|
|
}
|
|
resp = c.post("/auth", data, follow=True)
|
|
self.assertGreater(
|
|
resp.content.find('<!-- Current language: en -->'), -1)
|