code quality run

This commit is contained in:
2025-09-13 21:58:54 +02:00
parent 0dfd885847
commit 3eb18ef91e
63 changed files with 1802 additions and 1289 deletions

View File

@@ -1,8 +1,9 @@
# tests/test_export.py
import pytest
import csv
from io import StringIO
import pytest
from models.database_models import Product
@@ -15,7 +16,7 @@ class TestExportFunctionality:
assert response.headers["content-type"] == "text/csv; charset=utf-8"
# Parse CSV content
csv_content = response.content.decode('utf-8')
csv_content = response.content.decode("utf-8")
csv_reader = csv.reader(StringIO(csv_content))
# Check header row
@@ -35,10 +36,12 @@ class TestExportFunctionality:
db.add_all(products)
db.commit()
response = client.get("/api/v1/export-csv?marketplace=Amazon", headers=auth_headers)
response = client.get(
"/api/v1/export-csv?marketplace=Amazon", headers=auth_headers
)
assert response.status_code == 200
csv_content = response.content.decode('utf-8')
csv_content = response.content.decode("utf-8")
assert "EXP1" in csv_content
assert "EXP2" not in csv_content # Should be filtered out
@@ -50,7 +53,7 @@ class TestExportFunctionality:
product = Product(
product_id=f"PERF{i:04d}",
title=f"Performance Product {i}",
marketplace="Performance"
marketplace="Performance",
)
products.append(product)
@@ -58,10 +61,10 @@ class TestExportFunctionality:
db.commit()
import time
start_time = time.time()
response = client.get("/api/v1/export-csv", headers=auth_headers)
end_time = time.time()
assert response.status_code == 200
assert end_time - start_time < 10.0 # Should complete within 10 seconds