def test_record_job_stop_time(self): """ Test that job stop time is recorded properly. """ cstats = JobStats(oq_job=self.job, start_time=datetime.utcnow(), num_sites=10) cstats.save(using="job_superv") supervisor.record_job_stop_time(self.job.id) # Fetch the stats and check for the stop_time cstats = JobStats.objects.get(oq_job=self.job.id) self.assertTrue(cstats.stop_time is not None)
def test_record_job_stop_time(self): """ Test that job stop time is recorded properly. """ cstats = JobStats( oq_job=self.job, start_time=datetime.utcnow(), num_sites=10) cstats.save(using='job_superv') supervisor.record_job_stop_time(self.job.id) # Fetch the stats and check for the stop_time cstats = JobStats.objects.get(oq_job=self.job.id) self.assertTrue(cstats.stop_time is not None)
def _record_initial_stats(self): ''' Report initial job stats (such as start time) by adding a uiapi.job_stats record to the db. ''' job_stats = JobStats(oq_job=self.oq_job) job_stats.start_time = datetime.utcnow() job_stats.num_sites = len(self.sites_to_compute()) calc_mode = CALCULATION_MODE[self['CALCULATION_MODE']] if jobconf.HAZARD_SECTION in self.sections: if calc_mode != 'scenario': job_stats.realizations = self["NUMBER_OF_LOGIC_TREE_SAMPLES"] job_stats.save()