diff options
| author | Thom Nichols <tmnichols@gmail.com> | 2010-06-07 13:43:37 -0400 |
|---|---|---|
| committer | Thom Nichols <tmnichols@gmail.com> | 2010-06-07 13:43:37 -0400 |
| commit | 47f1fb16909d3baaec5822b1dcbca89491d0d18c (patch) | |
| tree | f60363662ece056558deb7e846d2f5177ac115d3 /tests/test_statemachine.py | |
| parent | 66cf0c20218657622a5d98aee155df1ea37d9a89 (diff) | |
| download | slixmpp-47f1fb16909d3baaec5822b1dcbca89491d0d18c.tar.gz slixmpp-47f1fb16909d3baaec5822b1dcbca89491d0d18c.tar.bz2 slixmpp-47f1fb16909d3baaec5822b1dcbca89491d0d18c.tar.xz slixmpp-47f1fb16909d3baaec5822b1dcbca89491d0d18c.zip | |
context manager now returns a boolean 'result' as the context variable to indicate whether the transition timed out or if you are actually locked when entering the context body
Diffstat (limited to 'tests/test_statemachine.py')
| -rw-r--r-- | tests/test_statemachine.py | 15 |
1 files changed, 9 insertions, 6 deletions
diff --git a/tests/test_statemachine.py b/tests/test_statemachine.py index 4cfb50de..0173ff09 100644 --- a/tests/test_statemachine.py +++ b/tests/test_statemachine.py @@ -223,11 +223,12 @@ class testStateMachine(unittest.TestCase): s = sm.StateMachine(('one','two','three')) - with s.transition_ctx('two','three') as _s: - self.assertTrue( _s['one'] ) - self.failIf( _s.current_state in ('two','three') ) + with s.transition_ctx('two','three') as result: + self.failIf( result ) + self.assertTrue( s['one'] ) + self.failIf( s.current_state in ('two','three') ) - self.assertTrue( _s['one'] ) + self.assertTrue( s['one'] ) def r1(): print 'thread 1 started' @@ -237,10 +238,12 @@ class testStateMachine(unittest.TestCase): def r2(): print 'thread 2 started' self.failIf( s['two'] ) - with s.transition_ctx('two','three', 10) as _s: - self.assertTrue( _s['two'] ) + with s.transition_ctx('two','three', 10) as result: + self.assertTrue( result ) + self.assertTrue( s['two'] ) print 'thread 2 will transition on exit from the context manager...' self.assertTrue( s['three'] ) + print 'transitioned to %s' % s.current_state() t1 = threading.Thread(target=r1) t2 = threading.Thread(target=r2) |