/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_transform.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2011-06-01 10:00:41 UTC
  • mfrom: (5946.1.3 integration)
  • Revision ID: pqm@pqm.ubuntu.com-20110601100041-xxi0zwzza9z4oz64
(gz) Handle bug #597686, if we open() try to delete the file,
        even if there was an error.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_transform.py
2464
2464
        self.assertPathExists('a/b')
2465
2465
 
2466
2466
 
 
2467
class TestFinalizeRobustness(tests.TestCaseWithTransport):
 
2468
    """Ensure treetransform creation errors can be safely cleaned up after"""
 
2469
 
 
2470
    def _override_globals_in_method(self, instance, method_name, globals):
 
2471
        """Replace method on instance with one with updated globals"""
 
2472
        import types
 
2473
        func = getattr(instance, method_name).im_func
 
2474
        new_globals = dict(func.func_globals)
 
2475
        new_globals.update(globals)
 
2476
        new_func = types.FunctionType(func.func_code, new_globals,
 
2477
            func.func_name, func.func_defaults)
 
2478
        setattr(instance, method_name,
 
2479
            types.MethodType(new_func, instance, instance.__class__))
 
2480
        self.addCleanup(delattr, instance, method_name)
 
2481
 
 
2482
    @staticmethod
 
2483
    def _fake_open_raises_before(name, mode):
 
2484
        """Like open() but raises before doing anything"""
 
2485
        raise RuntimeError
 
2486
 
 
2487
    @staticmethod
 
2488
    def _fake_open_raises_after(name, mode):
 
2489
        """Like open() but raises after creating file without returning"""
 
2490
        open(name, mode).close()
 
2491
        raise RuntimeError
 
2492
 
 
2493
    def create_transform_and_root_trans_id(self):
 
2494
        """Setup a transform creating a file in limbo"""
 
2495
        tree = self.make_branch_and_tree('.')
 
2496
        tt = TreeTransform(tree)
 
2497
        return tt, tt.create_path("a", tt.root)
 
2498
 
 
2499
    def create_transform_and_subdir_trans_id(self):
 
2500
        """Setup a transform creating a directory containing a file in limbo"""
 
2501
        tree = self.make_branch_and_tree('.')
 
2502
        tt = TreeTransform(tree)
 
2503
        d_trans_id = tt.create_path("d", tt.root)
 
2504
        tt.create_directory(d_trans_id)
 
2505
        f_trans_id = tt.create_path("a", d_trans_id)
 
2506
        tt.adjust_path("a", d_trans_id, f_trans_id)
 
2507
        return tt, f_trans_id
 
2508
 
 
2509
    def test_root_create_file_open_raises_before_creation(self):
 
2510
        tt, trans_id = self.create_transform_and_root_trans_id()
 
2511
        self._override_globals_in_method(tt, "create_file",
 
2512
            {"open": self._fake_open_raises_before})
 
2513
        self.assertRaises(RuntimeError, tt.create_file, ["contents"], trans_id)
 
2514
        path = tt._limbo_name(trans_id)
 
2515
        self.assertPathDoesNotExist(path)
 
2516
        tt.finalize()
 
2517
        self.assertPathDoesNotExist(tt._limbodir)
 
2518
 
 
2519
    def test_root_create_file_open_raises_after_creation(self):
 
2520
        tt, trans_id = self.create_transform_and_root_trans_id()
 
2521
        self._override_globals_in_method(tt, "create_file",
 
2522
            {"open": self._fake_open_raises_after})
 
2523
        self.assertRaises(RuntimeError, tt.create_file, ["contents"], trans_id)
 
2524
        path = tt._limbo_name(trans_id)
 
2525
        self.assertPathExists(path)
 
2526
        tt.finalize()
 
2527
        self.assertPathDoesNotExist(path)
 
2528
        self.assertPathDoesNotExist(tt._limbodir)
 
2529
 
 
2530
    def test_subdir_create_file_open_raises_before_creation(self):
 
2531
        tt, trans_id = self.create_transform_and_subdir_trans_id()
 
2532
        self._override_globals_in_method(tt, "create_file",
 
2533
            {"open": self._fake_open_raises_before})
 
2534
        self.assertRaises(RuntimeError, tt.create_file, ["contents"], trans_id)
 
2535
        path = tt._limbo_name(trans_id)
 
2536
        self.assertPathDoesNotExist(path)
 
2537
        tt.finalize()
 
2538
        self.assertPathDoesNotExist(tt._limbodir)
 
2539
 
 
2540
    def test_subdir_create_file_open_raises_after_creation(self):
 
2541
        tt, trans_id = self.create_transform_and_subdir_trans_id()
 
2542
        self._override_globals_in_method(tt, "create_file",
 
2543
            {"open": self._fake_open_raises_after})
 
2544
        self.assertRaises(RuntimeError, tt.create_file, ["contents"], trans_id)
 
2545
        path = tt._limbo_name(trans_id)
 
2546
        self.assertPathExists(path)
 
2547
        tt.finalize()
 
2548
        self.assertPathDoesNotExist(path)
 
2549
        self.assertPathDoesNotExist(tt._limbodir)
 
2550
 
 
2551
 
2467
2552
class TestTransformMissingParent(tests.TestCaseWithTransport):
2468
2553
 
2469
2554
    def make_tt_with_versioned_dir(self):