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

mq: use dirstateguard instead of dirstate.invalidate (qpush)...

mq: use dirstateguard instead of dirstate.invalidate (qpush) Before this patch, "mq.queue.apply()" uses "dirstate.invalidate()" as a kind of "restore .hg/dirstate to the original status" during afailure. But it just discards changes in memory, and doesn't actually restore ".hg/dirstate". Then, it can't work as expected, if "dirstate.write()" is executed while processing. This patch uses "dirstateguard" instead of "dirstate.invalidate()" to restore ".hg/dirstate" at failure even if "dirstate.write()" is executed before failure. This is a part of preparations to fix the issue that the recent (in memory) dirstate isn't visible to external processes (e.g. "precommit" hook).

Idan Kamara - - Load All Authors

File last commit:

r18666:fb9d1c28 default


                r24996:58308dde

default

Download file

             test-atomictempfile.py
        
                    42 lines
            
             | 1.3 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / tests / test-atomictempfile.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      import os

      import glob

      import unittest

      import silenttestrunner

      from mercurial.util import atomictempfile

      class testatomictempfile(unittest.TestCase):

          def test1_simple(self):

              if os.path.exists('foo'):

                  os.remove('foo')

              file = atomictempfile('foo')

              (dir, basename) = os.path.split(file._tempname)

              self.assertFalse(os.path.isfile('foo'))

              self.assertTrue(basename in glob.glob('.foo-*'))

              file.write('argh\n')

              file.close()

              self.assertTrue(os.path.isfile('foo'))

              self.assertTrue(basename not in glob.glob('.foo-*'))

          # discard() removes the temp file without making the write permanent

          def test2_discard(self):

              if os.path.exists('foo'):

                  os.remove('foo')

              file = atomictempfile('foo')

              (dir, basename) = os.path.split(file._tempname)

              file.write('yo\n')

              file.discard()

              self.assertFalse(os.path.isfile('foo'))

              self.assertTrue(basename not in os.listdir('.'))

          # if a programmer screws up and passes bad args to atomictempfile, they

          # get a plain ordinary TypeError, not infinite recursion

          def test3_oops(self):

              self.assertRaises(TypeError, atomictempfile)

      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

				import os
				import glob
				import unittest
				import silenttestrunner

				from mercurial.util import atomictempfile

				class testatomictempfile(unittest.TestCase):
				def test1_simple(self):
				if os.path.exists('foo'):
				os.remove('foo')
				file = atomictempfile('foo')
				(dir, basename) = os.path.split(file._tempname)
				self.assertFalse(os.path.isfile('foo'))
				self.assertTrue(basename in glob.glob('.foo-*'))

				file.write('argh\n')
				file.close()

				self.assertTrue(os.path.isfile('foo'))
				self.assertTrue(basename not in glob.glob('.foo-*'))

				# discard() removes the temp file without making the write permanent
				def test2_discard(self):
				if os.path.exists('foo'):
				os.remove('foo')
				file = atomictempfile('foo')
				(dir, basename) = os.path.split(file._tempname)

				file.write('yo\n')
				file.discard()

				self.assertFalse(os.path.isfile('foo'))
				self.assertTrue(basename not in os.listdir('.'))

				# if a programmer screws up and passes bad args to atomictempfile, they
				# get a plain ordinary TypeError, not infinite recursion
				def test3_oops(self):
				self.assertRaises(TypeError, atomictempfile)

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