upstream/mercurial-mirror Files · tests/test-ctxmanager.py

rebase: clean up rebasestate from active transaction...

rebase: clean up rebasestate from active transaction Previously, rebase assumes the following pattern: rebase: with transaction as tr: # top-level ... tr.__close__ writes rebasestate unlink('rebasestate') However it's possible that "rebase" was called inside a transaction: with transaction as tr1: rebase: with transaction as tr2: # not top-level ... tr2.__close__ does not write rebasestate unlink('rebasestate') tr1.__close__ writes rebasestate That leaves a rebasestate on disk incorrectly. This patch adds "removefilegenerator" to notify transaction code that the state file is no longer needed therefore fixes the issue.

Gregory Szorc - - Load All Authors

File last commit:

r32279:68c43a41 default


                r33056:2312e70c

default

Download file

             test-ctxmanager.py
        
                    77 lines
            
             | 2.4 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / tests / test-ctxmanager.py
          
                    History
                
                 |
                  Source
                 | Raw
                 |Copy content
                 |Copy permalink

        Bryan O'Sullivan
    
util: introduce ctxmanager, to avoid nested try/finally blocks...

              r27703
            
      from __future__ import absolute_import

      import silenttestrunner

      import unittest

        Yuya Nishihara
    
test-ctxmanager: stop direct symbol import of mercurial.util

              r28801
            
      from mercurial import util

        Bryan O'Sullivan
    
util: introduce ctxmanager, to avoid nested try/finally blocks...

              r27703
            
      class contextmanager(object):

          def __init__(self, name, trace):

              self.name = name

              self.entered = False

              self.exited = False

              self.trace = trace

          def __enter__(self):

              self.entered = True

              self.trace(('enter', self.name))

              return self

          def __exit__(self, exc_type, exc_val, exc_tb):

              self.exited = exc_type, exc_val, exc_tb

              self.trace(('exit', self.name))

          def __repr__(self):

              return '<ctx %r>' % self.name

      class ctxerror(Exception):

          pass

      class raise_on_enter(contextmanager):

          def __enter__(self):

              self.trace(('raise', self.name))

              raise ctxerror(self.name)

      class raise_on_exit(contextmanager):

          def __exit__(self, exc_type, exc_val, exc_tb):

              self.trace(('raise', self.name))

              raise ctxerror(self.name)

      def ctxmgr(name, trace):

          return lambda: contextmanager(name, trace)

      class test_ctxmanager(unittest.TestCase):

          def test_basics(self):

              trace = []

              addtrace = trace.append

        Yuya Nishihara
    
test-ctxmanager: stop direct symbol import of mercurial.util

              r28801
            
              with util.ctxmanager(ctxmgr('a', addtrace), ctxmgr('b', addtrace)) as c:

        Bryan O'Sullivan
    
util: rename ctxmanager's __call__ method to enter

              r27785
            
                  a, b = c.enter()

        Bryan O'Sullivan
    
util: introduce ctxmanager, to avoid nested try/finally blocks...

              r27703
            
                  c.atexit(addtrace, ('atexit', 'x'))

                  c.atexit(addtrace, ('atexit', 'y'))

              self.assertEqual(trace, [('enter', 'a'), ('enter', 'b'),

                                       ('atexit', 'y'), ('atexit', 'x'),

                                       ('exit', 'b'), ('exit', 'a')])

          def test_raise_on_enter(self):

              trace = []

              addtrace = trace.append

        Gregory Szorc
    
tests: use context manager form of assertRaises...

              r32279
            
              with self.assertRaises(ctxerror):

        Yuya Nishihara
    
test-ctxmanager: stop direct symbol import of mercurial.util

              r28801
            
                  with util.ctxmanager(ctxmgr('a', addtrace),

                                       lambda: raise_on_enter('b', addtrace)) as c:

        Bryan O'Sullivan
    
util: rename ctxmanager's __call__ method to enter

              r27785
            
                      c.enter()

        Bryan O'Sullivan
    
util: introduce ctxmanager, to avoid nested try/finally blocks...

              r27703
            
                      addtrace('unreachable')

              self.assertEqual(trace, [('enter', 'a'), ('raise', 'b'), ('exit', 'a')])

          def test_raise_on_exit(self):

              trace = []

              addtrace = trace.append

        Gregory Szorc
    
tests: use context manager form of assertRaises...

              r32279
            
              with self.assertRaises(ctxerror):

        Yuya Nishihara
    
