1
|
|
|
# -*- coding: utf-8 -*- |
2
|
|
|
# vi:si:et:sw=4:sts=4:ts=4 |
3
|
|
|
|
4
|
|
|
import asyncio |
5
|
|
|
from unittest import mock |
6
|
|
|
|
7
|
|
|
from loafer.exceptions import ConsumerError |
8
|
|
|
from loafer.dispatcher import LoaferDispatcher |
|
|
|
|
9
|
|
|
from loafer.manager import LoaferManager |
10
|
|
|
from loafer.route import Route |
|
|
|
|
11
|
|
|
|
12
|
|
|
from loafer.aws.consumer import Consumer as AWSConsumer |
|
|
|
|
13
|
|
|
|
14
|
|
|
|
15
|
|
|
def test_manager_with_no_routes(event_loop): |
16
|
|
|
manager = LoaferManager('test-queue', event_loop=event_loop) |
17
|
|
|
assert len(manager.routes) == 0 |
18
|
|
|
|
19
|
|
|
|
20
|
|
|
def test_manager_one_route(event_loop): |
21
|
|
|
route = Route('test-queue', lambda x: None) |
22
|
|
|
manager = LoaferManager('test-queue', event_loop=event_loop) |
23
|
|
|
manager.routes.append(route) |
24
|
|
|
|
25
|
|
|
assert len(manager.routes) == 1 |
26
|
|
|
|
27
|
|
|
|
28
|
|
|
def test_consumers(event_loop): |
29
|
|
|
manager = LoaferManager('test-queue', event_loop=event_loop) |
30
|
|
|
assert manager.consumers |
31
|
|
|
assert len(manager.consumers) == 1 |
32
|
|
|
|
33
|
|
|
|
34
|
|
|
def test_consumers_returns_default_if_not_manually_set(event_loop): |
|
|
|
|
35
|
|
|
manager = LoaferManager('test-queue', event_loop=event_loop) |
36
|
|
|
assert manager.consumers[0].__class__ == AWSConsumer |
37
|
|
|
|
38
|
|
|
|
39
|
|
|
def test_dispatcher(event_loop): |
40
|
|
|
manager = LoaferManager('test-queue', event_loop=event_loop) |
41
|
|
|
assert isinstance(manager.get_dispatcher(), LoaferDispatcher) |
42
|
|
|
|
43
|
|
|
|
44
|
|
|
def test_on_future_errors(event_loop): |
45
|
|
|
manager = LoaferManager('test-queue', event_loop=event_loop) |
46
|
|
|
manager.stop = mock.Mock() |
47
|
|
|
future = asyncio.Future() |
48
|
|
|
future.set_exception(ConsumerError) |
49
|
|
|
manager.on_future__errors(future) |
50
|
|
|
|
51
|
|
|
assert manager.stop.called |
52
|
|
|
assert manager.stop.called_once_with() |
53
|
|
|
|