Swiftgram/Tests/Common/ASDisplayNode+OCMock.m

32 lines
720 B
Objective-C

//
// ASDisplayNode+OCMock.m
// Texture
//
// Copyright (c) 2017-present, Pinterest, Inc. All rights reserved.
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
#import <AsyncDisplayKit/AsyncDisplayKit.h>
/**
* For some reason, when creating partial mocks of nodes, OCMock fails to find
* these class methods that it swizzled!
*/
@implementation ASDisplayNode (OCMock)
+ (Class)ocmock_replaced_viewClass
{
return [_ASDisplayView class];
}
+ (Class)ocmock_replaced_layerClass
{
return [_ASDisplayLayer class];
}
@end