2019-05-15 06:56:31 +08:00
|
|
|
# -*- coding: utf-8 -*-
|
2018-10-25 15:01:29 +08:00
|
|
|
from __future__ import absolute_import
|
|
|
|
from __future__ import division
|
|
|
|
from __future__ import print_function
|
|
|
|
|
2016-01-24 01:08:18 +08:00
|
|
|
import os
|
2018-11-01 01:22:42 +08:00
|
|
|
import re
|
2016-01-24 01:08:18 +08:00
|
|
|
import sys
|
2015-12-28 05:42:42 +08:00
|
|
|
import textwrap
|
|
|
|
|
2018-09-30 06:55:04 +08:00
|
|
|
import six
|
|
|
|
|
2010-11-18 05:12:16 +08:00
|
|
|
import pytest
|
2016-07-12 09:03:53 +08:00
|
|
|
from _pytest.monkeypatch import MonkeyPatch
|
2009-09-06 22:59:39 +08:00
|
|
|
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2016-07-12 09:03:53 +08:00
|
|
|
@pytest.fixture
|
2017-01-04 21:31:44 +08:00
|
|
|
def mp():
|
2012-01-04 19:40:57 +08:00
|
|
|
cwd = os.getcwd()
|
|
|
|
sys_path = list(sys.path)
|
2017-01-04 21:31:44 +08:00
|
|
|
yield MonkeyPatch()
|
|
|
|
sys.path[:] = sys_path
|
|
|
|
os.chdir(cwd)
|
2012-01-04 19:40:57 +08:00
|
|
|
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2009-09-06 22:59:39 +08:00
|
|
|
def test_setattr():
|
2017-02-17 02:41:51 +08:00
|
|
|
class A(object):
|
2009-09-06 22:59:39 +08:00
|
|
|
x = 1
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2009-09-07 23:53:50 +08:00
|
|
|
monkeypatch = MonkeyPatch()
|
2018-11-23 02:05:10 +08:00
|
|
|
pytest.raises(AttributeError, monkeypatch.setattr, A, "notexists", 2)
|
2018-05-23 22:48:46 +08:00
|
|
|
monkeypatch.setattr(A, "y", 2, raising=False)
|
2009-09-07 23:53:50 +08:00
|
|
|
assert A.y == 2
|
|
|
|
monkeypatch.undo()
|
2018-05-23 22:48:46 +08:00
|
|
|
assert not hasattr(A, "y")
|
2009-09-07 23:53:50 +08:00
|
|
|
|
2009-09-06 22:59:39 +08:00
|
|
|
monkeypatch = MonkeyPatch()
|
2018-05-23 22:48:46 +08:00
|
|
|
monkeypatch.setattr(A, "x", 2)
|
2009-09-06 22:59:39 +08:00
|
|
|
assert A.x == 2
|
2018-05-23 22:48:46 +08:00
|
|
|
monkeypatch.setattr(A, "x", 3)
|
2009-09-06 22:59:39 +08:00
|
|
|
assert A.x == 3
|
|
|
|
monkeypatch.undo()
|
|
|
|
assert A.x == 1
|
|
|
|
|
|
|
|
A.x = 5
|
2016-01-24 01:08:18 +08:00
|
|
|
monkeypatch.undo() # double-undo makes no modification
|
2009-09-06 22:59:39 +08:00
|
|
|
assert A.x == 5
|
|
|
|
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2017-02-17 02:41:51 +08:00
|
|
|
class TestSetattrWithImportPath(object):
|
2013-08-07 21:35:27 +08:00
|
|
|
def test_string_expression(self, monkeypatch):
|
2013-09-27 18:33:06 +08:00
|
|
|
monkeypatch.setattr("os.path.abspath", lambda x: "hello2")
|
2013-08-07 21:35:27 +08:00
|
|
|
assert os.path.abspath("123") == "hello2"
|
|
|
|
|
|
|
|
def test_string_expression_class(self, monkeypatch):
|
2013-09-27 18:33:06 +08:00
|
|
|
monkeypatch.setattr("_pytest.config.Config", 42)
|
2013-08-07 21:35:27 +08:00
|
|
|
import _pytest
|
2018-05-23 22:48:46 +08:00
|
|
|
|
2013-08-07 21:35:27 +08:00
|
|
|
assert _pytest.config.Config == 42
|
|
|
|
|
2013-10-21 20:01:02 +08:00
|
|
|
def test_unicode_string(self, monkeypatch):
|
2013-11-12 20:48:17 +08:00
|
|
|
monkeypatch.setattr("_pytest.config.Config", 42)
|
2013-10-21 20:01:02 +08:00
|
|
|
import _pytest
|
2018-05-23 22:48:46 +08:00
|
|
|
|
2013-10-21 20:01:02 +08:00
|
|
|
assert _pytest.config.Config == 42
|
2013-11-12 20:48:17 +08:00
|
|
|
monkeypatch.delattr("_pytest.config.Config")
|
2013-10-21 20:01:02 +08:00
|
|
|
|
2013-08-07 22:49:29 +08:00
|
|
|
def test_wrong_target(self, monkeypatch):
|
2013-09-27 18:33:06 +08:00
|
|
|
pytest.raises(TypeError, lambda: monkeypatch.setattr(None, None))
|
2013-08-07 22:49:29 +08:00
|
|
|
|
|
|
|
def test_unknown_import(self, monkeypatch):
|
2018-05-23 22:48:46 +08:00
|
|
|
pytest.raises(ImportError, lambda: monkeypatch.setattr("unkn123.classx", None))
|
2013-08-07 22:49:29 +08:00
|
|
|
|
|
|
|
def test_unknown_attr(self, monkeypatch):
|
2018-05-23 22:48:46 +08:00
|
|
|
pytest.raises(
|
|
|
|
AttributeError, lambda: monkeypatch.setattr("os.path.qweqwe", None)
|
|
|
|
)
|
2013-09-27 18:33:06 +08:00
|
|
|
|
2015-05-20 04:59:49 +08:00
|
|
|
def test_unknown_attr_non_raising(self, monkeypatch):
|
2015-06-16 05:28:31 +08:00
|
|
|
# https://github.com/pytest-dev/pytest/issues/746
|
2018-05-23 22:48:46 +08:00
|
|
|
monkeypatch.setattr("os.path.qweqwe", 42, raising=False)
|
2015-05-20 04:59:49 +08:00
|
|
|
assert os.path.qweqwe == 42
|
|
|
|
|
2013-09-27 18:33:06 +08:00
|
|
|
def test_delattr(self, monkeypatch):
|
|
|
|
monkeypatch.delattr("os.path.abspath")
|
|
|
|
assert not hasattr(os.path, "abspath")
|
|
|
|
monkeypatch.undo()
|
|
|
|
assert os.path.abspath
|
2013-08-07 22:49:29 +08:00
|
|
|
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2009-09-06 22:59:39 +08:00
|
|
|
def test_delattr():
|
2017-02-17 02:41:51 +08:00
|
|
|
class A(object):
|
2009-09-06 22:59:39 +08:00
|
|
|
x = 1
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2009-09-06 22:59:39 +08:00
|
|
|
monkeypatch = MonkeyPatch()
|
2018-05-23 22:48:46 +08:00
|
|
|
monkeypatch.delattr(A, "x")
|
|
|
|
assert not hasattr(A, "x")
|
2009-09-06 22:59:39 +08:00
|
|
|
monkeypatch.undo()
|
|
|
|
assert A.x == 1
|
|
|
|
|
|
|
|
monkeypatch = MonkeyPatch()
|
2018-05-23 22:48:46 +08:00
|
|
|
monkeypatch.delattr(A, "x")
|
2018-11-23 02:05:10 +08:00
|
|
|
pytest.raises(AttributeError, monkeypatch.delattr, A, "y")
|
2018-05-23 22:48:46 +08:00
|
|
|
monkeypatch.delattr(A, "y", raising=False)
|
|
|
|
monkeypatch.setattr(A, "x", 5, raising=False)
|
2009-09-06 22:59:39 +08:00
|
|
|
assert A.x == 5
|
|
|
|
monkeypatch.undo()
|
|
|
|
assert A.x == 1
|
|
|
|
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2009-09-06 22:59:39 +08:00
|
|
|
def test_setitem():
|
2018-05-23 22:48:46 +08:00
|
|
|
d = {"x": 1}
|
2009-09-06 22:59:39 +08:00
|
|
|
monkeypatch = MonkeyPatch()
|
2018-05-23 22:48:46 +08:00
|
|
|
monkeypatch.setitem(d, "x", 2)
|
|
|
|
monkeypatch.setitem(d, "y", 1700)
|
|
|
|
monkeypatch.setitem(d, "y", 1700)
|
|
|
|
assert d["x"] == 2
|
|
|
|
assert d["y"] == 1700
|
|
|
|
monkeypatch.setitem(d, "x", 3)
|
|
|
|
assert d["x"] == 3
|
2009-09-06 22:59:39 +08:00
|
|
|
monkeypatch.undo()
|
2018-05-23 22:48:46 +08:00
|
|
|
assert d["x"] == 1
|
|
|
|
assert "y" not in d
|
|
|
|
d["x"] = 5
|
2009-09-06 22:59:39 +08:00
|
|
|
monkeypatch.undo()
|
2018-05-23 22:48:46 +08:00
|
|
|
assert d["x"] == 5
|
2009-09-06 22:59:39 +08:00
|
|
|
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2011-12-17 06:41:23 +08:00
|
|
|
def test_setitem_deleted_meanwhile():
|
|
|
|
d = {}
|
|
|
|
monkeypatch = MonkeyPatch()
|
2018-05-23 22:48:46 +08:00
|
|
|
monkeypatch.setitem(d, "x", 2)
|
|
|
|
del d["x"]
|
2011-12-17 06:41:23 +08:00
|
|
|
monkeypatch.undo()
|
|
|
|
assert not d
|
|
|
|
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2011-12-28 23:49:13 +08:00
|
|
|
@pytest.mark.parametrize("before", [True, False])
|
|
|
|
def test_setenv_deleted_meanwhile(before):
|
|
|
|
key = "qwpeoip123"
|
|
|
|
if before:
|
|
|
|
os.environ[key] = "world"
|
2011-12-17 06:41:23 +08:00
|
|
|
monkeypatch = MonkeyPatch()
|
2018-05-23 22:48:46 +08:00
|
|
|
monkeypatch.setenv(key, "hello")
|
2011-12-28 23:49:13 +08:00
|
|
|
del os.environ[key]
|
2011-12-17 06:41:23 +08:00
|
|
|
monkeypatch.undo()
|
2011-12-28 23:49:13 +08:00
|
|
|
if before:
|
|
|
|
assert os.environ[key] == "world"
|
|
|
|
del os.environ[key]
|
|
|
|
else:
|
|
|
|
assert key not in os.environ
|
2011-12-17 06:41:23 +08:00
|
|
|
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2009-09-06 22:59:39 +08:00
|
|
|
def test_delitem():
|
2018-05-23 22:48:46 +08:00
|
|
|
d = {"x": 1}
|
2009-09-06 22:59:39 +08:00
|
|
|
monkeypatch = MonkeyPatch()
|
2018-05-23 22:48:46 +08:00
|
|
|
monkeypatch.delitem(d, "x")
|
|
|
|
assert "x" not in d
|
|
|
|
monkeypatch.delitem(d, "y", raising=False)
|
2018-11-23 02:05:10 +08:00
|
|
|
pytest.raises(KeyError, monkeypatch.delitem, d, "y")
|
2009-09-06 22:59:39 +08:00
|
|
|
assert not d
|
2018-05-23 22:48:46 +08:00
|
|
|
monkeypatch.setitem(d, "y", 1700)
|
|
|
|
assert d["y"] == 1700
|
|
|
|
d["hello"] = "world"
|
|
|
|
monkeypatch.setitem(d, "x", 1500)
|
|
|
|
assert d["x"] == 1500
|
2009-09-06 22:59:39 +08:00
|
|
|
monkeypatch.undo()
|
2018-05-23 22:48:46 +08:00
|
|
|
assert d == {"hello": "world", "x": 1}
|
2009-09-06 22:59:39 +08:00
|
|
|
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2009-09-06 22:59:39 +08:00
|
|
|
def test_setenv():
|
|
|
|
monkeypatch = MonkeyPatch()
|
2018-10-02 05:48:41 +08:00
|
|
|
with pytest.warns(pytest.PytestWarning):
|
|
|
|
monkeypatch.setenv("XYZ123", 2)
|
2009-09-06 22:59:39 +08:00
|
|
|
import os
|
2018-05-23 22:48:46 +08:00
|
|
|
|
|
|
|
assert os.environ["XYZ123"] == "2"
|
2009-09-06 22:59:39 +08:00
|
|
|
monkeypatch.undo()
|
2018-05-23 22:48:46 +08:00
|
|
|
assert "XYZ123" not in os.environ
|
2009-09-06 22:59:39 +08:00
|
|
|
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2009-09-06 22:59:39 +08:00
|
|
|
def test_delenv():
|
2018-05-23 22:48:46 +08:00
|
|
|
name = "xyz1234"
|
2010-07-27 03:15:15 +08:00
|
|
|
assert name not in os.environ
|
2009-09-06 22:59:39 +08:00
|
|
|
monkeypatch = MonkeyPatch()
|
2018-11-23 02:05:10 +08:00
|
|
|
pytest.raises(KeyError, monkeypatch.delenv, name, raising=True)
|
2009-09-06 22:59:39 +08:00
|
|
|
monkeypatch.delenv(name, raising=False)
|
|
|
|
monkeypatch.undo()
|
|
|
|
os.environ[name] = "1"
|
|
|
|
try:
|
|
|
|
monkeypatch = MonkeyPatch()
|
|
|
|
monkeypatch.delenv(name)
|
2010-07-27 03:15:15 +08:00
|
|
|
assert name not in os.environ
|
2009-09-06 22:59:39 +08:00
|
|
|
monkeypatch.setenv(name, "3")
|
|
|
|
assert os.environ[name] == "3"
|
|
|
|
monkeypatch.undo()
|
|
|
|
assert os.environ[name] == "1"
|
|
|
|
finally:
|
|
|
|
if name in os.environ:
|
|
|
|
del os.environ[name]
|
|
|
|
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2018-10-02 05:48:41 +08:00
|
|
|
class TestEnvironWarnings(object):
|
2018-09-30 06:55:04 +08:00
|
|
|
"""
|
2018-10-02 05:48:41 +08:00
|
|
|
os.environ keys and values should be native strings, otherwise it will cause problems with other modules (notably
|
|
|
|
subprocess). On Python 2 os.environ accepts anything without complaining, while Python 3 does the right thing
|
|
|
|
and raises an error.
|
2018-09-30 06:55:04 +08:00
|
|
|
"""
|
|
|
|
|
|
|
|
VAR_NAME = u"PYTEST_INTERNAL_MY_VAR"
|
|
|
|
|
2018-10-02 05:48:41 +08:00
|
|
|
@pytest.mark.skipif(six.PY3, reason="Python 2 only test")
|
2018-09-30 06:55:04 +08:00
|
|
|
def test_setenv_unicode_key(self, monkeypatch):
|
|
|
|
with pytest.warns(
|
|
|
|
pytest.PytestWarning,
|
|
|
|
match="Environment variable name {!r} should be str".format(self.VAR_NAME),
|
|
|
|
):
|
|
|
|
monkeypatch.setenv(self.VAR_NAME, "2")
|
|
|
|
|
2018-10-02 05:48:41 +08:00
|
|
|
@pytest.mark.skipif(six.PY3, reason="Python 2 only test")
|
2018-09-30 06:55:04 +08:00
|
|
|
def test_delenv_unicode_key(self, monkeypatch):
|
|
|
|
with pytest.warns(
|
|
|
|
pytest.PytestWarning,
|
|
|
|
match="Environment variable name {!r} should be str".format(self.VAR_NAME),
|
|
|
|
):
|
|
|
|
monkeypatch.delenv(self.VAR_NAME, raising=False)
|
|
|
|
|
2018-10-02 05:48:41 +08:00
|
|
|
def test_setenv_non_str_warning(self, monkeypatch):
|
|
|
|
value = 2
|
|
|
|
msg = (
|
2018-11-01 01:22:42 +08:00
|
|
|
"Value of environment variable PYTEST_INTERNAL_MY_VAR type should be str, "
|
|
|
|
"but got 2 (type: int); converted to str implicitly"
|
2018-10-02 05:48:41 +08:00
|
|
|
)
|
2018-11-01 01:22:42 +08:00
|
|
|
with pytest.warns(pytest.PytestWarning, match=re.escape(msg)):
|
2018-10-02 05:48:41 +08:00
|
|
|
monkeypatch.setenv(str(self.VAR_NAME), value)
|
|
|
|
|
2018-09-30 06:55:04 +08:00
|
|
|
|
2009-09-06 22:59:39 +08:00
|
|
|
def test_setenv_prepend():
|
|
|
|
import os
|
2018-05-23 22:48:46 +08:00
|
|
|
|
2009-09-06 22:59:39 +08:00
|
|
|
monkeypatch = MonkeyPatch()
|
2018-10-02 05:48:41 +08:00
|
|
|
with pytest.warns(pytest.PytestWarning):
|
|
|
|
monkeypatch.setenv("XYZ123", 2, prepend="-")
|
2018-05-23 22:48:46 +08:00
|
|
|
assert os.environ["XYZ123"] == "2"
|
2018-10-02 05:48:41 +08:00
|
|
|
with pytest.warns(pytest.PytestWarning):
|
|
|
|
monkeypatch.setenv("XYZ123", 3, prepend="-")
|
2018-05-23 22:48:46 +08:00
|
|
|
assert os.environ["XYZ123"] == "3-2"
|
2009-09-06 22:59:39 +08:00
|
|
|
monkeypatch.undo()
|
2018-05-23 22:48:46 +08:00
|
|
|
assert "XYZ123" not in os.environ
|
2009-09-06 22:59:39 +08:00
|
|
|
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2009-09-06 22:59:39 +08:00
|
|
|
def test_monkeypatch_plugin(testdir):
|
2018-05-23 22:48:46 +08:00
|
|
|
reprec = testdir.inline_runsource(
|
|
|
|
"""
|
2009-09-06 22:59:39 +08:00
|
|
|
def test_method(monkeypatch):
|
2016-07-12 09:03:53 +08:00
|
|
|
assert monkeypatch.__class__.__name__ == "MonkeyPatch"
|
2018-05-23 22:48:46 +08:00
|
|
|
"""
|
|
|
|
)
|
2009-09-06 22:59:39 +08:00
|
|
|
res = reprec.countoutcomes()
|
|
|
|
assert tuple(res) == (1, 0, 0), res
|
|
|
|
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2012-01-04 19:40:57 +08:00
|
|
|
def test_syspath_prepend(mp):
|
2009-09-06 22:59:39 +08:00
|
|
|
old = list(sys.path)
|
2018-05-23 22:48:46 +08:00
|
|
|
mp.syspath_prepend("world")
|
|
|
|
mp.syspath_prepend("hello")
|
2012-01-04 19:40:57 +08:00
|
|
|
assert sys.path[0] == "hello"
|
|
|
|
assert sys.path[1] == "world"
|
|
|
|
mp.undo()
|
|
|
|
assert sys.path == old
|
|
|
|
mp.undo()
|
|
|
|
assert sys.path == old
|
|
|
|
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2012-01-04 19:40:57 +08:00
|
|
|
def test_syspath_prepend_double_undo(mp):
|
2018-08-26 05:17:52 +08:00
|
|
|
old_syspath = sys.path[:]
|
|
|
|
try:
|
|
|
|
mp.syspath_prepend("hello world")
|
|
|
|
mp.undo()
|
|
|
|
sys.path.append("more hello world")
|
|
|
|
mp.undo()
|
|
|
|
assert sys.path[-1] == "more hello world"
|
|
|
|
finally:
|
|
|
|
sys.path[:] = old_syspath
|
2012-01-04 19:43:19 +08:00
|
|
|
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2012-01-04 19:43:19 +08:00
|
|
|
def test_chdir_with_path_local(mp, tmpdir):
|
|
|
|
mp.chdir(tmpdir)
|
|
|
|
assert os.getcwd() == tmpdir.strpath
|
|
|
|
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2012-01-04 19:43:19 +08:00
|
|
|
def test_chdir_with_str(mp, tmpdir):
|
|
|
|
mp.chdir(tmpdir.strpath)
|
|
|
|
assert os.getcwd() == tmpdir.strpath
|
|
|
|
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2012-01-04 19:43:19 +08:00
|
|
|
def test_chdir_undo(mp, tmpdir):
|
|
|
|
cwd = os.getcwd()
|
|
|
|
mp.chdir(tmpdir)
|
|
|
|
mp.undo()
|
|
|
|
assert os.getcwd() == cwd
|
|
|
|
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2012-01-04 19:43:19 +08:00
|
|
|
def test_chdir_double_undo(mp, tmpdir):
|
|
|
|
mp.chdir(tmpdir.strpath)
|
|
|
|
mp.undo()
|
|
|
|
tmpdir.chdir()
|
|
|
|
mp.undo()
|
|
|
|
assert os.getcwd() == tmpdir.strpath
|
2012-09-01 15:58:10 +08:00
|
|
|
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2012-09-01 15:58:10 +08:00
|
|
|
def test_issue185_time_breaks(testdir):
|
2018-05-23 22:48:46 +08:00
|
|
|
testdir.makepyfile(
|
|
|
|
"""
|
2012-09-01 15:58:10 +08:00
|
|
|
import time
|
|
|
|
def test_m(monkeypatch):
|
|
|
|
def f():
|
|
|
|
raise Exception
|
|
|
|
monkeypatch.setattr(time, "time", f)
|
2018-05-23 22:48:46 +08:00
|
|
|
"""
|
|
|
|
)
|
2012-09-01 15:58:10 +08:00
|
|
|
result = testdir.runpytest()
|
2018-05-23 22:48:46 +08:00
|
|
|
result.stdout.fnmatch_lines(
|
|
|
|
"""
|
2012-09-01 15:58:10 +08:00
|
|
|
*1 passed*
|
2018-05-23 22:48:46 +08:00
|
|
|
"""
|
|
|
|
)
|
2012-09-26 00:15:13 +08:00
|
|
|
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2015-12-28 05:42:42 +08:00
|
|
|
def test_importerror(testdir):
|
|
|
|
p = testdir.mkpydir("package")
|
2018-05-23 22:48:46 +08:00
|
|
|
p.join("a.py").write(
|
|
|
|
textwrap.dedent(
|
|
|
|
"""\
|
2015-12-28 05:42:42 +08:00
|
|
|
import doesnotexist
|
|
|
|
|
|
|
|
x = 1
|
2018-05-23 22:48:46 +08:00
|
|
|
"""
|
|
|
|
)
|
|
|
|
)
|
|
|
|
testdir.tmpdir.join("test_importerror.py").write(
|
|
|
|
textwrap.dedent(
|
2018-11-01 23:40:35 +08:00
|
|
|
"""\
|
2015-12-28 05:42:42 +08:00
|
|
|
def test_importerror(monkeypatch):
|
|
|
|
monkeypatch.setattr('package.a.x', 2)
|
2018-05-23 22:48:46 +08:00
|
|
|
"""
|
|
|
|
)
|
|
|
|
)
|
2015-12-28 05:42:42 +08:00
|
|
|
result = testdir.runpytest()
|
2018-05-23 22:48:46 +08:00
|
|
|
result.stdout.fnmatch_lines(
|
|
|
|
"""
|
2016-01-24 02:31:17 +08:00
|
|
|
*import error in package.a: No module named {0}doesnotexist{0}*
|
2018-05-23 22:48:46 +08:00
|
|
|
""".format(
|
|
|
|
"'" if sys.version_info > (3, 0) else ""
|
|
|
|
)
|
|
|
|
)
|
2012-09-26 00:15:13 +08:00
|
|
|
|
|
|
|
|
|
|
|
class SampleNew(object):
|
|
|
|
@staticmethod
|
|
|
|
def hello():
|
|
|
|
return True
|
|
|
|
|
|
|
|
|
|
|
|
class SampleNewInherit(SampleNew):
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
2017-02-17 02:41:51 +08:00
|
|
|
class SampleOld(object):
|
2016-01-24 01:08:18 +08:00
|
|
|
# oldstyle on python2
|
2012-09-26 00:15:13 +08:00
|
|
|
@staticmethod
|
|
|
|
def hello():
|
|
|
|
return True
|
|
|
|
|
2016-01-24 01:08:18 +08:00
|
|
|
|
2012-09-26 00:15:13 +08:00
|
|
|
class SampleOldInherit(SampleOld):
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
2018-05-23 22:48:46 +08:00
|
|
|
@pytest.mark.parametrize(
|
|
|
|
"Sample",
|
|
|
|
[SampleNew, SampleNewInherit, SampleOld, SampleOldInherit],
|
|
|
|
ids=["new", "new-inherit", "old", "old-inherit"],
|
|
|
|
)
|
2012-09-26 00:15:13 +08:00
|
|
|
def test_issue156_undo_staticmethod(Sample):
|
|
|
|
monkeypatch = MonkeyPatch()
|
|
|
|
|
2018-05-23 22:48:46 +08:00
|
|
|
monkeypatch.setattr(Sample, "hello", None)
|
2012-09-26 00:15:13 +08:00
|
|
|
assert Sample.hello is None
|
|
|
|
|
|
|
|
monkeypatch.undo()
|
|
|
|
assert Sample.hello()
|
2016-01-24 02:12:10 +08:00
|
|
|
|
2017-07-17 07:25:09 +08:00
|
|
|
|
2018-12-13 04:20:24 +08:00
|
|
|
def test_undo_class_descriptors_delattr():
|
|
|
|
class SampleParent(object):
|
|
|
|
@classmethod
|
|
|
|
def hello(_cls):
|
|
|
|
pass
|
|
|
|
|
|
|
|
@staticmethod
|
|
|
|
def world():
|
|
|
|
pass
|
|
|
|
|
|
|
|
class SampleChild(SampleParent):
|
|
|
|
pass
|
|
|
|
|
|
|
|
monkeypatch = MonkeyPatch()
|
|
|
|
|
|
|
|
original_hello = SampleChild.hello
|
|
|
|
original_world = SampleChild.world
|
|
|
|
monkeypatch.delattr(SampleParent, "hello")
|
|
|
|
monkeypatch.delattr(SampleParent, "world")
|
|
|
|
assert getattr(SampleParent, "hello", None) is None
|
|
|
|
assert getattr(SampleParent, "world", None) is None
|
|
|
|
|
|
|
|
monkeypatch.undo()
|
|
|
|
assert original_hello == SampleChild.hello
|
|
|
|
assert original_world == SampleChild.world
|
|
|
|
|
|
|
|
|
2016-01-24 02:12:10 +08:00
|
|
|
def test_issue1338_name_resolving():
|
2018-05-23 22:48:46 +08:00
|
|
|
pytest.importorskip("requests")
|
2016-01-24 02:12:10 +08:00
|
|
|
monkeypatch = MonkeyPatch()
|
|
|
|
try:
|
2018-05-23 22:48:46 +08:00
|
|
|
monkeypatch.delattr("requests.sessions.Session.request")
|
2016-01-24 02:12:10 +08:00
|
|
|
finally:
|
2016-07-21 09:05:49 +08:00
|
|
|
monkeypatch.undo()
|
2018-04-11 01:18:05 +08:00
|
|
|
|
|
|
|
|
2018-04-13 21:00:07 +08:00
|
|
|
def test_context():
|
|
|
|
monkeypatch = MonkeyPatch()
|
|
|
|
|
2018-04-11 01:18:05 +08:00
|
|
|
import functools
|
2018-04-13 21:00:07 +08:00
|
|
|
import inspect
|
2018-04-11 01:18:05 +08:00
|
|
|
|
2018-04-13 21:00:07 +08:00
|
|
|
with monkeypatch.context() as m:
|
|
|
|
m.setattr(functools, "partial", 3)
|
|
|
|
assert not inspect.isclass(functools.partial)
|
|
|
|
assert inspect.isclass(functools.partial)
|
2019-03-22 21:36:11 +08:00
|
|
|
|
|
|
|
|
|
|
|
def test_syspath_prepend_with_namespace_packages(testdir, monkeypatch):
|
|
|
|
for dirname in "hello", "world":
|
|
|
|
d = testdir.mkdir(dirname)
|
|
|
|
ns = d.mkdir("ns_pkg")
|
|
|
|
ns.join("__init__.py").write(
|
|
|
|
"__import__('pkg_resources').declare_namespace(__name__)"
|
|
|
|
)
|
|
|
|
lib = ns.mkdir(dirname)
|
|
|
|
lib.join("__init__.py").write("def check(): return %r" % dirname)
|
|
|
|
|
|
|
|
monkeypatch.syspath_prepend("hello")
|
|
|
|
import ns_pkg.hello
|
|
|
|
|
|
|
|
assert ns_pkg.hello.check() == "hello"
|
|
|
|
|
|
|
|
with pytest.raises(ImportError):
|
|
|
|
import ns_pkg.world
|
|
|
|
|
|
|
|
# Prepending should call fixup_namespace_packages.
|
|
|
|
monkeypatch.syspath_prepend("world")
|
|
|
|
import ns_pkg.world
|
|
|
|
|
|
|
|
assert ns_pkg.world.check() == "world"
|
2019-04-12 23:27:36 +08:00
|
|
|
|
|
|
|
# Should invalidate caches via importlib.invalidate_caches.
|
|
|
|
tmpdir = testdir.tmpdir
|
|
|
|
modules_tmpdir = tmpdir.mkdir("modules_tmpdir")
|
|
|
|
monkeypatch.syspath_prepend(str(modules_tmpdir))
|
|
|
|
modules_tmpdir.join("main_app.py").write("app = True")
|
|
|
|
from main_app import app # noqa: F401
|