258 lines
		
	
	
		
			7.5 KiB
		
	
	
	
		
			Python
		
	
	
	
			
		
		
	
	
			258 lines
		
	
	
		
			7.5 KiB
		
	
	
	
		
			Python
		
	
	
	
| import sys
 | |
| 
 | |
| import py, pytest
 | |
| import _pytest.assertion as plugin
 | |
| from _pytest.assertion import reinterpret, util
 | |
| 
 | |
| needsnewassert = pytest.mark.skipif("sys.version_info < (2,6)")
 | |
| 
 | |
| def interpret(expr):
 | |
|     return reinterpret.reinterpret(expr, py.code.Frame(sys._getframe(1)))
 | |
| 
 | |
| class TestBinReprIntegration:
 | |
|     pytestmark = needsnewassert
 | |
| 
 | |
|     def pytest_funcarg__hook(self, request):
 | |
|         class MockHook(object):
 | |
|             def __init__(self):
 | |
|                 self.called = False
 | |
|                 self.args = tuple()
 | |
|                 self.kwargs = dict()
 | |
| 
 | |
|             def __call__(self, op, left, right):
 | |
|                 self.called = True
 | |
|                 self.op = op
 | |
|                 self.left = left
 | |
|                 self.right = right
 | |
|         mockhook = MockHook()
 | |
|         monkeypatch = request.getfuncargvalue("monkeypatch")
 | |
|         monkeypatch.setattr(util, '_reprcompare', mockhook)
 | |
|         return mockhook
 | |
| 
 | |
|     def test_pytest_assertrepr_compare_called(self, hook):
 | |
|         interpret('assert 0 == 1')
 | |
|         assert hook.called
 | |
| 
 | |
| 
 | |
|     def test_pytest_assertrepr_compare_args(self, hook):
 | |
|         interpret('assert [0, 1] == [0, 2]')
 | |
|         assert hook.op == '=='
 | |
|         assert hook.left == [0, 1]
 | |
|         assert hook.right == [0, 2]
 | |
| 
 | |
|     def test_configure_unconfigure(self, testdir, hook):
 | |
|         assert hook == util._reprcompare
 | |
|         config = testdir.parseconfig()
 | |
|         plugin.pytest_configure(config)
 | |
|         assert hook != util._reprcompare
 | |
|         from _pytest.config import pytest_unconfigure
 | |
|         pytest_unconfigure(config)
 | |
|         assert hook == util._reprcompare
 | |
| 
 | |
| def callequal(left, right):
 | |
|     return plugin.pytest_assertrepr_compare('==', left, right)
 | |
| 
 | |
| class TestAssert_reprcompare:
 | |
|     def test_different_types(self):
 | |
|         assert callequal([0, 1], 'foo') is None
 | |
| 
 | |
|     def test_summary(self):
 | |
|         summary = callequal([0, 1], [0, 2])[0]
 | |
|         assert len(summary) < 65
 | |
| 
 | |
|     def test_text_diff(self):
 | |
|         diff = callequal('spam', 'eggs')[1:]
 | |
|         assert '- spam' in diff
 | |
|         assert '+ eggs' in diff
 | |
| 
 | |
|     def test_multiline_text_diff(self):
 | |
|         left = 'foo\nspam\nbar'
 | |
|         right = 'foo\neggs\nbar'
 | |
|         diff = callequal(left, right)
 | |
|         assert '- spam' in diff
 | |
|         assert '+ eggs' in diff
 | |
| 
 | |
|     def test_list(self):
 | |
|         expl = callequal([0, 1], [0, 2])
 | |
|         assert len(expl) > 1
 | |
| 
 | |
|     def test_list_different_lenghts(self):
 | |
|         expl = callequal([0, 1], [0, 1, 2])
 | |
|         assert len(expl) > 1
 | |
|         expl = callequal([0, 1, 2], [0, 1])
 | |
|         assert len(expl) > 1
 | |
| 
 | |
|     def test_dict(self):
 | |
|         expl = callequal({'a': 0}, {'a': 1})
 | |
|         assert len(expl) > 1
 | |
| 
 | |
|     def test_set(self):
 | |
|         expl = callequal(set([0, 1]), set([0, 2]))
 | |
|         assert len(expl) > 1
 | |
| 
 | |
|     def test_list_tuples(self):
 | |
|         expl = callequal([], [(1,2)])
 | |
|         assert len(expl) > 1
 | |
|         expl = callequal([(1,2)], [])
 | |
|         assert len(expl) > 1
 | |
| 
 | |
|     def test_list_bad_repr(self):
 | |
|         class A:
 | |
|             def __repr__(self):
 | |
|                 raise ValueError(42)
 | |
|         expl = callequal([], [A()])
 | |
|         assert 'ValueError' in "".join(expl)
 | |
|         expl = callequal({}, {'1': A()})
 | |
|         assert 'faulty' in "".join(expl)
 | |
| 
 | |
|     def test_one_repr_empty(self):
 | |
|         """
 | |
|         the faulty empty string repr did trigger
 | |
|         a unbound local error in _diff_text
 | |
|         """
 | |
|         class A(str):
 | |
|             def __repr__(self):
 | |
|                 return ''
 | |
|         expl = callequal(A(), '')
 | |
|         assert not expl
 | |
| 
 | |
|     def test_repr_no_exc(self):
 | |
|         expl = ' '.join(callequal('foo', 'bar'))
 | |
|         assert 'raised in repr()' not in expl
 | |
| 
 | |
| @pytest.mark.skipif("config._assertstate.mode != 'on'")
 | |
| def test_rewritten():
 | |
|     assert "@py_builtins" in globals()
 | |
| 
 | |
| def test_reprcompare_notin():
 | |
