Spring Boot抽象类及其实现类的单元测试实践

public abstract class CsvService<T extends CsvBean> {
    public List<T> readFromCsv(Class<T> type, CsvToBeanFilter filter) {
        // ... 省略了从资源读取并解析CSV的通用代码 ...
        // 最终会调用 getData(csvToBean) 方法处理解析后的数据
        return new ArrayList<>(); // 简化示例
    }
    protected abstract String getFileName();
    protected abstract List<T> getData(CsvToBean<T> csvToBean); // 假设此方法在 readFromCsv 内部被调用
}

@Service
public class AirportService extends CsvService<AirportBean> {
    @Override
    protected String getFileName() {
        return "airport.csv";
    }

    @Override
    protected List<AirportBean> getData(CsvToBean<AirportBean> csvToBean) {
        List<AirportBean> airports = new ArrayList<>();
        for (AirportBean bean : csvToBean) {
            // ... 省略了具体的业务处理逻辑 ...
            airports.add(bean);
        }
        return airports;
    }
}

import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.mockito.junit.jupiter.MockitoExtension;

import java.util.Arrays;
import java.util.List;

import static org.junit.jupiter.api.Assertions.*;
import static org.mockito.Mockito.*;

// 假设 CsvBean 和 CsvToBean 已经导入
// import com.opencsv.bean.CsvToBean;
// import com.example.model.AirportBean; // 你的数据模型

@ExtendWith(MockitoExtension.class) // JUnit 5
public class AirportServiceTest {

    @InjectMocks
    private AirportService airportService; // 被测试的具体服务

    // 如果 AirportService 自身有其他依赖，则在这里使用 @Mock
    // 例如：@Mock private SomeRepository someRepository;

    // 对于 getData() 方法，我们需要模拟传入的 CsvToBean 对象
    // 但它作为方法参数，不需要在这里声明为 @Mock 字段

    // ... 后续测试方法 ...
}

    @Test
    void testGetFileName() {
        String fileName = airportService.getFileName();
        assertEquals("airport.csv", fileName, "getFileName() 应该返回 'airport.csv'");
    }

    @Test
    void testGetData() {
        // 1. 准备模拟数据：创建预期从 CSV 中解析出的 AirportBean 列表
        AirportBean mockAirport1 = new AirportBean();
        mockAirport1.setCode("LAX");
        mockAirport1.setName("Los Angeles International Airport");
        // 假设 AirportBean 还有其他字段，例如：
        // mockAirport1.setCity("Los Angeles");

        AirportBean mockAirport2 = new AirportBean();
        mockAirport2.setCode("JFK");
        mockAirport2.setName("John F. Kennedy International Airport");
        // mockAirport2.setCity("New York");

        List<AirportBean> expectedAirportsFromCsv = Arrays.asList(mockAirport1, mockAirport2);

        // 2. 模拟 CsvToBean 对象及其行为
        // 创建一个 CsvToBean 的模拟实例
        CsvToBean<AirportBean> mockCsvToBean = Mockito.mock(CsvToBean.class);

        // 当 mockCsvToBean 的 iterator() 方法被调用时，返回我们预设数据列表的迭代器
        when(mockCsvToBean.iterator()).thenReturn(expectedAirportsFromCsv.iterator());

        // 3. 调用被测试方法
        List<AirportBean> actualAirports = airportService.getData(mockCsvToBean);

        // 4. 断言结果
        assertNotNull(actualAirports, "返回的机场列表不应为 null");
        assertEquals(2, actualAirports.size(), "返回的机场数量应为 2");

        // 验证返回的数据内容是否与预期相符
        assertEquals("LAX", actualAirports.get(0).getCode());
        assertEquals("Los Angeles International Airport", actualAirports.get(0).getName());

        assertEquals("JFK", actualAirports.get(1).getCode());
        assertEquals("John F. Kennedy International Airport", actualAirports.get(1).getName());

        // 验证 mockCsvToBean 的 iterator() 方法是否被调用了一次
        verify(mockCsvToBean, times(1)).iterator();
    }