Add pytest for API (ed9a60b4) · Commits · David Guo / TagThunder

tagthunder/api/tests/init.py

0 → 100644

+0 −0

Empty file added.

0 → 100644

+91 −0

Changes for tagthunder/api/tests/test_routes.py: 91 added lines, 0 removed lines.
Original line number	Diff line number	Diff line
		import io
		import zipfile

		import pytest
		from fastapi.testclient import TestClient

		from api.main import app
		from api.routers.routes import Routes, queries, router

		HEADERS = {"X-Token": "coneofsilence"}

		client = TestClient(app)


		def test_augmentation():
		body = {
		"url": "https://www.example.com",
		"algorithm": {
		"name": "puppeteer",
		"parameters": {
		"page_width": 1200,
		"page_height": 1200,
		"styles": [
		"display",
		"visibility"
		]
		}
		}
		}

		response = client.post(
		url=f'{router.prefix}{Routes.AUGMENTATION}',
		headers=HEADERS,
		json=body
		)

		assert response.status_code == 200


		def test_pipeline():
		body = {
		# "url": "https://www.example.com",
		"url": "https://www.calvados.fr",
		"htmlpp": "",
		"recompute": False,
		"augmentation": {
		"name": "puppeteer",
		"parameters": {
		"page_width": 1200,
		"page_height": 1200,
		"styles": [
		"display",
		"visibility"
		]
		}
		},
		"cleaning": {
		"name": "vision_based",
		"parameters": {}
		},
		"segmentation": {
		"name": "TDBU",
		"parameters": {
		"nb_zones": 5
		}
		},
		"extraction": {
		"name": "yake",
		"parameters": {
		"nb_keywords": 5,
		"max_ngram_size": 4,
		"window_size": 3,
		"language": "fr"
		}
		},
		"vocalization": {
		"name": "playback_speed_area_size",
		"parameters": {
		"lang_detector": "google_translate",
		"tts_engine": "google_tts"
		}
		}
		}

		response = client.post(
		url=f'{router.prefix}{Routes.PIPELINE}',
		headers=HEADERS,
		json=body
		)

		assert response.status_code == 200