96adb287f8
This lets jobs turn off and exit turbo-hipster once they are done. This is useful for when using nodepool or when a job leaves the environment dirty and we can't run more jobs on this worker. Change-Id: I823be4196a5bf9ca92a14d9caf26163398a9434c
69 lines
2.0 KiB
Python
69 lines
2.0 KiB
Python
#!/usr/bin/python2
|
|
#
|
|
# Copyright 2013 Rackspace Australia
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import gear
|
|
import json
|
|
import time
|
|
import uuid
|
|
|
|
|
|
class FakeJob(object):
|
|
def __init__(self):
|
|
pass
|
|
|
|
def sendWorkStatus(self, *args, **kwargs):
|
|
pass
|
|
|
|
|
|
class FakeZuul(object):
|
|
"""A fake zuul/gearman client to request work from gearman and check
|
|
results"""
|
|
def __init__(self, server, port):
|
|
self.gearman = gear.Client('FakeZuul')
|
|
self.gearman.addServer(server, port)
|
|
self.gearman.waitForServer()
|
|
self.job = None
|
|
|
|
def make_zuul_data(self, data={}):
|
|
defaults = {
|
|
'ZUUL_UUID': str(uuid.uuid1()),
|
|
'ZUUL_REF': 'a',
|
|
'ZUUL_COMMIT': 'a',
|
|
'ZUUL_PROJECT': 'a',
|
|
'ZUUL_PIPELINE': 'a',
|
|
'ZUUL_URL': 'http://localhost',
|
|
'BASE_LOG_PATH': '56/123456/8',
|
|
'LOG_PATH': '56/123456/8/check/job_name/uuid123'
|
|
}
|
|
defaults.update(data)
|
|
return defaults
|
|
|
|
def submit_job(self, name, data):
|
|
if not self.job:
|
|
self.job = gear.Job(name,
|
|
json.dumps(data),
|
|
unique=str(time.time()))
|
|
self.gearman.submitJob(self.job)
|
|
else:
|
|
raise Exception('A job already exists in self.job')
|
|
|
|
return self.job
|
|
|
|
def wait_for_completion(self):
|
|
if self.job:
|
|
while not self.job.complete:
|
|
time.sleep(0.1)
|