← 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-10-27 04:45:06 UTC
  • Revision ID: aaron.bentley@utoronto.ca-20051027044506-45f616c07537a1da
Got the shell basics working properly

Show diffs side-by-side

added added

removed removed

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