iOS中使用单元测试检查内存泄漏 iOS中使用单元测试检查内存泄

在任何的程序开发中，内存泄漏都是个需要令人重视的问题，因为它直接影响着程序的性能与质量，同时也影响着用户体验。要是用户用着用着，app内存占用过多被系统杀死，用户也懵了，不知道咋回事就闪退了，及其不好。
所以解决app中的内存泄漏问题，显得尤其重要。检查内存泄漏问题，可以试用IDE自带的工具，这里不做介绍。下面主要介绍一种比较新奇有意思的思路，通过单元测试来检测。
主要原理

weak修饰的对象在释放后会自动置为nil
给对象新赋值后，之前指向的对象会释放掉

解释下第二点

id obj = [NSObject new]; // 重新赋值，之前new的对象会被释放（如果没被其他对象持有） obj = [NSObject new];

所以，在单元测试中，可结合这两个原则，来思考。
delegate 一般的，我们会让delegate是weak的，如果不小心写成了strong，会引起循环引用。那么如何用单元测试来检查呢？
根据上面的原则，可以先创建一个对象obj后，赋给delegate，然后将obj重新赋值，由于之前的对象会释放掉，就意味着delegate所指向的对象也应该释放了，这时候如果delegate是weak修饰的，即可判断其是否为nil。如果不为nil，因为这里obj没有被其他对象引用，可以判定为会存在内存泄漏。

- (void)testDelagte { id obj = [Test new]; ViewController *vc = [ViewController new]; vc.delegate = obj; // obj被赋值，之前的对象会销毁，如果vc.delegate是weak修饰，则会变成nil obj = [Test new]; XCTAssertNil(vc.delegate); }

observer 【iOS中使用单元测试检查内存泄漏】在我们自己封装观察者模式时，会将观察者的实例保存起来，等有事件发生时，再逐一通知。这时候，是不希望强引用实例的。这里，也可以通过单元测试检测一波~。

- (void)testObserver { NotificationCenter *center = [NotificationCenter new]; Observer *observer = [Observer new]; weak weakObserver = observer; [center addObserver:observer]; observer = [Observer new]; XCTAssertNil(weakObserver) }

这里center是直接强引用observer的，所以测试不会通过。可以采用包装对象的方式，来避免强引用oberver。

@interface WrapperObj: NSObject @property (nonatomic, weak) id obj; @end

block block会capture内部使用的变量，进行retain，也是最容易造成循环引用的地方。一般的，在异步调用中，我们很多时候会使用block，保存回调，事情处理完后，再调用回调。比如网络请求。如果在网络请求结束后，还持有block，那么block内部的变量也不会释放掉，造成内存泄漏。所以，需要检测block是否有被清掉。
虽然使用weak可以避免，但是这里的检查重点是block是否清除的问题。
可以用上面的方式来写，只不过需要单独将block先赋给一个变量，稍显麻烦。

void (^completion)(void) = ^{ NSLog(@"hello"); }; __weak void (^weakCompletion)(void) = completion; completion = ^{}; NSLog(@"%@", weakCompletion);

还有另外一种方式。我们知道，在block被释放后，它持有的对象，引用计数会减1。因此，可以通过有意的让block持有对象，然后检查对象是否为nil，来得出block是否被移除。

id obj = [NSObject new]; __weak id weakObj = obj; NSString *url = @"xxx"; NSData *data = https://www.it610.com/article/[NSData dataWithContentsOfURL:[NSURL URLWithString:@"xxx"]]; // mock response NetworkResponse *response = [NetworkManager mockResponse:url data:data]; [NetworkManager loadUrl:url completion:^{ NSLog(@"%@", obj); }]; // response回来后，看是否会清除completion [response send]; obj = [NSObject new]; XCTAssertNil(weakObj);