~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test__dirstate_helpers.py

  • Committer: Ian Clatworthy
  • Date: 2007-12-11 02:07:30 UTC
  • mto: (3119.1.1 ianc-integration)
  • mto: This revision was merged to the branch mainline in revision 3120.
  • Revision ID: ian.clatworthy@internode.on.net-20071211020730-sdj4kj794dw0628e
make help topics more discoverable

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2007-2011 Canonical Ltd
 
1
# Copyright (C) 2007 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
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
"""Tests for the compiled dirstate helpers."""
18
18
 
19
19
import bisect
20
20
import os
21
 
import time
22
21
 
23
22
from bzrlib import (
24
23
    dirstate,
25
 
    errors,
26
 
    osutils,
27
24
    tests,
28
25
    )
29
 
from bzrlib.tests import (
30
 
    test_dirstate,
31
 
    )
32
 
from bzrlib.tests.test_osutils import dir_reader_scenarios
33
 
from bzrlib.tests.scenarios import (
34
 
    load_tests_apply_scenarios,
35
 
    multiply_scenarios,
36
 
    )
37
 
 
38
 
 
39
 
load_tests = load_tests_apply_scenarios
40
 
 
41
 
 
42
 
compiled_dirstate_helpers_feature = tests.ModuleAvailableFeature(
43
 
    'bzrlib._dirstate_helpers_pyx')
44
 
 
45
 
 
46
 
# FIXME: we should also parametrize against SHA1Provider !
47
 
 
48
 
ue_scenarios = [('dirstate_Python',
49
 
    {'update_entry': dirstate.py_update_entry})]
50
 
if compiled_dirstate_helpers_feature.available():
51
 
    update_entry = compiled_dirstate_helpers_feature.module.update_entry
52
 
    ue_scenarios.append(('dirstate_Pyrex', {'update_entry': update_entry}))
53
 
 
54
 
pe_scenarios = [('dirstate_Python',
55
 
    {'_process_entry': dirstate.ProcessEntryPython})]
56
 
if compiled_dirstate_helpers_feature.available():
57
 
    process_entry = compiled_dirstate_helpers_feature.module.ProcessEntryC
58
 
    pe_scenarios.append(('dirstate_Pyrex', {'_process_entry': process_entry}))
 
26
from bzrlib.tests import test_dirstate
 
27
 
 
28
 
 
29
class _CompiledDirstateHelpersFeature(tests.Feature):
 
30
    def _probe(self):
 
31
        try:
 
32
            import bzrlib._dirstate_helpers_c
 
33
        except ImportError:
 
34
            return False
 
35
        return True
 
36
 
 
37
    def feature_name(self):
 
38
        return 'bzrlib._dirstate_helpers_c'
 
39
 
 
40
CompiledDirstateHelpersFeature = _CompiledDirstateHelpersFeature()
59
41
 
60
42
 
61
43
class TestBisectPathMixin(object):
214
196
 
215
197
 
216
198
class TestBisectPathLeft(tests.TestCase, TestBisectPathMixin):
217
 
    """Run all Bisect Path tests against _bisect_path_left."""
 
199
    """Run all Bisect Path tests against _bisect_path_left_py."""
218
200
 
219
201
    def get_bisect_path(self):
220
 
        from bzrlib._dirstate_helpers_py import _bisect_path_left
221
 
        return _bisect_path_left
 
202
        from bzrlib._dirstate_helpers_py import _bisect_path_left_py
 
203
        return _bisect_path_left_py
222
204
 
223
205
    def get_bisect(self):
224
206
        return bisect.bisect_left, 0
225
207
 
226
208
 
227
209
class TestCompiledBisectPathLeft(TestBisectPathLeft):
228
 
    """Run all Bisect Path tests against _bisect_path_lect"""
 
210
    """Run all Bisect Path tests against _bisect_path_right_c"""
229
211
 
230
 
    _test_needs_features = [compiled_dirstate_helpers_feature]
 
212
    _test_needs_features = [CompiledDirstateHelpersFeature]
231
213
 
232
214
    def get_bisect_path(self):
233
 
        from bzrlib._dirstate_helpers_pyx import _bisect_path_left
234
 
        return _bisect_path_left
 
215
        from bzrlib._dirstate_helpers_c import _bisect_path_left_c
 
216
        return _bisect_path_left_c
235
217
 
236
218
 
237
219
class TestBisectPathRight(tests.TestCase, TestBisectPathMixin):
238
 
    """Run all Bisect Path tests against _bisect_path_right"""
 
220
    """Run all Bisect Path tests against _bisect_path_right_py"""
239
221
 
240
222
    def get_bisect_path(self):
241
 
        from bzrlib._dirstate_helpers_py import _bisect_path_right
242
 
        return _bisect_path_right
 
223
        from bzrlib._dirstate_helpers_py import _bisect_path_right_py
 
224
        return _bisect_path_right_py
243
225
 
244
226
    def get_bisect(self):
245
227
        return bisect.bisect_right, -1
246
228
 
247
229
 
248
230
class TestCompiledBisectPathRight(TestBisectPathRight):
249
 
    """Run all Bisect Path tests against _bisect_path_right"""
 
231
    """Run all Bisect Path tests against _bisect_path_right_c"""
250
232
 
251
 
    _test_needs_features = [compiled_dirstate_helpers_feature]
 
233
    _test_needs_features = [CompiledDirstateHelpersFeature]
252
234
 
253
235
    def get_bisect_path(self):
254
 
        from bzrlib._dirstate_helpers_pyx import _bisect_path_right
255
 
        return _bisect_path_right
 
236
        from bzrlib._dirstate_helpers_c import _bisect_path_right_c
 
237
        return _bisect_path_right_c
256
238
 
257
239
 
