83 lines
2.4 KiB
Python
83 lines
2.4 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import sys
|
|
if __name__=="__main__":
|
|
sys.path.extend([".", "app", "lib"])
|
|
|
|
import webnotes
|
|
from webnotes.model.meta import get_link_fields
|
|
from webnotes.model.code import load_doctype_module
|
|
|
|
def make_test_records(doctype, verbose=0):
|
|
webnotes.mute_emails = True
|
|
if not webnotes.conn:
|
|
webnotes.connect()
|
|
|
|
# also include doctype itself
|
|
options_list = list(set([options for fieldname, options, label
|
|
in get_link_fields(doctype)] + [doctype]))
|
|
|
|
for options in options_list:
|
|
if options.startswith("link:"):
|
|
options = options[5:]
|
|
if options == "[Select]":
|
|
continue
|
|
|
|
if options not in webnotes.test_objects:
|
|
webnotes.test_objects[options] = []
|
|
make_test_records(options, verbose)
|
|
|
|
load_module_and_make_records(options, verbose)
|
|
|
|
def load_module_and_make_records(options, verbose=0):
|
|
module = webnotes.conn.get_value("DocType", options, "module")
|
|
|
|
# get methods for either [doctype].py or test_[doctype].py
|
|
doctype_module = load_doctype_module(options, module)
|
|
test_module = load_doctype_module(options, module, "test_")
|
|
|
|
if hasattr(test_module, "make_test_records"):
|
|
webnotes.test_objects[options] += test_module.make_test_records(verbose)
|
|
|
|
elif hasattr(doctype_module, "make_test_records"):
|
|
webnotes.test_objects[options] += doctype_module.make_test_records(verbose)
|
|
|
|
elif hasattr(test_module, "test_records"):
|
|
webnotes.test_objects[options] += make_test_objects(test_module)
|
|
|
|
elif hasattr(doctype_module, "test_records"):
|
|
webnotes.test_objects[options] += make_test_objects(doctype_module)
|
|
|
|
elif verbose:
|
|
print_mandatory_fields(options)
|
|
|
|
def make_test_objects(obj):
|
|
if isinstance(obj, list):
|
|
test_records = obj
|
|
else:
|
|
# obj is a module object
|
|
test_records = obj.test_records
|
|
|
|
records = []
|
|
for doclist in test_records:
|
|
d = webnotes.model_wrapper(doclist)
|
|
if webnotes.test_objects.get(d.doc.doctype):
|
|
# do not create test records, if already exists
|
|
return []
|
|
d.insert()
|
|
records.append(d.doc.name)
|
|
return records
|
|
|
|
def print_mandatory_fields(doctype):
|
|
print "Please setup make_test_records for: " + doctype
|
|
print "-" * 60
|
|
doctype_obj = webnotes.get_doctype(doctype)
|
|
print "Autoname: " + (doctype_obj[0].autoname or "")
|
|
print "Mandatory Fields: "
|
|
for d in doctype_obj.get({"reqd":1}):
|
|
print d.parent + ":" + d.fieldname + " | " + d.fieldtype + " | " + (d.options or "")
|
|
print
|
|
|
|
if __name__=="__main__":
|
|
make_test_records(sys.argv[1], verbose=1)
|
|
|