tinybird 0.0.1.dev273__py3-none-any.whl → 0.0.1.dev274__py3-none-any.whl

tinybird/tb/__cli__.py

@@ -4,5 +4,5 @@ __description__ = 'Tinybird Command Line Tool'
 __url__ = 'https://www.tinybird.co/docs/forward/commands'
 __author__ = 'Tinybird'
 __author_email__ = 'support@tinybird.co'
-__version__ = '0.0.1.dev273'
-__revision__ = '9f4ea9e'
+__version__ = '0.0.1.dev274'
+__revision__ = 'c3d62b9'

tinybird/tb/modules/agent/agent.py

@@ -704,9 +704,9 @@ def get_connection_datafile_local(config: dict[str, Any], connection_name: str)
         return "Connection not found"
 
 
-def run_tests(client: TinyB, project: Project, pipe_name: Optional[str] = None) -> None:
+def run_tests(client: TinyB, project: Project, pipe_name: Optional[str] = None) -> Optional[str]:
     try:
-        run_tests_common(name=(pipe_name,) if pipe_name else (), project=project, client=client)
+        return run_tests_common(name=(pipe_name,) if pipe_name else (), project=project, client=client)
     except SystemExit as e:
         raise Exception(e.args[0])
 

tinybird/tb/modules/agent/prompts.py

