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

Change servicekeyworker to use a data interface

Browse source
This commit is contained in:
Joseph Schorr 2017-07-12 16:37:32 +03:00
parent 0afc222214
commit 932db23a5c
4 changed files with 53 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -6,18 +6,17 @@ from workers.servicekeyworker.servicekeyworker import ServiceKeyWorker
from util.morecollections import AttrDict
from test.fixtures import *
from workers.servicekeyworker.models_pre_oci import pre_oci_model as model
def test_refresh_service_key(initialized_db):
# Create a service key for testing.
test_key = model.service_keys.create_service_key('test', 'somekid', 'quay', '', {},
datetime.now() + timedelta(minutes=10))
instance_keys = AttrDict(dict(local_key_id=test_key.kid, service_key_expiration=10))
original_expiration = datetime.now() + timedelta(minutes=10)
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))
with patch('workers.servicekeyworker.servicekeyworker.instance_keys', instance_keys):
worker = ServiceKeyWorker()
worker._refresh_service_key()
# Ensure the key's expiration was changed.
updated_key = model.service_keys.get_service_key(test_key.kid, approved_only=False)
assert updated_key.expiration_date > test_key.expiration_date
assert model.get_service_key_expiration(test_key_kid) > original_expiration