class JobStatsTestCase(unittest.TestCase):
'''
Tests related to capturing job stats.
'''
def setUp(self):
# Test 'event-based' job
cfg_path = helpers.testdata_path("simplecase/config.gem")
base_path = helpers.testdata_path("simplecase")
oq_job = engine.prepare_job()
oq_job_profile, params, sections = engine.import_job_profile(
cfg_path, oq_job)
self.eb_job = JobContext(
params, oq_job.id, sections=sections, base_path=base_path,
oq_job_profile=oq_job_profile, oq_job=oq_job)
def test_record_initial_stats(self):
'''Verify that
:py:method:`openquake.engine.JobContext._record_initial_stats`
reports initial calculation stats.
As we add fields to the uiapi.job_stats table, this test will need to
be updated to check for this new information.
'''
self.eb_job._record_initial_stats()
actual_stats = models.JobStats.objects.get(oq_job=self.eb_job.job_id)
self.assertTrue(actual_stats.start_time is not None)
self.assertEqual(91, actual_stats.num_sites)
self.assertEqual(1, actual_stats.realizations)
def test_job_launch_calls_record_initial_stats(self):
'''When a job is launched, make sure that
:py:method:`openquake.engine.JobContext._record_initial_stats`
is called.
'''
# Mock out pieces of the test job so it doesn't actually run.
eb_haz_calc = ('openquake.calculators.hazard.event_based.core'
'.EventBasedHazardCalculator')
eb_risk_calc = ('openquake.calculators.risk.event_based.core'
'.EventBasedRiskCalculator')
methods = ('initialize', 'pre_execute', 'execute', 'post_execute')
haz_patchers = [patch('%s.%s' % (eb_haz_calc, m)) for m in methods]
risk_patchers = [patch('%s.%s' % (eb_risk_calc, m)) for m in methods]
for p in haz_patchers:
p.start()
for p in risk_patchers:
p.start()
try:
record = 'openquake.engine.JobContext._record_initial_stats'
with patch(record) as record_mock:
engine._launch_job(
self.eb_job, ['general', 'HAZARD', 'RISK'])
self.assertEqual(1, record_mock.call_count)
finally:
for p in haz_patchers:
p.stop()
for p in risk_patchers:
p.stop()
class JobStatsTestCase(unittest.TestCase):
'''
Tests related to capturing job stats.
'''
def setUp(self):
# Test 'event-based' job
cfg_path = helpers.testdata_path("simplecase/config.gem")
base_path = helpers.testdata_path("simplecase")
oq_job = engine.prepare_job()
oq_job_profile, params, sections = engine.import_job_profile(
cfg_path, oq_job)
self.eb_job = JobContext(
params, oq_job.id, sections=sections, base_path=base_path,
oq_job_profile=oq_job_profile, oq_job=oq_job)
def test_record_initial_stats(self):
'''Verify that
:py:method:`openquake.engine.JobContext._record_initial_stats`
reports initial calculation stats.
As we add fields to the uiapi.job_stats table, this test will need to
be updated to check for this new information.
'''
self.eb_job._record_initial_stats()
actual_stats = models.JobStats.objects.get(oq_job=self.eb_job.job_id)
self.assertTrue(actual_stats.start_time is not None)
self.assertEqual(91, actual_stats.num_sites)
self.assertEqual(1, actual_stats.realizations)
def test_job_launch_calls_record_initial_stats(self):
'''When a job is launched, make sure that
:py:method:`openquake.engine.JobContext._record_initial_stats`
is called.
'''
# Mock out pieces of the test job so it doesn't actually run.
eb_haz_calc = ('openquake.calculators.hazard.event_based.core'
'.EventBasedHazardCalculator')
eb_risk_calc = ('openquake.calculators.risk.event_based.core'
'.EventBasedRiskCalculator')
methods = ('initialize', 'pre_execute', 'execute', 'post_execute')
haz_patchers = [patch('%s.%s' % (eb_haz_calc, m)) for m in methods]
risk_patchers = [patch('%s.%s' % (eb_risk_calc, m)) for m in methods]
for p in haz_patchers:
p.start()
for p in risk_patchers:
p.start()
try:
record = 'openquake.engine.JobContext._record_initial_stats'
with patch(record) as record_mock:
engine._launch_job(
self.eb_job, ['general', 'HAZARD', 'RISK'])
self.assertEqual(1, record_mock.call_count)
finally:
for p in haz_patchers:
p.stop()
for p in risk_patchers:
p.stop()
