← Back to branch summary

~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to shelf_tests.py

  • Committer: Aaron Bentley
  • Date: 2006-02-01 03:40:19 UTC
  • Revision ID: aaron.bentley@utoronto.ca-20060201034019-6f9148fb037de1f1
got branch-history under test

Show diffs side-by-side

added added

removed removed

Lines of Context:
shelf_tests.py
1
1
#!/usr/bin/python
2
2
 
3
3
import bzrlib.tests
4
 
 
 
4
import os
5
5
class ShelfTests(bzrlib.tests.TestCaseInTempDir):
6
6
    ORIGINAL = '\n\nhello test world\n\n'
7
7
    MODIFIED = '\n\ngoodbye test world\n\n'
25
25
 
26
26
"""
27
27
    def test_shelf(self):
28
 
        self.__test_loop(1)
29
 
 
30
 
    def test_shelf_multi(self):
31
 
        self.__test_loop(10)
32
 
 
33
 
    def __test_loop(self, count):
34
28
        from bzrlib.branch import Branch
35
29
        b = Branch.initialize('.')
36
30
 
39
33
        b.working_tree().add('test_file')
40
34
        b.working_tree().commit(message='add test_file')
41
35
 
42
 
        while count > 0:
43
 
            count -= 1
44
 
 
45
 
            # Modify the test file
46
 
            file('test_file', 'w').write(self.MODIFIED)
47
 
 
48
 
            # Check the diff is right
49
 
            self.assertEqual(self.capture('diff', retcode=1),
50
 
                self.DIFF_HEADER + self.DIFF_1 + '\n')
51
 
 
52
 
            # Shelve the changes
53
 
            self.run_bzr('shelve', retcode=True)
54
 
 
55
 
            # Make sure there is no diff anymore
56
 
            self.assertEqual(self.capture('diff', retcode=0), '')
57
 
 
58
 
            # Make sure the file is actually back the way it was
59
 
            self.assertEqual(file('test_file').read(), self.ORIGINAL)
60
 
 
61
 
            # Check the shelf is right
62
 
            shelf = b._transport.get('.bzr/x-shelf/default/00').read()
63
 
            self.assertEqual(shelf, self.DIFF_1)
64
 
 
65
 
            # Unshelve
66
 
            self.run_bzr('unshelve', retcode=True)
67
 
 
68
 
            # Check the diff is right again
69
 
            self.assertEqual(self.capture('diff', retcode=1),
70
 
                self.DIFF_HEADER + self.DIFF_1 + '\n')
71
 
 
72
 
            # Make sure the file is back the way it should be
73
 
            self.assertEqual(file('test_file').read(), self.MODIFIED)
 
36
        # Modify the test file
 
37
        file('test_file', 'w').write(self.MODIFIED)
 
38
 
 
39
        # Check the diff is right
 
40
        self.assertEqual(self.capture('diff', retcode=1),
 
41
            self.DIFF_HEADER + self.DIFF_1 + '\n')
 
42
 
 
43
        # Shelve the changes
 
44
        self.run_bzr('shelve', '--all', retcode=1)
 
45
 
 
46
        # Make sure there is no diff anymore
 
47
        self.assertEqual(self.capture('diff'), '')
 
48
 
 
49
        # Make sure the file is actually back the way it was
 
50
        self.assertEqual(file('test_file').read(), self.ORIGINAL)
 
51
 
 
52
        # Check the shelf is right
 
53
        shelf = file('.bzr-shelf').read()
 
54
        self.assertEqual(shelf, self.DIFF_1)
 
55
 
 
56
        # Unshelve
 
57
        self.run_bzr('unshelve', retcode=1)
 
58
 
 
59
        # Check the diff is right again
 
60
        self.assertEqual(self.capture('diff', retcode=1),
 
61
            self.DIFF_HEADER + self.DIFF_1 + '\n')
 
62
 
 
63
        # Make sure the file is back the way it should be
 
64
        self.assertEqual(file('test_file').read(), self.MODIFIED)
74
65
 
75
66
    def test_shelf_nothing_to_shelve(self):
76
67
        import os.path
83
74
        b.working_tree().commit(message='add test_file')
84
75
 
85
76
        # Shelve the changes
86
 
        self.run_bzr('shelve', retcode=True)
 
77
        self.run_bzr('shelve', '--all')
87
78
 
88
 
        if b._transport.has('.bzr/x-shelf/default/00'):
 
79
        if os.path.exists('.bzr-shelf'):
89
80
            self.fail("Shelf exists, but it shouldn't")
90
81
 
91
82
    def test_shelf_with_revision(self):
102
93
        b.working_tree().commit(message='update test_file')
103
94
 
104
95
        # Shelve the changes
105
 
        self.run_bzr('shelve', '-r', '1', retcode=True)
 
96
        self.run_bzr('shelve', '-r', '1', '--all', retcode=1)
106
97
 
107
98
        # Check the diff is right
108
99
        self.assertEqual(self.capture('diff', retcode=1),
112
103
        self.assertEqual(file('test_file').read(), self.ORIGINAL)
113
104
 
114
105
        # Unshelve
115
 
        self.run_bzr('unshelve', retcode=True)
 
106
        self.run_bzr('unshelve', retcode=1)
116
107
 
117
108
        # Make sure the file is back the way it should be
118
109
        self.assertEqual(file('test_file').read(), self.MODIFIED)
125
116
        (stdout, stderr) = self.run_bzr_captured(cmd.split(), retcode=None)
126
117
 
127
118
        self.assertEqual(stderr.split('\n')[0],
128
 
            'bzr: ERROR: bzrlib.errors.BzrCommandError: shelve only ' \
129
 
                'accepts a single revision parameter.')
 
119
            'bzr: ERROR: shelve only accepts a single revision parameter.')
 
120
 
 
121
    def test_shelf_with_whitespace(self):
 
122
        """Shows that bzr doesn't handle whitespace well"""
 
123
        self.run_bzr('init')
 
124
        file('file name', 'wb').write('hello')
 
125
        self.run_bzr('add')
 
126
        self.run_bzr('commit', '-m', 'added')
 
127
        file('file name', 'wb').write('goodbye')
 
128
        self.run_bzr('shelve', '--all', 'file name', retcode=1)
 
129
 
 
130
    def test_whitespace_branches(self):
 
131
        os.mkdir('name with space')
 
132
        os.chdir('name with space')
 
133
        self.run_bzr('init')
 
134
        file('filename', 'wb').write('hello')
 
135
        self.run_bzr('add')
 
136
        self.run_bzr('commit', '-m', 'added')
 
137
        file('filename', 'wb').write('goodbye')
 
138
        self.run_bzr('shelve', '--all', 'filename', retcode=1)
 
139
        os.chdir('..')
 
140
 
 
141
    def test_long_filename(self):
 
142
        """Regression test for diffstat with long filenames.
 
143
 
 
144
        Create a branch with two files, one of which has a long name. Commit.
 
145
        Modify both files. Shelve the file with the short name. If diffstat
 
146
        has regressed, it will generate a diffstat of the file with the long
 
147
        name, and break.
 
148
        """
 
149
        self.run_bzr('init')
 
150
        filename = 'a' * 80
 
151
        file(filename, 'wb').write('hello')
 
152
        file('foo', 'wb').write('bar')
 
153
        self.run_bzr('add')
 
154
        self.run_bzr('commit', '-m', 'added')
 
155
        file(filename, 'wb').write('goodbye')
 
156
        file('foo', 'wb').write('baz')
 
157
        self.run_bzr('shelve', '--all', 'foo', retcode=1)
 
158