258
240
class TestBisectDirblock(tests.TestCase):
269
251
 
270
252
    def get_bisect_dirblock(self):
271
253
        """Return an implementation of bisect_dirblock"""
272
 
        from bzrlib._dirstate_helpers_py import bisect_dirblock
273
 
        return bisect_dirblock
 
254
        from bzrlib._dirstate_helpers_py import bisect_dirblock_py
 
255
        return bisect_dirblock_py
274
256
 
275
257
    def assertBisect(self, dirblocks, split_dirblocks, path, *args, **kwargs):
276
258
        """Assert that bisect_split works like bisect_left on the split paths.
360
342
    compiled version.
361
343
    """
362
344
 
363
 
    _test_needs_features = [compiled_dirstate_helpers_feature]
 
345
    _test_needs_features = [CompiledDirstateHelpersFeature]
364
346
 
365
347
    def get_bisect_dirblock(self):
366
 
        from bzrlib._dirstate_helpers_pyx import bisect_dirblock
367
 
        return bisect_dirblock
 
348
        from bzrlib._dirstate_helpers_c import bisect_dirblock_c
 
349
        return bisect_dirblock_c
368
350
 
369
351
 
370
352
class TestCmpByDirs(tests.TestCase):
379
361
 
380
362
    def get_cmp_by_dirs(self):
381
363
        """Get a specific implementation of cmp_by_dirs."""
382
 
        from bzrlib._dirstate_helpers_py import cmp_by_dirs
383
 
        return cmp_by_dirs
 
364
        from bzrlib._dirstate_helpers_py import cmp_by_dirs_py
 
365
        return cmp_by_dirs_py
384
366
 
385
367
    def assertCmpByDirs(self, expected, str1, str2):
386
368
        """Compare the two strings, in both directions.
482
464
class TestCompiledCmpByDirs(TestCmpByDirs):
483
465
    """Test the pyrex implementation of cmp_by_dirs"""
484
466
 
485
 
    _test_needs_features = [compiled_dirstate_helpers_feature]
 
467
    _test_needs_features = [CompiledDirstateHelpersFeature]
486
468
 
487
469
    def get_cmp_by_dirs(self):
488
 
        from bzrlib._dirstate_helpers_pyx import cmp_by_dirs
489
 
        return cmp_by_dirs
 
470
        from bzrlib._dirstate_helpers_c import cmp_by_dirs_c
 
471
        return cmp_by_dirs_c
490
472
 
491
473
 
492
474
class TestCmpPathByDirblock(tests.TestCase):
501
483
 
502
484
    def get_cmp_path_by_dirblock(self):
503
485
        """Get a specific implementation of _cmp_path_by_dirblock."""
504
 
        from bzrlib._dirstate_helpers_py import _cmp_path_by_dirblock
505
 
        return _cmp_path_by_dirblock
 
486
        from bzrlib._dirstate_helpers_py import _cmp_path_by_dirblock_py
 
487
        return _cmp_path_by_dirblock_py
506
488
 
507
489
    def assertCmpPathByDirblock(self, paths):
508
490
        """Compare all paths and make sure they evaluate to the correct order.
633
615
class TestCompiledCmpPathByDirblock(TestCmpPathByDirblock):
634
616
    """Test the pyrex implementation of _cmp_path_by_dirblock"""
635
617
 
636
 
    _test_needs_features = [compiled_dirstate_helpers_feature]
 
618
    _test_needs_features = [CompiledDirstateHelpersFeature]
637
619
 
638
620
    def get_cmp_by_dirs(self):
639
 
        from bzrlib._dirstate_helpers_pyx import _cmp_path_by_dirblock
640
 
        return _cmp_path_by_dirblock
 
621
        from bzrlib._dirstate_helpers_c import _cmp_path_by_dirblock_c
 
622
        return _cmp_path_by_dirblock_c
641
623
 
642
624
 
643
625
class TestMemRChr(tests.TestCase):
644
626
    """Test memrchr functionality"""
645
627
 
646
 
    _test_needs_features = [compiled_dirstate_helpers_feature]
 
628
    _test_needs_features = [CompiledDirstateHelpersFeature]
647
629
 
648
630
    def assertMemRChr(self, expected, s, c):
649
 
        from bzrlib._dirstate_helpers_pyx import _py_memrchr
 
631
        from bzrlib._dirstate_helpers_c import _py_memrchr
650
632
        self.assertEqual(expected, _py_memrchr(s, c))
651
633
 
652
634
    def test_missing(self):
693
675
    implementation.
694
676
    """
695
677
 
696
 
    # inherits scenarios from test_dirstate
697
 
 
698
678
    def get_read_dirblocks(self):
699
 
        from bzrlib._dirstate_helpers_py import _read_dirblocks
700
 
        return _read_dirblocks
 
679
        from bzrlib._dirstate_helpers_py import _read_dirblocks_py
 
680
        return _read_dirblocks_py
701
681
 
702
682
    def test_smoketest(self):
703
683
        """Make sure that we can create and read back a simple file."""
711
691
        self.assertEqual(dirstate.DirState.IN_MEMORY_UNMODIFIED,
712
692
                         state._dirblock_state)
713
693
 
714
 
    def test_trailing_garbage(self):
715
 
        tree, state, expected = self.create_basic_dirstate()
716
 
        # On Unix, we can write extra data as long as we haven't read yet, but
717
 
        # on Win32, if you've opened the file with FILE_SHARE_READ, trying to
718
 
        # open it in append mode will fail.
719
 
        state.unlock()
720
 
        f = open('dirstate', 'ab')
721
 
        try:
722
 
            # Add bogus trailing garbage
723
 
            f.write('bogus\n')
724
 
        finally:
725
 
            f.close()
726
 
            state.lock_read()
727
 
        e = self.assertRaises(errors.DirstateCorrupt,
728
 
                              state._read_dirblocks_if_needed)
729
 
        # Make sure we mention the bogus characters in the error
730
 
        self.assertContainsRe(str(e), 'bogus')
731
 
 
732
694
 
733
695
class TestCompiledReadDirblocks(TestReadDirblocks):
734
696
    """Test the pyrex implementation of _read_dirblocks"""
735
697
 
736
 
    _test_needs_features = [compiled_dirstate_helpers_feature]
 
698
    _test_needs_features = [CompiledDirstateHelpersFeature]
737
699
 
738
700
    def get_read_dirblocks(self):
739
 
        from bzrlib._dirstate_helpers_pyx import _read_dirblocks
740
 
        return _read_dirblocks
 
701
        from bzrlib._dirstate_helpers_c import _read_dirblocks_c
 
702
        return _read_dirblocks_c
741
703
 
742
704
 
743
705
class TestUsingCompiledIfAvailable(tests.TestCase):
744
706
    """Check that any compiled functions that are available are the default.
