implement base picking command

Author: dana-yaish

codecov_cli/commands/base_picking.py

@@ -0,0 +1,95 @@
+import logging
+import typing
+import uuid
+
+import click
+
+from codecov_cli.fallbacks import CodecovOption, FallbackFieldEnum
+from codecov_cli.helpers.encoder import slug_is_invalid
+from codecov_cli.helpers.request import send_put_request
+
+logger = logging.getLogger("codecovcli")
+
+
+@click.command()
+@click.option(
+    "--base-sha",
+    help="Base commit SHA (with 40 chars)",
+    cls=CodecovOption,
+    fallback_field=FallbackFieldEnum.commit_sha,
+    required=True,
+)
+@click.option(
+    "--pr",
+    help="Pull Request id to associate commit with",
+    cls=CodecovOption,
+    fallback_field=FallbackFieldEnum.pull_request_number,
+)
+@click.option(
+    "--slug",
+    cls=CodecovOption,
+    fallback_field=FallbackFieldEnum.slug,
+    help="owner/repo slug",
+    envvar="CODECOV_SLUG",
+)
+@click.option(
+    "-t",
+    "--token",
+    help="Codecov upload token",
+    type=click.UUID,
+    envvar="CODECOV_TOKEN",
+)
+@click.option(
+    "--service",
+    cls=CodecovOption,
+    fallback_field=FallbackFieldEnum.service,
+    help="Specify the service provider of the repo e.g. github",
+)
+@click.pass_context
+def pr_base_picking(
+    ctx,
+    base_sha: str,
+    pr: typing.Optional[int],
+    slug: typing.Optional[str],
+    token: typing.Optional[uuid.UUID],
+    service: typing.Optional[str],
+):
+    logger.debug(
+        "Starting base picking process",
+        extra=dict(
+            extra_log_attributes=dict(
+                pr=pr,
+                slug=slug,
+                token=token,
+                service=service,
+            )
+        ),
+    )
+
+    if slug_is_invalid(slug):
+        logger.error(
+            "Slug is invalid. Slug should be in the form of owner_username/repo_name"
+        )
+        return
+
+    data = {
+        "user_provided_base_sha": base_sha,
+    }
+    headers = {"Authorization": f"token {token.hex}"}
+    url = f"https://api.codecov.io/api/v1/{service}/{slug}/pulls/{pr}"
+    sending_result = send_put_request(url=url, data=data, headers=headers)
+
+    if sending_result.warnings:
+        number_warnings = len(sending_result.warnings)
+        pluralization = "s" if number_warnings > 1 else ""
+        logger.info(
+            f"Base picking process had {number_warnings} warning{pluralization}",
+        )
+        for ind, w in enumerate(sending_result.warnings):
+            logger.warning(f"Warning {ind + 1}: {w}")
+    if sending_result.error is not None:
+        logger.error(f"Base picking failed: {sending_result.error.description}")
+        return
+
+    logger.info("Base picking finished successfully")
+    logger.info(sending_result.text)

codecov_cli/helpers/encoder.py

@@ -5,3 +5,9 @@ def encode_slug(slug: str):
     encoded_owner = ":::".join(owner.split("/"))
     encoded_slug = "::::".join([encoded_owner, repo])
     return encoded_slug
+
+
+def slug_is_invalid(slug: str):
+    if "/" not in slug or slug.count("/") > 1:
+        return True
+    return False

codecov_cli/main.py

@@ -5,6 +5,7 @@
 import click
 import yaml
 
+from codecov_cli.commands.base_picking import pr_base_picking
 from codecov_cli.commands.commit import create_commit
 from codecov_cli.commands.report import create_report
 from codecov_cli.commands.report_result import get_report_results
