def test_all_state_present(self):
check = RequiredState('thing', 'blah')
flow = MockFlow()
request = None
flow.state = {'thing': 1, 'blah': 'cake'}
self.assertTrue(check.process(request, flow) is None)
def test_missing_state(self):
check = RequiredState('thing', 'blah')
flow = MockFlow()
flow.state = {'thing': 1}
request = None
response = check.process(request, flow)
# ensure that we do something with the request rather than
# just let it pass through
self.assertTrue( isinstance(response, HttpResponse) )