官方文档

javaapi：https://javadoc.io/doc/org.mockito/mockito-core/latest/org/mockito/Mockito.html
官网：https://site.mockito.org/
github：https://github.com/mockito/mockito

如何编写一个单元测试用例

https://www.vogella.com/tutorials/Mockito/article.html#mockitousage

如何编写一个集成测试用例

这个不同框架不同，可以参考spring boot的写法

如何mock static的方法?

https://asolntsev.github.io/en/2020/07/11/mockito-static-methods/

注意：将业务逻辑写在 try 块中。
image.png

如何不启动spring整体工程，在本地实际连db测试dao层代码？

方法思路：

  • 采用jdbc 方式直连
  • 采用反射机制 动态 实例化类 及其属性

mybatis sqlSession相关解释见 https://www.yuque.com/antone/zv3ypu/tl15md

  2. @Slf4j
  3. public class MutiDataSourceTest {
  4.     private static final Logger LOGGER = LoggerFactory.getLogger(MutiDataSourceTest.class);
  6.     private static Map<DataSourceType, DataSource> dataSourceContext = new HashMap<>();
  7.     //@Getter
  8.     private static Map<DataSourceType, SqlSessionFactory> sqlSessionFactoryContext = new HashMap<>();
  9.     private static Map<DataSourceType, Properties> dataSourceTypeToConfig = new HashMap<>();
  10.     private static Map<Class<? extends Annotation>, DataSourceType> classToDataSourceType = new HashMap<>();
  11.     private static List<SqlSession> sqlSessions = new ArrayList<>();
  13.     public void logObject(Object o) {
  14.         //LOGGER.info("obj:{}", o);
  15.     }
  17.     static {
  18.         classToDataSourceType.put(EXPLORER.class, DataSourceType.EXPLORE);
  20.         // 数据库配置信息
  22.         Properties explorerConfig = new Properties();
  23.         explorerConfig.put("driver", "com.mysql.jdbc.Driver");
  24.         explorerConfig.put("url",
  25.             "jdbc:mysql://xxxx:3306/cheetah?useUnicode=true&characterEncoding=utf8"
  26.                 + "&rewriteBatchedStatements=true");
  27.         explorerConfig.put("username", "xxx");
  29.         explorerConfig.put("password", "xxx");
  31.         LoggerUtil.info(LOGGER, "adb config:{0}", explorerConfig);
  33.         dataSourceTypeToConfig.put(DataSourceType.EXPLORE, explorerConfig);
  34.     }
  36.     public static enum DataSourceType {
  37.         EXPLORE
  38.     }
  40.     public static SqlSession openSession(DataSourceType dataSourceType) {
  41.         SqlSessionFactory sqlSessionFactory = getSqlSessionFactoryByType(dataSourceType);
  42.         SqlSession sqlSession = sqlSessionFactory.openSession(null, true);
  43.         sqlSessions.add(sqlSession);
  44.         return sqlSession;
  45.     }
  47.     public static <T> T getMapper(Class<T> tClass) {
  48.         DataSourceType dataSourceType = parseType(tClass);
  49.         SqlSessionFactory sqlSessionFactory = getSqlSessionFactoryByType(dataSourceType);
  50.         MapperRegistry mapperRegistry = sqlSessionFactory.getConfiguration().getMapperRegistry();
  51.         if (!mapperRegistry.hasMapper(tClass)) {
  52.             sqlSessionFactory.getConfiguration().addMapper(tClass);
  53.         }
  54.         SqlSession sqlSession = openSession(dataSourceType);
  55.         return sqlSession.getMapper(tClass);
  56.     }
  58.     public static SqlSessionFactory getSqlSessionFactoryByType(DataSourceType dataSourceType) {
  59.         if (!sqlSessionFactoryContext.containsKey(dataSourceType)) {
  60.             DataSource dataSource = getDataSourceByType(dataSourceType);
  61.             Configuration configuration = new Configuration();
  62.             JdbcTransactionFactory jdbcTransactionFactory = new JdbcTransactionFactory();
  63.             Environment environment = new Environment("test", jdbcTransactionFactory, dataSource);
  64.             configuration.setEnvironment(environment);
  65.             configuration.getTypeAliasRegistry().registerAliases("com.alibaba.aliexpress.module");
  66.             configuration.setMapUnderscoreToCamelCase(true);
  68.             MybatisProperties mybatisProperties = new MybatisProperties();
  69.             mybatisProperties.setMapperLocations(
  70.                 new String[] {"classpath*:com/alipay/xxxx/dal/xxx/mapper/*.xml"}); //这里是mapper配置的地址
  72.             Resource[] resources = mybatisProperties.resolveMapperLocations();
  73.             for (Resource mapperLocation : resources) {
  74.                 XMLMapperBuilder xmlMapperBuilder = null;
  75.                 try {
  76.                     xmlMapperBuilder = new XMLMapperBuilder(mapperLocation.getInputStream(),
  77.                         configuration, mapperLocation.toString(), configuration.getSqlFragments());
  78.                 } catch (IOException e) {
  79.                     e.printStackTrace();
  80.                 }
  81.                 xmlMapperBuilder.parse();
  82.             }
  83.             DefaultSqlSessionFactory sqlSessionFactory = new DefaultSqlSessionFactory(configuration);
  84.             //LOGGER.info("{} sqlSessionFactory init success.", dataSourceType);
  85.             sqlSessionFactoryContext.put(dataSourceType, sqlSessionFactory);
  86.             //addIntercepter(sqlSessionFactory, dataSourceType);
  87.         }
  88.         return sqlSessionFactoryContext.get(dataSourceType);
  89.     }
  91.     public static DataSource getDataSourceByType(DataSourceType dataSourceType) {
  92.         if (!dataSourceContext.containsKey(dataSourceType)) {
  93.             DataSource dataSource = buildDataSouce(dataSourceType);
  94.             dataSourceContext.put(dataSourceType, dataSource);
  95.             //LOGGER.info("{} dataSource init success.", dataSourceType);
  96.         }
  97.         return dataSourceContext.get(dataSourceType);
  98.     }
  100.     private static DataSource buildDataSouce(DataSourceType dataSourceType) {
  101.         Properties properties = dataSourceTypeToConfig.get(dataSourceType);
  102.         UnpooledDataSource dataSource = new UnpooledDataSource();
  103.         dataSource.setDriver(properties.getProperty("driver"));
  104.         dataSource.setUrl(properties.getProperty("url"));
  105.         dataSource.setUsername(properties.getProperty("username"));
  106.         dataSource.setPassword(properties.getProperty("password"));
  107.         return dataSource;
  108.     }
  110.     private static <T> DataSourceType parseType(Class<T> tClass) {
  111.         Set<Class<? extends Annotation>> classes = classToDataSourceType.keySet();
  112.         for (Class<? extends Annotation> anotationClass : classes) {
  113.             if (anotationClass == Mapper.class) {
  114.                 continue;
  115.             }
  116.             if (tClass.isAnnotationPresent(anotationClass)) {
  117.                 return classToDataSourceType.get(anotationClass);
  118.             }
  119.         }
  120.         //return DataSourceType.TDDL;
  121.         return null;
  122.     }
  124.     @AfterClass
  125.     public static void destroy() {
  126.         sqlSessions.forEach(s -> {
  127.             if (s != null) {
  128.                 s.close();
  129.             }
  130.         });
  131.     }
  133. }
  2. public class ServiceTest extends MutiDataSourceTest {
  4.     private static final Logger logger = LoggerFactory.getLogger(ServiceTest.class);
  5.     private boolean added;
  7.     @Before
  8.     public void serviceTestInit() {
  9.         beforeInit();
  10.         populateField(this);
  11.         //afterPopulate(this);
  12.     }
  14.     private void beforeInit() {
  15.         addIntercepter();
  16.     }
  18.     public <T> T createService(Class<T> tClass) {
  19.         return (T)createObject(tClass);
  20.     }
  22.     private Object createObject(Class tClass) {
  23.         try {
  24.             //log.debug("create instance for class:{}", tClass.getName());
  25.             Class c = findImplClass(tClass);
  26.             if (c == null) {
  27.                 return null;
  28.             }
  29.             Object o = c.newInstance();
  30.             populateField(o);
  31.             //afterPopulate(o);
  32.             return o;
  33.         } catch (Exception e) {
  34.             throw new RuntimeException(e);
  35.         }
  36.     }
  38.     private void afterPopulate(Object o) {
  39.         //if (o instanceof BaseServiceImpl) {
  40.         //    try {
  41.         //        initBaseImpl(o);
  42.         //    } catch (InvocationTargetException e) {
  43.         //        e.printStackTrace();
  44.         //    } catch (IllegalAccessException e) {
  45.         //        e.printStackTrace();
  46.         //    }
  47.         //}
  49.     }
  51.     private void populateField(Object o) {
  52.         Field[] declaredFields = o.getClass().getDeclaredFields();
  53.         Arrays.stream(declaredFields).filter(
  54.             field -> field.isAnnotationPresent(Autowired.class) || field.isAnnotationPresent(
  55.                 Resource.class)).forEach(f -> populateField(f, o));
  56.     }
  58.     private <T> Class findImplClass(Class<T> tClass) {
  59.         if (tClass.isInterface()) {
  60.             String className = tClass.getSimpleName() + "Impl";
  61.             String packageName = tClass.getName().replace(tClass.getSimpleName(), "impl.");
  62.             String implClassName = packageName + className;
  63.             //String implClassName2 = packageName + className2;
  64.             //log.debug("find impl class for:{} is {}", tClass.getName(), implClassName);
  65.             Class<?> aClass = null;
  66.             try {
  67.                 aClass = Class.forName(implClassName);
  68.             } catch (ClassNotFoundException e) {
  69.                 LoggerUtil.warn(logger, "not found impl class for:{0},skip create instance", tClass.getName());
  70.                 // e.printStackTrace();
  72.             }
  73.             return aClass;
  74.         }
  75.         return tClass;
  76.     }
  78.     @SneakyThrows
  79.     private void populateField(Field field, Object target) {
  80.         Class<?> declaringClass = field.getType();
  81.         LoggerUtil.debug(logger, "field class:{0}", declaringClass);
  82.         Object fieldValue = null;
  83.         if (declaringClass.getName().contains("Service")) {
  84.             fieldValue = createService(declaringClass);
  85.         } else if ((declaringClass.isAnnotationPresent(EXPLORER.class))) {
  86.             fieldValue = super.getMapper(declaringClass);
  87.         } else {
  88.             fieldValue = createObject(declaringClass);
  89.         }
  90.         field.setAccessible(true);
  91.         field.set(target, fieldValue);
  93.         LoggerUtil.info(logger, "inject field:{0} value:{1}", field.getName(), fieldValue);
  94.     }
  96.     private void addIntercepter() {
  97.         if (this.added) {
  98.             return;
  99.         }
  100.         for (DataSourceType dataSourceType : DataSourceType.values()) {
  101.             getSqlSessionFactoryByType(dataSourceType);
  102.         }
  103.         //Map<DataSourceType, SqlSessionFactory> sqlSessionFactoryContext = getSqlSessionFactoryContext();
  104.         //for (Entry<DataSourceType, SqlSessionFactory> entry : sqlSessionFactoryContext.entrySet()) {
  105.         //    SqlSessionFactory sqlSessionFactory = entry.getValue();
  106.         //    DataSourceType dataSourceType = entry.getKey();
  107.         //    PageInterceptor interceptor = new PageInterceptor();
  108.         //    Properties properties = new Properties();
  109.         //    properties.put("helperDialect", "mysql");
  110.         //    properties.put("reasonable", "true");
  111.         //    properties.put("supportMethodsArguments", "true");
  112.         //    properties.put("params", "count=countSql;pageNum=pageNumber;pageSize=pageSize;");
  113.         //    interceptor.setProperties(properties);
  114.         //    sqlSessionFactory.getConfiguration().addInterceptor(interceptor);
  115.         //    if (dataSourceType == DataSourceType.ADB) {
  116.         //        PartitionParamInterceptor partitionParamInterceptor = new PartitionParamInterceptor();
  117.         //        partitionParamInterceptor.setPartitionService(createService(PartitionService.class));
  118.         //        sqlSessionFactory.getConfiguration().addInterceptor(partitionParamInterceptor);
  119.         //    }
  120.         //}
  121.         this.added = true;
  123.     }
  124. }

测试类只需要继承ServiceTest即可