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(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_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_put(self):
queue = StorageQueueContext(**testconfig)
queue.register_model(QueueOne())
message = QueueOne(epgid=1, resolution='test_put')
queue.put(message)
queue.delete(message)