ChatGPT学习之旅 (8) 单元测试助手

大家好，我是Edison。

本篇我们基于上一篇的基础，来写一个单元测试助手的prompt，让它帮我们写一些我们.NET开发者不太愿意编写的单元测试代码，进而提高我们的代码质量，同时还降低我们的开发工作量。

单元测试助手

这里我们基于上一篇中提到的万能语言辅助专家的提示词，稍作修改，形成我们的单元测试助手的提示词，如下所示，经过一些测试在GPT4-o模型下效果真的不错，建议收藏！

初始回复：“
**Hi I'm Unit Test Master，Created by Edison Zhou，V1.0，20230701**
### ⚙️ Preferences：
-  lang: <> else C#
- ⏲️ test: <> else xUnit
-  mock: <> else Moq
###  Menu：
请使用表格输出支持的`instructions`和对应名称，不需要解释具体含义，也不需要显示这句话：
---
请指出你的开发语言，E.g.：/lang C#。
回复1保持默认。
”

`preferences`
/lang:<开发语言偏好，默认为C#>
/test:<测试框架偏好，默认为xUnit>
/mock:<模拟框架偏好，默认为Moq>

`instructions`
/models：被测试代码涉及到的模型定义
用户输入一些被测试代码涉及到的Models如Entity，DTO, VO等，请按以下模版输出Models相关信息：
## Models：
用表格输出：**Model Name**
/constants：被测试代码涉及到的常量或枚举定义
用户输入一些被测试代码涉及到的Constants如Enum，Constant等，请按以下模版输出Constants相关信息：
## Constants：
用表格输出：**Constant Name**
/refer: 被测试代码依赖的的对象接口定义
用户输入一些被测试代码涉及到的对象接口如Repository，Gateway等，请按以下模版输出References相关信息：
## Refers：
用表格输出：**Reference Name**
/method：被测试方法的实际代码
结合之前的models,constants,refers定义，直接生成被测试方法的单元测试，无需再用表格输出相关信息。如果还有其他要求，用户会补充告诉你。
/help：输出支持的指令指引

`rules`
- 请使用用户设置的偏好的开发语言、测试框架和模拟框架实现
- 每个单元测试方法的命名请遵循格式："被测试方法名_测试场景_预期结果"
- 假如被测试方法中有try-catch，请考虑针对catch部分也编写单元测试用例
- 请一步一步思考，不需要解释代码，如果有错误，用户会纠正你

对话示例

假设我们有一个基于.NET开发的API项目，想要对其中某个Service的某个方法写写单元测试，我们只需要按照以下步骤即可生成可能会“一把过”的单元测试代码。

在我们的实践中，最好通过VS Code将上述的人设prompt编辑好，同时把下面需要喂给GPT的代码片段也准备好。

第一轮：偏好选择

直接回复1，即保持C#+xUnit+Moq的框架组合，默认偏好设置

第二轮：给GPT喂Models相关class的定义

例如下所示，如果model有继承一些接口或基类，最好也一起告诉GPT。

/models
public class AppTokenVo: VoBase, IVo
{
    public string TokenName { get; set; }
    public AppNamesapce Namespace { get; set; }
    public AppTokenType Type { get; set; }
    public string Remark { get; set; }
    public DateTime ExpireTime { get; set; }
}
public class VoBase : IIdentity<Guid>
{
    public virtual Guid Id { get; set; }
}
public class AppTokenEntity : EntityBase, IEntity
{
    ......
}
......

第三轮：给GPT喂Constants相关class/enum的定义

例如下所示，如果constant有继承一些接口或基类，最好也一起告诉GPT。

/constants
public enum AppStatusCode
{
    ......
    //// +++++++++++++++++++++++++++++++++++++
    //// Common error codes [-1..-99]
    //// +++++++++++++++++++++++++++++++++++++
    UnexpectedException = -1,
    OperationFinishedWithErrors = -2,
    ValidationFailed = -3,
    CouldNotSaveEntityInDb = -4,
    EntityNotFoundInDb = -5,
    EntityAlreadyExistInDb = -6
}
......

第四轮：给GPT喂被测试方法依赖的一些对象的定义，如Repository或Gateway等

例如下所示，如果被依赖的对象有继承一些接口或基类，最好也一起告诉GPT。

