frappe/webnotes/test_runner.py

from __future__ import unicode_literals

import sys
if __name__=="__main__":
	sys.path.extend([".", "app", "lib"])

import webnotes
import unittest

from webnotes.model.meta import has_field
from webnotes.model.code import load_doctype_module, get_module_name
from webnotes.model.doctype import get_link_fields


def make_test_records(doctype, verbose=0):
	webnotes.mute_emails = True
	if not webnotes.conn:
		webnotes.connect()

	for options in get_dependencies(doctype):

		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)
			make_test_records_for_doctype(options, verbose)

def get_modules(doctype):
	module = webnotes.conn.get_value("DocType", doctype, "module")
	test_module = load_doctype_module(doctype, module, "test_")

	return module, test_module

def get_dependencies(doctype):
	module, test_module = get_modules(doctype)

	options_list = list(set([df.options for df in get_link_fields(doctype)] + [doctype]))
	
	if hasattr(test_module, "test_dependencies"):
		options_list += test_module.test_dependencies
	
	if hasattr(test_module, "test_ignore"):
		for doctype_name in test_module.test_ignore:
			if doctype_name in options_list:
				options_list.remove(doctype_name)
	
	return options_list

def make_test_records_for_doctype(doctype, verbose=0):
	module, test_module = get_modules(doctype)

	if hasattr(test_module, "make_test_records"):
		webnotes.test_objects[doctype] += test_module.make_test_records(verbose)

	elif hasattr(test_module, "test_records"):
		webnotes.test_objects[doctype] += make_test_objects(doctype, test_module.test_records)

	elif verbose:
		print_mandatory_fields(doctype)


def make_test_objects(doctype, test_records):		
	records = []
		
	for doclist in test_records:
		if not "doctype" in doclist[0]:
			doclist[0]["doctype"] = doctype
		d = webnotes.bean((webnotes.doclist(doclist)).copy())
		if webnotes.test_objects.get(d.doc.doctype):
			# do not create test records, if already exists
			return []
		if has_field(d.doc.doctype, "naming_series"):
			if not d.doc.naming_series:
				d.doc.naming_series = "_T-" + d.doc.doctype + "-"

		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
	
def export_doc(doctype, docname):
	import json
	doclist = []
	ignore_list = ["name", "owner", "creation", "modified", "modified_by", "idx", "naming_series",
		"parenttype", "parent", "docstatus"]
	
	make_test_records(doctype)
	meta = webnotes.get_doctype(doctype)
	
	for d in webnotes.bean(doctype, docname):
		new_doc = {}
		for key, val in d.fields.iteritems():
			if val and key not in ignore_list:
				df = meta.get_field(key, d.parent or None, d.parentfield or None)
				if df and df.fieldtype == "Link":
					val = (webnotes.test_objects.get(df.options) or [val])[0]
				elif df and df.fieldtype == "Select" and df.options and df.options.startswith("link:"):
					val = (webnotes.test_objects.get(df.options[5:]) or [val])[0]
				if not df or df.reqd == 1:
					new_doc[key] = val
		doclist.append(new_doc)
		
	print json.dumps(doclist, indent=4, sort_keys=True).replace("    ", "\t")
		

def run_unittest(doctype, verbose=False):
	module = webnotes.conn.get_value("DocType", doctype, "module")
	test_module = get_module_name(doctype, module, "test_")
	make_test_records(args.doctype[0], verbose=verbose)

	try:
		exec ('from %s import *' % test_module) in globals()		
		del sys.argv[1:]
		unittest.main()
				
	except ImportError, e:
		print "No test module."
		
def run_all_tests(verbose):
	import os, imp
	from webnotes.modules.utils import peval_doclist

	for path, folders, files in os.walk("."):
		for filename in files:
			if filename.startswith("test_") and filename.endswith(".py"):
				test_suite = unittest.TestSuite()
				if os.path.basename(os.path.dirname(path))=="doctype":
					txt_file = os.path.join(path, filename[5:].replace(".py", ".txt"))
					with open(txt_file, 'r') as f:
						doctype_doclist = peval_doclist(f.read())
					doctype = doctype_doclist[0]["name"]
					make_test_records(doctype, verbose)
					
				module = imp.load_source('test', os.path.join(path, filename))
				test_suite.addTest(unittest.TestLoader().loadTestsFromModule(module))
				unittest.TextTestRunner(verbosity=1+(verbose and 1 or 0)).run(test_suite)
	
if __name__=="__main__":
	import argparse
	
	parser = argparse.ArgumentParser(description='Run tests.')
	parser.add_argument('-d', '--doctype', nargs=1, metavar = "DOCTYPE",
		help="test for doctype")
	parser.add_argument('-v', '--verbose', default=False, action="store_true")
	parser.add_argument('-e', '--export', nargs=2, metavar="DOCTYPE DOCNAME")
	parser.add_argument('-a', '--all', default=False, action="store_true")
	parser.add_argument('-m', '--module', default=1, metavar="MODULE")

	args = parser.parse_args()
	webnotes.print_messages = args.verbose
	
	if not webnotes.conn:
		webnotes.connect()
	
	if args.doctype:
		run_unittest(args.doctype[0], verbose=args.verbose)
	elif args.all:
		run_all_tests(args.verbose)
	elif args.export:
		export_doc(args.export[0], args.export[1])
	elif args.module:
		test_suite = unittest.TestSuite()
		__import__(args.module)
		test_suite.addTest(unittest.TestLoader().loadTestsFromModule(sys.modules[args.module]))
		unittest.TextTestRunner(verbosity=1+(args.verbose and 1 or 0)).run(test_suite)