def test_augmenting_missing_src_status(self):
obj = Transition('foo', src_status_title='Bar',
dest_status_title='Baz')
states = {'Baz': Status('Baz', [])}
with self.assertRaises(ValueError) as cm:
obj.augment_states(states)
self.assertEquals('No such src_status "Bar" (foo).',
str(cm.exception))
def test_augmenting_states(self):
obj = Transition('foo', src_status_title='Bar',
dest_status_title='Baz')
states = {'Bar': Status('Bar', []),
'Baz': Status('Baz', [])}
obj.augment_states(states)
self.assertEquals(obj.src_status, states['Bar'])
self.assertEquals(obj.dest_status, states['Baz'])