@@ -612,6 +612,7 @@ test_instructions = """
 - MANDATORY: Before creating the test, analyze the fixture files that the tables of the endpoint are using so you can create relevant tests.
 - IMPORTANT: expected_result field should always be an empty string, because it will be filled by the `create_test` tool.
 - If the endpoint does not have parameters, you can omit parameters and generate a single test.
+- If some tests are skipped, it is because some test names do not match any pipe name. Rename the test file to match the pipe name.
 - The format of the test file is the following:
 <test_file_format>
 - name: kpis_single_day

tinybird/tb/modules/agent/testing_agent.py

@@ -5,8 +5,7 @@ from pydantic_ai.usage import Usage
 from tinybird.tb.modules.agent.animations import ThinkingAnimation
 from tinybird.tb.modules.agent.models import create_model
 from tinybird.tb.modules.agent.prompts import test_instructions, tests_files_prompt, tone_and_style_instructions
-from tinybird.tb.modules.agent.tools.test import create_tests as create_tests_tool
-from tinybird.tb.modules.agent.tools.test import run_tests as run_tests_tool
+from tinybird.tb.modules.agent.tools.test import create_tests, remove_test, rename_test, run_tests
 from tinybird.tb.modules.agent.utils import TinybirdAgentContext
 from tinybird.tb.modules.project import Project
 
@@ -42,13 +41,17 @@ You can do the following:
 - Create new test files.
 - Update existing test files.
 - Run tests.
+- Rename test files.
+- Remove test files.
 """,
                 tone_and_style_instructions,
                 test_instructions,
             ],
             tools=[
-                Tool(create_tests_tool, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
-                Tool(run_tests_tool, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
+                Tool(create_tests, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
+                Tool(run_tests, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
+                Tool(rename_test, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
+                Tool(remove_test, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
             ],
         )
 

tinybird/tb/modules/agent/tools/test.py

@@ -1,3 +1,4 @@
+import os
 from pathlib import Path
 from typing import Optional
 
@@ -122,7 +123,6 @@ def run_tests(ctx: RunContext[TinybirdAgentContext], pipe_name: Optional[str] =
             return f"User did not confirm the proposed changes and gave the following feedback: {feedback}"
 
         test_output = ctx.deps.run_tests(pipe_name=pipe_name)
-        click.echo(test_output)
         ctx.deps.thinking_animation.start()
         if pipe_name:
             return f"Tests for '{pipe_name}' endpoint in {path} and ran successfully\n{test_output}"
@@ -146,3 +146,111 @@ def run_tests(ctx: RunContext[TinybirdAgentContext], pipe_name: Optional[str] =
             error_message = error_message.replace(test_exit_code, "")
         ctx.deps.thinking_animation.start()
         return f"Error running tests: {error_message}"
+
+
+def rename_test(ctx: RunContext[TinybirdAgentContext], path: str, new_path: str) -> str:
+    """Renames a test file.
+
+    Args:
+        path (str): The path to the test file to rename. Required.
+        new_path (str): The new path to the test file. Required.
+
+    Returns:
+        str: Result of the rename operation.
+    """
+    try:
+        ctx.deps.thinking_animation.stop()
+        confirmation = show_confirmation(
+            title=f"Rename '{path}' to '{new_path}'?",
+            skip_confirmation=ctx.deps.dangerously_skip_permissions,
+        )
+
+        if confirmation == "review":
+            feedback = show_input(ctx.deps.workspace_name)
+            ctx.deps.thinking_animation.start()
+            return f"User did not confirm the proposed changes and gave the following feedback: {feedback}"
+
+        click.echo(FeedbackManager.highlight(message=f"» Renaming test file {path} to {new_path}..."))
+        old_path_full = Path(ctx.deps.folder) / path.removeprefix("/")
+        new_path_full = Path(ctx.deps.folder) / new_path.removeprefix("/")
+
+        if not old_path_full.exists():
+            click.echo(FeedbackManager.error(message=f"Error: Test file {path} not found"))
+            ctx.deps.thinking_animation.start()
+            return f"Error: Test file {path} not found (double check the file path)"
+
+        if new_path_full.exists():
+            click.echo(FeedbackManager.error(message=f"Error: Test file {new_path} already exists"))
+            ctx.deps.thinking_animation.start()
+            return f"Error: Test file {new_path} already exists"
+
+        # Ensure new file has .yaml extension for test files
+        if new_path_full.suffix != ".yaml":
+            new_path_full = new_path_full.with_suffix(".yaml")
+            new_path = str(new_path_full.relative_to(ctx.deps.folder))
+
+        parent_path = new_path_full.parent
+        parent_path.mkdir(parents=True, exist_ok=True)
+        os.rename(old_path_full, new_path_full)
+
+        click.echo(FeedbackManager.success(message=f"✓ {new_path} created"))
+        ctx.deps.thinking_animation.start()
+        return f"Renamed test file from {path} to {new_path}"
+    except AgentRunCancelled as e:
+        raise e
+    except FileNotFoundError:
+        ctx.deps.thinking_animation.start()
+        click.echo(FeedbackManager.error(message=f"Error: Test file {path} not found"))
+        return f"Error: Test file {path} not found (double check the file path)"
+    except Exception as e:
+        ctx.deps.thinking_animation.stop()
+        click.echo(FeedbackManager.error(message=e))
+        ctx.deps.thinking_animation.start()
+        return f"Error renaming test file {path} to {new_path}: {e}"
+
+
+def remove_test(ctx: RunContext[TinybirdAgentContext], path: str) -> str:
+    """Removes a test file from the project folder
+
+    Args:
+        path (str): The path to the test file to remove. Required.
+
+    Returns:
+        str: If the test file was removed successfully.
+    """
+    try:
+        ctx.deps.thinking_animation.stop()
+        path = path.removeprefix("/")
+        full_path = Path(ctx.deps.folder) / path
+
+        if not full_path.exists():
+            click.echo(FeedbackManager.error(message=f"Error: Test file {path} not found"))
+            ctx.deps.thinking_animation.start()
+            return f"Error: Test file {path} not found (double check the file path)"
+
+        confirmation = show_confirmation(
+            title=f"Delete '{path}'?",
+            skip_confirmation=ctx.deps.dangerously_skip_permissions,
+        )
+
+        if confirmation == "review":
+            feedback = show_input(ctx.deps.workspace_name)
+            ctx.deps.thinking_animation.start()
+            return f"User did not confirm the proposed changes and gave the following feedback: {feedback}"
+
+        click.echo(FeedbackManager.highlight(message=f"» Removing {path}..."))
+
+        # Remove the test file
+        full_path.unlink()
+
+        click.echo(FeedbackManager.success(message=f"✓ {path} removed"))
+        ctx.deps.thinking_animation.start()
+
+        return f"Removed test file {path}."
+    except AgentRunCancelled as e:
+        raise e
+    except Exception as e:
+        ctx.deps.thinking_animation.stop()
+        click.echo(FeedbackManager.error(message=e))
+        ctx.deps.thinking_animation.start()
+        return f"Error removing test file {path}: {e}"

tinybird/tb/modules/test_common.py

@@ -188,7 +188,7 @@ def update_test(pipe: str, project: Project, client: TinyB) -> None:
         cleanup_test_workspace(client, project.folder)
 
 
-def run_tests(name: Tuple[str, ...], project: Project, client: TinyB) -> None:
+def run_tests(name: Tuple[str, ...], project: Project, client: TinyB) -> Optional[str]:
     full_error = ""
     try:
         load_secrets(project=project, client=client)
@@ -203,19 +203,30 @@ def run_tests(name: Tuple[str, ...], project: Project, client: TinyB) -> None:
             endpoints if len(endpoints) > 0 else glob.glob(f"{project.path}/tests/**/*.y*ml", recursive=True)
         )
 
-        def run_test(test_file) -> Optional[str]:
+        def run_test(test_file) -> Tuple[Optional[str], int, int, bool]:
             test_file_path = Path(test_file)
-            click.echo(FeedbackManager.info(message=f"* {test_file_path.stem}{test_file_path.suffix}"))
             test_file_content = parse_tests(test_file_path.read_text())
+            total_tests = len(test_file_content)
+
+            # Check if pipe exists before processing any tests
+            try:
+                client._req(f"/v0/pipes/{test_file_path.stem}")
+                click.echo(FeedbackManager.info(message=f"* {test_file_path.stem}{test_file_path.suffix}"))
+            except Exception:
+                # Entire test file skipped because pipe doesn't exist
+                click.echo(FeedbackManager.info(message=f"* {test_file_path.stem}{test_file_path.suffix}"))
+                click.echo(
+                    FeedbackManager.warning(message=f"✗ All tests skipped ({test_file_path.stem}.pipe not found)")
+                )
+                return None, total_tests, total_tests, True  # True indicates file was skipped
+
             test_file_errors = ""
+            skipped_count = 0
+
             for test in test_file_content:
                 try:
                     test_params = test["parameters"].split("?")[1] if "?" in test["parameters"] else test["parameters"]
-                    response = None
-                    try:
-                        response = get_pipe_data(client, pipe_name=test_file_path.stem, test_params=test_params)
-                    except Exception:
-                        continue
+                    response = get_pipe_data(client, pipe_name=test_file_path.stem, test_params=test_params)
 
                     expected_result = response.text
                     if response.status_code >= 400:
@@ -237,23 +248,45 @@ def run_tests(name: Tuple[str, ...], project: Project, client: TinyB) -> None:
                 except Exception as e:
                     test_file_errors += f"✗ {test['name']} failed\n** Output and expected output are different: \n{e}"
                     click.echo(FeedbackManager.error(message=test_file_errors))
-                    return test_file_errors
-            return None
+                    return test_file_errors, skipped_count, total_tests, False
+            return None, skipped_count, total_tests, False  # False indicates file was not skipped
 
         failed_tests_count = 0
+        skipped_files_count = 0
+        total_tests_count = 0
         test_count = len(test_files)
-
+        output = ""
         for test_file in test_files:
-            if run_test_error := run_test(test_file):
+            run_test_error, skipped_count, individual_tests_count, file_skipped = run_test(test_file)
+            total_tests_count += individual_tests_count
+
+            if file_skipped:
+                skipped_files_count += 1
+            elif run_test_error:
                 full_error += f"\n{run_test_error}"
                 failed_tests_count += 1
 
+        runnable_files_count = test_count - skipped_files_count
+        passed_files_count = runnable_files_count - failed_tests_count
+
         if failed_tests_count:
-            error = f"\n✗ {test_count - failed_tests_count}/{test_count} passed"
+            error = f"\n✗ {passed_files_count}/{runnable_files_count} passed"
+            if skipped_files_count > 0:
+                error += f" ({skipped_files_count} skipped)"
+            output += f"{error}\n"
             click.echo(FeedbackManager.error(message=error))
             sys_exit("test_error", full_error)
         else:
-            click.echo(FeedbackManager.success(message=f"\n✓ {test_count}/{test_count} passed"))
+            if runnable_files_count == 0:
+                success_message = "\n✓ No tests to run"
+            else:
+                success_message = f"\n✓ {runnable_files_count}/{runnable_files_count} passed"
+            if skipped_files_count > 0:
+                success_message += f" ({skipped_files_count} skipped)"
+            message_color = FeedbackManager.success if runnable_files_count > 0 else FeedbackManager.warning
+            click.echo(message_color(message=success_message))
+            output += f"{success_message}\n"
+        return output
     except Exception as e:
         raise CLITestException(FeedbackManager.error(message=str(e)))
     finally:

{tinybird-0.0.1.dev273.dist-info → tinybird-0.0.1.dev274.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 0.0.1.dev273
+Version: 0.0.1.dev274
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/forward/commands
 Author: Tinybird

{tinybird-0.0.1.dev273.dist-info → tinybird-0.0.1.dev274.dist-info}/RECORD

@@ -17,7 +17,7 @@ tinybird/datafile/exceptions.py,sha256=8rw2umdZjtby85QbuRKFO5ETz_eRHwUY5l7eHsy1w
 tinybird/datafile/parse_connection.py,sha256=tRyn2Rpr1TeWet5BXmMoQgaotbGdYep1qiTak_OqC5E,1825
 tinybird/datafile/parse_datasource.py,sha256=ssW8QeFSgglVFi3sDZj_HgkJiTJ2069v2JgqnH3CkDE,1825
 tinybird/datafile/parse_pipe.py,sha256=xf4m0Tw44QWJzHzAm7Z7FwUoUUtr7noMYjU1NiWnX0k,3880
-tinybird/tb/__cli__.py,sha256=kp7873OcOrpDdwPcuqes22hVltt_E98SAn-s_yb-kQ4,247
+tinybird/tb/__cli__.py,sha256=Ua6pj01kXp3GMMZoliQZuqwIh1eCDW2n3F6DKULSfMk,247
 tinybird/tb/check_pypi.py,sha256=Gp0HkHHDFMSDL6nxKlOY51z7z1Uv-2LRexNTZSHHGmM,552
 tinybird/tb/cli.py,sha256=FdDFEIayjmsZEVsVSSvRiVYn_FHOVg_zWQzchnzfWho,1008
 tinybird/tb/client.py,sha256=IQRaInDjOwr9Fzaz3_xXc3aUGqh94tM2lew7IZbB9eM,53733
@@ -63,13 +63,13 @@ tinybird/tb/modules/sink.py,sha256=dK2s__my0ePIUYrqBzhPSgdWN9rbpvP1G4dT7DJzz80,3
 tinybird/tb/modules/table.py,sha256=4XrtjM-N0zfNtxVkbvLDQQazno1EPXnxTyo7llivfXk,11035
 tinybird/tb/modules/telemetry.py,sha256=T9gtsQffWqG_4hRBaUJPzOfMkPwz7mH-R6Bn1XRYViA,11482
 tinybird/tb/modules/test.py,sha256=O2-mS4uMU6nPi7yWPpWzshAgOlYKiGS-tkM12pXQGMI,1906
-tinybird/tb/modules/test_common.py,sha256=YZwAdSfYVXdvArfTc9tH-2QBOhb_XbnJ3eKvyXTJuEM,12717
+tinybird/tb/modules/test_common.py,sha256=kZ503FFXZwUkZ8zCGZKjBXmRKfF6ooqohkuFhzNRX48,14442
 tinybird/tb/modules/token.py,sha256=ZhW_o7XCr90wJRhMN6816vyo_TVfnzPXyIhrhzQ7oZ0,13807
 tinybird/tb/modules/watch.py,sha256=No0bK1M1_3CYuMaIgylxf7vYFJ72lTJe3brz6xQ-mJo,8819
 tinybird/tb/modules/workspace.py,sha256=Q_8HcxMsNg8QG9aBlwcWS2umrDP5IkTIHqqz3sfmGuc,11341
 tinybird/tb/modules/workspace_members.py,sha256=5JdkJgfuEwbq-t6vxkBhYwgsiTDxF790wsa6Xfif9nk,8608
 tinybird/tb/modules/agent/__init__.py,sha256=i3oe3vDIWWPaicdCM0zs7D7BJ1W0k7th93ooskHAV00,54
-tinybird/tb/modules/agent/agent.py,sha256=f0j0GScN3wdLSWRrxyjqlyPf1bTHgmJm62sM3XAAzQ8,31080
+tinybird/tb/modules/agent/agent.py,sha256=MjLpl2uNASKJiKH74U9HgvQk4h8gxii5u2KR1nJAXx4,31096
 tinybird/tb/modules/agent/animations.py,sha256=4WOC5_2BracttmMCrV0H91tXfWcUzQHBUaIJc5FA7tE,3490
 tinybird/tb/modules/agent/banner.py,sha256=l6cO5Fi7lbVKp-GsBP8jf3IkjOWxg2jpAt9NBCy0WR8,4085
 tinybird/tb/modules/agent/command_agent.py,sha256=NTzgb9qnuG-gDpk87VijKs1UUMukJPaJI5UiZtRWUoQ,2864
@@ -77,8 +77,8 @@ tinybird/tb/modules/agent/compactor.py,sha256=BK5AxZFhrp3xWnsRnYaleiYoIWtVNc-_m6
 tinybird/tb/modules/agent/explore_agent.py,sha256=HkzKmggfSMz7S3RSeKnZXufq-z_U0tTQJpF7JfNIaGQ,3504
 tinybird/tb/modules/agent/memory.py,sha256=vBewB_64L_wHoT4tLT6UX2uxcHwSY880QZ26F9rPqXs,3793
 tinybird/tb/modules/agent/models.py,sha256=IAxqlnHy8c2OeSnDrrSp2Mg38W9_r0GsDM87Wv-YNfM,925
-tinybird/tb/modules/agent/prompts.py,sha256=gIwYbfym7ydZIQQF0Xe4Nri4g-cyKlvR8UWUD8n-6QA,34934
-tinybird/tb/modules/agent/testing_agent.py,sha256=mjR9OJ_KzGnjCnjRxp5-sf01LSDkO4-QqknTbkZ-R-Q,2752
+tinybird/tb/modules/agent/prompts.py,sha256=PZ-WymaU1Dni-jh_xk8P_DO4mRtSN7Bv7tfoZ89nzJs,35066
+tinybird/tb/modules/agent/testing_agent.py,sha256=AtwtJViH7805i7djyBgDb7SSUtDyJnw0TWJu6lBFsrg,2953
 tinybird/tb/modules/agent/utils.py,sha256=U1s6_tGynArRA42GvDr5qz2acxVVKfO8NwBLMngyqSI,31788
 tinybird/tb/modules/agent/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tinybird/tb/modules/agent/tools/analyze.py,sha256=CR5LXg4fou-zYEksqnjpJ0icvxJVoKnTctoI1NRvqCM,3873
@@ -96,7 +96,7 @@ tinybird/tb/modules/agent/tools/plan.py,sha256=2KHLNkr2f1RfkbAR4mCVsv94LGosXd8-k
 tinybird/tb/modules/agent/tools/request_endpoint.py,sha256=xseEDQez2xfnPWNOoGnRmHB2KR9WLCx_q-vzS6NtaOY,3972
 tinybird/tb/modules/agent/tools/run_command.py,sha256=XjPDTTzkba9GOQBDiSTwddluyXkguVhxvXnRaC8m-Zc,1657
 tinybird/tb/modules/agent/tools/secret.py,sha256=UbF9YIW4zh5qdF7qCeMhbhsDt_2qdjjntJE1e8HSUG0,4292
-tinybird/tb/modules/agent/tools/test.py,sha256=CbGak_coopCTtqHoPWy-BwgLMIyEyeO34NTNkv18au4,6041
+tinybird/tb/modules/agent/tools/test.py,sha256=4XuEWVHLOTSO51Z9xJ08dTjk0j3IWY_JlPtSBO5aaUs,10373
 tinybird/tb/modules/datafile/build.py,sha256=NFKBrusFLU0WJNCXePAFWiEDuTaXpwc0lHlOQWEJ43s,51117
 tinybird/tb/modules/datafile/build_common.py,sha256=2yNdxe49IMA9wNvl25NemY2Iaz8L66snjOdT64dm1is,4511
 tinybird/tb/modules/datafile/build_datasource.py,sha256=Ra8pVQBDafbFRUKlhpgohhTsRyp_ADKZJVG8Gd69idY,17227
@@ -117,8 +117,8 @@ tinybird/tb_cli_modules/config.py,sha256=IsgdtFRnUrkY8-Zo32lmk6O7u3bHie1QCxLwgp4
 tinybird/tb_cli_modules/exceptions.py,sha256=pmucP4kTF4irIt7dXiG-FcnI-o3mvDusPmch1L8RCWk,3367
 tinybird/tb_cli_modules/regions.py,sha256=QjsL5H6Kg-qr0aYVLrvb1STeJ5Sx_sjvbOYO0LrEGMk,166
 tinybird/tb_cli_modules/telemetry.py,sha256=Hh2Io8ZPROSunbOLuMvuIFU4TqwWPmQTqal4WS09K1A,10449
-tinybird-0.0.1.dev273.dist-info/METADATA,sha256=axU5LHrxl4jCoiSzp8el1blBj3BGsqYMzLqtKmXXt_0,1763
-tinybird-0.0.1.dev273.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev273.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev273.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev273.dist-info/RECORD,,
+tinybird-0.0.1.dev274.dist-info/METADATA,sha256=AQKOg2W-tA_h-i8h1vfz2fzxThxcaz-LKFJmBB-KJXU,1763
+tinybird-0.0.1.dev274.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev274.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev274.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev274.dist-info/RECORD,,