model_visitor: reject is_before and is_after as they are tests

[nit.git] / src / model / model_visitor.nit

diff --git a/src/model/model_visitor.nit b/src/model/model_visitor.nit
index a2807de..9244055 100644 (file)
--- a/src/model/model_visitor.nit
+++ b/src/model/model_visitor.nit
@@ -109,7 +109,14 @@ abstract class ModelVisitor
        # Can we accept this `mentity` regarding its test suite status?
        fun accept_test(mentity: MEntity): Bool do
                if include_test then return true
-               if not mentity isa MModule then return true
+               if mentity isa MProperty then
+                       if mentity.is_before or mentity.is_before_all then return false
+                       if mentity.is_after or mentity.is_after_all then return false
+               end
+               if mentity isa MPropDef then
+                       if mentity.is_before or mentity.is_before_all then return false
+                       if mentity.is_after or mentity.is_after_all then return false
+               end
                return not mentity.is_test
        end