def test_get_nothide(self):
queue = StorageQueueContext(**testconfig)
queue.register_model(QueueOne())
message = QueueOne(epgid=1, resolution='test_get_nothide')
queue.put(message)
getmessage = queue.get(QueueOne(), hide=1)
assert getmessage.epgid == 1 and getmessage.resolution == 'test_get_nothide'
time.sleep(1)
testmessage = queue.get(QueueOne())
assert testmessage.epgid == 1 and getmessage.resolution == 'test_get_nothide'
queue.delete(testmessage)
def test_peek(self):
queue = StorageQueueContext(**testconfig)
queue.register_model(QueueOne())
message = QueueOne(epgid=1, resolution='test_peek')
message = queue.put(message)
assert not message is None
firstmessage = queue.peek(QueueOne())
assert firstmessage.resolution == 'test_peek'
queue.delete(message)
def test_update_nopop(self):
queue = StorageQueueContext(**testconfig)
queue.register_model(QueueOne())
message = QueueOne(epgid=1, resolution='test_get_nothide')
queue.put(message)
message.pop_receipt = None
with pytest.raises(AzureStorageWrapException):
updatemessage = queue.update(message)
def test_get_nothide_encrypted(self):
queue = StorageQueueContext(**testconfig)
message = QueueTwo()
queue.register_model(message)
message.server = 'encrypted server'
message.protocol = 'sftp'
message.user = '******'
message.password = '******'
queue.put(message)
message = queue.get(QueueTwo(), hide=1)
assert (message.server == 'encrypted server'
and message.protocol == 'sftp' and message.user == 'user'
and message.password == 'secret')
time.sleep(1)
testmessage = queue.get(QueueTwo())
queue.delete(testmessage)
def test_unregister_model(self):
queue = StorageQueueContext(**testconfig)
message = QueueOne()
queue.register_model(message)
assert message.__class__.__name__ in [
model['modelname'] for model in queue._modeldefinitions
]
queue.unregister_model(message)
assert not message.__class__.__name__ in [
model['modelname'] for model in queue._modeldefinitions
]
def test_unregister_model_delete(self):
queue = StorageQueueContext(**testconfig)
queue.register_model(QueueTwo())
queue.register_model(QueueOne())
assert ('QueueOne'
in [model['modelname']
for model in queue._modeldefinitions] and 'QueueTwo'
in [model['modelname'] for model in queue._modeldefinitions])
queue.unregister_model(QueueOne(), delete_queue=True)
queue.unregister_model(QueueTwo(), delete_queue=True)
assert (not 'QueueOne'
in [model['modelname']
for model in queue._modeldefinitions] and not 'QueueTwo'
in [model['modelname'] for model in queue._modeldefinitions])
def test_put(self):
queue = StorageQueueContext(**testconfig)
queue.register_model(QueueOne())
message = QueueOne(epgid=1, resolution='test_put')
queue.put(message)
queue.delete(message)
def test_register_model(self):
queue = StorageQueueContext(**testconfig)
queue.register_model(QueueOne())
assert 'QueueOne' in [
model['modelname'] for model in queue._modeldefinitions
]
def test_error_naming_convention(self):
queue = StorageQueueContext(**testconfig)
with pytest.raises(NameConventionError):
queue.register_model(QueueThree())