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

Update the search_config instantiated for tests in conftest

Browse Source
This commit is contained in:
Saba
2021-12-11 14:14:31 -05:00
parent 9ebf00e29b
commit ba8dc9ed5f
2 changed files with 20 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
tests/test_client.py
View File

@@ -8,13 +8,12 @@ from fastapi.testclient import TestClient
from src.main import app, model, config
from src.search_type import asymmetric, image_search
from src.utils.helpers import resolve_absolute_path
from src.utils.rawconfig import FullConfig
from src.utils.rawconfig import ContentTypeConfig
# Arrange
# ----------------------------------------------------------------------------------------------------
client = TestClient(app)
config = FullConfig()
# Test
# ----------------------------------------------------------------------------------------------------
@@ -30,9 +29,10 @@ def test_search_with_invalid_search_type():
# ----------------------------------------------------------------------------------------------------
def test_search_with_valid_search_type(search_config):
def test_search_with_valid_search_type(search_config: ContentTypeConfig):
# Arrange
config.content_type.image = search_config.image
config.content_type = search_config
# config.content_type.image = search_config.image
for search_type in ["notes", "ledger", "music", "image"]:
# Act
response = client.get(f"/search?q=random&t={search_type}")
@@ -50,9 +50,9 @@ def test_regenerate_with_invalid_search_type():
# ----------------------------------------------------------------------------------------------------
def test_regenerate_with_valid_search_type(search_config):
def test_regenerate_with_valid_search_type(search_config: ContentTypeConfig):
# Arrange
config.content_type.image = search_config.image
config.content_type = search_config
for search_type in ["notes", "ledger", "music", "image"]:
# Act
response = client.get(f"/regenerate?t={search_type}")
@@ -61,9 +61,9 @@ def test_regenerate_with_valid_search_type(search_config):
# ----------------------------------------------------------------------------------------------------
def test_image_search(search_config):
def test_image_search(search_config: ContentTypeConfig):
# Arrange
config.content_type.image = search_config.image
config.content_type = search_config
model.image_search = image_search.setup(search_config.image, regenerate=False)
query_expected_image_pairs = [("brown kitten next to fallen plant", "kitten_park.jpg"),
("a horse and dog on a leash", "horse_dog.jpg"),
@@ -83,9 +83,9 @@ def test_image_search(search_config):
# ----------------------------------------------------------------------------------------------------
def test_notes_search(search_config):
def test_notes_search(search_config: ContentTypeConfig):
# Arrange
model.notes_search = asymmetric.setup(search_config.notes, regenerate=False)
model.notes_search = asymmetric.setup(search_config.org, regenerate=False)
user_query = "How to git install application?"
# Act
@@ -99,9 +99,9 @@ def test_notes_search(search_config):
# ----------------------------------------------------------------------------------------------------
def test_notes_search_with_include_filter(search_config):
def test_notes_search_with_include_filter(search_config: ContentTypeConfig):
# Arrange
model.notes_search = asymmetric.setup(search_config.notes, regenerate=False)
model.notes_search = asymmetric.setup(search_config.org, regenerate=False)
user_query = "How to git install application? +Emacs"
# Act
@@ -115,9 +115,9 @@ def test_notes_search_with_include_filter(search_config):
# ----------------------------------------------------------------------------------------------------
def test_notes_search_with_exclude_filter(search_config):
def test_notes_search_with_exclude_filter(search_config: ContentTypeConfig):
# Arrange
model.notes_search = asymmetric.setup(search_config.notes, regenerate=False)
model.notes_search = asymmetric.setup(search_config.org, regenerate=False)
user_query = "How to git install application? -clone"
# Act