IOS swift：用MockSession进行unit test

最新推荐文章于 2024-09-03 08:33:33 发布

原创

最新推荐文章于 2024-09-03 08:33:33 发布 · 547 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#swift #IOS #MockSession #unit test #MVVM

在单元测试中，由于不能进行实际的网络请求，本文介绍了如何利用MockSession模拟网络调用来测试函数。MockSession包括模拟URLSession和DataTask两部分。测试时，创建MockSession并使用本地文件模拟JSON数据，确保测试快速且可靠。文中以一个具体项目为例，展示了如何在ViewModel中测试loadData函数，通过比较MockSession返回的数据和本地文件内容来验证测试结果。

在进行测试一个network call的时候，我们需要用mockSession来模拟这个api，然后在进行测试，因为在unit test里是不能进行真正的network call的。在unit test里，我们需要更快，更可靠的测试方法，所以一般在测试一个function里的network call的时候，我会建一个mocksession，来模拟api call的部分，接着再用一个本地文件，大多情况是txt文件，来模拟json数据的部分。Mocksession分为两个部分，一个是模拟URLsession，一个是模拟DataTask

import Foundation

class MockDataTask: URLSessionDataTask {
    override func resume() {
    }
}
class MockSession: URLSession {
    typealias CompletionHander = (Data?, URLResponse?, Error?) -> Void
    var data: Data?
    var error: Error?
    override func dataTask(with url: URL, completionHandler: @escaping CompletionHander) -> URLSessionDataTask {
        let data = self.data
        let error = self.error
        completionHandler(data, nil, error)
        return MockDataTask()
    }
}

模拟DataTask的部分，只需要写一个空的resume就行了，因为本身mockSession也不会做任何network call，我们只需要mocksession管理自己就行。接着是URLSession的部分，因为MockSession得到的数据是本地就有的数据，所以只需要得到本身的数据，在return mockDataTask就可以了。基本上这里是在模拟一个URLSessio