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 image search tests after update to API response for image search types

Browse Source 
- Look for 'entry' key in response json instead of 'Entry'
- Expect image where id = alphanumeric order of image name
This commit is contained in:
Debanjum Singh Solanky
2022-07-20 01:37:01 +04:00
parent 70221bb038
commit 68ee88cebc
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/test_client.py
View File

@@ -94,9 +94,9 @@ 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", "kitten_park.jpg"), query_expected_image_pairs = [("brown kitten next to fallen plant", "/static/0.jpg"), # kitten_park.jpg
("a horse and dog on a leash", "horse_dog.jpg"), ("a horse and dog on a leash", "static/1.jpg"), # horse_dog.jpg
("A guinea pig eating grass", "guineapig_grass.jpg")] ("A guinea pig eating grass", "static/2.jpg")] # guineapig_grass.jpg
# Act # Act
for query, expected_image_name in query_expected_image_pairs: for query, expected_image_name in query_expected_image_pairs:
@@ -104,8 +104,8 @@ def test_image_search(content_config: ContentConfig, search_config: SearchConfig
# Assert # Assert
assert response.status_code == 200 assert response.status_code == 200
actual_image = Path(response.json()[0]["Entry"]) actual_image = Path(response.json()[0]["entry"])
expected_image = resolve_absolute_path(content_config.image.input_directory.joinpath(expected_image_name)) expected_image = Path(expected_image_name)
# Assert # Assert
assert expected_image == actual_image assert expected_image == actual_image
@@ -123,7 +123,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 +139,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 +155,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