Browse Source

[fix] for test

version-14
Rushabh Mehta 8 years ago
parent
commit
79f51648a1
3 changed files with 15 additions and 7 deletions
  1. +9
    -4
      frappe/core/doctype/user/test_user.py
  2. +1
    -1
      frappe/utils/scheduler.py
  3. +5
    -2
      frappe/website/render.py

+ 9
- 4
frappe/core/doctype/user/test_user.py View File

@@ -8,9 +8,8 @@ import requests
from frappe.model.delete_doc import delete_doc from frappe.model.delete_doc import delete_doc
from frappe.utils.data import today, add_to_date from frappe.utils.data import today, add_to_date
from frappe import _dict from frappe import _dict
from frappe.limits import SiteExpiredError, update_limits, clear_limit
from frappe.limits import update_limits, clear_limit
from frappe.utils import get_url from frappe.utils import get_url
from frappe.installer import update_site_config
from frappe.core.doctype.user.user import MaxUsersReachedError from frappe.core.doctype.user.user import MaxUsersReachedError


test_records = frappe.get_test_records('User') test_records = frappe.get_test_records('User')
@@ -145,6 +144,7 @@ class TestUser(unittest.TestCase):
clear_limit('users') clear_limit('users')


# def test_deny_multiple_sessions(self): # def test_deny_multiple_sessions(self):
# from frappe.installer import update_site_config
# clear_limit('users') # clear_limit('users')
# #
# # allow one session # # allow one session
@@ -179,13 +179,18 @@ class TestUser(unittest.TestCase):
# test_request(conn1) # test_request(conn1)


def test_site_expiry(self): def test_site_expiry(self):
user = frappe.get_doc('User', 'test@example.com')
user.enabled = 1
user.new_password = 'testpassword'
user.save()

update_limits({'expiry': add_to_date(today(), days=-1)}) update_limits({'expiry': add_to_date(today(), days=-1)})
frappe.local.conf = _dict(frappe.get_site_config()) frappe.local.conf = _dict(frappe.get_site_config())


frappe.db.commit() frappe.db.commit()


res = requests.post(get_url(), params={'cmd': 'login', 'usr': 'test@example.com', 'pwd': 'testpassword',
'device': 'desktop'})
res = requests.post(get_url(), params={'cmd': 'login', 'usr':
'test@example.com', 'pwd': 'testpassword', 'device': 'desktop'})


# While site is expired status code returned is 417 Failed Expectation # While site is expired status code returned is 417 Failed Expectation
self.assertEqual(res.status_code, 417) self.assertEqual(res.status_code, 417)


+ 1
- 1
frappe/utils/scheduler.py View File

@@ -255,7 +255,7 @@ def reset_enabled_scheduler_events(login_manager):
frappe.db.set_global('enabled_scheduler_events', None) frappe.db.set_global('enabled_scheduler_events', None)
except MySQLdb.OperationalError, e: except MySQLdb.OperationalError, e:
if e.args[0]==1205: if e.args[0]==1205:
frappe.get_logger().error("Error in reset_enabled_scheduler_events")
frappe.log_error(frappe.get_traceback(), "Error in reset_enabled_scheduler_events")
else: else:
raise raise
else: else:


+ 5
- 2
frappe/website/render.py View File

@@ -244,5 +244,8 @@ def get_doctype_from_path(path):
return None, None return None, None


def add_csrf_token(data): def add_csrf_token(data):
return data.replace("<!-- csrf_token -->", '<script>frappe.csrf_token = "{0}";</script>'.format(
frappe.local.session.data.csrf_token))
if frappe.local.session:
return data.replace("<!-- csrf_token -->", '<script>frappe.csrf_token = "{0}";</script>'.format(
frappe.local.session.data.csrf_token))
else:
return data

Loading…
Cancel
Save