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()