1852e6bcc3
Signed-off-by: minmin-intel <minmin.hou@intel.com> Signed-off-by: Rita Brugarolas <rita.brugarolas.brufau@intel.com> Signed-off-by: rbrugaro <rita.brugarolas.brufau@intel.com> Co-authored-by: rbrugaro <rita.brugarolas.brufau@intel.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: lkk <33276950+lkk12014402@users.noreply.github.com> Co-authored-by: lkk12014402 <kaokao.lv@intel.com>
71 lines
2.4 KiB
Python
71 lines
2.4 KiB
Python
# Copyright (C) 2025 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import json
|
|
import os
|
|
|
|
import requests
|
|
|
|
|
|
def test_html(url, link_list):
|
|
proxies = {"http": ""}
|
|
payload = {"link_list": json.dumps(link_list)}
|
|
try:
|
|
resp = requests.post(url=url, data=payload, proxies=proxies)
|
|
print(resp.text)
|
|
resp.raise_for_status() # Raise an exception for unsuccessful HTTP status codes
|
|
print("Request successful!")
|
|
except requests.exceptions.RequestException as e:
|
|
print("An error occurred:", e)
|
|
|
|
|
|
def test_delete(url, filename):
|
|
proxies = {"http": ""}
|
|
payload = {"file_path": filename}
|
|
try:
|
|
resp = requests.post(url=url, json=payload, proxies=proxies)
|
|
print(resp.text)
|
|
resp.raise_for_status() # Raise an exception for unsuccessful HTTP status codes
|
|
print("Request successful!")
|
|
except requests.exceptions.RequestException as e:
|
|
print("An error occurred:", e)
|
|
|
|
|
|
def test_get(url):
|
|
proxies = {"http": ""}
|
|
try:
|
|
resp = requests.post(url=url, proxies=proxies)
|
|
print(resp.text)
|
|
resp.raise_for_status() # Raise an exception for unsuccessful HTTP status codes
|
|
print("Request successful!")
|
|
except requests.exceptions.RequestException as e:
|
|
print("An error occurred:", e)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import argparse
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--test_option", type=str, default="ingest", help="ingest, get, delete")
|
|
parser.add_argument("--port", type=str, default="6007", help="port number")
|
|
args = parser.parse_args()
|
|
|
|
port = args.port
|
|
|
|
if args.test_option == "ingest":
|
|
url = f"http://localhost:{port}/v1/dataprep/ingest"
|
|
link_list = [
|
|
"https://www.fool.com/earnings/call-transcripts/2025/03/06/costco-wholesale-cost-q2-2025-earnings-call-transc/",
|
|
"https://www.fool.com/earnings/call-transcripts/2025/03/07/gap-gap-q4-2024-earnings-call-transcript/",
|
|
]
|
|
test_html(url, link_list)
|
|
elif args.test_option == "delete":
|
|
url = f"http://localhost:{port}/v1/dataprep/delete"
|
|
filename = "Costco Wholesale"
|
|
test_delete(url, filename)
|
|
elif args.test_option == "get":
|
|
url = f"http://localhost:{port}/v1/dataprep/get"
|
|
test_get(url)
|
|
else:
|
|
raise ValueError("Invalid test_option value. Please choose from ingest, get, delete.")
|