38 lines
975 B
Python
38 lines
975 B
Python
|
import asyncio
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from mock import AsyncMock, Mock
|
||
|
|
|
||
|
|
import chweb.consumer
|
||
|
|
from chweb.consumer import DbWriter
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_db_setup(three_checks, config, event_loop):
|
||
|
|
queue = asyncio.Queue()
|
||
|
|
db_writer = DbWriter(config, Mock(), event_loop, queue)
|
||
|
|
assert db_writer.conn is None
|
||
|
|
|
||
|
|
chweb.consumer.asyncpg = AsyncMock()
|
||
|
|
await db_writer.connect()
|
||
|
|
chweb.consumer.asyncpg.connect.assert_called()
|
||
|
|
|
||
|
|
db_writer.conn = AsyncMock()
|
||
|
|
await db_writer.setup()
|
||
|
|
db_writer.conn.execute.assert_called()
|
||
|
|
|
||
|
|
for check in three_checks:
|
||
|
|
await queue.put(check)
|
||
|
|
print("&&&&&&&&&&")
|
||
|
|
print("&&&&&&&&&&")
|
||
|
|
print("&&&&&&&&&&")
|
||
|
|
print(queue.qsize())
|
||
|
|
print(config.postgres.batch_size)
|
||
|
|
print("&&&&&&&&&&")
|
||
|
|
print("&&&&&&&&&&")
|
||
|
|
print("&&&&&&&&&&")
|
||
|
|
task = event_loop.create_task(db_writer())
|
||
|
|
await asyncio.sleep(0.5)
|
||
|
|
db_writer.conn.copy_records_to_table.assert_called()
|
||
|
|
task.cancel()
|