def test_transitionRetryPolicyOverridesMachineLevelPolicy(self):
setUpByFilename(self, 'test-TaskQueueFSMTests.yaml')
transInitialToNormal = self.initialState._eventToTransition[
'next-event']
self.assertNotEquals(
self.machineConfig.taskRetryLimit,
transInitialToNormal.retryOptions.task_retry_limit)
def setUp(self):
super(StartStateMachineTests, self).setUp()
filename = 'test-TaskQueueFSMTests.yaml'
setUpByFilename(self, filename)
self.machineName = getMachineNameByFilename(filename)
self.mockQueue = TaskQueueDouble()
mock(name='Queue.add', returns_func=self.mockQueue.add, tracker=None)
def setUp(self):
super(StartStateMachineTests, self).setUp()
filename = 'test-TaskQueueFSMTests.yaml'
setUpByFilename(self, filename)
self.machineName = getMachineNameByFilename(filename)
self.mockQueue = TaskQueueDouble()
mock(name='Queue.add', returns_func=self.mockQueue.add, tracker=None)
def setUp(self):
super(DatastoreFSMContinuationBaseTests, self).setUp()
setUpByFilename(self, self.FILENAME, instanceName='instanceName', machineName=self.MACHINE_NAME)
self.mockQueue = TaskQueueDouble()
mock(name='Queue.add', returns_func=self.mockQueue.add, tracker=None)
self.loggingDouble = getLoggingDouble()
self.modelKeys = []
for i in range(10):
modelKey = TestModel().put()
self.modelKeys.append(modelKey)
def setUp(self):
super(FSMContextTests, self).setUp()
filename = 'test-FSMContextTests.yaml'
setUpByFilename(self, filename)
self.machineName = getMachineNameByFilename(filename)
self.mockQueue = TaskQueueDouble()
mock(name='Queue.add', returns_func=self.mockQueue.add, tracker=None)
# dispatch initial event to get context in correct state
self.taskName = 'foo'
self.obj = {TASK_NAME_PARAM: self.taskName}
self.context.dispatch(FSM.PSEUDO_INIT, self.obj)
def setUp(self):
super(FSMContextTests, self).setUp()
filename = 'test-FSMContextTests.yaml'
setUpByFilename(self, filename)
self.machineName = getMachineNameByFilename(filename)
self.mockQueue = TaskQueueDouble()
mock(name='Queue.add', returns_func=self.mockQueue.add, tracker=None)
# dispatch initial event to get context in correct state
self.taskName = 'foo'
self.obj = {TASK_NAME_PARAM: self.taskName}
self.context.dispatch(FSM.PSEUDO_INIT, self.obj)
def setUp(self):
super(TaskQueueFSMTests, self).setUp()
filename = 'test-TaskQueueFSMTests.yaml'
setUpByFilename(self, filename)
machineName = getMachineNameByFilename(filename)
self.stateInitial = self.factory.machines[machineName][MACHINE_STATES_ATTRIBUTE]['state-initial']
self.stateNormal = self.factory.machines[machineName][MACHINE_STATES_ATTRIBUTE]['state-normal']
self.stateFinal = self.factory.machines[machineName][MACHINE_STATES_ATTRIBUTE]['state-final']
self.transInitialToNormal = self.stateInitial._eventToTransition['next-event']
self.transNormalToFinal = self.stateNormal._eventToTransition['next-event']
def setUp(self):
super(DatastoreFSMContinuationBaseTests, self).setUp()
setUpByFilename(self,
self.FILENAME,
instanceName='instanceName',
machineName=self.MACHINE_NAME)
self.mockQueue = TaskQueueDouble()
mock(name='Queue.add', returns_func=self.mockQueue.add, tracker=None)
self.loggingDouble = getLoggingDouble()
self.modelKeys = []
for i in range(10):
modelKey = TestModel().put()
self.modelKeys.append(modelKey)
def setUp(self):
super(SpawnTests, self).setUp()
filename = 'test-TaskQueueFSMTests.yaml'
setUpByFilename(self, filename)
self.machineName = getMachineNameByFilename(filename)
self.mockQueue = TaskQueueDouble()
mock(name='Queue.add', returns_func=self.mockQueue.add, tracker=None)
# dispatch initial event to get context in correct state
self.taskName = 'foo'
self.obj = {TASK_NAME_PARAM: self.taskName}
self.context.dispatch(FSM.PSEUDO_INIT, self.obj)
# but now flush the task queue to remove the event that we just dispatched;
# we're only interested in testing spawn tasks
self.mockQueue.purge()
def setUp(self):
super(SpawnTests, self).setUp()
filename = 'test-TaskQueueFSMTests.yaml'
setUpByFilename(self, filename)
self.machineName = getMachineNameByFilename(filename)
self.mockQueue = TaskQueueDouble()
mock(name='Queue.add', returns_func=self.mockQueue.add, tracker=None)
# dispatch initial event to get context in correct state
self.taskName = 'foo'
self.obj = {TASK_NAME_PARAM: self.taskName}
self.context.dispatch(FSM.PSEUDO_INIT, self.obj)
# but now flush the task queue to remove the event that we just dispatched;
# we're only interested in testing spawn tasks
self.mockQueue.purge()
def setUp(self):
super(TaskQueueFSMTests, self).setUp()
filename = 'test-TaskQueueFSMTests.yaml'
setUpByFilename(self, filename)
machineName = getMachineNameByFilename(filename)
self.stateInitial = self.factory.machines[machineName][
MACHINE_STATES_ATTRIBUTE]['state-initial']
self.stateNormal = self.factory.machines[machineName][
MACHINE_STATES_ATTRIBUTE]['state-normal']
self.stateFinal = self.factory.machines[machineName][
MACHINE_STATES_ATTRIBUTE]['state-final']
self.transInitialToNormal = self.stateInitial._eventToTransition[
'next-event']
self.transNormalToFinal = self.stateNormal._eventToTransition[
'next-event']
def test_imports_and_machines(self):
setUpByFilename(self, 'test-YamlImport.yaml', machineName='Foo')
self.assertTrue('MyMachine' in self.currentConfig.machines)
self.assertTrue('TypeCoercionTests' in self.currentConfig.machines)
self.assertTrue('Foo' in self.currentConfig.machines)
def setUp(self):
super(ContextTypesCoercionTests, self).setUp()
setUpByFilename(self, 'test-TypeCoercionTests.yaml')
def test_transitionRetryPolicyOverridesMachineLevelPolicy(self):
setUpByFilename(self, 'test-TaskQueueFSMTests.yaml')
transInitialToNormal = self.initialState._eventToTransition['next-event']
self.assertNotEquals(self.machineConfig.taskRetryLimit, transInitialToNormal.retryOptions.task_retry_limit)
def test_TestYamlFileLocation(self):
setUpByFilename(self, 'test-TestYamlFileLocation.yaml', machineName='MyMachine')
def test_createFSMInstance_raises_UnknownStateError_for_currentState(self):
setUpByFilename(self, 'test-TaskQueueFSMTests.yaml')
self.assertRaises(UnknownStateError, self.factory.createFSMInstance, 'TaskQueueFSMTests',
currentStateName='foo')
def test_createFSMInstance_raises_UnknownMachineError(self):
setUpByFilename(self, 'test-TaskQueueFSMTests.yaml')
self.assertRaises(UnknownMachineError, self.factory.createFSMInstance,
'foo')
def test_TestYamlFileLocation(self):
setUpByFilename(self,
'test-TestYamlFileLocation.yaml',
machineName='MyMachine')
def test_createFSMInstance_initial_data(self):
setUpByFilename(self, 'test-TaskQueueFSMTests.yaml')
context = self.factory.createFSMInstance('TaskQueueFSMTests',
data={'a': 'b'})
self.assertEqual({'a': 'b'}, context)
def test_imports_and_machines(self):
setUpByFilename(self, 'test-YamlImport.yaml', machineName='Foo')
self.assertTrue('MyMachine' in self.currentConfig.machines)
self.assertTrue('TypeCoercionTests' in self.currentConfig.machines)
self.assertTrue('Foo' in self.currentConfig.machines)
def setUp(self):
super(ContextTypesCoercionTests, self).setUp()
setUpByFilename(self, 'test-TypeCoercionTests.yaml')
def setUp(self):
super(LoggerTestPersistent, self).setUp()
filename = 'test-FSMContextTests.yaml'
setUpByFilename(self, filename)
self.context.logger.persistentLogging = self.PERSISTENT_LOGGING
self.loggingDouble = getLoggingDouble()
def test_createFSMInstance_initial_data(self):
setUpByFilename(self, 'test-TaskQueueFSMTests.yaml')
context = self.factory.createFSMInstance('TaskQueueFSMTests', data={'a' : 'b'})
self.assertEqual({'a' : 'b'}, context)
def test_createFSMInstance_raises_UnknownStateError_for_currentState(self):
setUpByFilename(self, 'test-TaskQueueFSMTests.yaml')
self.assertRaises(UnknownStateError,
self.factory.createFSMInstance,
'TaskQueueFSMTests',
currentStateName='foo')
def test_TaskQueueFSMTests(self):
setUpByFilename(self, 'test-TaskQueueFSMTests.yaml')
def test_createFSMInstance_raises_UnknownMachineError(self):
setUpByFilename(self, 'test-TaskQueueFSMTests.yaml')
self.assertRaises(UnknownMachineError, self.factory.createFSMInstance, 'foo')
def setUp(self):
super(LoggerTestPersistent, self).setUp()
filename = 'test-FSMContextTests.yaml'
setUpByFilename(self, filename)
self.context.logger.persistentLogging = self.PERSISTENT_LOGGING
self.loggingDouble = getLoggingDouble()
def test_TaskQueueFSMTests(self):
setUpByFilename(self, 'test-TaskQueueFSMTests.yaml')
