在任何的程序开发中，内存泄漏都是个需要令人重视的问题，因为它直接影响着程序的性能与质量，同时也影响着用户体验。要是用户用着用着，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

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

- (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 = [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);
复制代码