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