def test_patch_unpatch(self): tracer = get_dummy_tracer() # Test patch idempotence patch() patch() client = mongoengine.connect(port=MONGO_CONFIG['port']) Pin.get_from(client).clone(tracer=tracer).onto(client) Artist.drop_collection() spans = tracer.writer.pop() assert spans, spans eq_(len(spans), 1) # Test unpatch mongoengine.connection.disconnect() unpatch() mongoengine.connect(port=MONGO_CONFIG['port']) Artist.drop_collection() spans = tracer.writer.pop() assert not spans, spans # Test patch again patch() client = mongoengine.connect(port=MONGO_CONFIG['port']) Pin.get_from(client).clone(tracer=tracer).onto(client) Artist.drop_collection() spans = tracer.writer.pop() assert spans, spans eq_(len(spans), 1)
def test_patch_unpatch(self): tracer = DummyTracer() # Test patch idempotence patch() patch() client = mongoengine.connect(port=MONGO_CONFIG["port"]) Pin.get_from(client).clone(tracer=tracer).onto(client) Artist.drop_collection() spans = tracer.pop() assert spans, spans assert len(spans) == 1 mongoengine.connection.disconnect() tracer.pop() # Test unpatch unpatch() mongoengine.connect(port=MONGO_CONFIG["port"]) Artist.drop_collection() spans = tracer.pop() assert not spans, spans # Test patch again patch() client = mongoengine.connect(port=MONGO_CONFIG["port"]) Pin.get_from(client).clone(tracer=tracer).onto(client) Artist.drop_collection() spans = tracer.pop() assert spans, spans assert len(spans) == 1
def tearDown(self): unpatch() # Disconnect and remove the client mongoengine.connection.disconnect()