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 testAssignProjectChangesDatabase( self ): resolver = MilestoneRelationshipResolver( 'test-milestone', self._db ) resolver.assignToProjectID( 'project-a' ) self.assertDatabaseContainsSingleRow( ( 'test-milestone', 'project', 'project-a' ))