def testUnassignProject( self ):
resolver = MilestoneRelationshipResolver( 'test-milestone', self._db )
resolver.assignToProjectID( 'test-project' )
projects = resolver.getAllAssignedProjectIDs()
self.assertEquals([ 'test-project' ], projects )
resolver.unassignFromProjectID( 'test-project' )
projects = resolver.getAllAssignedProjectIDs()
self.assertEquals( [], projects )
self.assertDatabaseContainsNoRow()
def testAssignProject( self ):
resolver1 = MilestoneRelationshipResolver( 'test-milestone', self._db )
resolver2 = MilestoneRelationshipResolver( 'test-milestone-2', self._db )
resolver1.assignToProjectID( 'test-project-name' )
resolver2.assignToProjectID( 'another-project-name' )
projects1 = resolver1.getAllAssignedProjectIDs()
projects2 = resolver2.getAllAssignedProjectIDs()
self.assertEquals([ 'test-project-name' ], projects1 )
self.assertEquals([ 'another-project-name' ], projects2 )
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 testGetAllAssignedProjects( self ):
resolver = MilestoneRelationshipResolver( 'test-milestone', self._db )
self.assertEquals( [], resolver.getAllAssignedProjectIDs() )
