Fix infrequent read preference test failure

This commit is contained in:
Bernie Hackett 2017-02-17 16:02:06 -08:00
parent c4091d8d75
commit 959b395da0

View File

@ -471,6 +471,31 @@ class TestCommandAndReadPreference(TestReplicaSetClientBase):
ctx.exit()
# Obedient commands.
# Geo stuff.
self.c.pymongo_test.test.create_index([('location', '2d')])
self.c.pymongo_test.test.create_index([('location', 'geoHaystack'),
('key', 1)], bucketSize=100)
# Attempt to await replication of indexes replicated.
self.c.pymongo_test.test2.insert({}, w=self.w)
self.c.pymongo_test.test2.remove({}, w=self.w)
self._test_fn(True, lambda: self.c.pymongo_test.command(
'geoNear', 'test', near=[0, 0]))
self._test_fn(True, lambda: self.c.pymongo_test.command(SON([
('geoNear', 'test'), ('near', [0, 0])])))
self._test_fn(
True, lambda: self.c.pymongo_test.command(
'geoSearch', 'test', near=[33, 33], maxDistance=6,
search={'type': 'restaurant'}, limit=30))
self._test_fn(
True, lambda: self.c.pymongo_test.command(SON([
('geoSearch', 'test'), ('near', [33, 33]), ('maxDistance', 6),
('search', {'type': 'restaurant'}), ('limit', 30)])))
self._test_fn(True, lambda: self.c.pymongo_test.command('group', {
'ns': 'test', 'key': {'a': 1}, '$reduce': 'function(obj, prev) { }',
'initial': {}}))
@ -498,31 +523,6 @@ class TestCommandAndReadPreference(TestReplicaSetClientBase):
self._test_fn(True, lambda: self.c.pymongo_test.command(SON([
('distinct', 'test'), ('key', 'a'), ('query', {'a': 1})])))
# Geo stuff.
self.c.pymongo_test.test.create_index([('location', '2d')])
self.c.pymongo_test.test.create_index([('location', 'geoHaystack'),
('key', 1)], bucketSize=100)
# Attempt to await replication of indexes replicated.
self.c.pymongo_test.test2.insert({}, w=self.w)
self.c.pymongo_test.test2.remove({}, w=self.w)
self._test_fn(True, lambda: self.c.pymongo_test.command(
'geoNear', 'test', near=[0, 0]))
self._test_fn(True, lambda: self.c.pymongo_test.command(SON([
('geoNear', 'test'), ('near', [0, 0])])))
self._test_fn(
True, lambda: self.c.pymongo_test.command(
'geoSearch', 'test', near=[33, 33], maxDistance=6,
search={'type': 'restaurant'}, limit=30))
self._test_fn(
True, lambda: self.c.pymongo_test.command(SON([
('geoSearch', 'test'), ('near', [33, 33]), ('maxDistance', 6),
('search', {'type': 'restaurant'}), ('limit', 30)])))
if (version.at_least(self.c, (2, 1, 0)) and
not version.at_least(self.c, (3, 5, 1))):
self._test_fn(