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() )