"""测试数据管理器单元测试""" import pytest from pathlib import Path from unittest.mock import Mock from apitest.data.test_data_manager import TestDataManager from apitest.models.test_models import TestCase, HTTPMethod from apitest.models.exceptions import TestRunException @pytest.fixture def mock_logger(): """创建模拟日志记录器""" return Mock() @pytest.fixture def data_manager(mock_logger): """创建测试数据管理器实例""" return TestDataManager(logger=mock_logger) @pytest.fixture def sample_test_cases(): """创建示例测试用例""" return [ TestCase( id="TC001", name="测试用例1", description="测试用例1", module="test", endpoint="/api/test1", method=HTTPMethod.GET, headers={}, enabled=True ), TestCase( id="TC002", name="测试用例2", description="测试用例2", module="test", endpoint="/api/test2", method=HTTPMethod.POST, headers={}, enabled=True ) ] def test_load_test_cases_from_json_success(data_manager, tmp_path): """测试从JSON文件加载测试用例（成功）""" import json test_data = [ { "id": "TC001", "name": "测试用例1", "description": "测试用例1", "module": "test", "endpoint": "/api/test1", "method": "GET", "headers": {}, "enabled": True }, { "id": "TC002", "name": "测试用例2", "description": "测试用例2", "module": "test", "endpoint": "/api/test2", "method": "POST", "headers": {}, "enabled": True } ] test_file = tmp_path / "test_cases.json" with open(test_file, "w", encoding="utf-8") as f: json.dump(test_data, f) test_cases = data_manager.load_test_cases_from_json(test_file) assert len(test_cases) == 2 assert test_cases[0].id == "TC001" assert test_cases[1].id == "TC002" assert test_cases[0].method == HTTPMethod.GET assert test_cases[1].method == HTTPMethod.POST def test_load_test_cases_from_json_file_not_found(data_manager, tmp_path): """测试从JSON文件加载测试用例（文件不存在）""" test_file = tmp_path / "nonexistent.json" with pytest.raises(TestRunException) as exc_info: data_manager.load_test_cases_from_json(test_file) assert "测试用例文件不存在" in str(exc_info.value) def test_load_test_cases_from_json_invalid_json(data_manager, tmp_path): """测试从JSON文件加载测试用例（无效JSON）""" test_file = tmp_path / "invalid.json" with open(test_file, "w", encoding="utf-8") as f: f.write("{ invalid json }") with pytest.raises(TestRunException) as exc_info: data_manager.load_test_cases_from_json(test_file) assert "JSON文件解析失败" in str(exc_info.value) def test_load_test_cases_from_json_invalid_method(data_manager, tmp_path): """测试从JSON文件加载测试用例（无效HTTP方法）""" import json test_data = [ { "id": "TC001", "name": "测试用例1", "description": "测试用例1", "module": "test", "endpoint": "/api/test1", "method": "INVALID", "headers": {}, "enabled": True } ] test_file = tmp_path / "test_cases.json" with open(test_file, "w", encoding="utf-8") as f: json.dump(test_data, f) test_cases = data_manager.load_test_cases_from_json(test_file) assert len(test_cases) == 1 assert test_cases[0].method == HTTPMethod.GET def test_load_test_data_from_csv_success(data_manager, tmp_path): """测试从CSV文件加载测试数据（成功）""" test_file = tmp_path / "test_data.csv" with open(test_file, "w", encoding="utf-8", newline="") as f: f.write("param1,param2,param3\n") f.write("value1,value2,value3\n") f.write("value4,value5,value6\n") test_data = data_manager.load_test_data_from_csv(test_file) assert len(test_data) == 2 assert test_data[0] == {"param1": "value1", "param2": "value2", "param3": "value3"} assert test_data[1] == {"param1": "value4", "param2": "value5", "param3": "value6"} def test_load_test_data_from_csv_file_not_found(data_manager, tmp_path): """测试从CSV文件加载测试数据（文件不存在）""" test_file = tmp_path / "nonexistent.csv" with pytest.raises(TestRunException) as exc_info: data_manager.load_test_data_from_csv(test_file) assert "测试数据文件不存在" in str(exc_info.value) def test_parameterize_test_case_success(data_manager, sample_test_cases): """测试参数化测试用例（成功）""" test_data = [ {"params": {"id": "1"}, "body": {"name": "test1"}}, {"params": {"id": "2"}, "body": {"name": "test2"}} ] parameterized_cases = data_manager.parameterize_test_case( sample_test_cases[0], test_data ) assert len(parameterized_cases) == 2 assert parameterized_cases[0].id == "TC001_1" assert parameterized_cases[0].name == "测试用例1 (数据集 1)" assert parameterized_cases[0].params == {"id": "1"} assert parameterized_cases[0].body == {"name": "test1"} assert parameterized_cases[1].id == "TC001_2" assert parameterized_cases[1].name == "测试用例1 (数据集 2)" assert parameterized_cases[1].params == {"id": "2"} assert parameterized_cases[1].body == {"name": "test2"} def test_parameterize_test_case_empty_data(data_manager, sample_test_cases): """测试参数化测试用例（空数据）""" test_data = [] parameterized_cases = data_manager.parameterize_test_case( sample_test_cases[0], test_data ) assert len(parameterized_cases) == 0 def test_save_test_cases_to_json_success(data_manager, sample_test_cases, tmp_path): """测试保存测试用例到JSON文件（成功）""" output_file = tmp_path / "output.json" data_manager.save_test_cases_to_json(sample_test_cases, output_file) assert output_file.exists() import json with open(output_file, "r", encoding="utf-8") as f: data = json.load(f) assert len(data) == 2 assert data[0]["id"] == "TC001" assert data[1]["id"] == "TC002" def test_save_test_cases_to_json_create_directory(data_manager, sample_test_cases, tmp_path): """测试保存测试用例到JSON文件（创建目录）""" output_file = tmp_path / "nested" / "dir" / "output.json" data_manager.save_test_cases_to_json(sample_test_cases, output_file) assert output_file.exists() assert output_file.parent.exists() def test_save_test_data_to_csv_success(data_manager, tmp_path): """测试保存测试数据到CSV文件（成功）""" test_data = [ {"param1": "value1", "param2": "value2"}, {"param1": "value3", "param2": "value4"} ] output_file = tmp_path / "output.csv" data_manager.save_test_data_to_csv(test_data, output_file) assert output_file.exists() import csv with open(output_file, "r", encoding="utf-8") as f: reader = csv.DictReader(f) rows = list(reader) assert len(rows) == 2 assert rows[0]["param1"] == "value1" assert rows[1]["param1"] == "value3" def test_save_test_data_to_csv_with_fieldnames(data_manager, tmp_path): """测试保存测试数据到CSV文件（指定字段名）""" test_data = [ {"param1": "value1", "param2": "value2", "param3": "value3"}, {"param1": "value4", "param2": "value5", "param3": "value6"} ] output_file = tmp_path / "output.csv" data_manager.save_test_data_to_csv( test_data, output_file, fieldnames=["param1", "param2"] ) assert output_file.exists() import csv with open(output_file, "r", encoding="utf-8") as f: reader = csv.DictReader(f) rows = list(reader) fieldnames = reader.fieldnames assert len(rows) == 2 assert fieldnames == ["param1", "param2"] def test_save_test_data_to_csv_empty_data(data_manager, tmp_path): """测试保存测试数据到CSV文件（空数据）""" test_data = [] output_file = tmp_path / "output.csv" with pytest.raises(TestRunException) as exc_info: data_manager.save_test_data_to_csv(test_data, output_file) assert "测试数据为空" in str(exc_info.value) def test_save_test_data_to_csv_create_directory(data_manager, tmp_path): """测试保存测试数据到CSV文件（创建目录）""" test_data = [{"param1": "value1"}] output_file = tmp_path / "nested" / "dir" / "output.csv" data_manager.save_test_data_to_csv(test_data, output_file) assert output_file.exists() assert output_file.parent.exists()