← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_atomicfile.py

  • Committer: John Arbash Meinel
  • Date: 2007-02-08 16:28:05 UTC
  • mto: This revision was merged to the branch mainline in revision 2278.
  • Revision ID: john@arbash-meinel.com-20070208162805-dcqiqrwjh9a5lo7n
``GPGStrategy.sign()`` will now raise ``BzrBadParameterUnicode`` if
you pass a Unicode string rather than an 8-bit string. It doesn't 
make sense to sign a Unicode string, and it turns out that some 
versions of python will write out the raw Unicode bytes rather than
encoding automatically. So fail and make callers do the right thing.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_atomicfile.py
 
1
# Copyright (C) 2006 Canonical Ltd
 
2
#
 
3
# This program is free software; you can redistribute it and/or modify
 
4
# it under the terms of the GNU General Public License as published by
 
5
# the Free Software Foundation; either version 2 of the License, or
 
6
# (at your option) any later version.
 
7
#
 
8
# This program is distributed in the hope that it will be useful,
 
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
11
# GNU General Public License for more details.
 
12
#
 
13
# You should have received a copy of the GNU General Public License
 
14
# along with this program; if not, write to the Free Software
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
"""Basic tests for AtomicFile"""
 
18
 
 
19
import os
 
20
import stat
 
21
import sys
 
22
 
 
23
from bzrlib import (
 
24
    atomicfile,
 
25
    errors,
 
26
    osutils,
 
27
    symbol_versioning,
 
28
    )
 
29
from bzrlib.tests import TestCaseInTempDir, TestSkipped
 
30
 
 
31
 
 
32
class TestAtomicFile(TestCaseInTempDir):
 
33
 
 
34
    def test_commit(self):
 
35
        f = atomicfile.AtomicFile('test')
 
36
        self.failIfExists('test')
 
37
        f.write('foo\n')
 
38
        f.commit()
 
39
 
 
40
        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)
 
44
        # close is re-entrant safe
 
45
        f.close()
 
46
 
 
47
    def test_abort(self):
 
48
        f = atomicfile.AtomicFile('test')
 
49
        f.write('foo\n')
 
50
        f.abort()
 
51
        self.assertEqual([], os.listdir('.'))
 
52
 
 
53
        self.assertRaises(errors.AtomicFileAlreadyClosed, f.abort)
 
54
        self.assertRaises(errors.AtomicFileAlreadyClosed, f.commit)
 
55
 
 
56
        # close is re-entrant safe
 
57
        f.close()
 
58
 
 
59
    def test_close(self):
 
60
        f = atomicfile.AtomicFile('test')
 
61
        f.write('foo\n')
 
62
        # close on an open file is an abort
 
63
        f.close()
 
64
        self.assertEqual([], os.listdir('.'))
 
65
 
 
66
        self.assertRaises(errors.AtomicFileAlreadyClosed, f.abort)
 
67
        self.assertRaises(errors.AtomicFileAlreadyClosed, f.commit)
 
68
 
 
69
        # close is re-entrant safe
 
70
        f.close()
 
71
        
 
72
    def test_text_mode(self):
 
73
        f = atomicfile.AtomicFile('test', mode='wt')
 
74
        f.write('foo\n')
 
75
        f.commit()
 
76
 
 
77
        contents = open('test', 'rb').read()
 
78
        if sys.platform == 'win32':
 
79
            self.assertEqual('foo\r\n', contents)
 
80
        else:
 
81
            self.assertEqual('foo\n', contents)
 
82
 
 
83
    def can_sys_preserve_mode(self):
 
84
        # PLATFORM DEFICIENCY/ TestSkipped
 
85
        return sys.platform not in ('win32',)
 
86
 
 
87
    def _test_mode(self, mode):
 
88
        if not self.can_sys_preserve_mode():
 
89
            raise TestSkipped("This test cannot be run on your platform")
 
90
        f = atomicfile.AtomicFile('test', mode='wb', new_mode=mode)
 
91
        f.write('foo\n')
 
92
        f.commit()
 
93
        st = os.lstat('test')
 
94
        self.assertEqualMode(mode, stat.S_IMODE(st.st_mode))
 
95
 
 
96
    def test_mode_0666(self):
 
97
        self._test_mode(0666)
 
98
 
 
99
    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)
 
107
 
 
108
    def test_mode_0640(self):
 
109
        self._test_mode(0640)
 
110
 
 
111
    def test_mode_0600(self):
 
112
        self._test_mode(0600)
 
113
 
 
114
    def test_mode_0400(self):
 
115
        self._test_mode(0400)
 
116
        # Make it read-write again so cleanup doesn't complain
 
117
        os.chmod('test', 0600)
 
118
 
 
119
    def test_no_mode(self):
 
120
        # The default file permissions should be based on umask
 
121
        umask = osutils.get_umask()
 
122
        f = atomicfile.AtomicFile('test', mode='wb')
 
123
        f.write('foo\n')
 
124
        f.commit()
 
125
        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)