|     detail = plugin.pytest_assertrepr_compare('not in', 'foo', 'aaafoobbb')[1:]
 | |
|     assert detail == ["'foo' is contained here:", '  aaafoobbb', '?    +++']
 | |
| 
 | |
| @needsnewassert
 | |
| def test_pytest_assertrepr_compare_integration(testdir):
 | |
|     testdir.makepyfile("""
 | |
|         def test_hello():
 | |
|             x = set(range(100))
 | |
|             y = x.copy()
 | |
|             y.remove(50)
 | |
|             assert x == y
 | |
|     """)
 | |
|     result = testdir.runpytest()
 | |
|     result.stdout.fnmatch_lines([
 | |
|         "*def test_hello():*",
 | |
|         "*assert x == y*",
 | |
|         "*E*Extra items*left*",
 | |
|         "*E*50*",
 | |
|     ])
 | |
| 
 | |
| @needsnewassert
 | |
| def test_sequence_comparison_uses_repr(testdir):
 | |
|     testdir.makepyfile("""
 | |
|         def test_hello():
 | |
|             x = set("hello x")
 | |
|             y = set("hello y")
 | |
|             assert x == y
 | |
|     """)
 | |
|     result = testdir.runpytest()
 | |
|     result.stdout.fnmatch_lines([
 | |
|         "*def test_hello():*",
 | |
|         "*assert x == y*",
 | |
|         "*E*Extra items*left*",
 | |
|         "*E*'x'*",
 | |
|         "*E*Extra items*right*",
 | |
|         "*E*'y'*",
 | |
|     ])
 | |
| 
 | |
| 
 | |
| def test_assertion_options(testdir):
 | |
|     testdir.makepyfile("""
 | |
|         def test_hello():
 | |
|             x = 3
 | |
|             assert x == 4
 | |
|     """)
 | |
|     result = testdir.runpytest()
 | |
|     assert "3 == 4" in result.stdout.str()
 | |
|     off_options = (("--no-assert",),
 | |
|                    ("--nomagic",),
 | |
|                    ("--no-assert", "--nomagic"),
 | |
|                    ("--assertmode=off",),
 | |
|                    ("--assertmode=off", "--no-assert"),
 | |
|                    ("--assertmode=off", "--nomagic"),
 | |
|                    ("--assertmode=off," "--no-assert", "--nomagic"))
 | |
|     for opt in off_options:
 | |
|         result = testdir.runpytest(*opt)
 | |
|         assert "3 == 4" not in result.stdout.str()
 | |
|     for mode in "on", "old":
 | |
|         for other_opt in off_options[:3]:
 | |
|             opt = ("--assertmode=" + mode,) + other_opt
 | |
|             result = testdir.runpytest(*opt)
 | |
|             assert result.ret == 3
 | |
|             assert "assertion options conflict" in result.stderr.str()
 | |
| 
 | |
| def test_old_assert_mode(testdir):
 | |
|     testdir.makepyfile("""
 | |
|         def test_in_old_mode():
 | |
|             assert "@py_builtins" not in globals()
 | |
|     """)
 | |
|     result = testdir.runpytest("--assertmode=old")
 | |
|     assert result.ret == 0
 | |
| 
 | |
| def test_triple_quoted_string_issue113(testdir):
 | |
|     testdir.makepyfile("""
 | |
|         def test_hello():
 | |
|             assert "" == '''
 | |
|     '''""")
 | |
|     result = testdir.runpytest("--fulltrace")
 | |
|     result.stdout.fnmatch_lines([
 | |
|         "*1 failed*",
 | |
|     ])
 | |
|     assert 'SyntaxError' not in result.stdout.str()
 | |
| 
 | |
| def test_traceback_failure(testdir):
 | |
|     p1 = testdir.makepyfile("""
 | |
|         def g():
 | |
|             return 2
 | |
|         def f(x):
 | |
|             assert x == g()
 | |
|         def test_onefails():
 | |
|             f(3)
 | |
|     """)
 | |
|     result = testdir.runpytest(p1)
 | |
|     result.stdout.fnmatch_lines([
 | |
|         "*test_traceback_failure.py F",
 | |
|         "====* FAILURES *====",
 | |
|         "____*____",
 | |
|         "",
 | |
|         "    def test_onefails():",
 | |
|         ">       f(3)",
 | |
|         "",
 | |
|         "*test_*.py:6: ",
 | |
|         "_ _ _ *",
 | |
|         #"",
 | |
|         "    def f(x):",
 | |
|         ">       assert x == g()",
 | |
|         "E       assert 3 == 2",
 | |
|         "E        +  where 2 = g()",
 | |
|         "",
 | |
|         "*test_traceback_failure.py:4: AssertionError"
 | |
|     ])
 | |
| 
 | |
| @pytest.mark.skipif("sys.version_info < (2,5) or '__pypy__' in sys.builtin_module_names")
 | |
| def test_warn_missing(testdir):
 | |
|     p1 = testdir.makepyfile("")
 | |
|     result = testdir.run(sys.executable, "-OO", "-m", "pytest", "-h")
 | |
|     result.stderr.fnmatch_lines([
 | |
|         "*WARNING*assertion*",
 | |
|     ])
 | |
|     result = testdir.run(sys.executable, "-OO", "-m", "pytest", "--no-assert")
 | |
|     result.stderr.fnmatch_lines([
 | |
|         "*WARNING*assertion*",
 | |
|     ])
 | |
| 
 | |
| def test_load_fake_pyc(testdir):
 | |
|     path = testdir.makepyfile("x = 'hello'")
 | |
|     co = compile("x = 'bye'", str(path), "exec")
 | |
|     plugin._write_pyc(co, path)
 | |
|     mod = path.pyimport()
 | |
|     assert mod.x == "bye"
 |