43-pytest-内置fixture之临时目录使用

---

前言

• 本篇来学习pytest中内置fixture中临时目录的使用

tmpdir

• tmpdir作用范围是函数级别，创建临时文件供单个测试点调用

# -*- coding: utf-8 -*-
# @Time    : 2022/10/20
# @Author  : 大海

import os


def test_tmpdir(tmpdir):
    """内置tmpdir fixture使用"""
    # 创建临时文件
    a_file = tmpdir.join('a.txt')
    # 写入内容
    a_file.write('A')

    # 创建临时目录
    a_sub_dir = tmpdir.mkdir('sub')
    sub_file = a_sub_dir.join('sub.txt')
    sub_file.write('sub')
    # 打印临时目录路径
    print(f"tmpdir:{a_file}")
    print(f"tmpdir:{a_sub_dir}")

    assert a_file.read() == 'A'
    assert sub_file.read() == 'sub'
    
if __name__ == '__main__':
    os.system('pytest -s -v')

tmpdir_factory

• tmpdir_factory作用范围是会话级别，主要针对创建临时目录的情况，可供多个测试点调用

# -*- coding: utf-8 -*-
# @Time    : 2022/10/20
# @Author  : 大海

import os

def test_create_file(tmpdir_factory):
    p = tmpdir_factory.mktemp("demo01").join("hello.txt")
    print(f"tmpdir:{p}")
    p.write("content")
    assert p.read() == "content"


def test_create_file2(tmpdir_factory):
    p = tmpdir_factory.mktemp("demo02").join("hello.txt")
    print(f"tmpdir:{p}")
    p.write("content")
    assert p.read() == "content"
    
if __name__ == '__main__':
    os.system('pytest -s -v')

tmp_path

• 测试用例级别，tmpdir 和tmp_path功能是一样的，唯一区别是tmpdir返回的是py.path.local类型，而tmp_path返回的是pathlib.Path类型

# -*- coding: utf-8 -*-
# @Time    : 2022/10/20
# @Author  : 大海

import os

def test_create_file_path(tmp_path):
    """临时路径"""
    d = tmp_path / "sub"
    print(f"temp_dir:{d}")
    d.mkdir()
    p = d / "hello.txt"
    str_txt = "hello world"
    p.write_text(str_txt)
    assert p.read_text() == str_txt
    assert len(list(tmp_path.iterdir())) == 1

if __name__ == '__main__':
    os.system('pytest -s -v')

tmp_path_factory

• 会话级别

# -*- coding: utf-8 -*-
# @Time    : 2022/10/20
# @Author  : 大海

import os

def test_create_file_path_factory(tmp_path_factory):
    """临时路径 会话级"""
    d = tmp_path_factory.mktemp("demo01") / "hello.txt"
    print(f"temp_dir:{d}")
    str_txt = "hello world"
    d.write_text(str_txt)
    assert d.read_text() == str_txt


def test_create_file2_path_factory(tmp_path_factory):
    d = tmp_path_factory.mktemp("demo02") / "hello.txt"
    print(f"temp_dir:{d}")
    str_txt = "hello world"
    d.write_text(str_txt)
    assert d.read_text() == str_txt

if __name__ == '__main__':
    os.system('pytest -s -v')

指定临时目录

• –basetemp = 临时路径

# -*- coding: utf-8 -*-
# @Time    : 2022/10/20
# @Author  : 大海

import os

def test_create_file_path(tmp_path):
    """临时路径"""
    d = tmp_path / "sub"
    print(f"temp_dir:{d}")
    d.mkdir()
    p = d / "hello.txt"
    str_txt = "hello world"
    p.write_text(str_txt)
    assert p.read_text() == str_txt
    assert len(list(tmp_path.iterdir())) == 1

if __name__ == '__main__':
    # 指定临时目录,确认为空目录 否则会被清空
    os.system('pytest -s -v --basetemp=./test_tmp')