1 files changed, 176 insertions, 0 deletions
diff --git a/test_pytest_catchlog.py b/test_pytest_catchlog.py
new file mode 100644
index 0000000..0e9d55f
--- /dev/null
+++ b/test_pytest_catchlog.py
@@ -0,0 +1,176 @@
+import py
+
+pytest_plugins = 'pytester', 'catchlog'
+
+
+def test_nothing_logged(testdir):
+    testdir.makepyfile('''
+        import sys
+        import logging
+
+        pytest_plugins = 'catchlog'
+
+        def test_foo():
+            sys.stdout.write('text going to stdout')
+            sys.stderr.write('text going to stderr')
+            assert False
+        ''')
+    result = testdir.runpytest()
+    assert result.ret == 1
+    result.stdout.fnmatch_lines(['*- Captured stdout call -*',
+                                 'text going to stdout'])
+    result.stdout.fnmatch_lines(['*- Captured stderr call -*',
+                                 'text going to stderr'])
+    py.test.raises(Exception, result.stdout.fnmatch_lines,
+                   ['*- Captured log -*'])
+
+
+def test_messages_logged(testdir):
+    testdir.makepyfile('''
+        import sys
+        import logging
+
+        pytest_plugins = 'catchlog'
+
+        def test_foo():
+            sys.stdout.write('text going to stdout')
+            sys.stderr.write('text going to stderr')
+            logging.getLogger().info('text going to logger')
+            assert False
+        ''')
+    result = testdir.runpytest()
+    assert result.ret == 1
+    result.stdout.fnmatch_lines(['*- Captured log -*',
+                                 '*text going to logger*'])
+    result.stdout.fnmatch_lines(['*- Captured stdout call -*',
+                                 'text going to stdout'])
+    result.stdout.fnmatch_lines(['*- Captured stderr call -*',
+                                 'text going to stderr'])
+
+
+def test_change_level(testdir):
+    testdir.makepyfile('''
+        import sys
+        import logging
+
+        pytest_plugins = 'catchlog'
+
+        def test_foo(caplog):
+            caplog.set_level(logging.INFO)
+            log = logging.getLogger()
+            log.debug('handler DEBUG level')
+            log.info('handler INFO level')
+
+            caplog.set_level(logging.CRITICAL, logger='root.baz')
+            log = logging.getLogger('root.baz')
+            log.warning('logger WARNING level')
+            log.critical('logger CRITICAL level')
+
+            assert False
+        ''')
+    result = testdir.runpytest()
+    assert result.ret == 1
+    result.stdout.fnmatch_lines(['*- Captured log -*',
+                                 '*handler INFO level*',
+                                 '*logger CRITICAL level*'])
+    py.test.raises(Exception, result.stdout.fnmatch_lines,
+                   ['*- Captured log -*', '*handler DEBUG level*'])
+    py.test.raises(Exception, result.stdout.fnmatch_lines,
+                   ['*- Captured log -*', '*logger WARNING level*'])
+
+
+@py.test.mark.skipif('sys.version_info < (2,5)')
+def test_with_statement(testdir):
+    testdir.makepyfile('''
+        from __future__ import with_statement
+        import sys
+        import logging
+
+        pytest_plugins = 'catchlog'
+
+        def test_foo(caplog):
+            with caplog.at_level(logging.INFO):
+                log = logging.getLogger()
+                log.debug('handler DEBUG level')
+                log.info('handler INFO level')
+
+                with caplog.at_level(logging.CRITICAL, logger='root.baz'):
+                    log = logging.getLogger('root.baz')
+                    log.warning('logger WARNING level')
+                    log.critical('logger CRITICAL level')
+
+            assert False
+        ''')
+    result = testdir.runpytest()
+    assert result.ret == 1
+    result.stdout.fnmatch_lines(['*- Captured log -*',
+                                 '*handler INFO level*',
+                                 '*logger CRITICAL level*'])
+    py.test.raises(Exception, result.stdout.fnmatch_lines,
+                   ['*- Captured log -*', '*handler DEBUG level*'])
+    py.test.raises(Exception, result.stdout.fnmatch_lines,
+                   ['*- Captured log -*', '*logger WARNING level*'])
+
+
+def test_log_access(testdir):
+    testdir.makepyfile('''
+        import sys
+        import logging
+
+        pytest_plugins = 'catchlog'
+
+        def test_foo(caplog):
+            logging.getLogger().info('boo %s', 'arg')
+            assert caplog.records()[0].levelname == 'INFO'
+            assert caplog.records()[0].msg == 'boo %s'
+            assert 'boo arg' in caplog.text()
+        ''')
+    result = testdir.runpytest()
+    assert result.ret == 0
+
+
+def test_funcarg_help(testdir):
+    result = testdir.runpytest('--funcargs')
+    result.stdout.fnmatch_lines(['*caplog*'])
+
+
+def test_record_tuples(testdir):
+    testdir.makepyfile('''
+        import sys
+        import logging
+
+        pytest_plugins = 'catchlog'
+
+        def test_foo(caplog):
+            logging.getLogger().info('boo %s', 'arg')
+
+            assert caplog.record_tuples() == [
+                ('root', logging.INFO, 'boo arg'),
+            ]
+        ''')
+    result = testdir.runpytest()
+    assert result.ret == 0
+
+
+def test_disable_log_capturing(testdir):
+    testdir.makepyfile('''
+        import sys
+        import logging
+
+        pytest_plugins = 'catchlog'
+
+        def test_foo(caplog):
+            sys.stdout.write('text going to stdout')
+            logging.getLogger().warning('catch me if you can!')
+            sys.stderr.write('text going to stderr')
+            assert False
+        ''')
+    result = testdir.runpytest('--no-print-logs')
+    print(result.stdout)
+    assert result.ret == 1
+    result.stdout.fnmatch_lines(['*- Captured stdout call -*',
+                                 'text going to stdout'])
+    result.stdout.fnmatch_lines(['*- Captured stderr call -*',
+                                 'text going to stderr'])
+    py.test.raises(Exception, result.stdout.fnmatch_lines,
+                   ['*- Captured log -*'])