def testUnassignCompoent( self ):
resolver = MilestoneRelationshipResolver( 'test-milestone', self._db )
resolver.assignToComponentID( 'test-component' )
components = resolver.getAllAssignedComponentIDs()
self.assertEquals([ 'test-component' ], components )
resolver.unassignFromComponentID( 'test-component' )
components = resolver.getAllAssignedComponentIDs()
self.assertEquals( [], components )
self.assertDatabaseContainsNoRow()
def testAssignComponent( self ):
resolver1 = MilestoneRelationshipResolver( 'test-milestone', self._db )
resolver2 = MilestoneRelationshipResolver( 'test-milestone-2', self._db )
resolver1.assignToComponentID( 'test-component-name' )
resolver2.assignToComponentID( 'another-component-name' )
components1 = resolver1.getAllAssignedComponentIDs()
components2 = resolver2.getAllAssignedComponentIDs()
self.assertEquals([ 'test-component-name' ], components1 )
self.assertEquals([ 'another-component-name' ], components2 )
def getByName( self, name ):
resolver = MilestoneRelationshipResolver( name, self._db )
componentIDs = resolver.getAllAssignedComponentIDs()
components = self._componentFactory.getAllByNames( componentIDs )
projectIDs = resolver.getAllAssignedProjectIDs()
projects = self._projectFactory.getAllByNames( projectIDs )
return Milestone( name, components, projects )
def testGetAllAssignedComponentes( self ):
resolver = MilestoneRelationshipResolver( 'test-milestone', self._db )
self.assertEquals( [], resolver.getAllAssignedComponentIDs() )