← 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: 2005-12-03 20:42:25 UTC
  • mfrom: (147.4.25 trunk)
  • mto: (147.4.27 trunk)
  • mto: This revision was merged to the branch mainline in revision 324.
  • Revision ID: aaron.bentley@utoronto.ca-20051203204225-25678bc921de4fc1
Merge from lifeless

Show diffs side-by-side

added added

removed removed

Lines of Context:
shelf_tests.py
1
1
#!/usr/bin/python
2
2
 
3
 
import bzrlib.selftest
4
 
 
5
 
class ShelfTests(bzrlib.selftest.TestCaseInTempDir):
 
3
import os
 
4
 
 
5
import bzrlib.tests
 
6
 
 
7
 
 
8
class ShelfTests(bzrlib.tests.TestCaseInTempDir):
6
9
    ORIGINAL = '\n\nhello test world\n\n'
7
10
    MODIFIED = '\n\ngoodbye test world\n\n'
8
11
    DIFF_HEADER = "=== modified file 'test_file'\n"
9
 
    DIFF_1 = """--- test_file
10
 
+++ test_file
 
12
    DIFF_1 = """--- test_file\t
 
13
+++ test_file\t
11
14
@@ -1,4 +1,4 @@
12
15
 
13
16
 
15
18
+goodbye test world
16
19
 
17
20
"""
18
 
    DIFF_2 = """--- test_file
19
 
+++ test_file
 
21
    DIFF_2 = """--- test_file\t
 
22
+++ test_file\t
20
23
@@ -1,4 +1,4 @@
21
24
 
22
25
 
26
29
"""
27
30
    def test_shelf(self):
28
31
        from bzrlib.branch import Branch
 
32
        from bzrlib.workingtree import WorkingTree
29
33
        b = Branch.initialize('.')
 
34
        tree = WorkingTree(b.base, b)
30
35
 
31
36
        # Create a test file and commit it
32
37
        file('test_file', 'w').write(self.ORIGINAL)
33
 
        b.add('test_file')
34
 
        b.commit(message='add test_file')
 
38
        tree.add('test_file')
 
39
        tree.commit(message='add test_file')
35
40
 
36
41
        # Modify the test file
37
42
        file('test_file', 'w').write(self.MODIFIED)
38
43
 
39
44
        # Check the diff is right
40
 
        self.assertEqual(self.capture('diff'),
 
45
        self.assertEqual(self.capture('diff', retcode=1),
41
46
            self.DIFF_HEADER + self.DIFF_1 + '\n')
42
47
 
43
48
        # Shelve the changes
44
 
        self.run_bzr('shelve', '--all', retcode=True)
 
49
        self.run_bzr('shelve', '--all', retcode=1)
45
50
 
46
51
        # Make sure there is no diff anymore
47
52
        self.assertEqual(self.capture('diff'), '')
54
59
        self.assertEqual(shelf, self.DIFF_1)
55
60
 
56
61
        # Unshelve
57
 
        self.run_bzr('unshelve', retcode=True)
 
62
        self.run_bzr('unshelve', retcode=1)
58
63
 
59
64
        # Check the diff is right again
60
 
        self.assertEqual(self.capture('diff'),
 
65
        self.assertEqual(self.capture('diff', retcode=1),
61
66
            self.DIFF_HEADER + self.DIFF_1 + '\n')
62
67
 
63
68
        # Make sure the file is back the way it should be
64
69
        self.assertEqual(file('test_file').read(), self.MODIFIED)
65
70
 
66
71
    def test_shelf_nothing_to_shelve(self):
67
 
        import os.path
68
72
        from bzrlib.branch import Branch
 
73
        from bzrlib.workingtree import WorkingTree
69
74
        b = Branch.initialize('.')
 
75
        tree = WorkingTree(b.base, b)
70
76
 
71
77
        # Create a test file and commit it
72
78
        file('test_file', 'w').write(self.ORIGINAL)
73
 
        b.add('test_file')
74
 
        b.commit(message='add test_file')
 
79
        tree.add('test_file')
 
80
        tree.commit(message='add test_file')
75
81
 
76
82
        # Shelve the changes
77
 
        self.run_bzr('shelve', '--all', retcode=True)
 
83
        self.run_bzr('shelve', '--all')
78
84
 
79
85
        if os.path.exists('.bzr-shelf'):
80
86
            self.fail("Shelf exists, but it shouldn't")
81
87
 
82
88
    def test_shelf_with_revision(self):
83
89
        from bzrlib.branch import Branch
 
90
        from bzrlib.workingtree import WorkingTree
84
91
        b = Branch.initialize('.')
 
92
        tree = WorkingTree(b.base, b)
85
93
 
86
94
        # Create a test file and commit it
87
95
        file('test_file', 'w').write(self.ORIGINAL)
88
 
        b.add('test_file')
89
 
        b.commit(message='add test_file')
 
96
        tree.add('test_file')
 
97
        tree.commit(message='add test_file')
90
98
 
91
99
        # Modify the test file and commit it
92
100
        file('test_file', 'w').write(self.MODIFIED)
93
 
        b.commit(message='update test_file')
 
101
        tree.commit(message='update test_file')
94
102
 
95
103
        # Shelve the changes
96
 
        self.run_bzr('shelve', '-r', '1', '--all', retcode=True)
 
104
        self.run_bzr('shelve', '-r', '1', '--all', retcode=1)
97
105
 
98
106
        # Check the diff is right
99
 
        self.assertEqual(self.capture('diff'),
 
107
        self.assertEqual(self.capture('diff', retcode=1),
100
108
            self.DIFF_HEADER + self.DIFF_2 + '\n')
101
109
 
102
110
        # Make sure the file is the way it should be
103
111
        self.assertEqual(file('test_file').read(), self.ORIGINAL)
104
112
 
105
113
        # Unshelve
106
 
        self.run_bzr('unshelve', retcode=True)
 
114
        self.run_bzr('unshelve', retcode=1)
107
115
 
108
116
        # Make sure the file is back the way it should be
109
117
        self.assertEqual(file('test_file').read(), self.MODIFIED)
116
124
        (stdout, stderr) = self.run_bzr_captured(cmd.split(), retcode=None)
117
125
 
118
126
        self.assertEqual(stderr.split('\n')[0],
119
 
            'bzr: ERROR: shelve only accepts a single revision parameter.')
 
127
            'bzr: ERROR: bzrlib.errors.BzrCommandError: shelve only accepts a single revision parameter.')
 
128
 
 
129
    def test_shelf_with_whitespace(self):
 
130
        """Shows that bzr doesn't handle whitespace well"""
 
131
        self.run_bzr('init')
 
132
        file('file name', 'wb').write('hello')
 
133
        self.run_bzr('add')
 
134
        self.run_bzr('commit', '-m', 'added')
 
135
        file('file name', 'wb').write('goodbye')
 
136
        self.run_bzr('shelve', '--all', 'file name', retcode=1)
 
137
 
 
138
    def test_whitespace_branches(self):
 
139
        os.mkdir('name with space')
 
140
        os.chdir('name with space')
 
141
        self.run_bzr('init')
 
142
        file('filename', 'wb').write('hello')
 
143
        self.run_bzr('add')
 
144
        self.run_bzr('commit', '-m', 'added')
 
145
        file('filename', 'wb').write('goodbye')
 
146
        self.run_bzr('shelve', '--all', 'filename', retcode=1)
 
147
        os.chdir('..')