71 lines
No EOL
2.4 KiB
Python
71 lines
No EOL
2.4 KiB
Python
import unittest
|
|
import frappe
|
|
|
|
from frappe.core.utils import find
|
|
from frappe.custom.doctype.property_setter.property_setter import make_property_setter
|
|
|
|
|
|
class TestDBUpdate(unittest.TestCase):
|
|
def test_db_update(self):
|
|
doctype = 'User'
|
|
frappe.reload_doctype('User', force=True)
|
|
frappe.model.meta.trim_tables('User')
|
|
make_property_setter(doctype, 'bio', 'fieldtype', 'Text', 'Data')
|
|
make_property_setter(doctype, 'middle_name', 'fieldtype', 'Data', 'Text')
|
|
make_property_setter(doctype, 'enabled', 'default', '1', 'Int')
|
|
|
|
frappe.db.updatedb(doctype)
|
|
|
|
field_defs = get_field_defs(doctype)
|
|
table_columns = frappe.db.get_table_columns_description('tab{}'.format(doctype))
|
|
|
|
self.assertEqual(len(field_defs), len(table_columns))
|
|
|
|
for field_def in field_defs:
|
|
fieldname = field_def.get('fieldname')
|
|
table_column = find(table_columns, lambda d: d.get('name') == fieldname)
|
|
|
|
fieldtype = get_fieldtype_from_def(field_def)
|
|
|
|
fallback_default = '0' if field_def.get('fieldtype') in frappe.model.numeric_fieldtypes else 'NULL'
|
|
default = field_def.default if field_def.default is not None else fallback_default
|
|
|
|
self.assertEqual(fieldtype, table_column.type)
|
|
self.assertIn(table_column.default or 'NULL', [default, "'{}'".format(default)])
|
|
|
|
def get_fieldtype_from_def(field_def):
|
|
fieldtuple = frappe.db.type_map.get(field_def.fieldtype, ('', 0))
|
|
fieldtype = fieldtuple[0]
|
|
if fieldtype in ('varchar', 'datetime', 'int'):
|
|
fieldtype += '({})'.format(field_def.length or fieldtuple[1])
|
|
return fieldtype
|
|
|
|
def get_field_defs(doctype):
|
|
meta = frappe.get_meta(doctype, cached=False)
|
|
field_defs = meta.get_fieldnames_with_value(True)
|
|
field_defs += get_other_fields_meta(meta)
|
|
return field_defs
|
|
|
|
def get_other_fields_meta(meta):
|
|
default_fields_map = {
|
|
'name': ('Data', 0),
|
|
'owner': ('Data', 0),
|
|
'parent': ('Data', 0),
|
|
'parentfield': ('Data', 0),
|
|
'modified_by': ('Data', 0),
|
|
'parenttype': ('Data', 0),
|
|
'creation': ('Datetime', 0),
|
|
'modified': ('Datetime', 0),
|
|
'idx': ('Int', 8),
|
|
'docstatus': ('Check', 0)
|
|
}
|
|
|
|
optional_fields = frappe.db.OPTIONAL_COLUMNS
|
|
if meta.track_seen:
|
|
optional_fields.append('_seen')
|
|
|
|
optional_fields_map = {field: ('Text', 0) for field in optional_fields}
|
|
fields = dict(default_fields_map, **optional_fields_map)
|
|
field_map = [frappe._dict({'fieldname': field, 'fieldtype': _type, 'length': _length}) for field, (_type, _length) in fields.items()]
|
|
|
|
return field_map |