vbatts/quay
Archived
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

yapf

Browse source
This commit is contained in:
Joseph Schorr 2017-07-12 16:37:51 +03:00
parent 932db23a5c
commit 4ed73d247b
4 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
workers/servicekeyworker/models_interface.py
View file

@ -1,6 +1,7 @@
from abc import ABCMeta, abstractmethod from abc import ABCMeta, abstractmethod
from six import add_metaclass from six import add_metaclass
@add_metaclass(ABCMeta) @add_metaclass(ABCMeta)
class ServiceKeyWorkerDataInterface(object): class ServiceKeyWorkerDataInterface(object):
""" """

6
workers/servicekeyworker/models_pre_oci.py
View file

@ -1,13 +1,14 @@
from data import model from data import model
from workers.servicekeyworker.models_interface import ServiceKeyWorkerDataInterface from workers.servicekeyworker.models_interface import ServiceKeyWorkerDataInterface
class PreOCIModel(ServiceKeyWorkerDataInterface): class PreOCIModel(ServiceKeyWorkerDataInterface):
def set_key_expiration(self, kid, expiration_date): def set_key_expiration(self, kid, expiration_date):
model.service_keys.set_key_expiration(kid, expiration_date) model.service_keys.set_key_expiration(kid, expiration_date)
def create_service_key_for_testing(self, expiration): def create_service_key_for_testing(self, expiration):
key = model.service_keys.create_service_key('test', 'somekid', 'quay', '', {}, expiration) key = model.service_keys.create_service_key('test', 'somekid', 'quay', '', {}, expiration)
return key.kid return key.kid
def get_service_key_expiration(self, kid): def get_service_key_expiration(self, kid):
try: try:
@ -16,4 +17,5 @@ class PreOCIModel(ServiceKeyWorkerDataInterface):
except model.ServiceKeyDoesNotExist: except model.ServiceKeyDoesNotExist:
return None return None
pre_oci_model = PreOCIModel() pre_oci_model = PreOCIModel()

4
workers/servicekeyworker/servicekeyworker.py
View file

@ -7,11 +7,12 @@ from workers.worker import Worker
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class ServiceKeyWorker(Worker): class ServiceKeyWorker(Worker):
def __init__(self): def __init__(self):
super(ServiceKeyWorker, self).__init__() super(ServiceKeyWorker, self).__init__()
self.add_operation(self._refresh_service_key, self.add_operation(self._refresh_service_key,
app.config.get('INSTANCE_SERVICE_KEY_REFRESH', 60)*60) app.config.get('INSTANCE_SERVICE_KEY_REFRESH', 60) * 60)
def _refresh_service_key(self): def _refresh_service_key(self):
""" """
@ -22,6 +23,7 @@ class ServiceKeyWorker(Worker):
model.set_key_expiration(instance_keys.local_key_id, datetime.now() + expiration) model.set_key_expiration(instance_keys.local_key_id, datetime.now() + expiration)
logger.debug('Finished refresh of automatic service keys') logger.debug('Finished refresh of automatic service keys')
if __name__ == "__main__": if __name__ == "__main__":
worker = ServiceKeyWorker() worker = ServiceKeyWorker()
worker.start() worker.start()

2
workers/servicekeyworker/test/test_servicekeyworker.py
View file

@ -13,7 +13,7 @@ def test_refresh_service_key(initialized_db):
original_expiration = datetime.now() + timedelta(minutes=10) original_expiration = datetime.now() + timedelta(minutes=10)
test_key_kid = model.create_service_key_for_testing(original_expiration) test_key_kid = model.create_service_key_for_testing(original_expiration)
instance_keys = AttrDict(dict(local_key_id=test_key_kid, service_key_expiration=10)) instance_keys = AttrDict(dict(local_key_id=test_key_kid, service_key_expiration=30))
with patch('workers.servicekeyworker.servicekeyworker.instance_keys', instance_keys): with patch('workers.servicekeyworker.servicekeyworker.instance_keys', instance_keys):
worker = ServiceKeyWorker() worker = ServiceKeyWorker()
worker._refresh_service_key() worker._refresh_service_key()