Finalized integration of coverage.py within the Appy framework.

gen/plone25/generator.py

@@ -325,6 +325,7 @@ class Generator(AbstractGenerator):
                 imports.append(importDef)
         repls = self.repls.copy()
         repls['imports'] = '\n'.join(imports)
+        repls['totalNumberOfTests'] = self.totalNumberOfTests
         self.copyFile('__init__.py', repls)
 
     def generateInstall(self):

gen/plone25/mixins/TestMixin.py

@@ -32,7 +32,6 @@ class TestMixin:
         # absence of names beginning with other chars than "__".
         for elem in moduleObj.__dict__.iterkeys():
             if not elem.startswith('__'):
-                print 'Element found in this module!!!', moduleObj, elem
                 res.append(moduleObj)
                 break
         # Include sub-modules if any
@@ -49,7 +48,8 @@ class TestMixin:
                     res += self.getNonEmptySubModules(subModuleName)
         return res
 
-    def getCovFolder(self):
+    @staticmethod
+    def getCovFolder():
         '''Returns the folder where to put the coverage folder if needed.'''
         for arg in sys.argv:
             if arg.startswith('[coverage'):
@@ -68,28 +68,18 @@ def beforeTest(test):
     test.createUser('admin', ('Member','Manager'))
     test.login('admin')
     g['t'] = g['test']
-    # Must we perform test coverage ?
-    covFolder = test.getCovFolder()
-    if covFolder:
-        try:
-            print 'COV!!!!', covFolder
-            import coverage
-            app = getattr(cfg, g['tool'].o.getAppName())
-            from coverage import coverage
-            cov = coverage()
-            g['cov'] = cov
-            g['covFolder'] = covFolder
-            cov.start()
-        except ImportError:
-            print 'You must install the "coverage" product.'
 
 def afterTest(test):
     '''Is executed after every test.'''
     g = test.globs
-    if g.has_key('covFolder'):
-        cov = g['cov']
+    appName = g['tool'].o.getAppName()
+    exec 'from Products.%s import cov, covFolder, totalNumberOfTests, ' \
+         'countTest' % appName
+    countTest()
+    exec 'from Products.%s import numberOfExecutedTests' % appName
+    if cov and (numberOfExecutedTests == totalNumberOfTests):
         cov.stop()
         # Dumps the coverage report
-        appModules = test.getNonEmptySubModules(g['tool'].o.getAppName())
-        cov.html_report(directory=g['covFolder'], morfs=appModules)
+        appModules = test.getNonEmptySubModules(appName)
+        cov.html_report(directory=covFolder, morfs=appModules)
 # ------------------------------------------------------------------------------

gen/plone25/templates/__init__.py

@@ -1,4 +1,30 @@
 <!codeHeader!>
+# Test coverage-related stuff --------------------------------------------------
+import sys
+from appy.gen.plone25.mixins.TestMixin import TestMixin
+covFolder = TestMixin.getCovFolder()
+# The previous method checks in sys.argv whether Zope was lauched for performing
+# coverage tests or not.
+cov = None # The main Coverage instance as created by the coverage program.
+totalNumberOfTests = <!totalNumberOfTests!>
+numberOfExecutedTests = 0
+if covFolder:
+    try:
+        import coverage
+        from coverage import coverage
+        cov = coverage()
+        cov.start()
+    except ImportError:
+        print 'COVERAGE KO! The "coverage" program is not installed. You can ' \
+              'download it from http://nedbatchelder.com/code/coverage.' \
+              '\nHit <enter> to execute the test suite without coverage.'
+        sys.stdin.readline()
+
+def countTest():
+    global numberOfExecutedTests
+    numberOfExecutedTests += 1
+
+# ------------------------------------------------------------------------------
 from config import *
 import logging
 try:
@@ -13,6 +39,7 @@ from Products.Archetypes import listTypes
 from appy.gen.plone25.installer import ZopeInstaller
 logger = logging.getLogger(PROJECTNAME)
 
+# Zope-level installation of the generated product. ----------------------------
 def initialize(context):
 <!imports!>
     # I need to do those imports here; else, types and add permissions will not
@@ -21,3 +48,4 @@ def initialize(context):
         <!applicationName!>Tool.<!applicationName!>Tool,
         DEFAULT_ADD_CONTENT_PERMISSION, ADD_CONTENT_PERMISSIONS,
         logger, globals()).install()
+# ------------------------------------------------------------------------------