前面的文章已經大概實現了普通方法的mock了。final方法大概要復雜一點,因為cglib天生不支持攔截final方法。
我們的處理方案也是非常簡單:把final方法改成非final方法。
當然這不是要求用戶改,而是通過徒手擼一個Mock框架(三)—— JUnit4Runner+ClassLoader=為所欲為定義的自定義類加載器來完成這件事情。
仿照那里的做法,我們只需要改進一下RemoveFinalFlagClassVisitor。現在的RemoveFinalFlagClassVisitor只處理了final的類聲明,類似地,我們可以也可以通過這個移除掉方法的final聲明。
只需要重載方法visitMethod:

在我們的測試里面,加上RunWith和PrepareForTest注解,準備自定義加載器的運行環(huán)境:

我們的測試是:

運行測試,一切okay。

代碼地址
徒手擼一個Mock框架(五)—— 復雜Matcher,Answer,void方法
徒手擼一個Mock框架(四)—— when...then...追上我就讓你嘿嘿嘿
徒手擼一個Mock框架(三)—— JUnit4Runner+ClassLoader=為所欲為
徒手擼一個Mock框架(二)——如何創(chuàng)建final類的代理
徒手擼一個Mock框架(一)——如何創(chuàng)建一個mock對象