745
707
 
746
708
    It is possible to have typos, etc in the import line, such that
747
 
    _dirstate_helpers_pyx is actually available, but the compiled functions are
 
709
    _dirstate_helpers_c is actually available, but the compiled functions are
748
710
    not being used.
749
711
    """
750
712
 
751
713
    def test_bisect_dirblock(self):
752
 
        if compiled_dirstate_helpers_feature.available():
753
 
            from bzrlib._dirstate_helpers_pyx import bisect_dirblock
 
714
        if CompiledDirstateHelpersFeature.available():
 
715
            from bzrlib._dirstate_helpers_c import bisect_dirblock_c
 
716
            self.assertIs(bisect_dirblock_c, dirstate.bisect_dirblock)
754
717
        else:
755
 
            from bzrlib._dirstate_helpers_py import bisect_dirblock
756
 
        self.assertIs(bisect_dirblock, dirstate.bisect_dirblock)
 
718
            from bzrlib._dirstate_helpers_py import bisect_dirblock_py
 
719
            self.assertIs(bisect_dirblock_py, dirstate.bisect_dirblock)
757
720
 
758
721
    def test__bisect_path_left(self):
759
 
        if compiled_dirstate_helpers_feature.available():
760
 
            from bzrlib._dirstate_helpers_pyx import _bisect_path_left
 
722
        if CompiledDirstateHelpersFeature.available():
 
723
            from bzrlib._dirstate_helpers_c import _bisect_path_left_c
 
724
            self.assertIs(_bisect_path_left_c, dirstate._bisect_path_left)
761
725
        else:
762
 
            from bzrlib._dirstate_helpers_py import _bisect_path_left
763
 
        self.assertIs(_bisect_path_left, dirstate._bisect_path_left)
 
726
            from bzrlib._dirstate_helpers_py import _bisect_path_left_py
 
727
            self.assertIs(_bisect_path_left_py, dirstate._bisect_path_left)
764
728
 
765
729
    def test__bisect_path_right(self):
766
 
        if compiled_dirstate_helpers_feature.available():
767
 
            from bzrlib._dirstate_helpers_pyx import _bisect_path_right
 
730
        if CompiledDirstateHelpersFeature.available():
 
731
            from bzrlib._dirstate_helpers_c import _bisect_path_right_c
 
732
            self.assertIs(_bisect_path_right_c, dirstate._bisect_path_right)
768
733
        else:
769
 
            from bzrlib._dirstate_helpers_py import _bisect_path_right
770
 
        self.assertIs(_bisect_path_right, dirstate._bisect_path_right)
 
734
            from bzrlib._dirstate_helpers_py import _bisect_path_right_py
 
735
            self.assertIs(_bisect_path_right_py, dirstate._bisect_path_right)
771
736
 
772
737
    def test_cmp_by_dirs(self):
773
 
        if compiled_dirstate_helpers_feature.available():
774
 
            from bzrlib._dirstate_helpers_pyx import cmp_by_dirs
 
738
        if CompiledDirstateHelpersFeature.available():
 
739
            from bzrlib._dirstate_helpers_c import cmp_by_dirs_c
 
740
            self.assertIs(cmp_by_dirs_c, dirstate.cmp_by_dirs)
775
741
        else:
776
 
            from bzrlib._dirstate_helpers_py import cmp_by_dirs
777
 
        self.assertIs(cmp_by_dirs, dirstate.cmp_by_dirs)
 
742
            from bzrlib._dirstate_helpers_py import cmp_by_dirs_py
 
743
            self.assertIs(cmp_by_dirs_py, dirstate.cmp_by_dirs)
778
744
 
779
745
    def test__read_dirblocks(self):
780
 
        if compiled_dirstate_helpers_feature.available():
781
 
            from bzrlib._dirstate_helpers_pyx import _read_dirblocks
782
 
        else:
783
 
            from bzrlib._dirstate_helpers_py import _read_dirblocks
784
 
        self.assertIs(_read_dirblocks, dirstate._read_dirblocks)
785
 
 
786
 
    def test_update_entry(self):
787
 
        if compiled_dirstate_helpers_feature.available():
788
 
            from bzrlib._dirstate_helpers_pyx import update_entry
789
 
        else:
790
 
            from bzrlib.dirstate import update_entry
791
 
        self.assertIs(update_entry, dirstate.update_entry)
792
 
 
793
 
    def test_process_entry(self):
794
 
        if compiled_dirstate_helpers_feature.available():
795
 
            from bzrlib._dirstate_helpers_pyx import ProcessEntryC
796
 
            self.assertIs(ProcessEntryC, dirstate._process_entry)
797
 
        else:
798
 
            from bzrlib.dirstate import ProcessEntryPython
799
 
            self.assertIs(ProcessEntryPython, dirstate._process_entry)
800
 
 
801
 
 
802
 
class TestUpdateEntry(test_dirstate.TestCaseWithDirState):
803
 
    """Test the DirState.update_entry functions"""
804
 
 
805
 
    scenarios = multiply_scenarios(
806
 
        dir_reader_scenarios(), ue_scenarios)
807
 
 
808
 
    # Set by load_tests
809
 
    update_entry = None
810
 
 
811
 
    def setUp(self):
812
 
        super(TestUpdateEntry, self).setUp()
813
 
        self.overrideAttr(dirstate, 'update_entry', self.update_entry)
814
 
 
815
 
    def get_state_with_a(self):
816
 
        """Create a DirState tracking a single object named 'a'"""
817
 
        state = test_dirstate.InstrumentedDirState.initialize('dirstate')
818
 
        self.addCleanup(state.unlock)
819
 
        state.add('a', 'a-id', 'file', None, '')
820
 
        entry = state._get_entry(0, path_utf8='a')
821
 
        return state, entry
822
 
 
823
 
    def test_observed_sha1_cachable(self):
824
 
        state, entry = self.get_state_with_a()
825
 
        atime = time.time() - 10
826
 
        self.build_tree(['a'])
827
 
        statvalue = os.lstat('a')
828
 
        statvalue = test_dirstate._FakeStat(statvalue.st_size, atime, atime,
829
 
            statvalue.st_dev, statvalue.st_ino, statvalue.st_mode)
830
 
        state._observed_sha1(entry, "foo", statvalue)
831
 
        self.assertEqual('foo', entry[1][0][1])
832
 
        packed_stat = dirstate.pack_stat(statvalue)
833
 
        self.assertEqual(packed_stat, entry[1][0][4])
834
 
 
835
 
    def test_observed_sha1_not_cachable(self):
836
 
        state, entry = self.get_state_with_a()
837
 
        oldval = entry[1][0][1]
838
 
        oldstat = entry[1][0][4]
839
 
        self.build_tree(['a'])
840
 
        statvalue = os.lstat('a')
841
 
        state._observed_sha1(entry, "foo", statvalue)
842
 
        self.assertEqual(oldval, entry[1][0][1])
843
 
        self.assertEqual(oldstat, entry[1][0][4])
844
 
 
845
 
    def test_update_entry(self):
846
 
        state, _ = self.get_state_with_a()
847
 
        tree = self.make_branch_and_tree('tree')
848
 
        tree.lock_write()
849
 
        empty_revid = tree.commit('empty')
850
 
        self.build_tree(['tree/a'])
851
 
        tree.add(['a'], ['a-id'])
852
 
        with_a_id = tree.commit('with_a')
853
 
        self.addCleanup(tree.unlock)
854
 
        state.set_parent_trees(
855
 
            [(empty_revid, tree.branch.repository.revision_tree(empty_revid))],
856
 
            [])
857
 
        entry = state._get_entry(0, path_utf8='a')
858
 
        self.build_tree(['a'])
859
 
        # Add one where we don't provide the stat or sha already
860
 
        self.assertEqual(('', 'a', 'a-id'), entry[0])
861
 
        self.assertEqual(('f', '', 0, False, dirstate.DirState.NULLSTAT),
862
 
                         entry[1][0])
863
 
        # Flush the buffers to disk
864
 
        state.save()
865
 
        self.assertEqual(dirstate.DirState.IN_MEMORY_UNMODIFIED,
866
 
                         state._dirblock_state)
867
 
 
868
 
        stat_value = os.lstat('a')
869
 
        packed_stat = dirstate.pack_stat(stat_value)
870
 
        link_or_sha1 = self.update_entry(state, entry, abspath='a',
871
 
                                          stat_value=stat_value)
872
 
        self.assertEqual(None, link_or_sha1)
873
 
 
874
 
        # The dirblock entry should not have computed or cached the file's
875
 
        # sha1, but it did update the files' st_size. However, this is not
876
 
        # worth writing a dirstate file for, so we leave the state UNMODIFIED
877
 
        self.assertEqual(('f', '', 14, False, dirstate.DirState.NULLSTAT),
878
 
                         entry[1][0])
879
 
        self.assertEqual(dirstate.DirState.IN_MEMORY_UNMODIFIED,
880
 
                         state._dirblock_state)
881
 
        mode = stat_value.st_mode
882
 
        self.assertEqual([('is_exec', mode, False)], state._log)
883
 
 
884
 
        state.save()
885
 
        self.assertEqual(dirstate.DirState.IN_MEMORY_UNMODIFIED,
886
 
                         state._dirblock_state)
887
 
 
888
 
        # Roll the clock back so the file is guaranteed to look too new. We
889
 
        # should still not compute the sha1.
890
 
        state.adjust_time(-10)
891
 
        del state._log[:]
892
 
 
893
 
        link_or_sha1 = self.update_entry(state, entry, abspath='a',
894
 
                                          stat_value=stat_value)
895
 
        self.assertEqual([('is_exec', mode, False)], state._log)
896
 
        self.assertEqual(None, link_or_sha1)
897
 
        self.assertEqual(dirstate.DirState.IN_MEMORY_UNMODIFIED,
898
 
                         state._dirblock_state)
899
 
        self.assertEqual(('f', '', 14, False, dirstate.DirState.NULLSTAT),
900
 
                         entry[1][0])
901
 
        state.save()
902
 
 
903
 
        # If it is cachable (the clock has moved forward) but new it still
904
 
        # won't calculate the sha or cache it.
905
 
        state.adjust_time(+20)
906
 
        del state._log[:]
907
 
        link_or_sha1 = dirstate.update_entry(state, entry, abspath='a',
908
 
                                          stat_value=stat_value)
909
 
        self.assertEqual(None, link_or_sha1)
910
 
        self.assertEqual([('is_exec', mode, False)], state._log)
911
 
        self.assertEqual(('f', '', 14, False, dirstate.DirState.NULLSTAT),
912
 
                         entry[1][0])
913
 
        self.assertEqual(dirstate.DirState.IN_MEMORY_UNMODIFIED,
914
 
                         state._dirblock_state)
915
 
 
916
 
        # If the file is no longer new, and the clock has been moved forward
917
 
        # sufficiently, it will cache the sha.
918
 
        del state._log[:]
919
 
        state.set_parent_trees(
920
 
            [(with_a_id, tree.branch.repository.revision_tree(with_a_id))],
921
 
            [])
922
 
        entry = state._get_entry(0, path_utf8='a')
923
 
 
924
 
        link_or_sha1 = self.update_entry(state, entry, abspath='a',
925
 
                                          stat_value=stat_value)
926
 
        self.assertEqual('b50e5406bb5e153ebbeb20268fcf37c87e1ecfb6',
927
 
                         link_or_sha1)
928
 
        self.assertEqual([('is_exec', mode, False), ('sha1', 'a')],
929
 
                          state._log)
930
 
        self.assertEqual(('f', link_or_sha1, 14, False, packed_stat),
931
 
                         entry[1][0])
932
 
 
933
 
        # Subsequent calls will just return the cached value
934
 
        del state._log[:]
935
 
        link_or_sha1 = self.update_entry(state, entry, abspath='a',
936
 
                                          stat_value=stat_value)
937
 
        self.assertEqual('b50e5406bb5e153ebbeb20268fcf37c87e1ecfb6',
938
 
                         link_or_sha1)
939
 
        self.assertEqual([], state._log)
940
 
        self.assertEqual(('f', link_or_sha1, 14, False, packed_stat),
941
 
                         entry[1][0])
942
 
 
943
 
    def test_update_entry_symlink(self):
944
 
        """Update entry should read symlinks."""
945
 
        self.requireFeature(tests.SymlinkFeature)
946
 
        state, entry = self.get_state_with_a()
947
 
        state.save()
948
 
        self.assertEqual(dirstate.DirState.IN_MEMORY_UNMODIFIED,
949
 
                         state._dirblock_state)
950
 
        os.symlink('target', 'a')
951
 
 
952
 
        state.adjust_time(-10) # Make the symlink look new
953
 
        stat_value = os.lstat('a')
954
 
        packed_stat = dirstate.pack_stat(stat_value)
955
 
        link_or_sha1 = self.update_entry(state, entry, abspath='a',
956
 
                                          stat_value=stat_value)
957
 
        self.assertEqual('target', link_or_sha1)
958
 
        self.assertEqual([('read_link', 'a', '')], state._log)
959
 
        # Dirblock is not updated (the link is too new)
960
 
        self.assertEqual([('l', '', 6, False, dirstate.DirState.NULLSTAT)],
961
 
                         entry[1])
962
 
        self.assertEqual(dirstate.DirState.IN_MEMORY_MODIFIED,
963
 
                         state._dirblock_state)
964
 
 
965
 
        # Because the stat_value looks new, we should re-read the target
966
 
        link_or_sha1 = self.update_entry(state, entry, abspath='a',
967
 
                                          stat_value=stat_value)
968
 
        self.assertEqual('target', link_or_sha1)
969
 
        self.assertEqual([('read_link', 'a', ''),
970
 
                          ('read_link', 'a', ''),
971
 
                         ], state._log)
972
 
        self.assertEqual([('l', '', 6, False, dirstate.DirState.NULLSTAT)],
973
 
                         entry[1])
974
 
        state.adjust_time(+20) # Skip into the future, all files look old
975
 
        link_or_sha1 = self.update_entry(state, entry, abspath='a',
976
 
                                          stat_value=stat_value)
977
 
        self.assertEqual('target', link_or_sha1)
978
 
        # We need to re-read the link because only now can we cache it
979
 
        self.assertEqual([('read_link', 'a', ''),
980
 
                          ('read_link', 'a', ''),
981
 
                          ('read_link', 'a', ''),
982
 
                         ], state._log)
983
 
        self.assertEqual([('l', 'target', 6, False, packed_stat)],
984
 
                         entry[1])
985
 
 
986
 
        # Another call won't re-read the link
987
 
        self.assertEqual([('read_link', 'a', ''),
988
 
                          ('read_link', 'a', ''),
989
 
                          ('read_link', 'a', ''),
990
 
                         ], state._log)
991
 
        link_or_sha1 = self.update_entry(state, entry, abspath='a',
992
 
                                          stat_value=stat_value)
993
 
        self.assertEqual('target', link_or_sha1)
994
 
        self.assertEqual([('l', 'target', 6, False, packed_stat)],
995
 
                         entry[1])
996
 
 
997
 
    def do_update_entry(self, state, entry, abspath):
998
 
        stat_value = os.lstat(abspath)
999
 
        return self.update_entry(state, entry, abspath, stat_value)
1000
 
 
1001
 
    def test_update_entry_dir(self):
1002
 
        state, entry = self.get_state_with_a()
1003
 
        self.build_tree(['a/'])
1004
 
        self.assertIs(None, self.do_update_entry(state, entry, 'a'))
1005
 
 
1006
 
    def test_update_entry_dir_unchanged(self):
1007
 
        state, entry = self.get_state_with_a()
1008
 
        self.build_tree(['a/'])
1009
 
        state.adjust_time(+20)
1010
 
        self.assertIs(None, self.do_update_entry(state, entry, 'a'))
1011
 
        # a/ used to be a file, but is now a directory, worth saving
1012
 
        self.assertEqual(dirstate.DirState.IN_MEMORY_MODIFIED,
1013
 
                         state._dirblock_state)
1014
 
        state.save()
1015
 
        self.assertEqual(dirstate.DirState.IN_MEMORY_UNMODIFIED,
1016
 
                         state._dirblock_state)
1017
 
        # No changes to a/ means not worth saving.
1018
 
        self.assertIs(None, self.do_update_entry(state, entry, 'a'))
1019
 
        self.assertEqual(dirstate.DirState.IN_MEMORY_UNMODIFIED,
1020
 
                         state._dirblock_state)
1021
 
        # Change the last-modified time for the directory
1022
 
        t = time.time() - 100.0
1023
 
        try:
1024
 
            os.utime('a', (t, t))
1025
 
        except OSError:
1026
 
            # It looks like Win32 + FAT doesn't allow to change times on a dir.
1027
 
            raise tests.TestSkipped("can't update mtime of a dir on FAT")
1028
 
        saved_packed_stat = entry[1][0][-1]
1029
 
        self.assertIs(None, self.do_update_entry(state, entry, 'a'))
1030
 
        # We *do* go ahead and update the information in the dirblocks, but we
1031
 
        # don't bother setting IN_MEMORY_MODIFIED because it is trivial to
1032
 
        # recompute.
1033
 
        self.assertNotEqual(saved_packed_stat, entry[1][0][-1])
1034
 
        self.assertEqual(dirstate.DirState.IN_MEMORY_UNMODIFIED,
1035
 
                         state._dirblock_state)
1036
 
 
1037
 
    def test_update_entry_file_unchanged(self):
1038
 
        state, _ = self.get_state_with_a()
1039
 
        tree = self.make_branch_and_tree('tree')
1040
 
        tree.lock_write()
1041
 
        self.build_tree(['tree/a'])
1042
 
        tree.add(['a'], ['a-id'])
1043
 
        with_a_id = tree.commit('witha')
1044
 
        self.addCleanup(tree.unlock)
1045
 
        state.set_parent_trees(
1046
 
            [(with_a_id, tree.branch.repository.revision_tree(with_a_id))],
1047
 
            [])
1048
 
        entry = state._get_entry(0, path_utf8='a')
1049
 
        self.build_tree(['a'])
1050
 
        sha1sum = 'b50e5406bb5e153ebbeb20268fcf37c87e1ecfb6'
1051
 
        state.adjust_time(+20)
1052
 
        self.assertEqual(sha1sum, self.do_update_entry(state, entry, 'a'))
1053
 
        self.assertEqual(dirstate.DirState.IN_MEMORY_MODIFIED,
1054
 
                         state._dirblock_state)
1055
 
        state.save()
1056
 
        self.assertEqual(dirstate.DirState.IN_MEMORY_UNMODIFIED,
1057
 
                         state._dirblock_state)
1058
 
        self.assertEqual(sha1sum, self.do_update_entry(state, entry, 'a'))
1059
 
        self.assertEqual(dirstate.DirState.IN_MEMORY_UNMODIFIED,
1060
 
                         state._dirblock_state)
1061
 
 
1062
 
    def test_update_entry_tree_reference(self):
1063
 
        state = test_dirstate.InstrumentedDirState.initialize('dirstate')
1064
 
        self.addCleanup(state.unlock)
1065
 
        state.add('r', 'r-id', 'tree-reference', None, '')
1066
 
        self.build_tree(['r/'])
1067
 
        entry = state._get_entry(0, path_utf8='r')
1068
 
        self.do_update_entry(state, entry, 'r')
1069
 
        entry = state._get_entry(0, path_utf8='r')
1070
 
        self.assertEqual('t', entry[1][0][0])
1071
 
 
1072
 
    def create_and_test_file(self, state, entry):
1073
 
        """Create a file at 'a' and verify the state finds it during update.
