This is an automated email from the ASF dual-hosted git repository.
weilee pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new 7904828f05 add delete index teardown to pinecone system tests (#40396)
7904828f05 is described below
commit 7904828f053cfb9a22275bfa35cf7f6a84e8a1eb
Author: vatsrahul1001 <[email protected]>
AuthorDate: Mon Jun 24 15:30:58 2024 +0530
add delete index teardown to pinecone system tests (#40396)
---
tests/system/providers/pinecone/example_create_pod_index.py | 13 ++++++++++++-
.../providers/pinecone/example_create_serverless_index.py | 13 ++++++++++++-
2 files changed, 24 insertions(+), 2 deletions(-)
diff --git a/tests/system/providers/pinecone/example_create_pod_index.py
b/tests/system/providers/pinecone/example_create_pod_index.py
index 9b6f7d7d88..a2d7f16c69 100644
--- a/tests/system/providers/pinecone/example_create_pod_index.py
+++ b/tests/system/providers/pinecone/example_create_pod_index.py
@@ -20,6 +20,7 @@ import os
from datetime import datetime
from airflow import DAG
+from airflow.decorators import task, teardown
from airflow.providers.pinecone.operators.pinecone import
CreatePodIndexOperator
index_name = os.getenv("INDEX_NAME", "test")
@@ -33,7 +34,7 @@ with DAG(
) as dag:
# [START howto_operator_create_pod_index]
# reference:
https://docs.pinecone.io/reference/api/control-plane/create_index
- CreatePodIndexOperator(
+ create_index = CreatePodIndexOperator(
task_id="pinecone_create_pod_index",
index_name=index_name,
dimension=3,
@@ -44,6 +45,16 @@ with DAG(
)
# [END howto_operator_create_pod_index]
+ @teardown
+ @task
+ def delete_index():
+ from airflow.providers.pinecone.hooks.pinecone import PineconeHook
+
+ hook = PineconeHook()
+ hook.delete_index(index_name=index_name)
+
+ create_index >> delete_index()
+
from tests.system.utils import get_test_run # noqa: E402
diff --git a/tests/system/providers/pinecone/example_create_serverless_index.py
b/tests/system/providers/pinecone/example_create_serverless_index.py
index a7924e63ef..cf1e2c5cee 100644
--- a/tests/system/providers/pinecone/example_create_serverless_index.py
+++ b/tests/system/providers/pinecone/example_create_serverless_index.py
@@ -20,6 +20,7 @@ import os
from datetime import datetime
from airflow import DAG
+from airflow.decorators import task, teardown
from airflow.providers.pinecone.operators.pinecone import
CreateServerlessIndexOperator
index_name = os.getenv("INDEX_NAME", "test")
@@ -33,7 +34,7 @@ with DAG(
) as dag:
# [START howto_operator_create_serverless_index]
# reference:
https://docs.pinecone.io/reference/api/control-plane/create_index
- CreateServerlessIndexOperator(
+ create_index = CreateServerlessIndexOperator(
task_id="pinecone_create_serverless_index",
index_name=index_name,
dimension=128,
@@ -43,6 +44,16 @@ with DAG(
)
# [END howto_operator_create_serverless_index]
+ @teardown
+ @task
+ def delete_index():
+ from airflow.providers.pinecone.hooks.pinecone import PineconeHook
+
+ hook = PineconeHook()
+ hook.delete_index(index_name=index_name)
+
+ create_index >> delete_index()
+
from tests.system.utils import get_test_run # noqa: E402
