无法将助手 class 与 pytests 一起使用

Question

帮手Class

    class UserBuilder:
        def __init__(self):
            self.user = CustomUser.objects.get(email='random_email@test.com')

        def get_user(self):
            return self.user

测试Class

    @pytest.mark.django_db
    @pytest.mark.parametrize('(user, output)', [
        (user_builder.change_bank_billdesk_bank_id(bank_name="IDN").get_user(), -1),
        (user_builder.change_bank_billdesk_bank_id(bank_name="IDB").get_user(), -1),
        (user_builder.change_bank_billdesk_bank_id(bank_name="SBI").get_user(), 1)
    ])
    def test_users_with_different_banks(self, user, output):

错误日志

    tests/trial.py:16: in <module>
        @pytest.mark.django_db
    tests/trial.py:100: in TestFunction
        user_builder = UserBuilder()
    tests/builder/user_builder.py:13: in __init__
        self.user = CustomUser.objects.get(email='random_email@test.com')
    ../.virtualenvs/local_dev/local/lib/python2.7/site-packages/django/db/models/manager.py:127: in manager_method
        return getattr(self.get_queryset(), name)(*args, **kwargs)
    ../.virtualenvs/local_dev/local/lib/python2.7/site-packages/django/db/models/query.py:328: in get
        num = len(clone)
    ../.virtualenvs/local_dev/local/lib/python2.7/site-packages/django/db/models/query.py:144: in __len__
        self._fetch_all()
    ../.virtualenvs/local_dev/local/lib/python2.7/site-packages/django/db/models/query.py:965: in _fetch_all
        self._result_cache = list(self.iterator())
    ../.virtualenvs/local_dev/local/lib/python2.7/site-packages/django/db/models/query.py:238: in iterator
        results = compiler.execute_sql()
    ../.virtualenvs/local_dev/local/lib/python2.7/site-packages/django/db/models/sql/compiler.py:838: in execute_sql
        cursor = self.connection.cursor()
    ../.virtualenvs/local_dev/local/lib/python2.7/site-packages/django/db/backends/base/base.py:162: in cursor
        cursor = self.make_debug_cursor(self._cursor())
    ../.virtualenvs/local_dev/local/lib/python2.7/site-packages/django/db/backends/base/base.py:135: in _cursor
        self.ensure_connection()
    E   Failed: Database access not allowed, use the "django_db" mark, or the "db" or "transactional_db" fixtures to enable it.

互联网上没有合适的解决方案，所以我尝试了所有遇到的问题。

我无法解决这个错误。任何帮助，将不胜感激。谢谢

Answer 1

您正试图在测试之外访问数据库。这是被禁止的。与其在测试之前尝试获取数据并将其作为参数传递，不如在测试中进行。这也将避免 parametrize 标记中的代码重复。

@pytest.mark.django_db
@pytest.mark.parametrize('bank_name,output', [("IDN", -1), ("IDB", -1), ("SBI", 1)])
def test_users_with_different_banks(self, bank_name, output):
    user = UserBuilder().change_bank_billdesk_bank_id(bank_name).get_user()
    assert user.id

无法将助手 class 与 pytests 一起使用

Unable to use helper class with pytests

django

pytest

pytest-django