test-ctxmanager: stop direct symbol import of mercurial.util

              r28801
            
                  with util.ctxmanager(ctxmgr('a', addtrace),

                                       lambda: raise_on_exit('b', addtrace)) as c:

        Bryan O'Sullivan
    
util: rename ctxmanager's __call__ method to enter

              r27785
            
                      c.enter()

        Bryan O'Sullivan
    
util: introduce ctxmanager, to avoid nested try/finally blocks...

              r27703
            
                      addtrace('running')

              self.assertEqual(trace, [('enter', 'a'), ('enter', 'b'), 'running',

                                       ('raise', 'b'), ('exit', 'a')])

      if __name__ == '__main__':

          silenttestrunner.main(__name__)

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

Bryan O'Sullivan util: introduce ctxmanager, to avoid nested try/finally blocks...	r27703	from __future__ import absolute_import

		import silenttestrunner
		import unittest

Yuya Nishihara test-ctxmanager: stop direct symbol import of mercurial.util	r28801	from mercurial import util
Bryan O'Sullivan util: introduce ctxmanager, to avoid nested try/finally blocks...	r27703
		class contextmanager(object):
		def __init__(self, name, trace):
		self.name = name
		self.entered = False
		self.exited = False
		self.trace = trace

		def __enter__(self):
		self.entered = True
		self.trace(('enter', self.name))
		return self

		def __exit__(self, exc_type, exc_val, exc_tb):
		self.exited = exc_type, exc_val, exc_tb
		self.trace(('exit', self.name))

		def __repr__(self):
		return '<ctx %r>' % self.name

		class ctxerror(Exception):
		pass

		class raise_on_enter(contextmanager):
		def __enter__(self):
		self.trace(('raise', self.name))
		raise ctxerror(self.name)

		class raise_on_exit(contextmanager):
		def __exit__(self, exc_type, exc_val, exc_tb):
		self.trace(('raise', self.name))
		raise ctxerror(self.name)

		def ctxmgr(name, trace):
		return lambda: contextmanager(name, trace)

		class test_ctxmanager(unittest.TestCase):
		def test_basics(self):
		trace = []
		addtrace = trace.append
Yuya Nishihara test-ctxmanager: stop direct symbol import of mercurial.util	r28801	with util.ctxmanager(ctxmgr('a', addtrace), ctxmgr('b', addtrace)) as c:
Bryan O'Sullivan util: rename ctxmanager's __call__ method to enter	r27785	a, b = c.enter()
Bryan O'Sullivan util: introduce ctxmanager, to avoid nested try/finally blocks...	r27703	c.atexit(addtrace, ('atexit', 'x'))
		c.atexit(addtrace, ('atexit', 'y'))
		self.assertEqual(trace, [('enter', 'a'), ('enter', 'b'),
		('atexit', 'y'), ('atexit', 'x'),
		('exit', 'b'), ('exit', 'a')])

		def test_raise_on_enter(self):
		trace = []
		addtrace = trace.append
Gregory Szorc tests: use context manager form of assertRaises...	r32279	with self.assertRaises(ctxerror):
Yuya Nishihara test-ctxmanager: stop direct symbol import of mercurial.util	r28801	with util.ctxmanager(ctxmgr('a', addtrace),
		lambda: raise_on_enter('b', addtrace)) as c:
Bryan O'Sullivan util: rename ctxmanager's __call__ method to enter	r27785	c.enter()
Bryan O'Sullivan util: introduce ctxmanager, to avoid nested try/finally blocks...	r27703	addtrace('unreachable')
		self.assertEqual(trace, [('enter', 'a'), ('raise', 'b'), ('exit', 'a')])

		def test_raise_on_exit(self):
		trace = []
		addtrace = trace.append
Gregory Szorc tests: use context manager form of assertRaises...	r32279	with self.assertRaises(ctxerror):
Yuya Nishihara test-ctxmanager: stop direct symbol import of mercurial.util	r28801	with util.ctxmanager(ctxmgr('a', addtrace),
		lambda: raise_on_exit('b', addtrace)) as c:
Bryan O'Sullivan util: rename ctxmanager's __call__ method to enter	r27785	c.enter()
Bryan O'Sullivan util: introduce ctxmanager, to avoid nested try/finally blocks...	r27703	addtrace('running')
		self.assertEqual(trace, [('enter', 'a'), ('enter', 'b'), 'running',
		('raise', 'b'), ('exit', 'a')])

		if __name__ == '__main__':
		silenttestrunner.main(__name__)