Fix unit test

This commit is contained in:
Thomas Conte 2017-03-17 10:01:58 +01:00
parent 08f040d218
commit 8526246f78

View File

@ -168,7 +168,7 @@ def test_initialize_replica_set(mock_cmd_line_opts):
] ]
# check that it returns # check that it returns
assert initialize_replica_set('host', 1337, 1000) is None assert initialize_replica_set('host', 1337, 1000, False, None, None) is None
# test it raises OperationError if anything wrong # test it raises OperationError if anything wrong
with mock.patch.object(Database, 'command') as mock_command: with mock.patch.object(Database, 'command') as mock_command:
@ -178,4 +178,4 @@ def test_initialize_replica_set(mock_cmd_line_opts):
] ]
with pytest.raises(pymongo.errors.OperationFailure): with pytest.raises(pymongo.errors.OperationFailure):
initialize_replica_set('host', 1337, 1000) initialize_replica_set('host', 1337, 1000, False, None, None)