klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-07 21:29:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix tests to handle updated response types by API

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2022-07-20 03:01:56 +04:00
parent bdc1b9f2bb
commit c9ff97451b
2 changed files with 22 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
tests/test_client.py
View File

@@ -1,5 +1,6 @@
# Standard Modules # Standard Modules
from pathlib import Path from io import BytesIO
from PIL import Image
# External Packages # External Packages
from fastapi.testclient import TestClient from fastapi.testclient import TestClient
@@ -94,18 +95,18 @@ def test_image_search(content_config: ContentConfig, search_config: SearchConfig
config.content_type = content_config config.content_type = content_config
config.search_type = search_config config.search_type = search_config
model.image_search = image_search.setup(content_config.image, search_config.image, regenerate=False) model.image_search = image_search.setup(content_config.image, search_config.image, regenerate=False)
query_expected_image_pairs = [("brown kitten next to fallen plant", "/static/0.jpg"), # kitten_park.jpg query_expected_image_pairs = [("kitten", "kitten_park.jpg"),
("a horse and dog on a leash", "static/1.jpg"), # horse_dog.jpg ("a horse and dog on a leash", "horse_dog.jpg"),
("A guinea pig eating grass", "static/2.jpg")] # guineapig_grass.jpg ("A guinea pig eating grass", "guineapig_grass.jpg")]
# Act
for query, expected_image_name in query_expected_image_pairs: for query, expected_image_name in query_expected_image_pairs:
# Act
response = client.get(f"/search?q={query}&n=1&t=image") response = client.get(f"/search?q={query}&n=1&t=image")
# Assert # Assert
assert response.status_code == 200 assert response.status_code == 200
actual_image = Path(response.json()[0]["entry"]) actual_image = Image.open(BytesIO(client.get(response.json()[0]["entry"]).content))
expected_image = Path(expected_image_name) expected_image = Image.open(content_config.image.input_directories[0].joinpath(expected_image_name))
# Assert # Assert
assert expected_image == actual_image assert expected_image == actual_image
@@ -123,7 +124,7 @@ def test_notes_search(content_config: ContentConfig, search_config: SearchConfig
# Assert # Assert
assert response.status_code == 200 assert response.status_code == 200
# assert actual_data contains "Khoj via Emacs" entry # assert actual_data contains "Khoj via Emacs" entry
search_result = response.json()[0]["entry"] search_result = response.json()[0]["Entry"]
assert "git clone" in search_result assert "git clone" in search_result
@@ -139,7 +140,7 @@ def test_notes_search_with_include_filter(content_config: ContentConfig, search_
# Assert # Assert
assert response.status_code == 200 assert response.status_code == 200
# assert actual_data contains explicitly included word "Emacs" # assert actual_data contains explicitly included word "Emacs"
search_result = response.json()[0]["entry"] search_result = response.json()[0]["Entry"]
assert "Emacs" in search_result assert "Emacs" in search_result
@@ -155,5 +156,5 @@ def test_notes_search_with_exclude_filter(content_config: ContentConfig, search_
# Assert # Assert
assert response.status_code == 200 assert response.status_code == 200
# assert actual_data does not contains explicitly excluded word "Emacs" # assert actual_data does not contains explicitly excluded word "Emacs"
search_result = response.json()[0]["entry"] search_result = response.json()[0]["Entry"]
assert "clone" not in search_result assert "clone" not in search_result

16
tests/test_image_search.py
View File

@@ -1,8 +1,12 @@
# Standard Modules
from pathlib import Path
from PIL import Image
# External Packages # External Packages
import pytest import pytest
# Internal Packages # Internal Packages
from src.main import model from src.main import model, web_directory
from src.search_type import image_search from src.search_type import image_search
from src.utils.helpers import resolve_absolute_path from src.utils.helpers import resolve_absolute_path
from src.utils.rawconfig import ContentConfig, SearchConfig from src.utils.rawconfig import ContentConfig, SearchConfig
@@ -24,8 +28,9 @@ def test_image_search_setup(content_config: ContentConfig, search_config: Search
@pytest.mark.skip(reason="results inconsistent currently") @pytest.mark.skip(reason="results inconsistent currently")
def test_image_search(content_config: ContentConfig, search_config: SearchConfig): def test_image_search(content_config: ContentConfig, search_config: SearchConfig):
# Arrange # Arrange
output_directory = resolve_absolute_path(web_directory)
model.image_search = image_search.setup(content_config.image, search_config.image, regenerate=False) model.image_search = image_search.setup(content_config.image, search_config.image, regenerate=False)
query_expected_image_pairs = [("brown kitten next to plant", "kitten_park.jpg"), query_expected_image_pairs = [("kitten", "kitten_park.jpg"),
("horse and dog in a farm", "horse_dog.jpg"), ("horse and dog in a farm", "horse_dog.jpg"),
("A guinea pig eating grass", "guineapig_grass.jpg")] ("A guinea pig eating grass", "guineapig_grass.jpg")]
@@ -39,11 +44,12 @@ def test_image_search(content_config: ContentConfig, search_config: SearchConfig
results = image_search.collate_results( results = image_search.collate_results(
hits, hits,
model.image_search.image_names, model.image_search.image_names,
content_config.image.input_directory, output_directory=output_directory,
static_files_url='/static',
count=1) count=1)
actual_image = results[0]["Entry"] actual_image = Image.open(output_directory.joinpath(Path(results[0]["entry"]).name))
expected_image = resolve_absolute_path(content_config.image.input_directory.joinpath(expected_image_name)) expected_image = Image.open(content_config.image.input_directories[0].joinpath(expected_image_name))
# Assert # Assert
assert expected_image == actual_image assert expected_image == actual_image