← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/blackbox/test_diff.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-09-01 08:02:42 UTC
  • mfrom: (5390.3.3 faster-revert-593560)
  • Revision ID: pqm@pqm.ubuntu.com-20100901080242-esg62ody4frwmy66
(spiv) Avoid repeatedly calling self.target.all_file_ids() in
 InterTree.iter_changes. (Andrew Bennetts)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/blackbox/test_diff.py
1
 
# Copyright (C) 2005, 2006 Canonical Ltd
 
1
# Copyright (C) 2006-2010 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
21
21
import os
22
22
import re
23
23
 
24
 
import bzrlib
25
 
from bzrlib import workingtree
26
 
from bzrlib.branch import Branch
27
 
from bzrlib.tests import TestSkipped
28
 
from bzrlib.tests.blackbox import ExternalBase
 
24
from bzrlib import (
 
25
    tests,
 
26
    workingtree,
 
27
    )
 
28
from bzrlib.diff import (
 
29
    DiffTree,
 
30
    format_registry as diff_format_registry,
 
31
    )
 
32
from bzrlib.tests import (
 
33
    features,
 
34
    )
29
35
 
30
36
 
31
37
def subst_dates(string):
34
40
                  'YYYY-MM-DD HH:MM:SS +ZZZZ', string)
35
41
 
36
42
 
37
 
class DiffBase(ExternalBase):
 
43
class DiffBase(tests.TestCaseWithTransport):
38
44
    """Base class with common setup method"""
39
45
 
40
46
    def make_example_branch(self):
133
139
        out, err = self.run_bzr('diff -r 1..23..123', retcode=3,
134
140
            error_regexes=('one or two revision specifiers',))
135
141
 
 
142
    def test_diff_using_and_format(self):
 
143
        out, err = self.run_bzr('diff --format=default --using=mydi', retcode=3,
 
144
            error_regexes=('are mutually exclusive',))
 
145
 
136
146
    def test_diff_nonexistent_revision(self):
137
147
        out, err = self.run_bzr('diff -r 123', retcode=3,
138
148
            error_regexes=("Requested revision: '123' does not "
298
308
        output = self.run_bzr('diff -r 1.. branch1', retcode=1)
299
309
        self.assertContainsRe(output[0], '\n\\-original line\n\\+repo line\n')
300
310
 
 
311
    def test_custom_format(self):
 
312
        class BooDiffTree(DiffTree):
 
313
 
 
314
            def show_diff(self, specific_files, extra_trees=None):
 
315
                self.to_file.write("BOO!\n")
 
316
                return super(BooDiffTree, self).show_diff(specific_files,
 
317
                    extra_trees)
 
318
 
 
319
        diff_format_registry.register("boo", BooDiffTree, 
 
320
            "Scary diff format")
 
321
        self.addCleanup(diff_format_registry.remove, "boo")
 
322
        self.make_example_branch()
 
323
        self.build_tree_contents([('hello', 'hello world!\n')])
 
324
        output = self.run_bzr('diff --format=boo', retcode=1)
 
325
        self.assertTrue("BOO!" in output[0])
301
326
 
302
327
class TestCheckoutDiff(TestDiff):
303
328
 
355
380
 
356
381
    def test_external_diff(self):
357
382
        """Test that we can spawn an external diff process"""
 
383
        self.disable_missing_extensions_warning()
358
384
        # We have to use run_bzr_subprocess, because we need to
359
385
        # test writing directly to stdout, (there was a bug in
360
386
        # subprocess.py that we had to workaround).
361
387
        # However, if 'diff' may not be available
362
388
        self.make_example_branch()
363
 
        orig_progress = os.environ.get('BZR_PROGRESS_BAR')
364
 
        try:
365
 
            os.environ['BZR_PROGRESS_BAR'] = 'none'
366
 
            out, err = self.run_bzr_subprocess('diff -r 1 --diff-options -ub',
367
 
                                               universal_newlines=True,
368
 
                                               retcode=None)
369
 
        finally:
370
 
            if orig_progress is None:
371
 
                del os.environ['BZR_PROGRESS_BAR']
372
 
            else:
373
 
                os.environ['BZR_PROGRESS_BAR'] = orig_progress
374
 
 
 
389
        # this will be automatically restored by the base bzr test class
 
390
        os.environ['BZR_PROGRESS_BAR'] = 'none'
 
391
        out, err = self.run_bzr_subprocess('diff -r 1 --diff-options -ub',
 
392
                                           universal_newlines=True,
 
393
                                           retcode=None)
375
394
        if 'Diff is not installed on this machine' in err:
376
 
            raise TestSkipped("No external 'diff' is available")
 
395
            raise tests.TestSkipped("No external 'diff' is available")
377
396
        self.assertEqual('', err)
378
397
        # We have to skip the stuff in the middle, because it depends
379
398
        # on time.time()
383
402
        self.assertEndsWith(out, "\n@@ -0,0 +1 @@\n"
384
403
                                 "+baz\n\n")
385
404
 
 
405
    def test_external_diff_options_and_using(self):
 
406
        """Test that the options are passed correctly to an external diff process"""
 
407
        self.requireFeature(features.diff_feature)
 
408
        self.make_example_branch()
 
409
        self.build_tree_contents([('hello', 'Foo\n')])
 
410
        out, err = self.run_bzr('diff --diff-options -i --using diff',
 
411
                                    retcode=1)
 
412
        self.assertEquals("=== modified file 'hello'\n", out)
 
413
        self.assertEquals('', err)
 
414
 
386
415
 
387
416
class TestDiffOutput(DiffBase):
388
417