personal memory agent
0
fork

Configure Feed

Select the types of activity you want to include in your feed.

Remove old vertex-credentials tests (endpoint removed in unified providers merge)

Tests referenced the removed /api/vertex-credentials POST/DELETE endpoint
and the old vertex_credentials config path. The unified providers API
handles vertex credentials through PUT /api/providers now.

-143
-55
tests/test_cli_provider.py
··· 741 741 assert "GOOGLE_GENAI_USE_VERTEXAI" not in env 742 742 assert env["ANTHROPIC_API_KEY"] == "sk-ant" 743 743 744 - def test_vertex_service_account_sets_gac(self): 745 - """Vertex + vertex_credentials sets GOOGLE_APPLICATION_CREDENTIALS.""" 746 - config = { 747 - "providers": { 748 - "google_backend": "vertex", 749 - "vertex_credentials": "/path/to/creds.json", 750 - "vertex_project": "my-project", 751 - "vertex_location": "us-central1", 752 - "auth": {"google": "platform"}, 753 - } 754 - } 755 - with ( 756 - patch.dict(os.environ, {"GOOGLE_API_KEY": "gk-test"}, clear=True), 757 - patch("think.utils.get_config", return_value=config), 758 - ): 759 - env = build_cogitate_env("GOOGLE_API_KEY") 760 - assert env["GOOGLE_GENAI_USE_VERTEXAI"] == "true" 761 - assert env["GOOGLE_APPLICATION_CREDENTIALS"] == "/path/to/creds.json" 762 - assert env["GOOGLE_CLOUD_PROJECT"] == "my-project" 763 - assert env["GOOGLE_CLOUD_LOCATION"] == "us-central1" 764 - assert "GOOGLE_API_KEY" not in env 765 - 766 - def test_vertex_service_account_no_api_key(self): 767 - """Vertex + service account works even without GOOGLE_API_KEY in env.""" 768 - config = { 769 - "providers": { 770 - "google_backend": "vertex", 771 - "vertex_credentials": "/path/to/creds.json", 772 - } 773 - } 774 - with ( 775 - patch.dict(os.environ, {}, clear=True), 776 - patch("think.utils.get_config", return_value=config), 777 - ): 778 - env = build_cogitate_env("GOOGLE_API_KEY") 779 - assert env["GOOGLE_GENAI_USE_VERTEXAI"] == "true" 780 - assert env["GOOGLE_APPLICATION_CREDENTIALS"] == "/path/to/creds.json" 781 - assert "GOOGLE_API_KEY" not in env 782 - 783 - def test_vertex_no_credentials_preserves_api_key(self): 784 - """Vertex without vertex_credentials preserves API key behavior.""" 785 - config = { 786 - "providers": { 787 - "google_backend": "vertex", 788 - "auth": {"google": "api_key"}, 789 - } 790 - } 791 - with ( 792 - patch.dict(os.environ, {"GOOGLE_API_KEY": "gk-test"}, clear=True), 793 - patch("think.utils.get_config", return_value=config), 794 - ): 795 - env = build_cogitate_env("GOOGLE_API_KEY") 796 - assert env["GOOGLE_GENAI_USE_VERTEXAI"] == "true" 797 - assert env["GOOGLE_API_KEY"] == "gk-test" 798 - assert "GOOGLE_APPLICATION_CREDENTIALS" not in env
-88
tests/test_validate_key.py
··· 477 477 assert "Invalid google_backend" in response.get_json()["error"] 478 478 479 479 480 - def test_vertex_credentials_upload_and_delete(settings_client): 481 - """POST/DELETE vertex-credentials saves and removes credentials file.""" 482 - client, journal = settings_client 483 - 484 - creds = { 485 - "type": "service_account", 486 - "project_id": "test-project", 487 - "client_email": "test@test-project.iam.gserviceaccount.com", 488 - "private_key": "-----BEGIN RSA PRIVATE KEY-----\ntest\n-----END RSA PRIVATE KEY-----\n", 489 - } 490 - 491 - # Upload 492 - response = client.post( 493 - "/app/settings/api/vertex-credentials", 494 - json={"credentials": creds}, 495 - ) 496 - assert response.status_code == 200 497 - payload = response.get_json() 498 - assert payload["success"] is True 499 - assert payload["client_email"] == "test@test-project.iam.gserviceaccount.com" 500 - 501 - # Verify file saved 502 - creds_path = journal / "config" / "vertex-credentials.json" 503 - assert creds_path.exists() 504 - saved = json.loads(creds_path.read_text()) 505 - assert saved["client_email"] == "test@test-project.iam.gserviceaccount.com" 506 - assert oct(creds_path.stat().st_mode)[-3:] == "600" 507 - 508 - # Verify config updated 509 - config = json.loads((journal / "config" / "journal.json").read_text()) 510 - assert Path(config["providers"]["vertex_credentials"]) == creds_path 511 - 512 - # GET providers returns email 513 - response = client.get("/app/settings/api/providers") 514 - payload = response.get_json() 515 - assert payload["vertex_credentials_email"] == "test@test-project.iam.gserviceaccount.com" 516 - 517 - # Delete 518 - response = client.delete("/app/settings/api/vertex-credentials") 519 - assert response.status_code == 200 520 - assert response.get_json()["success"] is True 521 - assert not creds_path.exists() 522 - 523 - config = json.loads((journal / "config" / "journal.json").read_text()) 524 - assert "vertex_credentials" not in config.get("providers", {}) 525 - 526 - # GET providers returns null email 527 - response = client.get("/app/settings/api/providers") 528 - payload = response.get_json() 529 - assert payload["vertex_credentials_email"] is None 530 - 531 - 532 - def test_vertex_credentials_upload_validation(settings_client): 533 - """POST vertex-credentials rejects invalid credentials JSON.""" 534 - client, _journal = settings_client 535 - 536 - # Missing required fields 537 - response = client.post( 538 - "/app/settings/api/vertex-credentials", 539 - json={"credentials": {"type": "service_account"}}, 540 - ) 541 - assert response.status_code == 400 542 - assert "Missing required fields" in response.get_json()["error"] 543 - 544 - # Wrong type 545 - response = client.post( 546 - "/app/settings/api/vertex-credentials", 547 - json={ 548 - "credentials": { 549 - "type": "authorized_user", 550 - "project_id": "p", 551 - "client_email": "e", 552 - "private_key": "k", 553 - } 554 - }, 555 - ) 556 - assert response.status_code == 400 557 - assert "service_account" in response.get_json()["error"] 558 - 559 - # Non-dict JSON (e.g., array) 560 - response = client.post( 561 - "/app/settings/api/vertex-credentials", 562 - json={"credentials": ["not", "a", "dict"]}, 563 - ) 564 - assert response.status_code == 400 565 - assert "JSON object" in response.get_json()["error"] 566 - 567 - 568 480 def test_validate_all_keys_with_vertex_credentials(settings_client): 569 481 """validate-all-keys validates vertex credentials when configured.""" 570 482 client, journal = settings_client