Use caplog to capture test records

Author: stijndehaes

tests/ext/aiohttp/test_client.py

@@ -145,7 +145,7 @@ async def test_no_segment_raise(loop, recorder):
                 pass
 
 
-async def test_no_segment_log_error(loop, recorder, capsys):
+async def test_no_segment_log_error(loop, recorder, caplog):
     xray_recorder.configure(context_missing='LOG_ERROR')
     trace_config = aws_xray_trace_config()
     status_code = 200
@@ -156,11 +156,10 @@ async def test_no_segment_log_error(loop, recorder, capsys):
 
     # Just check that the request was done correctly
     assert status_received == status_code
-    captured = capsys.readouterr()
-    assert MISSING_SEGMENT_MSG in captured.out
+    assert [MISSING_SEGMENT_MSG] == [rec.message for rec in caplog.records]
 
 
-async def test_no_segment_ignore(loop, recorder, capsys):
+async def test_no_segment_ignore(loop, recorder, caplog):
     xray_recorder.configure(context_missing='IGNORE')
     trace_config = aws_xray_trace_config()
     status_code = 200
@@ -171,5 +170,4 @@ async def test_no_segment_ignore(loop, recorder, capsys):
 
     # Just check that the request was done correctly
     assert status_received == status_code
-    captured = capsys.readouterr()
-    assert MISSING_SEGMENT_MSG not in captured.out
+    assert [] == [rec.message for rec in caplog.records]