[chore] fix CI errors

Author: Konboi

Pipfile

@@ -22,6 +22,7 @@ mypy = "<1.16.0"
 pre-commit = "*"
 responses = "*"
 types-click = "*"
+types-dataclasses = {markers = "python_version < '3.7'"}
 types-pkg_resources = "0.1.3"
 types-python-dateutil = "*"
 types-requests = "*"

launchable/commands/compare/subsets.py

@@ -1,20 +1,14 @@
+from dataclasses import dataclass
 from http import HTTPStatus
 from pathlib import Path
-from typing import Any, Generic, List, Optional, Sequence, Tuple, TypeVar, Union
+from typing import Any, Dict, Generic, List, Optional, Sequence, Tuple, TypeVar, Union
 
 import click
 from tabulate import tabulate
 
-from build.lib.launchable.testpath import unparse_test_path
+from launchable.testpath import unparse_test_path
 from launchable.utils.launchable_client import LaunchableClient
 
-try:
-    # for from 3.7
-    from dataclasses import dataclass
-except ImportError:
-    # for 3.6
-    from dataclasses import dataclass
-
 
 @dataclass(frozen=True)
 class SubsetResultBase:
@@ -29,7 +23,7 @@ class SubsetResult(SubsetResultBase):
     duration_sec: float
 
     @classmethod
-    def from_inspect_api(cls, result: dict[str, Any], order: int) -> "SubsetResult":
+    def from_inspect_api(cls, result: Dict[str, Any], order: int) -> "SubsetResult":
         test_path = result.get("testPath", []) or []
         name = unparse_test_path(test_path)
         density = float(result.get("density") or 0.0)
@@ -93,8 +87,8 @@ def load(cls, client: LaunchableClient, subset_id: int) -> "SubsetResults":
 
 
 @click.command()
-@click.argument('file_before', type=click.Path(exists=True))
-@click.argument('file_after', type=click.Path(exists=True))
+@click.argument('file_before', type=click.Path(exists=True), required=False)
+@click.argument('file_after', type=click.Path(exists=True), required=False)
 @click.option(
     '--subset-id-before',
     'subset_id_before',
@@ -107,6 +101,7 @@ def load(cls, client: LaunchableClient, subset_id: int) -> "SubsetResults":
     type=int,
     help='Subset ID for the second subset to compare',
     metavar="SUBSET_ID")
+@click.pass_context
 def subsets(context: click.core.Context, file_before, file_after, subset_id_before, subset_id_after):
     """Compare subsets sourced from files or remote subset IDs."""
 
@@ -125,7 +120,7 @@ def subsets(context: click.core.Context, file_before, file_after, subset_id_befo
 
     if from_subset_id:
 
-        client = LaunchableClient(app=context)
+        client = LaunchableClient(app=context.obj)
         # for type check
         assert subset_id_before is not None and subset_id_after is not None
         _from_subset_ids(client=client, subset_id_before=subset_id_before, subset_id_after=subset_id_after)
@@ -152,7 +147,7 @@ def _from_subset_ids(client: LaunchableClient, subset_id_before: int, subset_id_
     for result in after_subset.results:
         total += 1
         if result.reason.startswith("Changed file: "):
-            affected.add(result.reason.removeprefix("Changed file: "))
+            affected.add(result.reason[len("Changed file: "):])
 
         test_name = result.name
         after_order = result.order

setup.cfg

@@ -17,7 +17,7 @@ packages = find:
 install_requires =
     click>=8.0,<8.1;python_version=='3.6'
     click>=8.1,<8.2;python_version>'3.6'
-    dataclass
+    dataclasses;python_version=='3.6'
     requests>=2.25;python_version>='3.6'
     urllib3>=1.26
     junitparser>=4.0.0

tests/commands/compare/test_subsets.py

@@ -3,7 +3,7 @@
 
 import responses
 
-from smart_tests.utils.http_client import get_base_url
+from launchable.utils.http_client import get_base_url
 from tests.cli_test_case import CliTestCase
 
 
@@ -157,7 +157,7 @@ def tearDown(self):
         if os.path.exists("subset-after.txt"):
             os.remove("subset-after.txt")
 
-    @mock.patch.dict(os.environ, {"SMART_TESTS_TOKEN": CliTestCase.smart_tests_token})
+    @mock.patch.dict(os.environ, {"LAUNCHABLE_TOKEN": CliTestCase.launchable_token})
     @responses.activate
     def test_subsets_subset_ids(self):
         responses.add(