GBTestObjectsRegistry.m

//
//  GBTestObjectsRegistry.m
//  appledoc
//
//  Created by Tomaz Kragelj on 26.7.10.
//  Copyright (C) 2010, Gentle Bytes. All rights reserved.
//

#import "GBApplicationSettingsProvider.h"
#import "GBDataObjects.h"
#import "GBTestObjectsRegistry.h"

#define GBMOCKVALUE(variable) [NSValue value:&variable withObjCType:@encode(typeof(variable))]

@implementation GBTestObjectsRegistry

#pragma mark Common objects creation methods

+ (id)realSettingsProvider {
	GBApplicationSettingsProvider *result = [GBApplicationSettingsProvider provider];
	result.projectName = @"project";
	result.projectCompany = @"company";
	return result;
}

+ (OCMockObject *)mockSettingsProvider {
	OCMockObject *result = [OCMockObject niceMockForClass:[GBApplicationSettingsProvider class]];
	[[[result stub] andReturn:@"project"] projectName];
	[[[result stub] andReturn:@"company"] projectCompany];
	[[[result stub] andReturn:[GBCommentComponentsProvider provider]] commentComponents];
	[[[result stub] andReturn:[GBApplicationStringsProvider provider]] stringTemplates];
	return result;
}

+ (void)settingsProvider:(OCMockObject *)provider keepObjects:(BOOL)objects keepMembers:(BOOL)members {
	[[[provider stub] andReturnValue:[NSNumber numberWithBool:objects]] keepUndocumentedObjects];
	[[[provider stub] andReturnValue:[NSNumber numberWithBool:members]] keepUndocumentedMembers];
}

+ (void)registerComment:(id)comment forObject:(GBModelBase *)object {
	object.comment = comment;
}

#pragma mark GBIvarData creation methods

+ (GBIvarData *)ivarWithComponents:(NSString *)first, ... {
	va_list args;
	va_start(args, first);
	NSMutableArray *components = [NSMutableArray array];
	for (NSString *argument=first; argument!=nil; argument=va_arg(args, NSString*)) {
		[components addObject:argument];
	}
	va_end(args);
	return [GBIvarData ivarDataWithComponents:components];
}

#pragma mark GBMethodData creation methods

+ (GBMethodData *)instanceMethodWithName:(NSString *)name comment:(id)comment {
	GBMethodData *result = [self instanceMethodWithNames:name, nil];
	[self registerComment:comment forObject:result];
	return result;
}

+ (GBMethodData *)instanceMethodWithArguments:(GBMethodArgument *)first,... {
	va_list args;
	va_start(args, first);
	NSMutableArray *arguments = [NSMutableArray array];
	for (GBMethodArgument *argument=first; argument!=nil; argument=va_arg(args, GBMethodArgument*)) {
		[arguments addObject:argument];
	}
	va_end(args);
	return [GBMethodData methodDataWithType:GBMethodTypeInstance result:[NSArray arrayWithObject:@"void"] arguments:arguments];
}

+ (GBMethodData *)classMethodWithArguments:(GBMethodArgument *)first,... {
	va_list args;
	va_start(args, first);
	NSMutableArray *arguments = [NSMutableArray array];
	for (GBMethodArgument *argument=first; argument!=nil; argument=va_arg(args, GBMethodArgument*)) {
		[arguments addObject:argument];
	}
	va_end(args);
	return [GBMethodData methodDataWithType:GBMethodTypeClass result:[NSArray arrayWithObject:@"void"] arguments:arguments];
}

+ (GBMethodData *)instanceMethodWithNames:(NSString *)first,... {
	va_list args;
	va_start(args, first);
	NSMutableArray *arguments = [NSMutableArray array];
	for (NSString *name=first; name!=nil; name=va_arg(args, NSString*)) {
		GBMethodArgument *argument = [self typedArgumentWithName:name];
		[arguments addObject:argument];
	}
	va_end(args);
	return [GBMethodData methodDataWithType:GBMethodTypeInstance result:[NSArray arrayWithObject:@"void"] arguments:arguments];
}

+ (GBMethodData *)classMethodWithNames:(NSString *)first,... {
	va_list args;
	va_start(args, first);
	NSMutableArray *arguments = [NSMutableArray array];
	for (NSString *name=first; name!=nil; name=va_arg(args, NSString*)) {
		GBMethodArgument *argument = [self typedArgumentWithName:name];
		[arguments addObject:argument];
	}
	va_end(args);
	return [GBMethodData methodDataWithType:GBMethodTypeClass result:[NSArray arrayWithObject:@"void"] arguments:arguments];
}

