# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. ========= # 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. # ========= Copyright 2026 @ Strukto.AI All Rights Reserved. ========= from mirage.resource.gsheets.prompt import PROMPT, WRITE_PROMPT def test_prompt_includes_buckets_and_structure(): rendered = PROMPT.format(prefix="/gsheets") assert "owned/" in rendered assert "shared/" in rendered assert "shared with you by others" in rendered assert "still in owned/" in rendered assert "gsheet.json structure" in rendered assert ".sheets[].properties.title" in rendered assert "gws-sheets-read" in rendered def test_write_prompt_examples_match_actual_signatures(): assert "gws-sheets-write" in WRITE_PROMPT assert "--params" in WRITE_PROMPT assert "--json" in WRITE_PROMPT assert "spreadsheetId" in WRITE_PROMPT assert "valueInputOption" in WRITE_PROMPT assert "gws-sheets-append" in WRITE_PROMPT assert "--spreadsheet" in WRITE_PROMPT assert "--range" in WRITE_PROMPT assert "--values" in WRITE_PROMPT assert "--json-values" in WRITE_PROMPT assert "gws-sheets-spreadsheets-create" in WRITE_PROMPT assert '{"properties": {"title":' in WRITE_PROMPT assert "gws-sheets-spreadsheets-batchUpdate" in WRITE_PROMPT assert "spreadsheetId" in WRITE_PROMPT def test_write_prompt_documents_rm(): assert "rm " in WRITE_PROMPT assert ".gsheet.json" in WRITE_PROMPT