async def test_exception_classes(self):
channel = await self.create_channel()
master = Master(channel)
self.state = []
def worker_func(*, foo):
if foo < 50:
raise RejectMessage(requeue=False)
if foo > 100:
raise NackMessage(requeue=False)
self.state.append(foo)
worker = await master.create_worker('worker.foo',
worker_func,
auto_delete=True)
for item in range(200):
await master.proxy.worker.foo(foo=item)
await asyncio.sleep(2, loop=self.loop)
self.assertSequenceEqual(self.state, range(50, 101))
await worker.close()
async def test_simple_many(self):
channel = await self.create_channel()
master = Master(channel)
tasks = 100
self.state = []
def worker_func(*, foo):
nonlocal tasks
self.state.append(foo)
tasks -= 1
worker = await master.create_worker('worker.foo',
worker_func,
auto_delete=True)
for item in range(100):
await master.proxy.worker.foo(foo=item)
while tasks > 0:
await asyncio.sleep(0)
self.assertSequenceEqual(self.state, range(100))
await worker.close()
async def test_exception_classes(self, channel: aio_pika.Channel):
master = Master(channel)
counter = 200
self.state = []
def worker_func(*, foo):
nonlocal counter
counter -= 1
if foo < 50:
raise RejectMessage(requeue=False)
if foo > 100:
raise NackMessage(requeue=False)
self.state.append(foo)
worker = await master.create_worker(
"worker.foo", worker_func, auto_delete=True,
)
for item in range(200):
await master.proxy.worker.foo(foo=item)
while counter > 0:
await asyncio.sleep(0)
assert self.state == list(range(50, 101))
await worker.close()
async def test_simple_many(self, channel: aio_pika.Channel):
master = Master(channel)
tasks = 100
state = []
def worker_func(*, foo):
nonlocal tasks, state
state.append(foo)
tasks -= 1
worker = await master.create_worker(
"worker.foo", worker_func, auto_delete=True,
)
for item in range(100):
await master.proxy.worker.foo(foo=item)
while tasks > 0:
await asyncio.sleep(0)
assert state == list(range(100))
await worker.close()
def test_simple(self):
channel = yield from self.create_channel()
master = Master(channel)
self.state = []
def worker_func(*, foo, bar):
self.state.append((foo, bar))
worker = yield from master.create_worker('worker.foo',
worker_func,
auto_delete=True)
yield from master.proxy.worker.foo(foo=1, bar=2)
yield from asyncio.sleep(0.5, loop=self.loop)
self.assertSequenceEqual(self.state, [(1, 2)])
yield from worker.close()
def test_simple_many(self):
channel = yield from self.create_channel()
master = Master(channel)
self.state = []
def worker_func(*, foo):
self.state.append(foo)
worker = yield from master.create_worker('worker.foo',
worker_func,
auto_delete=True)
for item in range(1000):
yield from master.proxy.worker.foo(foo=item)
yield from asyncio.sleep(2, loop=self.loop)
self.assertSequenceEqual(self.state, range(1000))
yield from worker.close()
async def test_simple_coro(self):
channel = await self.create_channel()
master = Master(channel)
self.state = []
async def worker_func(*, foo, bar):
self.state.append((foo, bar))
worker = await master.create_worker('worker.foo',
worker_func,
auto_delete=True)
await master.proxy.worker.foo(foo=1, bar=2)
await asyncio.sleep(0.5, loop=self.loop)
self.assertSequenceEqual(self.state, [(1, 2)])
await worker.close()
async def test_simple(self, channel: aio_pika.Channel):
master = Master(channel)
event = asyncio.Event()
self.state = []
def worker_func(*, foo, bar):
nonlocal event
self.state.append((foo, bar))
event.set()
worker = await master.create_worker(
"worker.foo", worker_func, auto_delete=True,
)
await master.proxy.worker.foo(foo=1, bar=2)
await event.wait()
assert self.state == [(1, 2)]
await worker.close()
async def test_simple(self):
channel = await self.create_channel()
master = Master(channel)
event = asyncio.Event()
self.state = []
def worker_func(*, foo, bar):
nonlocal event
self.state.append((foo, bar))
event.set()
worker = await master.create_worker('worker.foo',
worker_func,
auto_delete=True)
await master.proxy.worker.foo(foo=1, bar=2)
await event.wait()
self.assertSequenceEqual(self.state, [(1, 2)])
await worker.close()