vassili/seitime-frappe
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

feat: add tests for JSON functions

Browse source
This commit is contained in:
Saqib Ansari 2026-04-23 10:29:58 +05:30
parent 11efbde9b9
commit 9d3f410037
1 changed files with 75 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

75
frappe/tests/test_query_builder.py
View file

@ -14,6 +14,9 @@ from frappe.query_builder.functions import (
CombineDatetime,
Date,
GroupConcat,
JSONContains,
JSONExtract,
JSONValue,
Match,
Round,
Truncate,
@ -176,6 +179,43 @@ class TestCustomFunctionsMariaDB(IntegrationTestCase):
query = frappe.qb.from_(note).select(Truncate(note.price, 3))
self.assertEqual("select truncate(`price`,3) from `tabnote`", str(query).lower())
def test_json_extract(self):
note = frappe.qb.DocType("Note")
# Simple get_sql
self.assertEqual("JSON_EXTRACT(content,'$.key')", JSONExtract(note.content, "$.key").get_sql())
# In a SELECT query
query = frappe.qb.from_(note).select(JSONExtract(note.content, "$.key"))
self.assertIn("json_extract(`content`,'$.key')", str(query).lower())
# In a WHERE clause
query = frappe.qb.from_(note).select(note.name).where(JSONExtract(note.content, "$.key") == "value")
self.assertIn("json_extract(`content`,'$.key')='value'", str(query).lower())
def test_json_value(self):
note = frappe.qb.DocType("Note")
# Simple get_sql
self.assertEqual(
"JSON_UNQUOTE(JSON_EXTRACT(content,'$.key'))", JSONValue(note.content, "$.key").get_sql()
)
# In a SELECT query
query = frappe.qb.from_(note).select(JSONValue(note.content, "$.key"))
self.assertIn("json_unquote(json_extract(`content`,'$.key'))", str(query).lower())
# In a WHERE clause
query = frappe.qb.from_(note).select(note.name).where(JSONValue(note.content, "$.key") == "value")
self.assertIn("json_unquote(json_extract(`content`,'$.key'))='value'", str(query).lower())
def test_json_contains(self):
note = frappe.qb.DocType("Note")
# With a plain string candidate (auto-wrapped as JSON)
self.assertEqual("JSON_CONTAINS(content,'\"value\"')", JSONContains(note.content, "value").get_sql())
# In a WHERE clause
query = frappe.qb.from_(note).select(note.name).where(JSONContains(note.content, "admin"))
self.assertIn("json_contains(`content`,'\"admin\"')", str(query).lower())
@run_only_if(db_type_is.POSTGRES)
class TestCustomFunctionsPostgres(IntegrationTestCase):
@ -313,6 +353,41 @@ class TestCustomFunctionsPostgres(IntegrationTestCase):
query = frappe.qb.from_(note).select(Truncate(note.price, 3))
self.assertEqual('select truncate("price",3) from "tabnote"', str(query).lower())
def test_json_extract(self):
note = frappe.qb.DocType("Note")
# Simple get_sql
self.assertEqual("\"content\"->'$.key'", JSONExtract(note.content, "$.key").get_sql())
# In a SELECT query
query = frappe.qb.from_(note).select(JSONExtract(note.content, "$.key"))
self.assertIn("\"content\"->'$.key'", str(query))
# In a WHERE clause
query = frappe.qb.from_(note).select(note.name).where(JSONExtract(note.content, "$.key") == "value")
self.assertIn("\"content\"->'$.key'='value'", str(query))
def test_json_value(self):
note = frappe.qb.DocType("Note")
# Simple get_sql
self.assertEqual("\"content\"->>'$.key'", JSONValue(note.content, "$.key").get_sql())
# In a SELECT query
query = frappe.qb.from_(note).select(JSONValue(note.content, "$.key"))
self.assertIn("\"content\"->>'$.key'", str(query))
# In a WHERE clause
query = frappe.qb.from_(note).select(note.name).where(JSONValue(note.content, "$.key") == "value")
self.assertIn("\"content\"->>'$.key'='value'", str(query))
def test_json_contains(self):
note = frappe.qb.DocType("Note")
# With a plain string candidate
self.assertEqual("\"content\"@>'admin'", JSONContains(note.content, "admin").get_sql())
# In a WHERE clause
query = frappe.qb.from_(note).select(note.name).where(JSONContains(note.content, "admin"))
self.assertIn("\"content\"@>'admin'", str(query))
class TestBuilderBase:
def test_adding_tabs(self):