/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to breezy/tests/test_atomicfile.py

  • Committer: Jelmer Vernooij
  • Date: 2019-06-15 13:39:46 UTC
  • mto: This revision was merged to the branch mainline in revision 7342.
  • Revision ID: jelmer@jelmer.uk-20190615133946-uywh9ix0lfpqw0hy
Install quilt.

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/tests/test_atomicfile.py
20
20
import stat
21
21
import sys
22
22
 
23
 
from bzrlib import (
 
23
from .. import (
24
24
    atomicfile,
25
 
    errors,
26
25
    osutils,
27
 
    symbol_versioning,
28
26
    )
29
 
from bzrlib.tests import TestCaseInTempDir, TestSkipped
 
27
from . import TestCaseInTempDir, TestSkipped
30
28
 
31
29
 
32
30
class TestAtomicFile(TestCaseInTempDir):
33
31
 
34
32
    def test_commit(self):
35
33
        f = atomicfile.AtomicFile('test')
36
 
        self.failIfExists('test')
37
 
        f.write('foo\n')
 
34
        self.assertPathDoesNotExist('test')
 
35
        f.write(b'foo\n')
38
36
        f.commit()
39
37
 
40
38
        self.assertEqual(['test'], os.listdir('.'))
41
 
        self.check_file_contents('test', 'foo\n')
42
 
        self.assertRaises(errors.AtomicFileAlreadyClosed, f.commit)
43
 
        self.assertRaises(errors.AtomicFileAlreadyClosed, f.abort)
 
39
        self.check_file_contents('test', b'foo\n')
 
40
        self.assertRaises(atomicfile.AtomicFileAlreadyClosed, f.commit)
 
41
        self.assertRaises(atomicfile.AtomicFileAlreadyClosed, f.abort)
44
42
        # close is re-entrant safe
45
43
        f.close()
46
44
 
47
45
    def test_abort(self):
48
46
        f = atomicfile.AtomicFile('test')
49
 
        f.write('foo\n')
 
47
        f.write(b'foo\n')
50
48
        f.abort()
51
49
        self.assertEqual([], os.listdir('.'))
52
50
 
53
 
        self.assertRaises(errors.AtomicFileAlreadyClosed, f.abort)
54
 
        self.assertRaises(errors.AtomicFileAlreadyClosed, f.commit)
 
51
        self.assertRaises(atomicfile.AtomicFileAlreadyClosed, f.abort)
 
52
        self.assertRaises(atomicfile.AtomicFileAlreadyClosed, f.commit)
55
53
 
56
54
        # close is re-entrant safe
57
55
        f.close()
58
56
 
59
57
    def test_close(self):
60
58
        f = atomicfile.AtomicFile('test')
61
 
        f.write('foo\n')
 
59
        f.write(b'foo\n')
62
60
        # close on an open file is an abort
63
61
        f.close()
64
62
        self.assertEqual([], os.listdir('.'))
65
63
 
66
 
        self.assertRaises(errors.AtomicFileAlreadyClosed, f.abort)
67
 
        self.assertRaises(errors.AtomicFileAlreadyClosed, f.commit)
 
64
        self.assertRaises(atomicfile.AtomicFileAlreadyClosed, f.abort)
 
65
        self.assertRaises(atomicfile.AtomicFileAlreadyClosed, f.commit)
68
66
 
69
67
        # close is re-entrant safe
70
68
        f.close()
71
69
 
72
70
    def test_text_mode(self):
73
71
        f = atomicfile.AtomicFile('test', mode='wt')
74
 
        f.write('foo\n')
 
72
        f.write(b'foo\n')
75
73
        f.commit()
76
74
 
77
75
        contents = open('test', 'rb').read()
78
76
        if sys.platform == 'win32':
79
 
            self.assertEqual('foo\r\n', contents)
 
77
            self.assertEqual(b'foo\r\n', contents)
80
78
        else:
81
 
            self.assertEqual('foo\n', contents)
 
79
            self.assertEqual(b'foo\n', contents)
82
80
 
83
81
    def can_sys_preserve_mode(self):
84
82
        # PLATFORM DEFICIENCY/ TestSkipped
88
86
        if not self.can_sys_preserve_mode():
89
87
            raise TestSkipped("This test cannot be run on your platform")
90
88
        f = atomicfile.AtomicFile('test', mode='wb', new_mode=mode)
91
 
        f.write('foo\n')
 
89
        f.write(b'foo\n')
92
90
        f.commit()
93
91
        st = os.lstat('test')
94
92
        self.assertEqualMode(mode, stat.S_IMODE(st.st_mode))
95
93
 
96
94
    def test_mode_0666(self):
97
 
        self._test_mode(0666)
 
95
        self._test_mode(0o666)
98
96
 
99
97
    def test_mode_0664(self):
100
 
        self._test_mode(0664)
101
 
 
102
 
    def test_mode_0660(self):
103
 
        self._test_mode(0660)
104
 
 
105
 
    def test_mode_0660(self):
106
 
        self._test_mode(0660)
 
98
        self._test_mode(0o664)
 
99
 
 
100
    def test_mode_0660(self):
 
101
        self._test_mode(0o660)
 
102
 
 
103
    def test_mode_0660(self):
 
104
        self._test_mode(0o660)
107
105
 
108
106
    def test_mode_0640(self):
109
 
        self._test_mode(0640)
 
107
        self._test_mode(0o640)
110
108
 
111
109
    def test_mode_0600(self):
112
 
        self._test_mode(0600)
 
110
        self._test_mode(0o600)
113
111
 
114
112
    def test_mode_0400(self):
115
 
        self._test_mode(0400)
 
113
        self._test_mode(0o400)
116
114
        # Make it read-write again so cleanup doesn't complain
117
 
        os.chmod('test', 0600)
 
115
        os.chmod('test', 0o600)
118
116
 
119
117
    def test_no_mode(self):
120
118
        # The default file permissions should be based on umask
121
119
        umask = osutils.get_umask()
122
120
        f = atomicfile.AtomicFile('test', mode='wb')
123
 
        f.write('foo\n')
 
121
        f.write(b'foo\n')
124
122
        f.commit()
125
123
        st = os.lstat('test')
126
 
        self.assertEqualMode(0666 & ~umask, stat.S_IMODE(st.st_mode))
127
 
 
128
 
    def test_closed(self):
129
 
        local_warnings = []
130
 
        def capture_warnings(msg, cls, stacklevel=None):
131
 
            self.assertEqual(cls, DeprecationWarning)
132
 
            local_warnings.append(msg)
133
 
 
134
 
        method = symbol_versioning.warn
135
 
        try:
136
 
            symbol_versioning.set_warning_method(capture_warnings)
137
 
            f = atomicfile.AtomicFile('test', mode='wb')
138
 
            self.assertEqual(False, f.closed)
139
 
            f.abort()
140
 
            self.assertEqual(True, f.closed)
141
 
 
142
 
            f = atomicfile.AtomicFile('test', mode='wb')
143
 
            f.close()
144
 
            self.assertEqual(True, f.closed)
145
 
 
146
 
            f = atomicfile.AtomicFile('test', mode='wb')
147
 
            f.commit()
148
 
            self.assertEqual(True, f.closed)
149
 
        finally:
150
 
            symbol_versioning.set_warning_method(method)
151
 
 
152
 
        txt = 'AtomicFile.closed deprecated in bzr 0.10'
153
 
        self.assertEqual([txt]*4, local_warnings)
 
124
        self.assertEqualMode(0o666 & ~umask, stat.S_IMODE(st.st_mode))
 
125
 
 
126
    def test_context_manager_commit(self):
 
127
        with atomicfile.AtomicFile('test') as f:
 
128
            self.assertPathDoesNotExist('test')
 
129
            f.write(b'foo\n')
 
130
 
 
131
        self.assertEqual(['test'], os.listdir('.'))
 
132
        self.check_file_contents('test', b'foo\n')
 
133
 
 
134
    def test_context_manager_abort(self):
 
135
        def abort():
 
136
            with atomicfile.AtomicFile('test') as f:
 
137
                f.write(b'foo\n')
 
138
                raise AssertionError
 
139
        self.assertRaises(AssertionError, abort)
 
140
        self.assertEqual([], os.listdir('.'))