From 89cd1bdcf4a814051ca9f345861aff3fb645cd9e Mon Sep 17 00:00:00 2001
From: Jean Privat <jean@pryen.org>
Date: Mon, 15 Dec 2014 15:33:39 -0500
Subject: [PATCH] nitunit: enable the execution of simple docunit in a single
 program

Signed-off-by: Jean Privat <jean@pryen.org>
---
 src/testing/testing_doc.nit |   97 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 97 insertions(+)

diff --git a/src/testing/testing_doc.nit b/src/testing/testing_doc.nit
index 2d04a9c..2bdbb02 100644
--- a/src/testing/testing_doc.nit
+++ b/src/testing/testing_doc.nit
@@ -116,12 +116,109 @@ class NitUnitExecutor
 	# Execute all the docunits
 	fun run_tests
 	do
+		var simple_du = new Array[DocUnit]
 		for du in docunits do
+			var ast = toolcontext.parse_something(du.block)
+			if ast isa AExpr then
+				simple_du.add du
+			else
 				test_single_docunit(du)
+			end
+		end
+
+		test_simple_docunits(simple_du)
+	end
+
+	# Executes multiples doc-units in a shared program.
+	# Used for docunits simple block of code (without modules, classes, functions etc.)
+	#
+	# In case of compilation error, the method fallbacks to `test_single_docunit` to
+	# * locate exactly the compilation problem in the problematic docunit.
+	# * permit the execution of the other docunits that may be correct.
+	fun test_simple_docunits(dus: Array[DocUnit])
+	do
+		if dus.is_empty then return
+
+		var file = "{prefix}-0.nit"
+
+		var dir = file.dirname
+		if dir != "" then dir.mkdir
+		var f
+		f = new OFStream.open(file)
+		f.write("# GENERATED FILE\n")
+		f.write("# Docunits extracted from comments\n")
+		f.write("import {mmodule.name}\n")
+		f.write("\n")
+		var i = 0
+		for du in dus do
+
+			i += 1
+			f.write("fun run_{i} do\n")
+			f.write("# {du.testcase.attrs["name"]}\n")
+			f.write(du.block)
+			f.write("end\n")
+		end
+		f.write("var a = args.first.to_i\n")
+		for j in [1..i[ do
+			f.write("if a == {j} then run_{j}\n")
+		end
+		f.close
+
+		if toolcontext.opt_noact.value then return
+
+		var nit_dir = toolcontext.nit_dir
+		var nitg = nit_dir/"bin/nitg"
+		if not nitg.file_exists then
+			toolcontext.error(null, "Cannot find nitg. Set envvar NIT_DIR.")
+			toolcontext.check_errors
+		end
+		var cmd = "{nitg} --ignore-visibility --no-color '{file}' -I {mmodule.location.file.filename.dirname} >'{file}.out1' 2>&1 </dev/null -o '{file}.bin'"
+		var res = sys.system(cmd)
+
+		if res != 0 then
+			# Compilation error.
+			# Fall-back to individual modes:
+			for du in dus do
+				test_single_docunit(du)
+			end
+			return
+		end
+
+		i = 0
+		for du in dus do
+			var tc = du.testcase
+			toolcontext.modelbuilder.unit_entities += 1
+			i += 1
+			toolcontext.info("Execute doc-unit {du.testcase.attrs["name"]} in {file} {i}", 1)
+			var res2 = sys.system("{file.to_program_name}.bin {i} >>'{file}.out1' 2>&1 </dev/null")
+
+			var msg
+			f = new IFStream.open("{file}.out1")
+			var n2
+			n2 = new HTMLTag("system-err")
+			tc.add n2
+			msg = f.read_all
+			f.close
+
+			n2 = new HTMLTag("system-out")
+			tc.add n2
+			n2.append(du.block)
+
+			if res2 != 0 then
+				var ne = new HTMLTag("error")
+				ne.attr("message", msg)
+				tc.add ne
+				toolcontext.warning(du.ndoc.location, "error", "ERROR: {tc.attrs["classname"]}.{tc.attrs["name"]} (in {file}): {msg}")
+				toolcontext.modelbuilder.failed_entities += 1
+			end
+			toolcontext.check_errors
+
+			testsuite.add(tc)
 		end
 	end
 
 	# Executes a single doc-unit in its own program.
+	# Used for docunits larger than a single block of code (with modules, classes, functions etc.)
 	fun test_single_docunit(du: DocUnit)
 	do
 		var tc = du.testcase
-- 
1.7.9.5