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

View file

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