def test_pop_messages(self): """ MESSAGE (CORE): Test retrieve and delete messages """ truncate_messages() for i in range(10): add_message(event_type='TEST', payload={ 'foo': True, 'monty': 'python', 'number': i }) tmp = retrieve_messages(10) to_delete = [] for i in tmp: assert isinstance(i['payload'], dict) assert i['payload']['foo'] is True assert i['payload']['monty'] == 'python' assert i['payload']['number'] in list(range(100)) to_delete.append({ 'id': i['id'], 'created_at': i['created_at'], 'updated_at': i['created_at'], 'payload': str(i['payload']), 'event_type': i['event_type'] }) delete_messages(to_delete) assert retrieve_messages() == []
def test_add_message(self): """ MESSAGE (CORE): Test valid and invalid message """ truncate_messages() add_message(event_type='NEW_DID', payload={'name': 'name', 'name_Y': 'scope_X', 'type': 'file'}) with assert_raises(InvalidObject): add_message(event_type='NEW_DID', payload={'name': 'name', 'name_Y': 'scope_X', 'type': int})
def test_pop_messages(self): """ MESSAGE (CORE): Test retrieve and delete messages """ truncate_messages() for i in xrange(10): add_message(event_type='TEST', payload={'foo': True, 'monty': 'python', 'number': i}) tmp = retrieve_messages(10) to_delete = [] for i in tmp: assert_is_instance(i['payload'], dict) assert_equal(i['payload']['foo'], True) assert_equal(i['payload']['monty'], 'python') assert_in(i['payload']['number'], xrange(100)) to_delete.append(i['id']) delete_messages(to_delete) assert_equal(retrieve_messages(), [])
def test_pop_messages(self): """ MESSAGE (CORE): Test retrieve and delete messages """ truncate_messages() for i in xrange(10): add_message(event_type='TEST', payload={ 'foo': True, 'monty': 'python', 'number': i }) tmp = retrieve_messages(10) to_delete = [] for i in tmp: assert_is_instance(i['payload'], dict) assert_equal(i['payload']['foo'], True) assert_equal(i['payload']['monty'], 'python') assert_in(i['payload']['number'], xrange(100)) to_delete.append(i['id']) delete_messages(to_delete) assert_equal(retrieve_messages(), [])