//

// SZTestSuiteDataSource.m

//

// Xcode Unit Test GUI

// Copyright (c) 2009 Jon Nall, STUNTAZ!!!

// All rights reserved.

//

// Permission is hereby granted, free of charge, to any person

// obtaining a copy of this software and associated documentation

// files (the "Software"), to deal in the Software without

// restriction, including without limitation the rights to use,

// copy, modify, merge, publish, distribute, sublicense, and/or sell

// copies of the Software, and to permit persons to whom the

// Software is furnished to do so, subject to the following

// conditions:

//

// The above copyright notice and this permission notice shall be

// included in all copies or substantial portions of the Software.

//

// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,

// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES

// OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND

// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT

// HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,

// WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING

// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR

// OTHER DEALINGS IN THE SOFTWARE.

#import "SZTestSuiteDataSource.h"

#import "SZTestDescriptor.h"

NSString* const kszAllTests = @"All Tests";

NSString* const kszSelectedTests = @"Selected Tests";

NSString* const kszFailingTests = @"Failing Tests";

@implementation SZTestSuiteDataSource

@synthesize suites;

@synthesize tests;

-(id)init

{

self = [super init];

if(self != nil)

{

testCell = [[NSBrowserCell alloc] init];

[testCell setLeaf:YES];

enableCell = [[NSButtonCell alloc] init];

[enableCell setEnabled:YES];

[enableCell setAllowsMixedState:YES];

[enableCell setButtonType:NSSwitchButton];

[enableCell setStringValue:@""];

[enableCell setAlternateTitle:@""];

[enableCell setAttributedTitle:[[[NSAttributedString alloc] initWithString:@""] autorelease]];

[enableCell setTarget:self];

[enableCell setAction:@selector(enableClicked:)];

passedImage = [[NSImage imageNamed:@"testStatePassed"] retain];

failedImage = [[NSImage imageNamed:@"testStateFailed"] retain];

unknownImage = [[NSImage imageNamed:@"testStateUnknown"] retain];

}

return self;

}

-(void)dealloc

{

[testCell release];

[enableCell release];

[super dealloc];

}

-(void)updateSuiteState:(SZTestDescriptor*)theSuite

{

// Update suite state

NSArray* testArray = [tests objectAtIndex:theSuite.index];

NSInteger buttonState = NSOnState;

BOOL foundOn = NO;

for(SZTestDescriptor* test in testArray)

{

if(test.enabled == NO)

{

// If we find anything off, transition to mixed

buttonState = NSMixedState;

}

// Track if we found any that were on

foundOn |= test.enabled;

}

// If we found none on, transition to off

if(foundOn == NO)

{

buttonState = NSOffState;

}

theSuite.enabled = buttonState;

}

-(void)enableClicked:(id)sender

{

const NSUInteger row = [outlineView clickedRow];

SZTestDescriptor* theTest = [outlineView itemAtRow:row];

if(theTest.type == TestcaseType)

{

assert(theTest.enabled == NSOnState || theTest.enabled == NSOffState);

theTest.enabled = (theTest.enabled == NSOnState) ? NSOffState : NSOnState;

SZTestDescriptor* suite = [outlineView parentForItem:theTest];

[self updateSuiteState:suite];

}

else

{

BOOL updateTests = NO;

NSInteger newState = theTest.enabled;

switch(theTest.enabled)

{

case NSOffState:

{

newState = NSOnState;

updateTests = YES;

break;

}

case NSOnState:

{

newState = NSOffState;

updateTests = YES;

break;

}

case NSMixedState:

// Do nothing

updateTests = NO;

break;

default:

NSBeep();

NSLog(@"Unexpected Button State: %d", theTest.enabled);

}

if(updateTests)

{

NSArray* testArray = [tests objectAtIndex:theTest.index];

for(SZTestDescriptor* test in testArray)

{

assert(newState != NSMixedState);

test.enabled = newState;

}

theTest.enabled = newState;

}

}

BOOL allOn = YES;

for(SZTestDescriptor* suite in suites)

{

if(suite.enabled != NSOnState)

{

[runTypeButton selectItemWithTitle:kszSelectedTests];

allOn = NO;

break;

}

}

if(allOn)

{

[runTypeButton selectItemWithTitle:kszAllTests];

}

[outlineView reloadData];

}

-(NSImage*)stateToImage:(const enum TestState)theState

{

switch(theState)

{

case TestPassed: return passedImage;

case TestFailed: return failedImage;

case TestUnknown: return unknownImage;

default:

{

NSBeep();

NSLog(@"Unexpected state: %d", theState);

}

}

return nil;

}

-(id)outlineView:(NSOutlineView*)outlineView

child:(NSInteger)index

ofItem:(id)item

{

if(item == nil)

{

return [suites objectAtIndex:index];

}

else

{

SZTestDescriptor* t = item;

return [[tests objectAtIndex:t.index] objectAtIndex:index];

}

return nil;

}

-(BOOL)outlineView:(NSOutlineView*)outlineView

isItemExpandable:(id)item

{

SZTestDescriptor* t = item;

return t.type == TestsuiteType;

}

-(NSInteger)outlineView:(NSOutlineView*)outlineView

numberOfChildrenOfItem:(id)item

{

if(item == nil)

{

return [suites count];

}

else

{

SZTestDescriptor* t = item;

return [[tests objectAtIndex:t.index] count];

}

return 0;

}

-(void)outlineView:(NSOutlineView*)outlineView

willDisplayCell:(id)theCell

forTableColumn:(NSTableColumn*)tableColumn

item:(id)item

{

SZTestDescriptor* test = item;

if([[tableColumn identifier] isEqualToString:@"tests"])

{

NSBrowserCell* c = theCell;

[c setImage:[self stateToImage:test.state]];

[c setStringValue:test.name];

}

else

{

NSButtonCell* c = theCell;

[c setState:test.enabled];

}

}

-(id)outlineView:(NSOutlineView*)outlineView

objectValueForTableColumn:(NSTableColumn*)tableColumn

byItem:(id)item

{

return item;

}

-(NSCell*)outlineView:(NSOutlineView*)outlineView

dataCellForTableColumn:(NSTableColumn*)tableColumn

item:(id)item

{

if(tableColumn == nil)

{

// Use different cells for each column

return nil;

}

if([[tableColumn identifier] isEqualToString:@"tests"])

{

return testCell;

}

else

{

return enableCell;

}

}

-(BOOL)outlineView:(NSOutlineView*)outlineView

shouldEditTableColumn:(NSTableColumn *)tableColumn

item:(id)item

{

// No editing ever allowed

return NO;

}

-(void)invalidateStates

{

for(SZTestDescriptor* t in suites)

{

t.state = TestUnknown;

}

for(NSArray* a in tests)

{

for(SZTestDescriptor* t in a)

{

t.state = TestUnknown;

}

}

}

@end