← Back to branch summary

~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to tests/shelf_tests.py

  • Committer: Aaron Bentley
  • Date: 2009-06-16 05:05:04 UTC
  • Revision ID: aaron@aaronbentley.com-20090616050504-wexqxdlrkpljs0cv
Remove obsolete aliases for shelve1 and unshelve1

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/shelf_tests.py
7
7
    UnshelveHunkSelector,
8
8
    )
9
9
from bzrlib.plugins.bzrtools.errors import NoColor
 
10
from bzrlib.plugins.bzrtools.command_classes import cmd_shelf1
10
11
 
11
12
 
12
13
class ShelfTests(bzrlib.tests.TestCaseWithTransport):
42
43
            new_date = _patch_header_date(self.tree,
43
44
                self.tree.inventory.path2id(filename), filename)
44
45
        finally:
 
46
            self.tree.unlock()
45
47
            old_tree.unlock()
46
 
        keys = { 'filename' : filename , 'old_date': old_date, 
 
48
        keys = { 'filename' : filename , 'old_date': old_date,
47
49
                 'new_date': new_date}
48
50
        hdr  = self.DIFF_HEADER % keys
49
51
        diff = diff % keys
50
 
        self.assertEqual(self.capture('diff', retcode=1), hdr + diff + '\n')
 
52
        self.assertEqual(self.run_bzr('diff', retcode=1)[0], hdr + diff + '\n')
51
53
 
52
54
    def _check_shelf(self, idx, diff=DIFF_1, filename='test_file',
53
55
                     new_date=None):
75
77
    def __test_loop(self, count):
76
78
        self.tree = self.make_branch_and_tree('.')
77
79
        self.__create_and_add_test_file()
78
 
 
79
 
        while count > 0:
80
 
            count -= 1
81
 
 
 
80
        for counter in range(count):
82
81
            # Modify the test file
83
 
            # write in binary mode because on win32 line-endings should be LF
 
82
            # write in binary mode because on win32 line-endings should be
 
83
            # LF
84
84
            f = file('test_file', 'wb')
85
85
            f.write(self.MODIFIED)
86
86
            f.close()
87
87
 
88
88
            self._check_diff()
89
 
            
90
 
            new_date = _patch_header_date(self.tree, 
91
 
                self.tree.inventory.path2id('test_file'), 'test_file')
 
89
 
 
90
            self.tree.lock_write()
 
91
            try:
 
92
                new_date = _patch_header_date(self.tree,
 
93
                    self.tree.inventory.path2id('test_file'), 'test_file')
 
94
            finally:
 
95
                self.tree.unlock()
92
96
 
93
97
            # Shelve the changes
94
 
            self.run_bzr('shelve', '--all', retcode=0)
 
98
            self.run_bzr('shelve1 --all', retcode=0)
95
99
 
96
100
            # Make sure there is no diff anymore
97
 
            self.assertEqual(self.capture('diff', retcode=0), '')
 
101
            self.assertEqual(self.run_bzr('diff', retcode=0)[0], '')
98
102
 
99
103
            # Make sure the file is actually back the way it was
100
104
            self.assertEqual(file('test_file').read(), self.ORIGINAL)
102
106
            self._check_shelf('00', new_date=new_date)
103
107
 
104
108
            # Unshelve
105
 
            self.run_bzr('unshelve', '--all', retcode=0)
 
109
            self.run_bzr('unshelve1 --all', retcode=0)
106
110
 
107
111
            self._check_diff()
108
112
 
118
122
        self.__create_and_add_test_file()
119
123
 
120
124
        # Shelve the changes
121
 
        self.run_bzr('shelve', '--all', retcode=3)
 
125
        self.run_bzr('shelve1 --all', retcode=3)
122
126
 
123
127
        if os.path.exists(os.path.join(self.tree.branch.base,
124
128
                '.shelf/shelves/default/00')):
142
146
        self.tree.commit(message='update test_file')
143
147
 
144
148
        # Shelve the changes
145
 
        self.run_bzr('shelve', '--all', '-r', '1', retcode=0)
 
149
        self.run_bzr('shelve1 --all -r 1', retcode=0)
146
150
 
147
151
        self._check_diff(self.DIFF_2)
148
152
 
150
154
        self.assertEqual(file('test_file').read(), self.ORIGINAL)
151
155
 
152
156
        # Unshelve
153
 
        self.run_bzr('unshelve', '--all', retcode=0)
 
157
        self.run_bzr('unshelve1 --all', retcode=0)
154
158
 
155
159
        # Make sure the file is back the way it should be
156
160
        self.assertEqual(file('test_file').read(), self.MODIFIED)
158
162
    def test_shelf_with_two_revisions(self):
159
163
        self.tree = self.make_branch_and_tree('.')
160
164
 
161
 
        cmd = 'shelve --all -r 1..2'
162
 
        (stdout, stderr) = self.run_bzr_captured(cmd.split(), retcode=None)
 
165
        stdout, stderr = self.run_bzr('shelve1 --all -r 1..2', retcode=None)
163
166
 
164
167
        self.assertEqual(stderr.split('\n')[0],
165
168
            'bzr: ERROR: shelve only accepts a single revision parameter.')
174
177
        self.build_tree_contents([('test_file', 'patch %s\n' % patch)])
175
178
 
176
179
        # Shelve the changes
177
 
        self.run_bzr('shelve', '--all', retcode=0)
 
180
        self.run_bzr('shelve1 --all', retcode=0)
178
181
 
179
182
        # Make sure there is no diff anymore
180
 
        self.assertEqual(self.capture('diff', retcode=0), '')
 
183
        self.assertEqual(self.run_bzr('diff', retcode=0)[0], '')
181
184
 
182
185
        # Check the shelf is right
183
186
        shelf = open(os.path.join(self.tree.basedir,
185
188
        self.assertTrue('patch %s' % patch in shelf)
186
189
 
187
190
        # Check the shown output is right
188
 
        shown = self.capture('shelf show %s' % patch, retcode=0)
 
191
        shown = self.run_bzr('shelf1 show %s' % patch, retcode=0)[0]
189
192
        self.assertEqual(shown, shelf)
190
193
 
191
194
    def test_shelf_show_multi(self):
201
204
        self.assertTrue('patch 00' in shelf)
202
205
 
203
206
        # Check the shown output is right
204
 
        shown = self.capture('shelf show 00', retcode=0)
 
207
        shown = self.run_bzr('shelf1 show 00', retcode=0)[0]
205
208
        self.assertEqual(shown, shelf)
206
209
 
207
210
    def test_shelf_show_unspecified(self):
217
220
        self.assertTrue('patch 02' in shelf)
218
221
 
219
222
        # Check the shown output is right
220
 
        shown = self.capture('shelf show', retcode=0)
 
223
        shown = self.run_bzr('shelf1 show', retcode=0)[0]
221
224
        self.assertEqual(shown, shelf)
222
225
 
223
226
    def test_shelf_show_with_no_patch(self):
224
227
        self.tree = self.make_branch_and_tree('.')
225
 
        stderr = self.run_bzr_captured(['shelf', 'show', '00'], retcode=None)[1]
 
228
        stderr = self.run_bzr('shelf1 show 00', retcode=None)[1]
226
229
        self.assertTrue("Patch '00' doesn't exist on shelf default!" in stderr)
227
230
 
228
231
    def test_shelf_unshelve_failure(self):
234
237
        file('test_file', 'w').write(self.MODIFIED)
235
238
 
236
239
        # Shelve the changes
237
 
        self.run_bzr('shelve', '--all', retcode=0)
 
240
        self.run_bzr('shelve1 --all', retcode=0)
238
241
 
239
242
        # Write an unapplyable patch into the shelf
240
243
        shelf = open(os.path.join(self.tree.basedir,
243
246
        shelf.close()
244
247
 
245
248
        # Unshelve, should fail
246
 
        self.run_bzr('unshelve', '--all', retcode=3)
 
249
        self.run_bzr('unshelve1 --all', retcode=3)
247
250
 
248
251
        # Make sure the patch is still there, eventhough it's broken
249
252
        shelf = open(os.path.join(self.tree.basedir,
251
254
        self.assertEqual(shelf, self.DIFF_2)
252
255
 
253
256
        # Working tree should be unchanged
254
 
        diff = self.capture('diff', retcode=0)
 
257
        diff = self.run_bzr('diff', retcode=0)[0]
255
258
        self.assertEqual(diff, '')
256
259
 
257
260
    def test_shelf_unshelve_failure_two_hunks(self):
265
268
        file('test_file2', 'w').write(self.MODIFIED)
266
269
 
267
270
        # Shelve the changes
268
 
        self.run_bzr('shelve', '--all', retcode=0)
 
271
        self.run_bzr('shelve1 --all', retcode=0)
269
272
 
270
273
        # Put the changes to test_file back, the shelved patch won't apply now
271
274
        file('test_file', 'w').write(self.MODIFIED)
272
275
        self.tree.commit(message='screw up test_file')
273
276
 
274
277
        # Unshelve, should fail
275
 
        self.run_bzr('unshelve', '--all', retcode=3)
 
278
        self.run_bzr('unshelve1 --all', retcode=3)
276
279
 
277
280
        # Working tree should be unchanged
278
 
        diff = self.capture('diff', retcode=0)
 
281
        diff = self.run_bzr('diff', retcode=0)[0]
279
282
        self.assertEqual(diff, '')
280
283
 
281
284
        # Force should succeed and modify test_file2, but leave shelf
282
 
        self.run_bzr('unshelve', '--force', '--all', retcode=0)
 
285
        self.run_bzr('unshelve1 --force --all', retcode=0)
283
286
        self.assertEqual(open('test_file2').read(), self.MODIFIED)
284
287
        self.assertTrue(os.path.exists('.shelf/shelves/default/00'))
285
288
 
294
297
        file('test_file2', 'w').write(self.MODIFIED)
295
298
 
296
299
        # Shelve the changes
297
 
        self.run_bzr('shelve', '--all', 'test_file', retcode=0)
298
 
        self.run_bzr('shelve', '--all', 'test_file2', retcode=0)
 
300
        self.run_bzr('shelve1 --all test_file', retcode=0)
 
301
        self.run_bzr('shelve1 --all test_file2', retcode=0)
299
302
 
300
303
        # Unshelve
301
 
        self.run_bzr('unshelve', '--all', retcode=0)
 
304
        self.run_bzr('unshelve1 --all', retcode=0)
302
305
 
303
306
        # We should now have 00 and 01~
304
307
        self.assertTrue(os.path.exists('.shelf/shelves/default/00'))
305
308
        self.assertTrue(os.path.exists('.shelf/shelves/default/01~'))
306
309
 
307
310
        # Check ls works
308
 
        lines = self.capture('shelf ls', retcode=0).split('\n')
 
311
        lines = self.run_bzr('shelf1 ls', retcode=0)[0].split('\n')
309
312
        for line in lines:
310
313
            self.assertFalse(line.startswith(' 01'))
311
314
 
312
315
        # Unshelve, if unshelve is confused by the backup it will fail
313
 
        self.run_bzr('unshelve', '--all', retcode=0)
 
316
        self.run_bzr('unshelve1 --all', retcode=0)
314
317
 
315
318
    def test_shelf_delete(self):
316
319
        self.tree = self.make_branch_and_tree('.')
333
336
            self.tree.unlock()
334
337
 
335
338
        # Shelve the changes
336
 
        self.run_bzr('shelve', '--all', 'test_file', retcode=0)
337
 
        self.run_bzr('shelve', '--all', 'test_file2', retcode=0)
 
339
        self.run_bzr('shelve1 --all test_file', retcode=0)
 
340
        self.run_bzr('shelve1 --all test_file2', retcode=0)
338
341
 
339
342
        self._check_shelf('00', new_date=new_date)
340
343
 
341
344
        # Delete 00
342
 
        self.run_bzr('shelf', 'delete', '00', retcode=0)
 
345
        self.run_bzr('shelf1 delete 00', retcode=0)
343
346
 
344
347
        # We should now have 01 but not 00, but we should have 00~
345
348
        self.assertFalse(os.path.exists('.shelf/shelves/default/00'))
350
353
        self._check_shelf('00~', new_date=new_date)
351
354
 
352
355
        # Check ls works
353
 
        lines = self.capture('shelf ls', retcode=0).split('\n')
 
356
        lines = self.run_bzr('shelf1 ls', retcode=0)[0].split('\n')
354
357
        for line in lines:
355
358
            self.assertFalse(line.startswith(' 00'))
356
359
 
357
360
        # Unshelve should unshelve 01
358
 
        self.run_bzr('unshelve', '--all', retcode=0)
 
361
        self.run_bzr('unshelve1 --all', retcode=0)
359
362
        self.assertEqual(file('test_file2').read(), self.MODIFIED)
360
363
 
361
364
    def test_shelf_gaps(self):
362
365
        self.tree = self.make_branch_and_tree('.')
363
366
        self.__create_and_add_test_file()
364
367
        file('test_file', 'w').write(self.MODIFIED)
365
 
        self.run_bzr('shelve', '--all', 'test_file', retcode=0)
 
368
        self.run_bzr('shelve1 --all test_file', retcode=0)
366
369
        file('test_file', 'w').write(self.MODIFIED)
367
 
        self.run_bzr('shelve', '--all', 'test_file', retcode=0)
 
370
        self.run_bzr('shelve1 --all test_file', retcode=0)
368
371
 
369
372
        # Now delete 00, leaving 01, next shelve should go into 02
370
 
        self.run_bzr('shelf', 'delete', '0', retcode=0)
 
373
        self.run_bzr('shelf1 delete 0', retcode=0)
371
374
        self.assertFalse(os.path.exists('.shelf/shelves/default/00'))
372
375
        self.assertFalse(os.path.exists('.shelf/shelves/default/02'))
373
376
        file('test_file', 'w').write(self.MODIFIED)
374
 
        self.run_bzr('shelve', '--all', 'test_file', retcode=0)
 
377
        self.run_bzr('shelve1 --all test_file', retcode=0)
375
378
        self.assertFalse(os.path.exists('.shelf/shelves/default/00'))
376
379
        self.assertTrue(os.path.exists('.shelf/shelves/default/02'))
377
380
 
382
385
 
383
386
        # Modify then shelve, so we're not upgrading to 00, just for kicks
384
387
        file('test_file', 'w').write(self.MODIFIED)
385
 
        self.run_bzr('shelve', '--all', 'test_file', retcode=0)
 
388
        self.run_bzr('shelve1 --all test_file', retcode=0)
386
389
 
387
390
        open('.bzr-shelf', 'w').write('First old shelf')
388
391
        open('.bzr-shelf-1', 'w').write('Second old shelf')
390
393
 
391
394
        # shelve and unshelve should bitch and do nothing
392
395
        file('test_file', 'w').write('blah blah blah')
393
 
        self.run_bzr('shelve', '--all', retcode=3)
 
396
        self.run_bzr('shelve1 --all', retcode=3)
394
397
        self.assertFalse(os.path.exists('.shelf/shelves/default/01'))
395
398
        self.assertEqual(file('test_file').read(), 'blah blah blah')
396
 
        self.run_bzr('unshelve', '--all', retcode=3)
 
399
        self.run_bzr('unshelve1 --all', retcode=3)
397
400
        self.assertTrue(os.path.exists('.shelf/shelves/default/00'))
398
401
 
399
402
        # Upgrade, make sure it worked
400
 
        self.run_bzr('shelf', 'upgrade', retcode=0)
 
403
        self.run_bzr('shelf1 upgrade', retcode=0)
401
404
        self.assertEqual(open('.shelf/shelves/default/01').read(),
402
405
                'First old shelf')
403
406
        self.assertEqual(open('.shelf/shelves/default/02').read(),
414
417
        self.assertFalse(os.path.exists('.bzr-shelf-3'))
415
418
 
416
419
        # Shelve should work now
417
 
        self.run_bzr('shelve', '--all', retcode=0)
 
420
        self.run_bzr('shelve1 --all', retcode=0)
418
421
 
419
422
    def test_shelf_p1_patch(self):
420
423
        self.tree = self.make_branch_and_tree('.')
422
425
        self.__create_and_add_test_file()
423
426
 
424
427
        # Run a benign shelf command to setup .shelf for us
425
 
        self.run_bzr('shelf', 'ls', retcode=0)
 
428
        self.run_bzr('shelf1 ls', retcode=0)
426
429
 
427
430
        old_tree = self.tree.basis_tree()
428
431
        old_tree.lock_read()
444
447
        open('.shelf/shelves/default/00', 'w').write(diff)
445
448
 
446
449
        # This should work
447
 
        self.run_bzr('unshelve', '--all', retcode=0)
 
450
        self.run_bzr('unshelve1 --all', retcode=0)
448
451
 
449
452
        self._check_diff()
450
453
 
465
468
        f.close()
466
469
 
467
470
        # Shelve the changes
468
 
        self.run_bzr('shelve', '--all', retcode=0)
 
471
        self.run_bzr('shelve1 --all', retcode=0)
469
472
 
470
473
        # Working tree should be unchanged
471
 
        diff = self.capture('diff', retcode=0)
 
474
        diff = self.run_bzr('diff', retcode=0)[0]
472
475
        self.assertEqual(diff, '')
473
476
 
474
477
        # Unshelve, should succeed
475
 
        self.run_bzr('unshelve', '--all', retcode=0)
 
478
        self.run_bzr('unshelve1 --all', retcode=0)
476
479
 
477
480
        self._check_diff(filename='subdir/test_file')
478
481
 
479
482
        # Make sure relative filenames work ok
480
 
        self.run_bzr('shelve', 'test_file', '--all', retcode=0)
 
483
        self.run_bzr('shelve1 test_file --all', retcode=0)
481
484
 
482
485
    def test_shelf_shelf_bogus_subcommand(self):
483
486
        self.tree = self.make_branch_and_tree('.')
484
 
        self.run_bzr('shelf', 'foo', retcode=3) # <- retcode == 3
 
487
        self.run_bzr('shelf1 foo', retcode=3) # <- retcode == 3
485
488
 
486
489
    def test_shelf_OOO_unshelve(self):
487
490
        self.tree = self.make_branch_and_tree('.')
495
498
 
496
499
        # Shelve the changes
497
500
        for i in range(1, 5):
498
 
            self.run_bzr('shelve', '--all', 'test_file%d' % i, retcode=0)
 
501
            self.run_bzr(['shelve1', '--all', 'test_file%d' % i], retcode=0)
499
502
 
500
503
        # Check shelving worked
501
504
        for i in range(1, 5):
506
509
            self.assertTrue(os.path.exists('.shelf/shelves/default/0%d' % i))
507
510
 
508
511
        # Unshelve 00
509
 
        self.run_bzr('unshelve', '--all', '00', retcode=0)
 
512
        self.run_bzr('unshelve1 --all 00', retcode=0)
510
513
        self.assertEqual(file('test_file1').read(), self.MODIFIED)
511
514
 
512
515
        # Check ls works
513
 
        lines = self.capture('shelf ls', retcode=0).split('\n')
 
516
        lines = self.run_bzr('shelf1 ls', retcode=0)[0].split('\n')
514
517
        for line in lines:
515
518
            self.assertFalse(line.startswith(' 00'))
516
519
 
517
520
        # Check we can reshelve once we've unshelved out of order, should be 04
518
521
        self.assertFalse(os.path.exists('.shelf/shelves/default/04'))
519
 
        self.run_bzr('shelve', '--all')
 
522
        self.run_bzr('shelve1 --all')
520
523
        self.assertTrue(os.path.exists('.shelf/shelves/default/04'))
521
524
 
522
525
        # Check ls works
523
 
        text = self.capture('shelf ls', retcode=0)
 
526
        text = self.run_bzr('shelf1 ls', retcode=0)[0]
524
527
        for line in text.split('\n'):
525
528
            self.assertFalse(line.startswith(' 00'))
526
529
 
527
530
        # We now have 01,02,03,04
528
531
        # Unshelve 02
529
 
        self.run_bzr('unshelve', '--all', '02', retcode=0)
 
532
        self.run_bzr('unshelve1 --all 02', retcode=0)
530
533
        self.assertEqual(file('test_file3').read(), self.MODIFIED)
531
534
 
532
535
        # Unshelve the default, this is the reshelved 00, hence modifies file 1
533
 
        self.run_bzr('unshelve', '--all', retcode=0)
 
536
        self.run_bzr('unshelve1 --all', retcode=0)
534
537
        self.assertEqual(file('test_file1').read(), self.MODIFIED)
535
538
 
536
539
    def test_shelf_switch_basic(self):
539
542
 
540
543
        # This should go to "default"
541
544
        file('test_file', 'w').write(self.MODIFIED)
542
 
        self.run_bzr('shelve', '--all', 'test_file', retcode=0)
 
545
        self.run_bzr('shelve1 --all test_file', retcode=0)
543
546
 
544
547
        # Switch to "other"
545
 
        self.run_bzr('shelf', 'switch', 'other', retcode=0)
 
548
        self.run_bzr('shelf1 switch other', retcode=0)
546
549
        file('test_file', 'w').write(self.MODIFIED)
547
 
        self.run_bzr('shelve', '--all', 'test_file', retcode=0)
 
550
        self.run_bzr('shelve1 --all test_file', retcode=0)
548
551
 
549
552
        # Check it worked
550
553
        self.assertTrue(os.path.exists('.shelf/shelves/default/00'))
552
555
        self.assertTrue(os.path.exists('.shelf/shelves/other/00'))
553
556
 
554
557
        # Switch back
555
 
        self.run_bzr('shelf', 'switch', 'default', retcode=0)
 
558
        self.run_bzr('shelf1 switch default', retcode=0)
556
559
        file('test_file', 'w').write(self.MODIFIED)
557
 
        self.run_bzr('shelve', '--all', 'test_file', retcode=0)
 
560
        self.run_bzr('shelve1 --all test_file', retcode=0)
558
561
 
559
562
        # Check that worked
560
563
        self.assertTrue(os.path.exists('.shelf/shelves/default/01'))
561
564
        self.assertFalse(os.path.exists('.shelf/shelves/other/01'))
562
565
 
 
566
    def test_shelf_binary(self):
 
567
        self.tree = self.make_branch_and_tree('.')
 
568
        self.build_tree_contents([('file', '\x00')])
 
569
        self.tree.add('file')
 
570
        self.run_bzr_error(['Changes involve binary files.'], 'shelve1 --all')
 
571
 
563
572
    def test_shelf_bad_patch_arg(self):
564
573
        self.tree = self.make_branch_and_tree('.')
565
574
 
566
575
        # Check the bad arg handling
567
 
        stdout, error = self.run_bzr_captured(['unshelve', '01'], retcode=3)
 
576
        stdout, error = self.run_bzr('unshelve1 01', retcode=3)
568
577
        self.assertTrue("Patch '01' doesn't exist on shelf" in error)
569
578
 
570
 
        stdout, error = self.run_bzr_captured(['unshelve', 'foo'], retcode=3)
 
579
        stdout, error = self.run_bzr('unshelve1 foo', retcode=3)
571
580
        self.assertTrue("Invalid patch name 'foo'" in error)
572
581
 
573
582
        # Hex and is cracky, so it shouldn't work
574
 
        stdout, error = self.run_bzr_captured(['unshelve', '0x00'], retcode=3)
 
583
        stdout, error = self.run_bzr(['unshelve1', '0x00'], retcode=3)
575
584
        self.assertTrue("Invalid patch name '0x00'" in error)
576
585
 
577
586
    def test_color_hunk_selector(self):
609
618
        f = file('test_file', 'wb')
610
619
        f.write(self.MODIFIED)
611
620
        f.close()
612
 
        stdout, error = self.run_bzr_captured(['shelve', '--all', 
613
 
                                               '--no-color'])
614
 
        stdout, error = self.run_bzr_captured(['unshelve', '--all', 
615
 
                                               '--no-color'])
 
621
        stdout, error = self.run_bzr('shelve1 --all --no-color')
 
622
        stdout, error = self.run_bzr('unshelve1 --all --no-color')
 
623
 
 
624
    def test_shelf_help(self):
 
625
        self.assertContainsRe(cmd_shelf1().help(),
 
626
                              'list\n.*List the patches on the current shelf')
 
627
 
 
628
    def test_show_empty_shelf(self):
 
629
        self.tree = self.make_branch_and_tree('.')
 
630
        self.run_bzr_error(('No patches on shelf.',), 'shelf1 show')