tinybird-cli 3.2.0__tar.gz → 3.2.1.dev2__tar.gz

{tinybird-cli-3.2.0 → tinybird-cli-3.2.1.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tinybird-cli
-Version: 3.2.0
+Version: 3.2.1.dev2
 Summary: Tinybird Command Line Tool
 Home-page: https://docs.tinybird.co/cli.html
 Author: Tinybird
@@ -19,6 +19,11 @@ Changelog
 
 ---------
 
+3.2.1.dev2
+************
+
+- `Added` Data quality tests inside a `skip_in_branch` folder inside the `tests` folder are skipped when running on a branch. Use this to skip tests in the CI piipeline that should only run with production data.
+
 3.2.0
 ************
 

{tinybird-cli-3.2.0 → tinybird-cli-3.2.1.dev2}/tinybird/__cli__.py

@@ -4,5 +4,5 @@ __description__ = 'Tinybird Command Line Tool'
 __url__ = 'https://docs.tinybird.co/cli.html'
 __author__ = 'Tinybird'
 __author_email__ = 'support@tinybird.co'
-__version__ = '3.2.0'
-__revision__ = '3067b41'
+__version__ = '3.2.1.dev2'
+__revision__ = '3862bf2'

{tinybird-cli-3.2.0 → tinybird-cli-3.2.1.dev2}/tinybird/tb_cli_modules/test.py

@@ -4,13 +4,13 @@
 # - But please, **do not** interleave utility functions and command definitions.
 
 import glob
-from typing import Iterable, List, Tuple
+from typing import Any, Dict, Iterable, List, Tuple
 
 import click
 
 from tinybird.feedback_manager import FeedbackManager
 from tinybird.tb_cli_modules.cli import cli
-from tinybird.tb_cli_modules.common import coro, create_tb_client, gather_with_concurrency
+from tinybird.tb_cli_modules.common import coro, create_tb_client, gather_with_concurrency, get_config_and_hosts
 from tinybird.tb_cli_modules.exceptions import CLIException
 from tinybird.tb_cli_modules.tinyunit.tinyunit import (
     TestSummaryResults,
@@ -27,7 +27,10 @@ def test(ctx: click.Context) -> None:
     """Test commands."""
 
 
-@test.command(name="run", help="Run the test suite, a file, or a test.")
+@test.command(
+    name="run",
+    help="Run the test suite, a file, or a test. To skip test to run in branches and CI put them in a 'skip_in_branch' folder.",
+)
 @click.argument("file", nargs=-1)
 @click.option("-v", "--verbose", is_flag=True, default=False, help="Enable verbose (show results)", type=bool)
 @click.option("--fail", "only_fail", is_flag=True, default=False, help="Showy onl failed/error tests", type=bool)
@@ -39,11 +42,24 @@ async def test_run(ctx: click.Context, file: Tuple[str, ...], verbose: bool, onl
 
     try:
         tb_client = create_tb_client(ctx)
+        config, _, _ = await get_config_and_hosts(ctx, check_token_exists=True)
+        workspaces: List[Dict[str, Any]] = (await tb_client.user_workspaces_and_branches()).get("workspaces", [])
+        current_ws: Dict[str, Any] = next(
+            (workspace for workspace in workspaces if config and workspace.get("id", ".") == config.get("id", "..")), {}
+        )
     except Exception as e:
         raise CLIException(FeedbackManager.error_exception(error=e))
 
     file_list: Iterable[str] = file if len(file) > 0 else glob.glob("./tests/**/*.y*ml", recursive=True)
+    click.echo(FeedbackManager.info_skipping_resource(resource="regression.yaml"))
     file_list = [f for f in file_list if not f.endswith("regression.yaml")]
+    final_file_list = []
+    for f in file_list:
+        if "skip_in_branch" in f and current_ws and current_ws.get("is_branch"):
+            click.echo(FeedbackManager.info_skipping_resource(resource=f))
+        else:
+            final_file_list.append(f)
+    file_list = final_file_list
 
     async def run_test(tb_client, test_file, results):
         try:

{tinybird-cli-3.2.0 → tinybird-cli-3.2.1.dev2}/tinybird_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tinybird-cli
-Version: 3.2.0
+Version: 3.2.1.dev2
 Summary: Tinybird Command Line Tool
 Home-page: https://docs.tinybird.co/cli.html
 Author: Tinybird
@@ -19,6 +19,11 @@ Changelog
 
 ---------
 
+3.2.1.dev2
+************
+
+- `Added` Data quality tests inside a `skip_in_branch` folder inside the `tests` folder are skipped when running on a branch. Use this to skip tests in the CI piipeline that should only run with production data.
+
 3.2.0
 ************