3446026555
The license.txt file has been replaced with LICENSE for quite a while now. INAL but it didn't seem accurate to say "hey, checkout license.txt although there's no such file". Apart from this, there were inconsistencies in the headers altogether...this change brings consistency.
125 lines
3.3 KiB
Python
125 lines
3.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright (c) 2019, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# License: MIT. See LICENSE
|
|
|
|
import unittest
|
|
import frappe
|
|
import frappe.recorder
|
|
from frappe.utils import set_request
|
|
from frappe.website.serve import get_response_content
|
|
|
|
import sqlparse
|
|
|
|
class TestRecorder(unittest.TestCase):
|
|
def setUp(self):
|
|
frappe.recorder.stop()
|
|
frappe.recorder.delete()
|
|
set_request()
|
|
frappe.recorder.start()
|
|
frappe.recorder.record()
|
|
|
|
def test_start(self):
|
|
frappe.recorder.dump()
|
|
requests = frappe.recorder.get()
|
|
self.assertEqual(len(requests), 1)
|
|
|
|
def test_do_not_record(self):
|
|
frappe.recorder.do_not_record(frappe.get_all)('DocType')
|
|
frappe.recorder.dump()
|
|
requests = frappe.recorder.get()
|
|
self.assertEqual(len(requests), 0)
|
|
|
|
def test_get(self):
|
|
frappe.recorder.dump()
|
|
|
|
requests = frappe.recorder.get()
|
|
self.assertEqual(len(requests), 1)
|
|
|
|
request = frappe.recorder.get(requests[0]['uuid'])
|
|
self.assertTrue(request)
|
|
|
|
def test_delete(self):
|
|
frappe.recorder.dump()
|
|
|
|
requests = frappe.recorder.get()
|
|
self.assertEqual(len(requests), 1)
|
|
|
|
frappe.recorder.delete()
|
|
|
|
requests = frappe.recorder.get()
|
|
self.assertEqual(len(requests), 0)
|
|
|
|
def test_record_without_sql_queries(self):
|
|
frappe.recorder.dump()
|
|
|
|
requests = frappe.recorder.get()
|
|
request = frappe.recorder.get(requests[0]['uuid'])
|
|
|
|
self.assertEqual(len(request['calls']), 0)
|
|
|
|
def test_record_with_sql_queries(self):
|
|
frappe.get_all('DocType')
|
|
frappe.recorder.dump()
|
|
|
|
requests = frappe.recorder.get()
|
|
request = frappe.recorder.get(requests[0]['uuid'])
|
|
|
|
self.assertNotEqual(len(request['calls']), 0)
|
|
|
|
def test_explain(self):
|
|
frappe.db.sql('SELECT * FROM tabDocType')
|
|
frappe.db.sql('COMMIT')
|
|
frappe.recorder.dump()
|
|
|
|
requests = frappe.recorder.get()
|
|
request = frappe.recorder.get(requests[0]['uuid'])
|
|
|
|
self.assertEqual(len(request['calls'][0]['explain_result']), 1)
|
|
self.assertEqual(len(request['calls'][1]['explain_result']), 0)
|
|
|
|
|
|
def test_multiple_queries(self):
|
|
queries = [
|
|
{'mariadb': 'SELECT * FROM tabDocType', 'postgres': 'SELECT * FROM "tabDocType"'},
|
|
{'mariadb': 'SELECT COUNT(*) FROM tabDocType', 'postgres': 'SELECT COUNT(*) FROM "tabDocType"'},
|
|
{'mariadb': 'COMMIT', 'postgres': 'COMMIT'},
|
|
]
|
|
|
|
sql_dialect = frappe.db.db_type or 'mariadb'
|
|
for query in queries:
|
|
frappe.db.sql(query[sql_dialect])
|
|
|
|
frappe.recorder.dump()
|
|
|
|
requests = frappe.recorder.get()
|
|
request = frappe.recorder.get(requests[0]['uuid'])
|
|
|
|
self.assertEqual(len(request['calls']), len(queries))
|
|
|
|
for query, call in zip(queries, request['calls']):
|
|
self.assertEqual(call['query'], sqlparse.format(query[sql_dialect].strip(), keyword_case='upper', reindent=True))
|
|
|
|
def test_duplicate_queries(self):
|
|
queries = [
|
|
('SELECT * FROM tabDocType', 2),
|
|
('SELECT COUNT(*) FROM tabDocType', 1),
|
|
('select * from tabDocType', 2),
|
|
('COMMIT', 3),
|
|
('COMMIT', 3),
|
|
('COMMIT', 3),
|
|
]
|
|
for query in queries:
|
|
frappe.db.sql(query[0])
|
|
|
|
frappe.recorder.dump()
|
|
|
|
requests = frappe.recorder.get()
|
|
request = frappe.recorder.get(requests[0]['uuid'])
|
|
|
|
for query, call in zip(queries, request['calls']):
|
|
self.assertEqual(call['exact_copies'], query[1])
|
|
|
|
def test_error_page_rendering(self):
|
|
content = get_response_content("error")
|
|
self.assertIn("Error", content)
|