/refers
public interface IOperationResult<out T> where T : class
{
    int StatusCode { get; }

    string ErrorMessage { get; }

    T Content { get; }

    bool HasContent { get; }

    bool IsSuccess { get; }
}
public interface IAppTokenRepository : IEfCoreRepositoryBase<AppTokenEntity>
{
}
public interface IEfCoreRepositoryBase<TEntity> where TEntity : EntityBase, IEntity
{
    Task<IEnumerable<TEntity>> GetAllAsync(Expression<Func<TEntity, bool>> predicate, bool asNoTracking = true);

    Task<TEntity> GetAsync(Guid id, bool asNoTracking = false);

    Task<TEntity> GetAsync(Expression<Func<TEntity, bool>> predicate, bool asNoTracking = false);

    Task<PagedEntityResult<TEntity>> GetPagedDataAsync(Expression<Func<TEntity, bool>> predicate, int limit, int skip);

    Task<PagedEntityResult<TEntity>> GetPagedDataAsync(Expression<Func<TEntity, bool>> predicate, string orderField, bool orderByAsc, int limit, int skip);

    TEntity Add(TEntity entity);

    TEntity Update(TEntity entity);

    TEntity Delete(TEntity entity);
}

第四轮：告诉GPT被测试的方法代码

例如下所示，最好将该service的定义和构造函数也一起告诉GPT，可以将该service的其他方法移除掉再告诉GPT，让其保持专注。

当然，这也是我们为什么需要通过编辑器将其编辑好，再统一发给GPT的原因。

/method
public class AppTokenService : AppServiceBase<AppTokenService>, IAppTokenService
{
    public AppTokenService(
           IAppUnitOfWork unitOfWork,
           IMapper mapper,
           ILogger<AppTokenService> logger)
       : base(unitOfWork, mapper, logger)
    {
    }

    public async Task<IOperationResult<IEnumerable<AppTokenVo>>> GetAllAppTokensAsync()
    {
        Logger.LogInformation(LoggingConstants.FunctionCalled,
            nameof(AppTokenService), nameof(GetAllAppTokensAsync));
        try
        {
            ......
            return OperationResult<IEnumerable<AppTokenVo>>.Success(appTokens);
        }
        catch (Exception ex)
        {
            Logger.LogError(ex, LoggingConstants.Exception,
                nameof(AppTokenService), nameof(GetAllAppTokensAsync));
            return OperationResult<IEnumerable<AppTokenVo>>.Error((int)AppStatusCode.UnexpectedException);
        }
        finally
        {
            Logger.LogInformation(LoggingConstants.FunctionFinished,
                nameof(AppTokenService), nameof(GetAllAppTokensAsync));
        }
    }
}

随后，GPT就会输出一些高质量的单元测试代码，你可以将其复制出来做一些验证：

（1）首先看看有没有编译错误，如果有，那一定是GPT虚构了某些依赖对象的接口定义，换言之，你忘记告诉GPT准确的定义了，因此你需要补充告诉GPT。

（2）其次如果没有编译错误了，那么恭喜你，你基本可以得到一个全部Pass的结果。但是，请再次review一下它的Assert有没有满足你的需求，如果没有，请一定反馈给GPT，按照你的需求做一些完善。

（3）最后建议跑一下代码覆盖率，看看还有没有没有覆盖的分支语句和代码行，然后反馈给GPT，直到满足你的需求再结束，比如Line Coverage和Branch Coverage都达到了80%及以上。

最后，你可能会发现你这么准备下来，可能自己手写单元测试也差不多写完了，但是这毕竟是你第一次调教GPT帮你写单元测试，当你熟悉这个套路之后，有了自己的固定模板，以后的单元测试就会越来越快，而且你几乎不需要写一行代码了，是不是很酷？

小结

本篇，我们了解了如何基于ChatGPT中的参数化表达沟通，实现一个可以帮我们开发者编写单元测试代码的单元测试助手。当然，这个Prompt我们的参数化设置，我们还可以将其用于C#之外的其他编程语言和测试框架，只要你愿意，你可以根据本文中的模板去修改和完善。

本文工具

本文示例大模型版本：gpt4-o

作者：周旭龙

出处：https://edisonchou.cnblogs.com

本文版权归作者和博客园共有，欢迎转载，但未经作者同意必须保留此段声明，且在文章页面明显位置给出原文链接。