← 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 21:13:56 UTC
  • mfrom: (256)
  • mto: (147.4.14)
  • mto: This revision was merged to the branch mainline in revision 324.
  • Revision ID: abentley@panoramicfeedback.com-20051027211356-0d4699554bcfbc7b
Merge from mainline

Show diffs side-by-side

added added

removed removed

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