def setUp(self):
super(TestPymongoSocketTracing, self).setUp()
patch()
# Override server pin's tracer with our dummy tracer
Pin.override(pymongo.server.Server, tracer=self.tracer)
# maxPoolSize controls the number of sockets that the client can instantiate
# and choose from to perform classic operations. For the sake of our tests,
# let's limit this number to 1
self.client = pymongo.MongoClient(port=MONGO_CONFIG["port"], maxPoolSize=1)
# Override TracedMongoClient's pin's tracer with our dummy tracer
Pin.override(self.client, tracer=self.tracer, service="testdb")
def setUp(self):
patch()
self.tracer = get_dummy_tracer()
# Override server pin's tracer with our dummy tracer
Pin.override(pymongo.server.Server, tracer=self.tracer)
# maxPoolSize controls the number of sockets that the client can instanciate
# and choose from to perform classic operations. For the sake of our tests,
# let's limit this number to 1
self.client = pymongo.MongoClient(port=MONGO_CONFIG['port'],
maxPoolSize=1)
# Override TracedMongoClient's pin's tracer with our dummy tracer
Pin.override(self.client, tracer=self.tracer)
def test_patch_unpatch(self):
tracer = get_dummy_tracer()
writer = tracer.writer
# Test patch idempotence
patch()
patch()
client = pymongo.MongoClient(port=MONGO_CONFIG['port'])
Pin.get_from(client).clone(tracer=tracer).onto(client)
client['testdb'].drop_collection('whatever')
spans = writer.pop()
assert spans, spans
assert len(spans) == 1
# Test unpatch
unpatch()
client = pymongo.MongoClient(port=MONGO_CONFIG['port'])
client['testdb'].drop_collection('whatever')
spans = writer.pop()
assert not spans, spans
# Test patch again
patch()
client = pymongo.MongoClient(port=MONGO_CONFIG['port'])
Pin.get_from(client).clone(tracer=tracer).onto(client)
client['testdb'].drop_collection('whatever')
spans = writer.pop()
assert spans, spans
assert len(spans) == 1
def test_patch_unpatch(self):
tracer = get_dummy_tracer()
writer = tracer.writer
# Test patch idempotence
patch()
patch()
client = pymongo.MongoClient(port=MONGO_CONFIG['port'])
Pin.get_from(client).clone(tracer=tracer).onto(client)
client["testdb"].drop_collection("whatever")
spans = writer.pop()
assert spans, spans
eq_(len(spans), 1)
# Test unpatch
unpatch()
client = pymongo.MongoClient(port=MONGO_CONFIG['port'])
client["testdb"].drop_collection("whatever")
spans = writer.pop()
assert not spans, spans
# Test patch again
patch()
client = pymongo.MongoClient(port=MONGO_CONFIG['port'])
Pin.get_from(client).clone(tracer=tracer).onto(client)
client["testdb"].drop_collection("whatever")
spans = writer.pop()
assert spans, spans
eq_(len(spans), 1)
def test_patch_unpatch(self):
# Test patch idempotence
patch()
patch()
client = pymongo.MongoClient(port=MONGO_CONFIG["port"])
Pin.get_from(client).clone(tracer=self.tracer).onto(client)
# We do not wish to trace tcp spans here
Pin.get_from(pymongo.server.Server).remove_from(pymongo.server.Server)
client["testdb"].drop_collection("whatever")
spans = self.pop_spans()
assert spans, spans
assert len(spans) == 1
# Test unpatch
unpatch()
client = pymongo.MongoClient(port=MONGO_CONFIG["port"])
client["testdb"].drop_collection("whatever")
spans = self.pop_spans()
assert not spans, spans
# Test patch again
patch()
client = pymongo.MongoClient(port=MONGO_CONFIG["port"])
Pin.get_from(client).clone(tracer=self.tracer).onto(client)
# We do not wish to trace tcp spans here
Pin.get_from(pymongo.server.Server).remove_from(pymongo.server.Server)
client["testdb"].drop_collection("whatever")
spans = self.pop_spans()
assert spans, spans
assert len(spans) == 1
def setUp(self):
patch()
def setUp(self):
super(TestPymongoPatchConfigured, self).setUp()
patch()
def setUp(self):
patch()
