add wrap_all test

lithoxyl/tests/test_wrap_all.py

@@ -0,0 +1,24 @@
+
+import json
+
+from lithoxyl import Logger, AggregateSink
+from lithoxyl.utils import wrap_all, unwrap_all
+
+
+def test_wrap_all_json():
+    agg_sink = AggregateSink()
+    log = Logger('wrapper_log', sinks=[agg_sink])
+
+    wrap_all(log, 'info', json)
+
+    json.loads('{}')
+
+    assert agg_sink.begin_events[0].name == 'json.loads'
+
+    unwrap_all(json)
+
+    json.loads('{}')
+
+    assert len(agg_sink.begin_events) == 1
+
+    return