Pytest/Locust: ModuleNotFoundError 没有命名的模块

Question

我试图找到关于类似主题的答案，但是......没有任何帮助。
当我使用pytest -m blablabla运行我的常规测试时 - 没有问题，但是
当我通过命令运行蝗虫时：

蝗虫-f my_locustfiles/instr_performance.py

比得到这个：

(venv) evgen@TLL amapitest % locust -f my_locustfiles/instr_performance.py
Traceback (most recent call last):
  File "/Users/evgen/venv/bin/locust", line 8, in <module>
    sys.exit(main())
  File "/Users/evgen/venv/lib/python3.10/site-packages/locust/main.py", line 70, in main
    docstring, _user_classes, shape_class = load_locustfile(_locustfile)
  File "/Users/evgen/venv/lib/python3.10/site-packages/locust/util/load_locustfile.py", line 58, in load_locustfile
    imported = source.load_module()
  File "<frozen importlib._bootstrap_external>", line 548, in _check_name_wrapper
  File "<frozen importlib._bootstrap_external>", line 1063, in load_module
  File "<frozen importlib._bootstrap_external>", line 888, in load_module
  File "<frozen importlib._bootstrap>", line 290, in _load_module_shim
  File "<frozen importlib._bootstrap>", line 719, in _load
  File "<frozen importlib._bootstrap>", line 688, in _load_unlocked
  File "<frozen importlib._bootstrap_external>", line 883, in exec_module
  File "<frozen importlib._bootstrap>", line 241, in _call_with_frames_removed
  File "/Users/evgen/PycharmProjects/api-testing/amapitest/my_locustfiles/instr_performance.py", line 8, in <module>
    from amapitest.src.helpers.jwt_generator import generate_json_web_token
ModuleNotFoundError: No module named 'amapitest.src.helpers'

我的项目结构：

├── amapitest  
│   ├── my_locustfiles  
│   │   ├── __init__.py  
│   │   └── instr_performance.py  
│   ├── src  
│   │   ├── configs  
│   │   ├── dao  
│   │   ├── helpers  
│   │   └── utilities  
│   ├── tests  
│   └── tmp  
├── amapitest.egg-info  
├── build  
├── dist  
├── env.sh  
├── requirements.txt  
└── setup.py  

附加信息：
蝗虫 2.11.0
pytest 7.1.2
python 3.10
macOS 蒙特雷

Answer 1

根据文档https://docs.locust.io/en/stable/writing-a-locustfile.html#how-to-structure-your-test-code ，您的当前目录会被 locust 自动添加到sys.path

尝试转到父目录并运行

locust -f amapitest/my_locustfiles/instr_performance.py

Answer 2

我通过将“my_locustfiles”文件夹移动到项目的根目录并从该目录运行测试解决了我的问题。

我仍然不明白为什么当 locust 文件位于“amapitest”中时，locust 找不到导入的方法

但是……无论如何，谢谢大家

Answer 3

1. print(sys.path)查看帮助 Python 查找模块的路径。

2. 如果找到路径。 例如，假设“/Users/user/SomeFolder/amapitest/” ，尝试@Cyberwiz 的评论，在每个文件夹中添加__init__.py文件以将文件夹转换为模块。 在amapitest文件夹中执行 locust

如果你想在子文件夹中执行 locust 并且仍然找不到模块，请尝试步骤 3。

3. 在instr_performance.py中，append 这样的父路径

   sys.path.append( os.path.join( os.path.dirname(__file__), '..' ) )

   你会看到'/Users/user/SomeFolder/amapitest/my_locustfiles/..' when print(sys.path) ，我认为这个解决方法可能有帮助