|
7 | 7 | import os.path as osp
|
8 | 8 | from pathlib import Path
|
9 | 9 | import subprocess
|
10 |
| -import tempfile |
11 | 10 |
|
12 | 11 | from git.objects import Tree, Blob
|
13 |
| -from git.util import cwd, rmtree |
14 |
| -from test.lib import TestBase |
| 12 | +from git.util import cwd |
| 13 | +from test.lib import TestBase, with_rw_directory |
15 | 14 |
|
16 | 15 |
|
17 | 16 | class TestTree(TestBase):
|
@@ -43,39 +42,35 @@ def test_serializable(self):
|
43 | 42 | testtree._deserialize(stream)
|
44 | 43 | # END for each item in tree
|
45 | 44 |
|
46 |
| - @staticmethod |
47 |
| - def _get_git_ordered_files(): |
| 45 | + @with_rw_directory |
| 46 | + def _get_git_ordered_files(self, rw_dir): |
48 | 47 | """Get files as git orders them, to compare in test_tree_modifier_ordering."""
|
49 |
| - with tempfile.TemporaryDirectory() as tdir: |
50 |
| - # Create directory contents. |
51 |
| - Path(tdir, "file").mkdir() |
52 |
| - for filename in ( |
53 |
| - "bin", |
54 |
| - "bin.d", |
55 |
| - "file.to", |
56 |
| - "file.toml", |
57 |
| - "file.toml.bin", |
58 |
| - "file0", |
59 |
| - ): |
60 |
| - Path(tdir, filename).touch() |
61 |
| - Path(tdir, "file", "a").touch() |
62 |
| - |
63 |
| - try: |
64 |
| - with cwd(tdir): |
65 |
| - # Prepare the repository. |
66 |
| - subprocess.run(["git", "init", "-q"], check=True) |
67 |
| - subprocess.run(["git", "add", "."], check=True) |
68 |
| - subprocess.run(["git", "commit", "-m", "c1"], check=True) |
69 |
| - |
70 |
| - # Get git output from which an ordered file list can be parsed. |
71 |
| - rev_parse_command = ["git", "rev-parse", "HEAD^{tree}"] |
72 |
| - tree_hash = subprocess.check_output(rev_parse_command).decode().strip() |
73 |
| - cat_file_command = ["git", "cat-file", "-p", tree_hash] |
74 |
| - cat_file_output = subprocess.check_output(cat_file_command).decode() |
75 |
| - finally: |
76 |
| - rmtree(Path(tdir, ".git")) |
77 |
| - |
78 |
| - return [line.split()[-1] for line in cat_file_output.split("\n") if line] |
| 48 | + # Create directory contents. |
| 49 | + Path(rw_dir, "file").mkdir() |
| 50 | + for filename in ( |
| 51 | + "bin", |
| 52 | + "bin.d", |
| 53 | + "file.to", |
| 54 | + "file.toml", |
| 55 | + "file.toml.bin", |
| 56 | + "file0", |
| 57 | + ): |
| 58 | + Path(rw_dir, filename).touch() |
| 59 | + Path(rw_dir, "file", "a").touch() |
| 60 | + |
| 61 | + with cwd(rw_dir): |
| 62 | + # Prepare the repository. |
| 63 | + subprocess.run(["git", "init", "-q"], check=True) |
| 64 | + subprocess.run(["git", "add", "."], check=True) |
| 65 | + subprocess.run(["git", "commit", "-m", "c1"], check=True) |
| 66 | + |
| 67 | + # Get git output from which an ordered file list can be parsed. |
| 68 | + rev_parse_command = ["git", "rev-parse", "HEAD^{tree}"] |
| 69 | + tree_hash = subprocess.check_output(rev_parse_command).decode().strip() |
| 70 | + cat_file_command = ["git", "cat-file", "-p", tree_hash] |
| 71 | + cat_file_output = subprocess.check_output(cat_file_command).decode() |
| 72 | + |
| 73 | + return [line.split()[-1] for line in cat_file_output.split("\n") if line] |
79 | 74 |
|
80 | 75 | def test_tree_modifier_ordering(self):
|
81 | 76 | """TreeModifier.set_done() sorts files in the same order git does."""
|
|
0 commit comments