- Timestamp:
- 2 Nov 2012, 13:40:10 (12 years ago)
- File:
-
- 1 edited
Legend:
- Unmodified
- Added
- Removed
-
main/waeup.kofa/trunk/src/waeup/kofa/tests/test_reports.py
r9344 r9511 1 1 import datetime 2 2 import grok 3 import logging 3 4 import pytz 4 5 import unittest 6 from cStringIO import StringIO 7 from grok.interfaces import IContainer 5 8 from zc.async.interfaces import IJob 6 from zope.component import getGlobalSiteManager 9 from zc.async.testing import wait_for_result 10 from zope.component import getGlobalSiteManager, getUtility 11 from zope.component.hooks import setSite 7 12 from zope.interface.verify import verifyClass, verifyObject 8 from waeup.kofa.interfaces import IJobManager 13 from waeup.kofa.interfaces import IJobManager, IKofaPluggable 9 14 from waeup.kofa.reports import ( 10 IReport, IReportGenerator, IReportJob, IReportJobContainer,) 15 IReport, IReportGenerator, IReportJob, IReportJobContainer, 16 IReportsContainer,) 11 17 from waeup.kofa.reports import ( 12 18 Report, ReportGenerator, get_generators, report_job, AsyncReportJob, 13 ReportJobContainer,) 14 from waeup.kofa.testing import FakeJob, FakeJobManager 19 ReportJobContainer, ReportsContainer, ReportsContainerPlugin) 20 from waeup.kofa.testing import FakeJob, FakeJobManager, FunctionalLayer 21 from waeup.kofa.tests.test_async import FunctionalAsyncTestCase 15 22 16 23 class FakeReportGenerator(ReportGenerator): … … 18 25 19 26 def __init__(self, name=None, perm_create=None, perm_view=None): 20 self.name = name27 self.name = 'Report 1' 21 28 self.perm_create = perm_create 22 29 self.perm_view = perm_view … … 24 31 25 32 def __eq__(self, obj): 33 if getattr(obj, 'name', None) is None: 34 return False 26 35 return self.name == obj.name 27 36 28 def generate(self, site): 29 return Report() 37 def generate(self, site, args=[], kw={}): 38 result = Report() 39 result.args = args 40 result.kw = kw 41 return result 30 42 31 43 class ReportTests(unittest.TestCase): … … 47 59 return 48 60 61 def test_args(self): 62 # a report stores the args and kw used for generation 63 report1 = Report() 64 report2 = Report(args=[1, 2], kwargs=dict(a=1, b=2)) 65 self.assertEqual(report1.args, []) 66 self.assertEqual(report1.kwargs, dict()) 67 self.assertEqual(report2.args, [1, 2]) 68 self.assertEqual(report2.kwargs, dict(a=1, b=2)) 69 return 70 71 def test_create_pdf(self): 72 # trying to create a pdf results in an error 73 report = Report() 74 self.assertRaises(NotImplementedError, report.create_pdf) 75 return 76 49 77 class ReportGeneratorTest(unittest.TestCase): 50 78 def setUp(self): … … 58 86 return 59 87 60 61 class HelpersTests(unittest.TestCase): 62 # Tests for helper functions 63 64 def setUp(self): 65 grok.testing.grok('waeup.kofa.reports') # register utils 66 self.registered = [] # report utils registrations 88 def test_generate(self): 89 # the base report generator delivers reports 90 obj = ReportGenerator() 91 result = obj.generate(None) 92 self.assertTrue(IReport.providedBy(result)) 93 return 94 95 class GeneratorRegistrar(object): 96 # Mix-in providing report generator registrations 97 def setUp(self): 98 self.registered = [] 67 99 68 100 def tearDown(self): … … 81 113 return generator 82 114 115 class HelpersTests(GeneratorRegistrar, unittest.TestCase): 116 # Tests for helper functions 117 118 def setUp(self): 119 grok.testing.grok('waeup.kofa.reports') # register utils 120 super(HelpersTests, self).setUp() 121 return 122 83 123 def test_get_generators_none(self): 84 124 # we get no generators if none was registered … … 106 146 return 107 147 108 class ReportJobTests(HelpersTests): 148 149 class ReportJobTests(FunctionalAsyncTestCase, GeneratorRegistrar): 109 150 # Test asynchronous report functionality (simple cases) 151 152 layer = FunctionalLayer 153 154 def setUp(self): 155 super(ReportJobTests, self).setUp() 156 GeneratorRegistrar.setUp(self) 157 self.root_folder = self.getRootFolder() 110 158 111 159 def test_report_job_func(self): … … 125 173 return 126 174 175 def test_finished(self): 176 # AsyncReportJobs signal with a bool whether they`re finished 177 job = AsyncReportJob(self.root_folder, None) 178 setSite(self.root_folder) 179 self.assertEqual(job.finished, False) 180 manager = getUtility(IJobManager) 181 manager.put(job) 182 wait_for_result(job) 183 self.assertEqual(job.finished, True) 184 return 185 186 def test_failed_true(self): 187 # We can test whether a job failed 188 job = AsyncReportJob(self.root_folder, None) # no report generator 189 setSite(self.root_folder) 190 # while a job is not finished, `failed` is ``None`` 191 self.assertTrue(job.failed is None) 192 manager = getUtility(IJobManager) 193 manager.put(job) 194 wait_for_result(job) 195 # the finished job failed 196 self.assertEqual(job.failed, True) 197 return 198 199 def test_failed_false(self): 200 # We can test whether a job failed 201 self.register_generator('report1') 202 job = AsyncReportJob(self.root_folder, 'report1') 203 setSite(self.root_folder) 204 # while a job is not finished, `failed` is ``None`` 205 self.assertTrue(job.failed is None) 206 manager = getUtility(IJobManager) 207 manager.put(job) 208 wait_for_result(job) 209 # the finished job failed 210 self.assertEqual(job.failed, False) 211 return 212 213 def test_description(self): 214 # IReportJobs provide a description of the started job 215 self.register_generator('report1') 216 args, kw = ['a', 'b'], dict(a=1, b=2) 217 job = AsyncReportJob(self.root_folder, 'report1', args=args, kw=kw) 218 self.assertEqual( 219 job.description, 220 "Report 1 ('a', 'b', a=1, b=2)") 221 # w/o args nor kwargs 222 job = AsyncReportJob(self.root_folder, 'report1') 223 self.assertEqual( 224 job.description, 225 'Report 1 ()') 226 # with args only 227 job = AsyncReportJob(self.root_folder, 'report1', args=args) 228 self.assertEqual( 229 job.description, "Report 1 ('a', 'b')") 230 # with keywords only 231 job = AsyncReportJob(self.root_folder, 'report1', kw=kw) 232 self.assertEqual( 233 job.description, "Report 1 (a=1, b=2)") 234 return 235 236 def test_description_invalid_generator(self): 237 # We can get a description even with an invalid generator 238 job = AsyncReportJob(self.root_folder, 'NOT EXISTENT') 239 self.assertEqual( 240 job.description, u'Invalid Report Generator ()') 241 # with args set (no kws) 242 job = AsyncReportJob( 243 self.root_folder, 'NOT EXISTENT', args=['a', 'b']) 244 self.assertEqual( 245 job.description, u"Invalid Report Generator ('a', 'b')") 246 # with kw set (no args) 247 job = AsyncReportJob( 248 self.root_folder, 'NOT_EXISTENT', kw=dict(a=1, b=2)) 249 self.assertEqual( 250 job.description, u'Invalid Report Generator (a=1, b=2)') 251 # with args and kws set 252 job = AsyncReportJob( 253 self.root_folder, 'NOT_EXISTENT', args=['a'], kw=dict(b=2)) 254 self.assertEqual( 255 job.description, u"Invalid Report Generator ('a', b=2)") 256 return 257 127 258 128 259 class FakeJobWithResult(FakeJob): 129 260 130 def __init__(self ):261 def __init__(self, args=[], kw={}): 131 262 #self.dir_path = tempfile.mkdtemp() 132 263 #self.result = os.path.join(self.dir_path, 'fake.csv') 133 264 #open(self.result, 'wb').write('a fake result') 134 265 self.result = Report() 266 self.result.args = args 267 self.result.kw = kw 135 268 return 136 269 … … 225 358 self.assertEqual( 226 359 result, 227 [('new', u'new', u' unnamed'),228 ('completed', u'completed', u' unnamed')]360 [('new', u'new', u'Unnamed Report'), 361 ('completed', u'completed', u'Unnamed Report')] 229 362 ) 230 363 return … … 257 390 self.assertEqual(result2, ('4', 'report3', 'bob')) 258 391 self.assertEqual(result3, None) 259 #shutil.rmtree(fake_job.dir_path) 260 return 392 return 393 394 395 class ReportsContainerTests(unittest.TestCase): 396 # Tests for ReportsContainer 397 398 def test_iface(self): 399 # ReportsContainers really provide the promised interfaces 400 obj = ReportsContainer() 401 verifyClass(IReportsContainer, ReportsContainer) 402 verifyClass(IContainer, ReportsContainer) 403 verifyObject(IReportsContainer, obj) 404 verifyObject(IContainer, obj) 405 return 406 407 class ReportsContainerPluginTests(unittest.TestCase): 408 # Tests for ReportsContainerPlugin 409 410 def create_logger(self): 411 # create a logger suitable for local tests. 412 test_logger = logging.getLogger('waeup.kofa.reports.testlogger') 413 log = StringIO() 414 handler = logging.StreamHandler(log) 415 handler.setLevel(logging.DEBUG) 416 test_logger.addHandler(handler) 417 test_logger.setLevel(logging.DEBUG) 418 self.logger = test_logger 419 self.log = log 420 self.handler = handler 421 return self.logger 422 423 def remove_logger(self): 424 del self.handler 425 del self.logger 426 del self.log 427 pass 428 429 def get_log(self): 430 self.log.seek(0) 431 return self.log.read() 432 433 def setUp(self): 434 self.create_logger() 435 return 436 437 def tearDown(self): 438 self.remove_logger() 439 return 440 441 def test_iface(self): 442 # make sure we fullfill the promised interfaces 443 obj = ReportsContainerPlugin() 444 verifyClass(IKofaPluggable, ReportsContainerPlugin) 445 verifyObject(IKofaPluggable, obj) 446 return 447 448 def test_get_as_utility(self): 449 # make sure we can get the plugin as utility 450 grok.testing.grok('waeup.kofa.reports') 451 util = getUtility(IKofaPluggable, name='reports') 452 self.assertTrue(util is not None) 453 return 454 455 def test_update_no_container(self): 456 # we can update an existing site 457 fake_site = grok.Container() 458 plugin = ReportsContainerPlugin() 459 plugin.update(fake_site, 'app', self.logger) 460 log = self.get_log() 461 self.assertEqual( 462 log, 'Added reports container for site "app"\n') 463 self.assertTrue('reports' in fake_site.keys()) 464 self.assertTrue(IReportsContainer.providedBy(fake_site['reports'])) 465 return 466 467 def test_update_uptodate_site(self): 468 # we leave already existing reports containers in place 469 fake_site = grok.Container() 470 plugin = ReportsContainerPlugin() 471 fake_site['reports'] = ReportsContainer() 472 plugin.update(fake_site, 'app', self.logger) 473 log = self.get_log() 474 self.assertEqual(log, '') # no log message 475 return 476 477 def test_setup_new_site(self): 478 # if we setup a site, we always install a fresh reports container 479 fake_site = grok.Container() 480 plugin = ReportsContainerPlugin() 481 plugin.setup(fake_site, 'app', self.logger) 482 log1 = self.get_log() 483 result1 = fake_site.get('reports', None) 484 plugin.setup(fake_site, 'app', self.logger) # replace old container 485 log2 = self.get_log() 486 result2 = fake_site.get('reports', None) 487 self.assertTrue(result1 is not result2) 488 self.assertEqual(log1, 489 'Added reports container for site "app"\n') 490 self.assertEqual(log2, 491 'Added reports container for site "app"\n' 492 'Removed reports container for site "app"\n' 493 'Added reports container for site "app"\n') 494 return
Note: See TracChangeset for help on using the changeset viewer.