def setUp(self): self.mAwaiter = Awaiter()
class Awaiter_KeepWaiting(unittest.TestCase): def setUp(self): self.mAwaiter = Awaiter() def testKeepWaitingReturnsFalseIfThereIsOnlyOneNotStartedState(self): workStates = \ { 'Work1': GraphRiteSession.STATE_NOT_STARTED } self.assertFalse(self.mAwaiter.keepWaiting(workStates)) def testKeepWaitingReturnsTrueIfThereIsOnlyOneStartedState(self): workStates = \ { 'Work1': GraphRiteSession.STATE_STARTED } self.assertTrue(self.mAwaiter.keepWaiting(workStates)) def testKeepWaitingReturnsFalseIfThereIsOnlyOneSucceedState(self): workStates = \ { 'Work1': GraphRiteSession.STATE_SUCCEED } self.assertFalse(self.mAwaiter.keepWaiting(workStates)) def testKeepWaitingReturnsFalseIfThereIsOnlyOneFailedState(self): workStates = \ { 'Work1': GraphRiteSession.STATE_FAILED } self.assertFalse(self.mAwaiter.keepWaiting(workStates)) def testKeepWaitingReturnsFalseIfThereAreOnlyNotStartedStates(self): workStates = \ { 'Work1': GraphRiteSession.STATE_NOT_STARTED, 'Work2': GraphRiteSession.STATE_NOT_STARTED, 'Work3': GraphRiteSession.STATE_NOT_STARTED, 'Work4': GraphRiteSession.STATE_NOT_STARTED } self.assertFalse(self.mAwaiter.keepWaiting(workStates)) def testKeepWaitingReturnsTrueIfThereAreOnlyStartedStates(self): workStates = \ { 'Work1': GraphRiteSession.STATE_STARTED, 'Work2': GraphRiteSession.STATE_STARTED, 'Work3': GraphRiteSession.STATE_STARTED, 'Work4': GraphRiteSession.STATE_STARTED } self.assertTrue(self.mAwaiter.keepWaiting(workStates)) def testKeepWaitingReturnsFalseIfThereAreOnlySucceedStates(self): workStates = \ { 'Work1': GraphRiteSession.STATE_FAILED, 'Work2': GraphRiteSession.STATE_FAILED, 'Work3': GraphRiteSession.STATE_FAILED, 'Work4': GraphRiteSession.STATE_FAILED } self.assertFalse(self.mAwaiter.keepWaiting(workStates)) def testKeepWaitingReturnsFalseIfThereAreOnlyFailedStates(self): workStates = \ { 'Work1': GraphRiteSession.STATE_FAILED, 'Work2': GraphRiteSession.STATE_FAILED, 'Work3': GraphRiteSession.STATE_FAILED, 'Work4': GraphRiteSession.STATE_FAILED } self.assertFalse(self.mAwaiter.keepWaiting(workStates)) def testKeepAwaitingReturnsTrueIfThereAreStartedStates_OneStartedOneAnother(self): workStates = \ { 'Work1': GraphRiteSession.STATE_NOT_STARTED, 'Work2': GraphRiteSession.STATE_STARTED } self.assertTrue(self.mAwaiter.keepWaiting(workStates)) def testKeepAwaitingReturnsTrueIfThereAreStartedStates_OneStartedManyAnother(self): workStates = \ { 'Work1': GraphRiteSession.STATE_STARTED, 'Work2': GraphRiteSession.STATE_SUCCEED, 'Work3': GraphRiteSession.STATE_FAILED, 'Work4': GraphRiteSession.STATE_NOT_STARTED } self.assertTrue(self.mAwaiter.keepWaiting(workStates)) def testKeepAwaitingReturnsTrueIfThereAreStartedStates_ManyStartedOneAnother(self): workStates = \ { 'Work1': GraphRiteSession.STATE_STARTED, 'Work2': GraphRiteSession.STATE_STARTED, 'Work3': GraphRiteSession.STATE_FAILED, 'Work4': GraphRiteSession.STATE_STARTED } self.assertTrue(self.mAwaiter.keepWaiting(workStates)) def testKeepAwaitingReturnsTrueIfThereAreStartedStates_ManyStartedManyAnother(self): workStates = \ { 'Work1': GraphRiteSession.STATE_STARTED, 'Work2': GraphRiteSession.STATE_SUCCEED, 'Work3': GraphRiteSession.STATE_FAILED, 'Work4': GraphRiteSession.STATE_STARTED } self.assertTrue(self.mAwaiter.keepWaiting(workStates))