1074
 
 
1075
 
        The state should already be versioning *something* at 'a'. This makes
1076
 
        sure that state.update_entry recognizes it as a file.
1077
 
        """
1078
 
        self.build_tree(['a'])
1079
 
        stat_value = os.lstat('a')
1080
 
        packed_stat = dirstate.pack_stat(stat_value)
1081
 
 
1082
 
        link_or_sha1 = self.do_update_entry(state, entry, abspath='a')
1083
 
        self.assertEqual(None, link_or_sha1)
1084
 
        self.assertEqual([('f', '', 14, False, dirstate.DirState.NULLSTAT)],
1085
 
                         entry[1])
1086
 
        return packed_stat
1087
 
 
1088
 
    def create_and_test_dir(self, state, entry):
1089
 
        """Create a directory at 'a' and verify the state finds it.
1090
 
 
1091
 
        The state should already be versioning *something* at 'a'. This makes
1092
 
        sure that state.update_entry recognizes it as a directory.
1093
 
        """
1094
 
        self.build_tree(['a/'])
1095
 
        stat_value = os.lstat('a')
1096
 
        packed_stat = dirstate.pack_stat(stat_value)
1097
 
 
1098
 
        link_or_sha1 = self.do_update_entry(state, entry, abspath='a')
1099
 
        self.assertIs(None, link_or_sha1)
1100
 
        self.assertEqual([('d', '', 0, False, packed_stat)], entry[1])
1101
 
 
1102
 
        return packed_stat
1103
 
 
1104
 
    # FIXME: Add unicode version
1105
 
    def create_and_test_symlink(self, state, entry):
1106
 
        """Create a symlink at 'a' and verify the state finds it.
