chore: import upstream snapshot with attribution

tests/unittests/tools/bigquery/test_bigquery_skill.py

@@ -0,0 +1,116 @@
+# Copyright 2026 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Tests for the pre-packaged BigQuery skill."""
+
+from __future__ import annotations
+
+import re
+
+from google.adk.skills._utils import _validate_skill_dir
+from google.adk.tools.bigquery.bigquery_skill import _SKILL_DIR
+from google.adk.tools.bigquery.bigquery_skill import get_bigquery_skill
+from google.adk.tools.skill_toolset import ListSkillsTool
+from google.adk.tools.skill_toolset import LoadSkillResourceTool
+from google.adk.tools.skill_toolset import LoadSkillTool
+from google.adk.tools.skill_toolset import RunSkillScriptTool
+from google.adk.tools.skill_toolset import SkillToolset
+import pytest
+
+
+def test_get_bigquery_skill_returns_valid_skill():
+  """Verify get_bigquery_skill returns a Skill with expected fields."""
+  skill = get_bigquery_skill()
+
+  assert skill.name == "bigquery-ai-ml"
+  assert skill.description
+  assert len(skill.description) > 0
+  assert skill.instructions
+  assert len(skill.instructions) > 0
+
+
+def test_skill_name_matches_spec():
+  """Verify skill name is kebab-case and matches directory name."""
+  skill = get_bigquery_skill()
+
+  # Name must be kebab-case
+  assert re.fullmatch(r"[a-z][a-z0-9]*(-[a-z0-9]+)*", skill.name)
+
+  # Name must match the directory name
+  assert skill.name == _SKILL_DIR.name
+
+
+def test_skill_has_expected_references():
+  """Verify all expected reference files are present and non-empty."""
+  skill = get_bigquery_skill()
+
+  expected_refs = {
+      "bigquery_ai_classify.md",
+      "bigquery_ai_detect_anomalies.md",
+      "bigquery_ai_forecast.md",
+      "bigquery_ai_generate.md",
+      "bigquery_ai_generate_bool.md",
+      "bigquery_ai_generate_double.md",
+      "bigquery_ai_generate_int.md",
+      "bigquery_ai_if.md",
+      "bigquery_ai_score.md",
+      "bigquery_ai_search.md",
+      "bigquery_ai_similarity.md",
+  }
+  actual_refs = set(skill.resources.list_references())
+
+  assert expected_refs == actual_refs
+
+  for ref_name in expected_refs:
+    content = skill.resources.get_reference(ref_name)
+    assert content is not None, f"Reference {ref_name} returned None"
+    assert len(content) > 0, f"Reference {ref_name} is empty"
+
+
+@pytest.mark.asyncio
+async def test_skill_works_with_skill_toolset():
+  """Verify the skill integrates with SkillToolset and produces 4 tools."""
+  skill = get_bigquery_skill()
+  toolset = SkillToolset(skills=[skill])
+
+  tools = await toolset.get_tools()
+  assert len(tools) == 4
+
+  tool_types = {type(t) for t in tools}
+  expected_types = {
+      ListSkillsTool,
+      LoadSkillTool,
+      LoadSkillResourceTool,
+      RunSkillScriptTool,
+  }
+  assert tool_types == expected_types
+
+
+def test_skill_passes_validation():
+  """Verify the skill directory passes ADK's built-in validator."""
+  problems = _validate_skill_dir(_SKILL_DIR)
+  assert not problems, f"Validation problems: {problems}"
+
+
+def test_skill_frontmatter_has_license():
+  """Verify the skill includes a license field."""
+  skill = get_bigquery_skill()
+  assert skill.frontmatter.license == "Apache-2.0"
+
+
+def test_skill_frontmatter_has_metadata():
+  """Verify the skill includes author and version metadata."""
+  skill = get_bigquery_skill()
+  assert "author" in skill.frontmatter.metadata
+  assert "version" in skill.frontmatter.metadata