+ (GBMethodData *)propertyMethodWithArgument:(NSString *)name {
	GBMethodArgument *argument = [GBMethodArgument methodArgumentWithName:name];
	return [GBMethodData methodDataWithType:GBMethodTypeProperty result:[NSArray arrayWithObject:@"int"] arguments:[NSArray arrayWithObject:argument]];
}

+ (GBMethodArgument *)typedArgumentWithName:(NSString *)name {
	return [GBMethodArgument methodArgumentWithName:name types:[NSArray arrayWithObject:@"id"] var:name];
}

#pragma mark Store objects creation methods

+ (GBClassData *)classWithName:(NSString *)name methods:(GBMethodData *)first,... {
	GBClassData *result = [GBClassData classDataWithName:name];
	va_list args;
	va_start(args, first);
	for (GBMethodData *method=first; method!=nil; method=va_arg(args, GBMethodData*)) {
		[result.methods registerMethod:method];
	}
	va_end(args);
	return result;
}

+ (GBCategoryData *)categoryWithName:(NSString *)name className:(NSString *)className methods:(GBMethodData *)first,... {
	GBCategoryData *result = [GBCategoryData categoryDataWithName:name className:className];
	va_list args;
	va_start(args, first);
	for (GBMethodData *method=first; method!=nil; method=va_arg(args, GBMethodData*)) {
		[result.methods registerMethod:method];
	}
	va_end(args);
	return result;
}

+ (GBProtocolData *)protocolWithName:(NSString *)name methods:(GBMethodData *)first,... {
	GBProtocolData *result = [GBProtocolData protocolDataWithName:name];
	va_list args;
	va_start(args, first);
	for (GBMethodData *method=first; method!=nil; method=va_arg(args, GBMethodData*)) {
		[result.methods registerMethod:method];
	}
	va_end(args);
	return result;
}

#pragma mark GBStore creation methods

+ (GBStore *)store {
	return [[[GBStore alloc] init] autorelease];
}

+ (GBStore *)storeWithClassWithName:(NSString *)name {
	GBClassData *class = [GBClassData classDataWithName:name];
	return [self storeByPerformingSelector:@selector(registerClass:) withObject:class];
}

+ (GBStore *)storeWithClassWithComment:(id)comment {
	GBClassData *class = [GBClassData classDataWithName:@"Class"];
	[self registerComment:comment forObject:class];
	return [self storeByPerformingSelector:@selector(registerClass:) withObject:class];
}

+ (GBStore *)storeWithCategoryWithComment:(id)comment {
	GBCategoryData *category = [GBCategoryData categoryDataWithName:@"Category" className:@"Class"];
	[self registerComment:comment forObject:category];
	return [self storeByPerformingSelector:@selector(registerCategory:) withObject:category];
}

+ (GBStore *)storeWithProtocolWithComment:(id)comment {
	GBProtocolData *protocol = [GBProtocolData protocolDataWithName:@"Protocol"];
	[self registerComment:comment forObject:protocol];
	return [self storeByPerformingSelector:@selector(registerProtocol:) withObject:protocol];
}

+ (GBStore *)storeWithDocumentWithComment:(id)comment {
	// Note that we still assign the comment so that we can use mocks for testing - because of that we can safely pass arbitrary string to contents!
	GBDocumentData *document = [GBDocumentData documentDataWithContents:@"contents" path:@"path"];
	[self registerComment:comment forObject:document];
	return [self storeByPerformingSelector:@selector(registerDocument:) withObject:document];
}

+ (GBStore *)storeWithObjects:(id)first, ... {
	// If mocks are passed, they are registered as classes!
	GBStore *result = [self store];
	va_list args;
	va_start(args, first);
	for (id object=first; object!=nil; object=va_arg(args, id)) {
		if ([object isKindOfClass:[GBCategoryData class]])
			[result registerCategory:object];
		else if ([object isKindOfClass:[GBProtocolData class]])
			[result registerProtocol:object];
		else if ([object isKindOfClass:[GBDocumentData class]])
			[result registerDocument:object];
		else
			[result registerClass:object];
	}
	va_end(args);
	return result;
}

+ (GBStore *)storeByPerformingSelector:(SEL)selector withObject:(id)object {
	GBStore *result = [self store];
	[result performSelector:selector withObject:object];
	return result;
}

@end