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'])