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 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 _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('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)
