~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test__dirstate_helpers.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2008-09-03 22:30:56 UTC
  • mfrom: (3644.2.13 index_builder_cleanup)
  • Revision ID: pqm@pqm.ubuntu.com-20080903223056-b108iytb38xkznci
(jam) Streamline BTreeBuilder.add_node et al to make btree creation
        faster.

Show diffs side-by-side

added added

removed removed

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