← Back to branch summary

~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to shelf_tests.py

  • Committer: Robert Collins
  • Date: 2005-09-28 05:43:19 UTC
  • mto: (147.2.6) (364.1.3 bzrtools)
  • mto: This revision was merged to the branch mainline in revision 324.
  • Revision ID: robertc@robertcollins.net-20050928054319-2c2e9e3048bbc215
find_branch -> open_containing change

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.')