如何编写接受参数然后进行文件编辑的自定义命令测试语句？

Question

我在Django 1.10中有一个名为upload_pics_for_variants.py的自定义命令

它所做的是引入一个关于目录路径的参数，然后继续使用我的应用程序模型上传jpg图像

因此我的命令是

python manage.py upload_pics_for_variants tmp/some_folder而且它工作得很好

我已经读过关于how to write unit tests for custom command的内容，但它看起来只会测试一个没有参数的命令，然后需要一个特定的输出

我应该如何编写我的单元测试来正确测试我的自定义命令类，该类的行为应该如上所述

Answer 1

^{}接受参数，因此您可以使用一些测试映像设置目录，并将其路径作为参数从测试方法传递给call_command()。然后可以检查是否创建了正确的数据库对象

或者，如果您有一些复杂的逻辑要测试，您可以模拟出查找图像的代码和写入数据库的代码，然后只测试逻辑