1 # This file is part of NIT ( http://www.nitlanguage.org ).
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
7 # http://www.apache.org/licenses/LICENSE-2.0
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
15 # Program to extract and execute unit tests from nit source files
18 import modelize_property
19 intrude import markdown
22 # Extractor, Executor an Reporter for the tests in a module
26 # The name of the module to import
29 # The prefix of the generated Nit source-file
32 # The XML node associated to the module
33 var testsuite
: HTMLTag
36 init(toolcontext
: ToolContext, prefix
: String, modname
: String, testsuite
: HTMLTag)
40 self.modname
= modname
41 self.testsuite
= testsuite
44 # All blocks of code from a same `ADoc`
45 var block
= new Array[String]
47 redef fun process_code
(n
: HTMLTag, text
: String)
50 var ast
= toolcontext
.parse_something
(text
)
52 # We want executable code
53 if not (ast
isa AModule or ast
isa ABlockExpr or ast
isa AExpr) then return
55 # Search `assert` in the AST
56 var v
= new SearchAssertVisitor
58 if not v
.foundit
then return
64 # used to generate distinct names
67 # The entry point for a new `ndoc` node
68 # Fill the prepated `tc` (testcase) XTM node
69 fun extract
(ndoc
: ADoc, tc
: HTMLTag)
75 if block
.is_empty
then return
78 var file
= "{prefix}{cpt}.nit"
80 toolcontext
.info
("Execute {tc.attrs["classname"]}.{tc.attrs["name"]} in {file}", 2)
82 var dir
= file
.dirname
83 if dir
!= "" then dir
.mkdir
85 f
= new OFStream.open
(file
)
86 f
.write
("# GENERATED FILE\n")
87 f
.write
("# Example extracted from a documentation\n")
88 var modname
= self.modname
89 f
.write
("import {modname}\n")
96 var cmd
= "../bin/nitg --no-color '{file}' -I . >'{file}.out1' 2>&1 </dev/null -o '{file}.bin'"
97 var res
= sys
.system
(cmd
)
100 res2
= sys
.system
("./{file}.bin >>'{file}.out1' 2>&1 </dev/null")
104 f
= new IFStream.open
("{file}.out1")
106 n2
= new HTMLTag("system-err")
111 n2
= new HTMLTag("system-out")
113 for text
in block
do n2
.append
(text
)
117 var ne
= new HTMLTag("failure")
118 ne
.attr
("message", msg
)
120 toolcontext
.warning
(ndoc
.location
, "FAILURE: {tc.attrs["classname"]}.{tc.attrs["name"]} (in {file}): {msg}")
121 else if res2
!= 0 then
122 var ne
= new HTMLTag("error")
123 ne
.attr
("message", msg
)
125 toolcontext
.warning
(ndoc
.location
, "ERROR: {tc.attrs["classname"]}.{tc.attrs["name"]} (in {file}): {msg}")
127 toolcontext
.check_errors
133 class SearchAssertVisitor
136 redef fun visit
(node
)
140 else if node
isa AAssertExpr then
149 redef class ModelBuilder
150 fun test_markdown
(mmodule
: MModule): HTMLTag
152 var ts
= new HTMLTag("testsuite")
153 toolcontext
.info
("nitunit: {mmodule}", 2)
154 if not mmodule2nmodule
.has_key
(mmodule
) then return ts
156 var nmodule
= mmodule2nmodule
[mmodule
]
157 assert nmodule
!= null
159 # what module to import in the unit test.
160 # try to detect the main module of the project
161 # TODO do things correctly once the importation of arbitraty nested module is legal
165 o
= get_mmodule_by_name
(nmodule
, mmodule
, g
.mproject
.name
).as(not null)
168 ts
.attr
("package", mmodule
.full_name
)
170 var prefix
= toolcontext
.opt_dir
.value
171 if prefix
== null then prefix
= ".nitunit"
172 prefix
= prefix
.join_path
(mmodule
.to_s
)
173 var d2m
= new NitUnitExecutor(toolcontext
, prefix
, o
.name
, ts
)
178 var nmoduledecl
= nmodule
.n_moduledecl
179 if nmoduledecl
== null then break label x
180 var ndoc
= nmoduledecl
.n_doc
181 if ndoc
== null then break label x
182 tc
= new HTMLTag("testcase")
183 # NOTE: jenkins expects a '.' in the classname attr
184 tc
.attr
("classname", mmodule
.full_name
+ ".<module>")
185 tc
.attr
("name", "<module>")
186 d2m
.extract
(ndoc
, tc
)
188 for nclassdef
in nmodule
.n_classdefs
do
189 var mclassdef
= nclassdef
.mclassdef
.as(not null)
190 if nclassdef
isa AStdClassdef then
191 var ndoc
= nclassdef
.n_doc
193 tc
= new HTMLTag("testcase")
194 tc
.attr
("classname", mmodule
.full_name
+ "." + mclassdef
.mclass
.full_name
)
195 tc
.attr
("name", "<class>")
196 d2m
.extract
(ndoc
, tc
)
199 for npropdef
in nclassdef
.n_propdefs
do
200 var mpropdef
= npropdef
.mpropdef
.as(not null)
201 var ndoc
= npropdef
.n_doc
203 tc
= new HTMLTag("testcase")
204 tc
.attr
("classname", mmodule
.full_name
+ "." + mclassdef
.mclass
.full_name
)
205 tc
.attr
("name", mpropdef
.mproperty
.full_name
)
206 d2m
.extract
(ndoc
, tc
)
215 redef class ToolContext
216 var opt_full
= new OptionBool("Process also imported modules", "--full")
217 var opt_output
= new OptionString("Output name (default is 'nitunit.xml')", "-o", "--output")
218 var opt_dir
= new OptionString("Working directory (default is '.nitunit')", "--dir")
221 var toolcontext
= new ToolContext
223 toolcontext
.option_context
.add_option
(toolcontext
.opt_full
, toolcontext
.opt_output
, toolcontext
.opt_dir
)
226 toolcontext
.process_options
227 var args
= toolcontext
.option_context
.rest
228 if args
.is_empty
or toolcontext
.opt_help
.value
then
229 print
"usage: nitunit [options] file.nit..."
230 toolcontext
.option_context
.usage
234 var model
= new Model
235 var modelbuilder
= new ModelBuilder(model
, toolcontext
)
237 var mmodules
= modelbuilder
.parse
(args
)
238 modelbuilder
.run_phases
240 var page
= new HTMLTag("testsuites")
242 if toolcontext
.opt_full
.value
then mmodules
= model
.mmodules
245 page
.add modelbuilder
.test_markdown
(m
)
248 var file
= toolcontext
.opt_output
.value
249 if file
== null then file
= "nitunit.xml"
250 page
.write_to_file
(file
)