audrius/reporting-engine
1
0
Fork 1
mirror of https://github.com/OCA/reporting-engine.git synced 2025-02-16 16:30:38 +02:00
Code Issues Packages Projects Releases Wiki Activity

[IMP] sql_export: black, isort, prettier

Browse Source
This commit is contained in:
Florian da Costa
2021-03-22 16:15:37 +01:00
committed by Sander Lienaerts
parent ad918b8386
commit 89ab0494e4
9 changed files with 306 additions and 205 deletions
Download Patch File Download Diff File Expand all files Collapse all files

60
sql_export/tests/test_sql_query.py
View File

@@ -3,28 +3,30 @@
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
import base64
from odoo.tests.common import TransactionCase, post_install
from odoo.exceptions import UserError
from odoo import fields
from odoo.exceptions import UserError
from odoo.tests.common import TransactionCase, post_install
@post_install(True)
class TestExportSqlQuery(TransactionCase):
def setUp(self):
super(TestExportSqlQuery, self).setUp()
self.sql_export_obj = self.env['sql.export']
self.wizard_obj = self.env['sql.file.wizard']
self.sql_report_demo = self.env.ref('sql_export.sql_export_partner')
self.sql_export_obj = self.env["sql.export"]
self.wizard_obj = self.env["sql.file.wizard"]
self.sql_report_demo = self.env.ref("sql_export.sql_export_partner")
def test_sql_query(self):
wizard = self.wizard_obj.create({
'sql_export_id': self.sql_report_demo.id,
})
wizard = self.wizard_obj.create(
{
"sql_export_id": self.sql_report_demo.id,
}
)
wizard.export_sql()
export = base64.b64decode(wizard.binary_file).decode('utf-8')
self.assertEqual(export.split(';')[0], 'name')
self.assertTrue(len(export.split(';')) > 6)
export = base64.b64decode(wizard.binary_file).decode("utf-8")
self.assertEqual(export.split(";")[0], "name")
self.assertTrue(len(export.split(";")) > 6)
def test_prohibited_queries(self):
prohibited_queries = [
@@ -39,9 +41,9 @@ class TestExportSqlQuery(TransactionCase):
]
for query in prohibited_queries:
with self.assertRaises(UserError):
sql_export = self.sql_export_obj.create({
'name': 'test_prohibited',
'query': query})
sql_export = self.sql_export_obj.create(
{"name": "test_prohibited", "query": query}
)
sql_export.button_validate_sql_expression()
def test_authorized_queries(self):
@@ -50,23 +52,25 @@ class TestExportSqlQuery(TransactionCase):
]
for query in authorized_queries:
sql_export = self.sql_export_obj.create({
'name': 'test_authorized',
'query': query})
sql_export = self.sql_export_obj.create(
{"name": "test_authorized", "query": query}
)
sql_export.button_validate_sql_expression()
self.assertEqual(
sql_export.state, 'sql_valid',
"%s is a valid request" % (query))
sql_export.state, "sql_valid", "%s is a valid request" % (query)
)
def test_sql_query_with_params(self):
query = self.env.ref('sql_export.sql_export_partner_with_variables')
categ_id = self.env.ref('base.res_partner_category_0').id
wizard = self.wizard_obj.create({
'sql_export_id': query.id,
'x_date': fields.Date.today(),
'x_id': 1,
'x_partner_categ_ids': [(6, 0, [categ_id])]
})
query = self.env.ref("sql_export.sql_export_partner_with_variables")
categ_id = self.env.ref("base.res_partner_category_0").id
wizard = self.wizard_obj.create(
{
"sql_export_id": query.id,
"x_date": fields.Date.today(),
"x_id": 1,
"x_partner_categ_ids": [(6, 0, [categ_id])],
}
)
wizard.export_sql()
export = base64.b64decode(wizard.binary_file)
self.assertTrue(export)