diff options
Diffstat (limited to 'tests/test_subcmds_upload.py')
| -rw-r--r-- | tests/test_subcmds_upload.py | 39 |
1 files changed, 37 insertions, 2 deletions
diff --git a/tests/test_subcmds_upload.py b/tests/test_subcmds_upload.py index 4c9008f55..53cc651a1 100644 --- a/tests/test_subcmds_upload.py +++ b/tests/test_subcmds_upload.py | |||
| @@ -116,11 +116,46 @@ def test_GatherOne_returns_resolved_current_branch( | |||
| 116 | "get_parallel_context", | 116 | "get_parallel_context", |
| 117 | lambda: {"projects": [project]}, | 117 | lambda: {"projects": [project]}, |
| 118 | ) | 118 | ) |
| 119 | opt = mock.MagicMock(current_branch=True) | 119 | opt = mock.MagicMock(current_branch=True, ignore_remote_commits=False) |
| 120 | 120 | ||
| 121 | assert upload.Upload._GatherOne(opt, 0) == (0, [branch], "topic") | 121 | assert upload.Upload._GatherOne(opt, 0) == (0, [branch], "topic") |
| 122 | 122 | ||
| 123 | project.GetUploadableBranch.assert_called_once_with("topic") | 123 | project.GetUploadableBranch.assert_called_once_with( |
| 124 | "topic", ignore_remote_commits=False | ||
| 125 | ) | ||
| 126 | |||
| 127 | |||
| 128 | def test_GatherOne_forwards_ignore_remote_commits( | ||
| 129 | cmd: upload.Upload, | ||
| 130 | monkeypatch: pytest.MonkeyPatch, | ||
| 131 | ) -> None: | ||
| 132 | """_GatherOne forwards --ignore-remote-commits to project methods.""" | ||
| 133 | project = mock.MagicMock() | ||
| 134 | project.CurrentBranch = "topic" | ||
| 135 | branch = mock.sentinel.branch | ||
| 136 | project.GetUploadableBranch.return_value = branch | ||
| 137 | project.GetUploadableBranches.return_value = [branch] | ||
| 138 | monkeypatch.setattr( | ||
| 139 | upload.Upload, | ||
| 140 | "get_parallel_context", | ||
| 141 | lambda: {"projects": [project]}, | ||
| 142 | ) | ||
| 143 | |||
| 144 | opt_cbr, _ = cmd.OptionParser.parse_args( | ||
| 145 | ["--cbr", "--ignore-remote-commits"] | ||
| 146 | ) | ||
| 147 | assert upload.Upload._GatherOne(opt_cbr, 0) == (0, [branch], "topic") | ||
| 148 | project.GetUploadableBranch.assert_called_once_with( | ||
| 149 | "topic", ignore_remote_commits=True | ||
| 150 | ) | ||
| 151 | |||
| 152 | opt_br, _ = cmd.OptionParser.parse_args( | ||
| 153 | ["--br", "topic", "--ignore-remote-commits"] | ||
| 154 | ) | ||
| 155 | assert upload.Upload._GatherOne(opt_br, 0) == (0, [branch], None) | ||
| 156 | project.GetUploadableBranches.assert_called_once_with( | ||
| 157 | "topic", ignore_remote_commits=True | ||
| 158 | ) | ||
| 124 | 159 | ||
| 125 | 160 | ||
| 126 | def _create_mock_branch( | 161 | def _create_mock_branch( |
