diff options
| author | Mike Frysinger <vapier@google.com> | 2025-04-30 13:29:20 -0400 |
|---|---|---|
| committer | LUCI <gerrit-scoped@luci-project-accounts.iam.gserviceaccount.com> | 2025-04-30 12:25:15 -0700 |
| commit | 1acbc14c34f264e0158436dc3d0265d500848462 (patch) | |
| tree | 789df4c5324fe0b485ef502f6aac8e9e20314943 /tests | |
| parent | c448ba9cc7c68b91a122e293402dcc96f511b655 (diff) | |
| download | git-repo-1acbc14c34f264e0158436dc3d0265d500848462.tar.gz | |
manifest: generalize --json as --format=<format>
This will make it easier to add more formats without exploding the
common --xxx space and checking a large set of boolean flags.
Also fill out the test coverage while we're here.
Bug: b/412725063
Change-Id: I754013dc6cb3445f8a0979cefec599d55dafdcff
Reviewed-on: https://gerrit-review.googlesource.com/c/git-repo/+/471941
Reviewed-by: Gavin Mak <gavinmak@google.com>
Commit-Queue: Mike Frysinger <vapier@google.com>
Tested-by: Mike Frysinger <vapier@google.com>
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_subcmds_manifest.py | 156 |
1 files changed, 156 insertions, 0 deletions
diff --git a/tests/test_subcmds_manifest.py b/tests/test_subcmds_manifest.py new file mode 100644 index 00000000..9b1ffb30 --- /dev/null +++ b/tests/test_subcmds_manifest.py | |||
| @@ -0,0 +1,156 @@ | |||
| 1 | # Copyright (C) 2025 The Android Open Source Project | ||
| 2 | # | ||
| 3 | # Licensed under the Apache License, Version 2.0 (the "License"); | ||
| 4 | # you may not use this file except in compliance with the License. | ||
| 5 | # You may obtain a copy of the License at | ||
| 6 | # | ||
| 7 | # http://www.apache.org/licenses/LICENSE-2.0 | ||
| 8 | # | ||
| 9 | # Unless required by applicable law or agreed to in writing, software | ||
| 10 | # distributed under the License is distributed on an "AS IS" BASIS, | ||
| 11 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| 12 | # See the License for the specific language governing permissions and | ||
| 13 | # limitations under the License. | ||
| 14 | |||
| 15 | """Unittests for the subcmds/manifest.py module.""" | ||
| 16 | |||
| 17 | import json | ||
| 18 | from pathlib import Path | ||
| 19 | from unittest import mock | ||
| 20 | |||
| 21 | import manifest_xml | ||
| 22 | from subcmds import manifest | ||
| 23 | |||
| 24 | |||
| 25 | _EXAMPLE_MANIFEST = """\ | ||
| 26 | <?xml version="1.0" encoding="UTF-8"?> | ||
| 27 | <manifest> | ||
| 28 | <remote name="test-remote" fetch="http://localhost" /> | ||
| 29 | <default remote="test-remote" revision="refs/heads/main" /> | ||
| 30 | <project name="repohooks" path="src/repohooks"/> | ||
| 31 | <repo-hooks in-project="repohooks" enabled-list="a, b"/> | ||
| 32 | </manifest> | ||
| 33 | """ | ||
| 34 | |||
| 35 | |||
| 36 | def _get_cmd(repodir: Path) -> manifest.Manifest: | ||
| 37 | """Instantiate a manifest command object to test.""" | ||
| 38 | manifests_git = repodir / "manifests.git" | ||
| 39 | manifests_git.mkdir() | ||
| 40 | (manifests_git / "config").write_text( | ||
| 41 | """ | ||
| 42 | [remote "origin"] | ||
| 43 | \turl = http://localhost/manifest | ||
| 44 | """ | ||
| 45 | ) | ||
| 46 | client = manifest_xml.RepoClient(repodir=str(repodir)) | ||
| 47 | git_event_log = mock.MagicMock(ErrorEvent=mock.Mock(return_value=None)) | ||
| 48 | return manifest.Manifest( | ||
| 49 | repodir=client.repodir, | ||
| 50 | client=client, | ||
| 51 | manifest=client.manifest, | ||
| 52 | outer_client=client, | ||
| 53 | outer_manifest=client.manifest, | ||
| 54 | git_event_log=git_event_log, | ||
| 55 | ) | ||
| 56 | |||
| 57 | |||
| 58 | def test_output_format_xml_file(tmp_path): | ||
| 59 | """Test writing XML to a file.""" | ||
| 60 | path = tmp_path / "manifest.xml" | ||
| 61 | path.write_text(_EXAMPLE_MANIFEST) | ||
| 62 | outpath = tmp_path / "output.xml" | ||
| 63 | cmd = _get_cmd(tmp_path) | ||
| 64 | opt, args = cmd.OptionParser.parse_args(["--output-file", str(outpath)]) | ||
| 65 | cmd.Execute(opt, args) | ||
| 66 | # Normalize the output a bit as we don't exactly care. | ||
| 67 | normalize = lambda data: "\n".join( | ||
| 68 | x.strip() for x in data.splitlines() if x.strip() | ||
| 69 | ) | ||
| 70 | assert ( | ||
| 71 | normalize(outpath.read_text()) | ||
| 72 | == """<?xml version="1.0" encoding="UTF-8"?> | ||
| 73 | <manifest> | ||
| 74 | <remote name="test-remote" fetch="http://localhost"/> | ||
| 75 | <default remote="test-remote" revision="refs/heads/main"/> | ||
| 76 | <project name="repohooks" path="src/repohooks"/> | ||
| 77 | <repo-hooks in-project="repohooks" enabled-list="a b"/> | ||
| 78 | </manifest>""" | ||
| 79 | ) | ||
| 80 | |||
| 81 | |||
| 82 | def test_output_format_xml_stdout(tmp_path, capsys): | ||
| 83 | """Test writing XML to stdout.""" | ||
| 84 | path = tmp_path / "manifest.xml" | ||
| 85 | path.write_text(_EXAMPLE_MANIFEST) | ||
| 86 | cmd = _get_cmd(tmp_path) | ||
| 87 | opt, args = cmd.OptionParser.parse_args(["--format", "xml"]) | ||
| 88 | cmd.Execute(opt, args) | ||
| 89 | # Normalize the output a bit as we don't exactly care. | ||
| 90 | normalize = lambda data: "\n".join( | ||
| 91 | x.strip() for x in data.splitlines() if x.strip() | ||
| 92 | ) | ||
| 93 | stdout = capsys.readouterr().out | ||
| 94 | assert ( | ||
| 95 | normalize(stdout) | ||
| 96 | == """<?xml version="1.0" encoding="UTF-8"?> | ||
| 97 | <manifest> | ||
| 98 | <remote name="test-remote" fetch="http://localhost"/> | ||
| 99 | <default remote="test-remote" revision="refs/heads/main"/> | ||
| 100 | <project name="repohooks" path="src/repohooks"/> | ||
| 101 | <repo-hooks in-project="repohooks" enabled-list="a b"/> | ||
| 102 | </manifest>""" | ||
| 103 | ) | ||
| 104 | |||
| 105 | |||
| 106 | def test_output_format_json(tmp_path, capsys): | ||
| 107 | """Test writing JSON.""" | ||
| 108 | path = tmp_path / "manifest.xml" | ||
| 109 | path.write_text(_EXAMPLE_MANIFEST) | ||
| 110 | cmd = _get_cmd(tmp_path) | ||
| 111 | opt, args = cmd.OptionParser.parse_args(["--format", "json"]) | ||
| 112 | cmd.Execute(opt, args) | ||
| 113 | obj = json.loads(capsys.readouterr().out) | ||
| 114 | assert obj == { | ||
| 115 | "default": {"remote": "test-remote", "revision": "refs/heads/main"}, | ||
| 116 | "project": [{"name": "repohooks", "path": "src/repohooks"}], | ||
| 117 | "remote": [{"fetch": "http://localhost", "name": "test-remote"}], | ||
| 118 | "repo-hooks": {"enabled-list": "a b", "in-project": "repohooks"}, | ||
| 119 | } | ||
| 120 | |||
| 121 | |||
| 122 | def test_output_format_json_pretty(tmp_path, capsys): | ||
| 123 | """Test writing pretty JSON.""" | ||
| 124 | path = tmp_path / "manifest.xml" | ||
| 125 | path.write_text(_EXAMPLE_MANIFEST) | ||
| 126 | cmd = _get_cmd(tmp_path) | ||
| 127 | opt, args = cmd.OptionParser.parse_args(["--format", "json", "--pretty"]) | ||
| 128 | cmd.Execute(opt, args) | ||
| 129 | stdout = capsys.readouterr().out | ||
| 130 | assert ( | ||
| 131 | stdout | ||
| 132 | == """\ | ||
| 133 | { | ||
| 134 | "default": { | ||
| 135 | "remote": "test-remote", | ||
| 136 | "revision": "refs/heads/main" | ||
| 137 | }, | ||
| 138 | "project": [ | ||
| 139 | { | ||
| 140 | "name": "repohooks", | ||
| 141 | "path": "src/repohooks" | ||
| 142 | } | ||
| 143 | ], | ||
| 144 | "remote": [ | ||
| 145 | { | ||
| 146 | "fetch": "http://localhost", | ||
| 147 | "name": "test-remote" | ||
| 148 | } | ||
| 149 | ], | ||
| 150 | "repo-hooks": { | ||
| 151 | "enabled-list": "a b", | ||
| 152 | "in-project": "repohooks" | ||
| 153 | } | ||
| 154 | } | ||
| 155 | """ | ||
| 156 | ) | ||
