def _send_records_to_master_helper( self, jobs, hqes, shard_hostname='host1', exception_to_throw=error.UnallowedRecordsSentToMaster, aborted=False): job_id = rpc_interface.create_job(name='dummy', priority='Medium', control_file='foo', control_type=SERVER, test_retry=10, hostless=True) job = models.Job.objects.get(pk=job_id) shard = models.Shard.objects.create(hostname='host1') job.shard = shard job.save() if aborted: job.hostqueueentry_set.update(aborted=True) job.shard = None job.save() hqe = job.hostqueueentry_set.all()[0] if not exception_to_throw: self._do_heartbeat_and_assert_response( shard_hostname=shard_hostname, upload_jobs=jobs, upload_hqes=hqes) else: self.assertRaises(exception_to_throw, self._do_heartbeat_and_assert_response, shard_hostname=shard_hostname, upload_jobs=jobs, upload_hqes=hqes)
def test_one_time_hosts(self): job = rpc_interface.create_job('test', 'Medium', 'control file', 'Server', one_time_hosts=['testhost']) host = models.Host.objects.get(hostname='testhost') self.assertEquals(host.invalid, True) self.assertEquals(host.labels.count(), 0) self.assertEquals(host.aclgroup_set.count(), 0)
def _createJobForLabel(self, label): job_id = rpc_interface.create_job(name='dummy', priority='Medium', control_file='foo', control_type=CLIENT, meta_hosts=[label.name], dependencies=(label.name, )) return models.Job.objects.get(id=job_id)
def test_job_keyvals(self): keyval_dict = {'mykey': 'myvalue'} job_id = rpc_interface.create_job(name='test', priority='Medium', control_file='foo', control_type='Client', hosts=['host1'], keyvals=keyval_dict) jobs = rpc_interface.get_jobs(id=job_id) self.assertEquals(len(jobs), 1) self.assertEquals(jobs[0]['keyvals'], keyval_dict)
def test_test_retry(self): job_id = rpc_interface.create_job(name='flake', priority='Medium', control_file='foo', control_type=CLIENT, hosts=['host1'], test_retry=10) jobs = rpc_interface.get_jobs(id=job_id) self.assertEquals(len(jobs), 1) self.assertEquals(jobs[0]['test_retry'], 10)
def _create_job_helper(self, **kwargs): return rpc_interface.create_job('test', 'Medium', 'control file', 'Server', **kwargs)
def _create_job_helper(self, **kwargs): return rpc_interface.create_job(name='test', priority='Medium', control_file='control file', control_type=SERVER, **kwargs)