Mirroristas/bigchaindb
2
0
Fork 0
mirror of https://github.com/bigchaindb/bigchaindb.git synced 2024-10-13 13:34:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix test on changefeed reconnection

Browse Source
This commit is contained in:
vrde 2017-02-03 11:58:18 +01:00
parent 98135c7df9
commit 7f50d76d9e
No known key found for this signature in database
GPG Key ID: 6581C7C39B3D397D
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/backend/mongodb/test_changefeed.py
View File

@ -150,16 +150,15 @@ def test_changefeed_prefeed(mock_cursor_next, mock_cursor_alive,
@pytest.mark.bdb @pytest.mark.bdb
@mock.patch('pymongo.cursor.Cursor.alive', new_callable=mock.PropertyMock) @mock.patch('bigchaindb.backend.mongodb.changefeed.MongoDBChangeFeed.run_changefeed') # noqa
@mock.patch('bigchaindb.backend.mongodb.connection.MongoDBConnection.run') # noqa def test_connection_failure(mock_run_changefeed):
def test_connection_failure(mock_run_changefeed, mock_cursor_alive):
from bigchaindb.backend import get_changefeed, connect from bigchaindb.backend import get_changefeed, connect
from bigchaindb.backend.exceptions import ConnectionError from bigchaindb.backend.exceptions import ConnectionError
from bigchaindb.backend.changefeed import ChangeFeed from bigchaindb.backend.changefeed import ChangeFeed
conn = connect() conn = connect()
mock_cursor_alive.return_value = False mock_run_changefeed.side_effect = [ConnectionError(),
mock_run_changefeed.side_effect = [ConnectionError(), mock.DEFAULT] mock.DEFAULT]
changefeed = get_changefeed(conn, 'backlog', ChangeFeed.INSERT) changefeed = get_changefeed(conn, 'backlog', ChangeFeed.INSERT)
changefeed.run_forever() changefeed.run_forever()