Esempio n. 1
0
def test_send_notices(mocker, dbmock):
    TEST_NOTICES = [1, 3, 2]

    queue_mock = mocker.MagicMock()
    dbmock._conn.notices = TEST_NOTICES
    DB._send_notices(dbmock._conn, queue_mock)

    # Not so ideal, makes assumptions about the implementation
    queue_mock.put_nowait.assert_has_calls(
        [mocker.call(x) for x in TEST_NOTICES])

    assert not dbmock._conn.notices