@@ -47,6 +48,7 @@ def cli(
 cli.add_command(create_commit)
 cli.add_command(create_report)
 cli.add_command(get_report_results)
+cli.add_command(pr_base_picking)
 
 
 def run():

tests/helpers/test_encoder.py

@@ -1,6 +1,6 @@
 import pytest
 
-from codecov_cli.helpers.encoder import encode_slug
+from codecov_cli.helpers.encoder import encode_slug, slug_is_invalid
 
 
 def test_invalid_slug():
@@ -9,13 +9,23 @@ def test_invalid_slug():
         encode_slug(slug)
 
 
-def test_owner_repo_slug():
+def test_encode_slug():
     slug = "owner/repo"
     encoded_slug = encode_slug(slug)
     assert encoded_slug == "owner::::repo"
 
 
-def test_owner_with_subgroups_slug():
+def test_encode_owner_with_subgroups_slug():
     slug = "owner/subgroup/repo"
     encoded_slug = encode_slug(slug)
     assert encoded_slug == "owner:::subgroup::::repo"
+
+
+def test_invalid_slug2():
+    slug = "invalid_slug"
+    assert slug_is_invalid(slug)
+
+
+def test_valid_slug():
+    slug = "owner/repo"
+    assert not slug_is_invalid(slug)

tests/test_base_picking.py

@@ -0,0 +1,140 @@
+import uuid
+
+from click.testing import CliRunner
+
+from codecov_cli.commands.base_picking import pr_base_picking
+from codecov_cli.main import cli
+from codecov_cli.types import RequestError, RequestResult
+
+
+def test_base_picking_command(mocker):
+    mocked_response = mocker.patch(
+        "codecov_cli.commands.base_picking.send_put_request",
+        return_value=RequestResult(status_code=200, error=None, warnings=[], text=""),
+    )
+    token = uuid.uuid4()
+    runner = CliRunner()
+    result = runner.invoke(
+        pr_base_picking,
+        [
+            "-t",
+            token,
+            "--pr",
+            "11",
+            "--base-sha",
+            "9a6902ee94c18e8e27561ce316b16d75a02c7bc1",
+            "--service",
+            "github",
+            "--slug",
+            "owner/repo",
+        ],
+    )
+    assert result.exit_code == 0
+    assert (
+        f'debug: Starting base picking process --- {{"pr": "11", "slug": "owner/repo", "token": "{token}", "service": "github"}}'
+        in result.output
+    )
+    assert "info: Base picking finished successfully" in result.output
+    mocked_response.assert_called_once()
+
+
+def test_base_picking_command_slug_invalid(mocker):
+    token = uuid.uuid4()
+    runner = CliRunner()
+    result = runner.invoke(
+        pr_base_picking,
+        [
+            "-t",
+            token,
+            "--pr",
+            "11",
+            "--base-sha",
+            "9a6902ee94c18e8e27561ce316b16d75a02c7bc1",
+            "--service",
+            "github",
+            "--slug",
+            "owner-repo",
+        ],
+    )
+    assert result.exit_code == 0
+    assert (
+        "error: Slug is invalid. Slug should be in the form of owner_username/repo_name"
+        in result.output
+    )
+
+
+def test_base_picking_command_warnings(mocker):
+    mocked_response = mocker.patch(
+        "codecov_cli.commands.base_picking.send_put_request",
+        return_value=RequestResult(
+            status_code=200, error=None, warnings=["some random warning"], text=""
+        ),
+    )
+    token = uuid.uuid4()
+    runner = CliRunner()
+    result = runner.invoke(
+        pr_base_picking,
+        [
+            "-t",
+            token,
+            "--pr",
+            "11",
+            "--base-sha",
+            "9a6902ee94c18e8e27561ce316b16d75a02c7bc1",
+            "--service",
+            "github",
+            "--slug",
+            "owner/repo",
+        ],
+    )
+    assert result.exit_code == 0
+    assert (
+        f'debug: Starting base picking process --- {{"pr": "11", "slug": "owner/repo", "token": "{token}", "service": "github"}}'
+        in result.output
+    )
+    assert "info: Base picking process had 1 warning" in result.output
+    assert "Warning 1: some random warning" in result.output
+    assert "info: Base picking finished successfully" in result.output
+    mocked_response.assert_called_once()
+
+
+def test_base_picking_command_error(mocker):
+    mocked_response = mocker.patch(
+        "codecov_cli.commands.base_picking.send_put_request",
+        return_value=RequestResult(
+            status_code=401,
+            error=RequestError(
+                code="HTTP Error 401",
+                description="Unauthorized",
+                params={},
+            ),
+            warnings=[],
+            text="",
+        ),
+    )
+    token = uuid.uuid4()
+    runner = CliRunner()
+    result = runner.invoke(
+        pr_base_picking,
+        [
+            "-t",
+            token,
+            "--pr",
+            "11",
+            "--base-sha",
+            "9a6902ee94c18e8e27561ce316b16d75a02c7bc1",
+            "--service",
+            "github",
+            "--slug",
+            "owner/repo",
+        ],
+    )
+    mocked_response.assert_called_once()
+    print(result.output)
+    assert result.exit_code == 0
+    assert (
+        f'debug: Starting base picking process --- {{"pr": "11", "slug": "owner/repo", "token": "{token}", "service": "github"}}'
+        in result.output
+    )
+    assert "error: Base picking failed: Unauthorized" in result.output
+    assert "info: Base picking finished successfully" not in result.output

tests/test_codecov_cli.py

@@ -11,4 +11,5 @@ def test_existing_commands():
         "create-report",
         "do-upload",
         "get-report-results",
+        "pr-base-picking",
     ]