class-dump/UnitTests/TestBlockSignature.m
2019-03-01 23:58:53 -07:00

80 lines
2.8 KiB
Objective-C

// -*- mode: ObjC -*-
// This file is part of class-dump, a utility for examining the Objective-C segment of Mach-O files.
// Copyright (C) 1997-2019 Steve Nygard.
#import <XCTest/XCTest.h>
#import "CDType.h"
#import "CDTypeName.h"
@interface CDType (UnitTests)
- (NSString *)blockSignatureString;
@end
@interface TestBlockSignature : XCTestCase
@end
@implementation TestBlockSignature
- (void)testZeroArguments;
{
NSMutableArray *types = [NSMutableArray new];
[types addObject:[[CDType alloc] initSimpleType:'v']];
[types addObject:[[CDType alloc] initBlockTypeWithTypes:nil]];
CDType *blockType = [[CDType alloc] initBlockTypeWithTypes:types];
NSString *blockSignatureString = [blockType blockSignatureString];
XCTAssertEqualObjects(blockSignatureString, @"void (^)(void)", @"");
}
- (void)testOneArgument;
{
NSMutableArray *types = [NSMutableArray new];
[types addObject:[[CDType alloc] initSimpleType:'v']];
[types addObject:[[CDType alloc] initBlockTypeWithTypes:nil]];
CDTypeName *typeName = [CDTypeName new];
typeName.name = @"NSData";
[types addObject:[[CDType alloc] initIDType:typeName]];
CDType *blockType = [[CDType alloc] initBlockTypeWithTypes:types];
NSString *blockSignatureString = [blockType blockSignatureString];
XCTAssertEqualObjects(blockSignatureString, @"void (^)(NSData *)", @"");
}
- (void)testTwoArguments;
{
NSMutableArray *types = [NSMutableArray new];
[types addObject:[[CDType alloc] initSimpleType:'v']];
[types addObject:[[CDType alloc] initBlockTypeWithTypes:nil]];
[types addObject:[[CDType alloc] initIDType:nil]];
CDTypeName *typeName = [CDTypeName new];
typeName.name = @"NSError";
[types addObject:[[CDType alloc] initIDType:typeName]];
CDType *blockType = [[CDType alloc] initBlockTypeWithTypes:types];
NSString *blockSignatureString = [blockType blockSignatureString];
XCTAssertEqualObjects(blockSignatureString, @"void (^)(id, NSError *)", @"");
}
- (void)testBlockArgument;
{
NSMutableArray *types = [NSMutableArray new];
[types addObject:[[CDType alloc] initSimpleType:'v']];
[types addObject:[[CDType alloc] initBlockTypeWithTypes:nil]];
[types addObject:[[CDType alloc] initBlockTypeWithTypes:[types copy]]];
CDType *blockType = [[CDType alloc] initBlockTypeWithTypes:types];
NSString *blockSignatureString = [blockType blockSignatureString];
XCTAssertEqualObjects(blockSignatureString, @"void (^)(void (^)(void))", @"");
}
- (void)testBOOLArgument;
{
NSMutableArray *types = [NSMutableArray new];
[types addObject:[[CDType alloc] initSimpleType:'v']];
[types addObject:[[CDType alloc] initBlockTypeWithTypes:nil]];
[types addObject:[[CDType alloc] initSimpleType:'c']];
CDType *blockType = [[CDType alloc] initBlockTypeWithTypes:types];
NSString *blockSignatureString = [blockType blockSignatureString];
XCTAssertEqualObjects(blockSignatureString, @"void (^)(BOOL)", @"");
}
@end