X-Git-Url: http://nitlanguage.org

diff --git a/src/testing/testing_base.nit b/src/testing/testing_base.nit
index 243a51e..414de9a 100644
--- a/src/testing/testing_base.nit
+++ b/src/testing/testing_base.nit
@@ -15,8 +15,8 @@
 # Base options for testing tools.
 module testing_base
 
-import modelize_property
-import parser_util
+import modelize
+private import parser_util
 
 redef class ToolContext
 	# opt --full
@@ -27,6 +27,8 @@ redef class ToolContext
 	var opt_dir = new OptionString("Working directory (default is '.nitunit')", "--dir")
 	# opt --no-act
 	var opt_noact = new OptionBool("Does not compile and run tests", "--no-act")
+	# opt --nitc
+	var opt_nitc = new OptionString("nitc compiler to use", "--nitc")
 
 	# Working directory for testing.
 	fun test_dir: String do
@@ -34,4 +36,96 @@ redef class ToolContext
 		if dir == null then return ".nitunit"
 		return dir
 	end
+
+	# Search the `nitc` compiler to use
+	#
+	# If not `nitc` is suitable, then prints an error and quit.
+	fun find_nitc: String
+	do
+		var nitc = opt_nitc.value
+		if nitc != null then
+			if not nitc.file_exists then
+				fatal_error(null, "error: cannot find `{nitc}` given by --nitc.")
+				abort
+			end
+			return nitc
+		end
+
+		nitc = "NITC".environ
+		if nitc != "" then
+			if not nitc.file_exists then
+				fatal_error(null, "error: cannot find `{nitc}` given by NITC.")
+				abort
+			end
+			return nitc
+		end
+
+		var nit_dir = nit_dir
+		nitc = nit_dir/"bin/nitc"
+		if not nitc.file_exists then
+			fatal_error(null, "Error: cannot find nitc. Set envvar NIT_DIR or NITC or use the --nitc option.")
+			abort
+		end
+		return nitc
+	end
+
+	# Execute a system command in a more safe context than `Sys::system`.
+	fun safe_exec(command: String): Int
+	do
+		info(command, 2)
+		var real_command = """
+bash -c "
+ulimit -f {{{ulimit_file}}} 2> /dev/null
+ulimit -t {{{ulimit_usertime}}} 2> /dev/null
+{{{command}}}
+"
+"""
+		return system(real_command)
+	end
+
+	# The maximum size (in KB) of files written by a command executed trough `safe_exec`
+	#
+	# Default: 64MB
+	var ulimit_file = 65536 is writable
+
+	# The maximum amount of cpu time (in seconds) for a command executed trough `safe_exec`
+	#
+	# Default: 10 CPU minute
+	var ulimit_usertime = 600 is writable
+end
+
+redef class String
+	# If needed, truncate `self` at `max_length` characters and append an informative `message`.
+	#
+	# ~~~
+	# assert "hello".trunc(10) == "hello"
+	# assert "hello".trunc(2) == "he[truncated. Full size is 5]"
+	# assert "hello".trunc(2, "...") == "he..."
+	# ~~~
+	fun trunc(max_length: Int, message: nullable String): String
+	do
+		if length <= max_length then return self
+		if message == null then message = "[truncated. Full size is {length}]"
+		return substring(0, max_length) + message
+	end
+
+	# Use a special notation for whitespace characters that are not `'\n'` (LFD) or `' '` (space).
+	#
+	# ~~~
+	# assert "hello".filter_nonprintable == "hello"
+	# assert "\r\n\t".filter_nonprintable == "^13\n^9"
+	# ~~~
+	fun filter_nonprintable: String
+	do
+		var buf = new Buffer
+		for c in self do
+			var cp = c.code_point
+			if cp < 32 and c != '\n' then
+				buf.append "^{cp}"
+			else
+				buf.add c
+			end
+		end
+		return buf.to_s
+	end
 end