Example #1
0
    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() == []
Example #2
0
    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})
Example #3
0
    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})
Example #4
0
    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(), [])
Example #5
0
    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(), [])