Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
44 changes: 44 additions & 0 deletions tests/unit/vertexai/genai/replays/test_skills_delete.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,44 @@
"""Tests the skills.delete() method against the autopush endpoint."""

from tests.unit.vertexai.genai.replays import pytest_helper
from vertexai._genai import types
from google.genai import errors
import pytest

pytestmark = pytest_helper.setup(
file=__file__,
globals_for_file=globals(),
)


def test_delete_skill(client, tmp_path):
# Target the autopush sandbox endpoint for the Skill Registry API
client._api_client._http_options.base_url = (
"https://us-central1-autopush-aiplatform.sandbox.googleapis.com"
)

# 1. Create a fresh unique skill first
with open(tmp_path / "SKILL.md", "w") as f:
f.write("# Test Skill\nTo be deleted.")

created_skill = client.skills.create(
display_name="To Be Deleted Skill",
description="Skill to be deleted",
config=types.CreateSkillConfig(
local_path=str(tmp_path), wait_for_completion=True
),
)

assert created_skill.name is not None

# 2. Delete the skill and wait for LRO completion
client.skills.delete(
name=created_skill.name,
config=types.DeleteSkillConfig(wait_for_completion=True),
)

# 3. Verify the skill is successfully deleted (Getting it should raise NotFound)
with pytest.raises(errors.ClientError) as exc_info:
client.skills.get(name=created_skill.name)

assert exc_info.value.code == 404
21 changes: 21 additions & 0 deletions tests/unit/vertexai/genai/replays/test_skills_list.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
"""Tests the skills.list() method against the autopush endpoint."""

from tests.unit.vertexai.genai.replays import pytest_helper
from vertexai._genai import types

pytestmark = pytest_helper.setup(
file=__file__,
globals_for_file=globals(),
)


def test_list_skills(client):
# Target the autopush sandbox endpoint for the Skill Registry API
client._api_client._http_options.base_url = (
"https://us-central1-autopush-aiplatform.sandbox.googleapis.com"
)

skills = client.skills.list()
for skill in skills:
assert isinstance(skill, types.Skill)
assert skill.name is not None
103 changes: 103 additions & 0 deletions tests/unit/vertexai/genai/replays/test_skills_update.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,103 @@
# 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 the skills.update() method against the Vertex AI endpoint using replays."""

import io
import os
import zipfile

from tests.unit.vertexai.genai.replays import pytest_helper
from vertexai._genai import types

# MANDATORY: Initialize the replay test framework for this module
pytestmark = pytest_helper.setup(
file=__file__,
globals_for_file=globals(),
)

PROJECT_ID = os.environ.get("GOOGLE_CLOUD_PROJECT", "srbai-testing")
REGION = "us-central1"


def test_update_skill(client, tmp_path):
# Target the autopush sandbox endpoint for the Skill Registry API
client._api_client._http_options.base_url = (
"https://us-central1-autopush-aiplatform.sandbox.googleapis.com"
)

# 1. Create a fresh unique skill first
with open(tmp_path / "SKILL.md", "w") as f:
f.write("# Test Skill\nInitial content.")

created_skill = client.skills.create(
display_name="Original Skill",
description="Original Description",
config=types.CreateSkillConfig(
local_path=str(tmp_path), wait_for_completion=True
),
)

# 2. Perform the metadata-only update on the new skill
updated_skill = client.skills.update(
name=created_skill.name,
config=types.UpdateSkillConfig(
display_name="My Updated Replay Skill",
description="My Updated Replay Skill Description",
wait_for_completion=True,
),
)

assert updated_skill.name == created_skill.name
assert updated_skill.display_name == "My Updated Replay Skill"
assert updated_skill.description == "My Updated Replay Skill Description"


def test_update_skill_with_zipped_bytes(client, tmp_path):
# Target the autopush sandbox endpoint for the Skill Registry API
client._api_client._http_options.base_url = (
"https://us-central1-autopush-aiplatform.sandbox.googleapis.com"
)

# 1. Create a fresh unique skill first
with open(tmp_path / "SKILL.md", "w") as f:
f.write("# Test Skill\nInitial content.")

created_skill = client.skills.create(
display_name="Original Skill",
description="Original Description",
config=types.CreateSkillConfig(
local_path=str(tmp_path), wait_for_completion=True
),
)

# 2. Prepare zipped bytes for update
zip_buffer = io.BytesIO()
zinfo = zipfile.ZipInfo("SKILL.md", date_time=(1980, 1, 1, 0, 0, 0))
with zipfile.ZipFile(zip_buffer, "w") as zip_file:
zip_file.writestr(zinfo, "# My Updated Zipped Replay Skill\nThis is updated.")
zipped_bytes = zip_buffer.getvalue()

# 3. Update the skill with new zipped bytes
updated_skill = client.skills.update(
name=created_skill.name,
config=types.UpdateSkillConfig(
zipped_filesystem=zipped_bytes, wait_for_completion=True
),
)

assert updated_skill.name == created_skill.name
assert (
updated_skill.display_name == "Original Skill"
) # Display name remains unchanged
Loading
Loading