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 blocks
= new Array[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
60 # Create a first block
61 # Or create a new block for modules that are more than a main part
62 if blocks
.is_empty
or ast
isa AModule then
63 blocks
.add
(new Array[String])
70 # used to generate distinct names
73 # The entry point for a new `ndoc` node
74 # Fill the prepated `tc` (testcase) XTM node
75 fun extract
(ndoc
: ADoc, tc
: HTMLTag)
81 if blocks
.is_empty
then return
83 for block
in blocks
do test_block
(ndoc
, tc
, block
)
87 fun test_block
(ndoc
: ADoc, tc
: HTMLTag, block
: Array[String])
89 toolcontext
.modelbuilder
.unit_entities
+= 1
92 var file
= "{prefix}{cpt}.nit"
94 toolcontext
.info
("Execute {tc.attrs["classname"]}.{tc.attrs["name"]} in {file}", 1)
96 var dir
= file
.dirname
97 if dir
!= "" then dir
.mkdir
99 f
= new OFStream.open
(file
)
100 f
.write
("# GENERATED FILE\n")
101 f
.write
("# Example extracted from a documentation\n")
102 var modname
= self.modname
103 f
.write
("import {modname}\n")
110 if toolcontext
.opt_noact
.value
then return
112 var nit_dir
= toolcontext
.nit_dir
113 var nitg
= "{nit_dir}/bin/nitg"
114 if nit_dir
== null or not nitg
.file_exists
then
115 toolcontext
.error
(null, "Cannot find nitg. Set envvar NIT_DIR.")
116 toolcontext
.check_errors
118 var cmd
= "{nitg} --ignore-visibility --no-color '{file}' -I . >'{file}.out1' 2>&1 </dev/null -o '{file}.bin'"
119 var res
= sys
.system
(cmd
)
122 res2
= sys
.system
("./{file}.bin >>'{file}.out1' 2>&1 </dev/null")
126 f
= new IFStream.open
("{file}.out1")
128 n2
= new HTMLTag("system-err")
133 n2
= new HTMLTag("system-out")
135 for text
in block
do n2
.append
(text
)
139 var ne
= new HTMLTag("failure")
140 ne
.attr
("message", msg
)
142 toolcontext
.warning
(ndoc
.location
, "FAILURE: {tc.attrs["classname"]}.{tc.attrs["name"]} (in {file}): {msg}")
143 toolcontext
.modelbuilder
.failed_entities
+= 1
144 else if res2
!= 0 then
145 var ne
= new HTMLTag("error")
146 ne
.attr
("message", msg
)
148 toolcontext
.warning
(ndoc
.location
, "ERROR: {tc.attrs["classname"]}.{tc.attrs["name"]} (in {file}): {msg}")
149 toolcontext
.modelbuilder
.failed_entities
+= 1
151 toolcontext
.check_errors
157 class SearchAssertVisitor
160 redef fun visit
(node
)
164 else if node
isa AAssertExpr then
173 redef class ModelBuilder
174 var total_entities
= 0
176 var unit_entities
= 0
177 var failed_entities
= 0
179 fun test_markdown
(mmodule
: MModule): HTMLTag
181 var ts
= new HTMLTag("testsuite")
182 toolcontext
.info
("nitunit: {mmodule}", 2)
183 if not mmodule2nmodule
.has_key
(mmodule
) then return ts
185 var nmodule
= mmodule2nmodule
[mmodule
]
186 assert nmodule
!= null
188 # what module to import in the unit test.
189 # try to detect the main module of the project
190 # TODO do things correctly once the importation of arbitraty nested module is legal
194 o
= get_mmodule_by_name
(nmodule
, g
, g
.mproject
.name
).as(not null)
197 ts
.attr
("package", mmodule
.full_name
)
199 var prefix
= toolcontext
.opt_dir
.value
200 if prefix
== null then prefix
= ".nitunit"
201 prefix
= prefix
.join_path
(mmodule
.to_s
)
202 var d2m
= new NitUnitExecutor(toolcontext
, prefix
, o
.name
, ts
)
208 var nmoduledecl
= nmodule
.n_moduledecl
209 if nmoduledecl
== null then break label x
210 var ndoc
= nmoduledecl
.n_doc
211 if ndoc
== null then break label x
213 tc
= new HTMLTag("testcase")
214 # NOTE: jenkins expects a '.' in the classname attr
215 tc
.attr
("classname", mmodule
.full_name
+ ".<module>")
216 tc
.attr
("name", "<module>")
217 d2m
.extract
(ndoc
, tc
)
219 for nclassdef
in nmodule
.n_classdefs
do
220 var mclassdef
= nclassdef
.mclassdef
.as(not null)
221 if nclassdef
isa AStdClassdef then
223 var ndoc
= nclassdef
.n_doc
226 tc
= new HTMLTag("testcase")
227 tc
.attr
("classname", mmodule
.full_name
+ "." + mclassdef
.mclass
.full_name
)
228 tc
.attr
("name", "<class>")
229 d2m
.extract
(ndoc
, tc
)
232 for npropdef
in nclassdef
.n_propdefs
do
233 var mpropdef
= npropdef
.mpropdef
.as(not null)
235 var ndoc
= npropdef
.n_doc
238 tc
= new HTMLTag("testcase")
239 tc
.attr
("classname", mmodule
.full_name
+ "." + mclassdef
.mclass
.full_name
)
240 tc
.attr
("name", mpropdef
.mproperty
.full_name
)
241 d2m
.extract
(ndoc
, tc
)
250 redef class ToolContext
251 var opt_full
= new OptionBool("Process also imported modules", "--full")
252 var opt_output
= new OptionString("Output name (default is 'nitunit.xml')", "-o", "--output")
253 var opt_dir
= new OptionString("Working directory (default is '.nitunit')", "--dir")
254 var opt_noact
= new OptionBool("Does not compile and run tests", "--no-act")
257 var toolcontext
= new ToolContext
259 toolcontext
.option_context
.add_option
(toolcontext
.opt_full
, toolcontext
.opt_output
, toolcontext
.opt_dir
, toolcontext
.opt_noact
)
260 toolcontext
.tooldescription
= "Usage: nitunit [OPTION]... <file.nit>...\nExecutes the unit tests from Nit source files."
262 toolcontext
.process_options
(args
)
263 var args
= toolcontext
.option_context
.rest
265 var model
= new Model
266 var modelbuilder
= new ModelBuilder(model
, toolcontext
)
268 var mmodules
= modelbuilder
.parse
(args
)
269 modelbuilder
.run_phases
271 var page
= new HTMLTag("testsuites")
273 if toolcontext
.opt_full
.value
then mmodules
= model
.mmodules
276 page
.add modelbuilder
.test_markdown
(m
)
279 var file
= toolcontext
.opt_output
.value
280 if file
== null then file
= "nitunit.xml"
281 page
.write_to_file
(file
)
282 print
"Results saved in {file}"
284 if modelbuilder
.unit_entities
== 0 then
285 print
"No nitunits found"
286 else if modelbuilder
.failed_entities
== 0 and not toolcontext
.opt_noact
.value
then
289 print
"Entities: {modelbuilder.total_entities}; Documented ones: {modelbuilder.doc_entities}; With nitunits: {modelbuilder.unit_entities}; Failures: {modelbuilder.failed_entities}"