1107
 
 
1108
 
        The state should already be versioning *something* at 'a'. This makes
1109
 
        sure that state.update_entry recognizes it as a symlink.
1110
 
 
1111
 
        This should not be called if this platform does not have symlink
1112
 
        support.
1113
 
        """
1114
 
        # caller should care about skipping test on platforms without symlinks
1115
 
        os.symlink('path/to/foo', 'a')
1116
 
 
1117
 
        stat_value = os.lstat('a')
1118
 
        packed_stat = dirstate.pack_stat(stat_value)
1119
 
 
1120
 
        link_or_sha1 = self.do_update_entry(state, entry, abspath='a')
1121
 
        self.assertEqual('path/to/foo', link_or_sha1)
1122
 
        self.assertEqual([('l', 'path/to/foo', 11, False, packed_stat)],
1123
 
                         entry[1])
1124
 
        return packed_stat
1125
 
 
1126
 
    def test_update_file_to_dir(self):
1127
 
        """If a file changes to a directory we return None for the sha.
1128
 
        We also update the inventory record.
1129
 
        """
1130
 
        state, entry = self.get_state_with_a()
1131
 
        # The file sha1 won't be cached unless the file is old
1132
 
        state.adjust_time(+10)
1133
 
        self.create_and_test_file(state, entry)
1134
 
        os.remove('a')
1135
 
        self.create_and_test_dir(state, entry)
1136
 
 
1137
 
    def test_update_file_to_symlink(self):
1138
 
        """File becomes a symlink"""
1139
 
        self.requireFeature(tests.SymlinkFeature)
1140
 
        state, entry = self.get_state_with_a()
1141
 
        # The file sha1 won't be cached unless the file is old
1142
 
        state.adjust_time(+10)
1143
 
        self.create_and_test_file(state, entry)
1144
 
        os.remove('a')
1145
 
        self.create_and_test_symlink(state, entry)
1146
 
 
1147
 
    def test_update_dir_to_file(self):
1148
 
        """Directory becoming a file updates the entry."""
1149
 
        state, entry = self.get_state_with_a()
1150
 
        # The file sha1 won't be cached unless the file is old
1151
 
        state.adjust_time(+10)
1152
 
        self.create_and_test_dir(state, entry)
1153
 
        os.rmdir('a')
1154
 
        self.create_and_test_file(state, entry)
1155
 
 
1156
 
    def test_update_dir_to_symlink(self):
1157
 
        """Directory becomes a symlink"""
1158
 
        self.requireFeature(tests.SymlinkFeature)
1159
 
        state, entry = self.get_state_with_a()
1160
 
        # The symlink target won't be cached if it isn't old
1161
 
        state.adjust_time(+10)
1162
 
        self.create_and_test_dir(state, entry)
1163
 
        os.rmdir('a')
1164
 
        self.create_and_test_symlink(state, entry)
1165
 
 
1166
 
    def test_update_symlink_to_file(self):
1167
 
        """Symlink becomes a file"""
1168
 
        self.requireFeature(tests.SymlinkFeature)
1169
 
        state, entry = self.get_state_with_a()
1170
 
        # The symlink and file info won't be cached unless old
1171
 
        state.adjust_time(+10)
1172
 
        self.create_and_test_symlink(state, entry)
1173
 
        os.remove('a')
1174
 
        self.create_and_test_file(state, entry)
1175
 
 
1176
 
    def test_update_symlink_to_dir(self):
1177
 
        """Symlink becomes a directory"""
1178
 
        self.requireFeature(tests.SymlinkFeature)
1179
 
        state, entry = self.get_state_with_a()
1180
 
        # The symlink target won't be cached if it isn't old
1181
 
        state.adjust_time(+10)
1182
 
        self.create_and_test_symlink(state, entry)
1183
 
        os.remove('a')
1184
 
        self.create_and_test_dir(state, entry)
1185
 
 
1186
 
    def test__is_executable_win32(self):
1187
 
        state, entry = self.get_state_with_a()
1188
 
        self.build_tree(['a'])
1189
 
 
1190
 
        # Make sure we are using the win32 implementation of _is_executable
1191
 
        state._is_executable = state._is_executable_win32
1192
 
 
1193
 
        # The file on disk is not executable, but we are marking it as though
1194
 
        # it is. With _is_executable_win32 we ignore what is on disk.
1195
 
        entry[1][0] = ('f', '', 0, True, dirstate.DirState.NULLSTAT)
1196
 
 
1197
 
        stat_value = os.lstat('a')
1198
 
        packed_stat = dirstate.pack_stat(stat_value)
1199
 
 
1200
 
        state.adjust_time(-10) # Make sure everything is new
1201
 
        self.update_entry(state, entry, abspath='a', stat_value=stat_value)
1202
 
 
1203
 
        # The row is updated, but the executable bit stays set.
1204
 
        self.assertEqual([('f', '', 14, True, dirstate.DirState.NULLSTAT)],
1205
 
                         entry[1])
1206
 
 
1207
 
        # Make the disk object look old enough to cache (but it won't cache the
1208
 
        # sha as it is a new file).
1209
 
        state.adjust_time(+20)
1210
 
        digest = 'b50e5406bb5e153ebbeb20268fcf37c87e1ecfb6'
1211
 
        self.update_entry(state, entry, abspath='a', stat_value=stat_value)
1212
 
        self.assertEqual([('f', '', 14, True, dirstate.DirState.NULLSTAT)],
1213
 
            entry[1])
1214
 
 
1215
 
    def _prepare_tree(self):
1216
 
        # Create a tree
1217
 
        text = 'Hello World\n'
1218
 
        tree = self.make_branch_and_tree('tree')
1219
 
        self.build_tree_contents([('tree/a file', text)])
1220
 
        tree.add('a file', 'a-file-id')
1221
 
        # Note: dirstate does not sha prior to the first commit
1222
 
        # so commit now in order for the test to work
1223
 
        tree.commit('first')
1224
 
        return tree, text
1225
 
 
1226
 
    def test_sha1provider_sha1_used(self):
1227
 
        tree, text = self._prepare_tree()
1228
 
        state = dirstate.DirState.from_tree(tree, 'dirstate',
1229
 
            UppercaseSHA1Provider())
1230
 
        self.addCleanup(state.unlock)
1231
 
        expected_sha = osutils.sha_string(text.upper() + "foo")
1232
 
        entry = state._get_entry(0, path_utf8='a file')
1233
 
        state._sha_cutoff_time()
1234
 
        state._cutoff_time += 10
1235
 
        sha1 = self.update_entry(state, entry, 'tree/a file',
1236
 
                                 os.lstat('tree/a file'))
1237
 
        self.assertEqual(expected_sha, sha1)
1238
 
 
1239
 
    def test_sha1provider_stat_and_sha1_used(self):
1240
 
        tree, text = self._prepare_tree()
1241
 
        tree.lock_write()
1242
 
        self.addCleanup(tree.unlock)
1243
 
        state = tree._current_dirstate()
1244
 
        state._sha1_provider = UppercaseSHA1Provider()
1245
 
        # If we used the standard provider, it would look like nothing has
1246
 
        # changed
1247
 
        file_ids_changed = [change[0] for change
1248
 
                            in tree.iter_changes(tree.basis_tree())]
1249
 
        self.assertEqual(['a-file-id'], file_ids_changed)
1250
 
 
1251
 
 
1252
 
class UppercaseSHA1Provider(dirstate.SHA1Provider):
1253
 
    """A custom SHA1Provider."""
1254
 
 
1255
 
    def sha1(self, abspath):
1256
 
        return self.stat_and_sha1(abspath)[1]
1257
 
 
1258
 
    def stat_and_sha1(self, abspath):
1259
 
        file_obj = file(abspath, 'rb')
1260
 
        try:
1261
 
            statvalue = os.fstat(file_obj.fileno())
1262
 
            text = ''.join(file_obj.readlines())
1263
 
            sha1 = osutils.sha_string(text.upper() + "foo")
1264
 
        finally:
1265
 
            file_obj.close()
1266
 
        return statvalue, sha1
1267
 
 
1268
 
 
1269
 
class TestProcessEntry(test_dirstate.TestCaseWithDirState):
1270
 
 
1271
 
    scenarios = multiply_scenarios(dir_reader_scenarios(), pe_scenarios)
1272
 
 
1273
 
    # Set by load_tests
1274
 
    _process_entry = None
1275
 
 
1276
 
    def setUp(self):
1277
 
        super(TestProcessEntry, self).setUp()
1278
 
        self.overrideAttr(dirstate, '_process_entry', self._process_entry)
1279
 
 
1280
 
    def assertChangedFileIds(self, expected, tree):
1281
 
        tree.lock_read()
1282
 
        try:
1283
 
            file_ids = [info[0] for info
1284
 
                        in tree.iter_changes(tree.basis_tree())]
1285
 
        finally:
1286
 
            tree.unlock()
1287
 
        self.assertEqual(sorted(expected), sorted(file_ids))
1288
 
 
1289
 
    def test_exceptions_raised(self):
1290
 
        # This is a direct test of bug #495023, it relies on osutils.is_inside
1291
 
        # getting called in an inner function. Which makes it a bit brittle,
1292
 
        # but at least it does reproduce the bug.
1293
 
        tree = self.make_branch_and_tree('tree')
1294
 
        self.build_tree(['tree/file', 'tree/dir/', 'tree/dir/sub',
1295
 
                         'tree/dir2/', 'tree/dir2/sub2'])
1296
 
        tree.add(['file', 'dir', 'dir/sub', 'dir2', 'dir2/sub2'])
1297
 
        tree.commit('first commit')
1298
 
        tree.lock_read()
1299
 
        self.addCleanup(tree.unlock)
1300
 
        basis_tree = tree.basis_tree()
1301
 
        def is_inside_raises(*args, **kwargs):
1302
 
            raise RuntimeError('stop this')
1303
 
        self.overrideAttr(osutils, 'is_inside', is_inside_raises)
1304
 
        self.assertListRaises(RuntimeError, tree.iter_changes, basis_tree)
1305
 
 
1306
 
    def test_simple_changes(self):
1307
 
        tree = self.make_branch_and_tree('tree')
1308
 
        self.build_tree(['tree/file'])
1309
 
        tree.add(['file'], ['file-id'])
1310
 
        self.assertChangedFileIds([tree.get_root_id(), 'file-id'], tree)
1311
 
        tree.commit('one')
1312
 
        self.assertChangedFileIds([], tree)
1313
 
 
1314
 
    def test_sha1provider_stat_and_sha1_used(self):
1315
 
        tree = self.make_branch_and_tree('tree')
1316
 
        self.build_tree(['tree/file'])
1317
 
        tree.add(['file'], ['file-id'])
1318
 
        tree.commit('one')
1319
 
        tree.lock_write()
1320
 
        self.addCleanup(tree.unlock)
1321
 
        state = tree._current_dirstate()
1322
 
        state._sha1_provider = UppercaseSHA1Provider()
1323
 
        self.assertChangedFileIds(['file-id'], tree)
 
746
        if CompiledDirstateHelpersFeature.available():
 
747
            from bzrlib._dirstate_helpers_c import _read_dirblocks_c
 
748
            self.assertIs(_read_dirblocks_c, dirstate._read_dirblocks)
 
749
        else:
 
750
            from bzrlib._dirstate_helpers_py import _read_dirblocks_py
 
751
            self.assertIs(_read_dirblocks_py, dirstate._read_dirblocks)
1324
752