1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
import json
from unittest.mock import MagicMock, patch
import pytest
from lunr import __TARGET_JS_VERSION__
from lunr.index import Index
from lunr.exceptions import BaseLunrException
from tests.utils import assert_vectors_equal
class TestIndex:
def test_create_query_default_fields(self, index):
query = index.create_query()
assert query.all_fields == index.fields
def test_create_query_subset_of_fields(self, index):
query = index.create_query([index.fields[0]])
assert query.all_fields == [index.fields[0]]
def test_create_query_non_contained_fields(self, index):
with pytest.raises(BaseLunrException):
index.create_query(["foo"])
def test_query_no_arguments_warns_and_returns_no_results(self, monkeypatch, index):
from lunr.index import logger
mock_logger = MagicMock()
monkeypatch.setattr(logger, "warning", mock_logger)
results = index.query()
assert results == []
mock_logger.assert_called_once()
def test_query_callback_argument_is_query_with_fields(self, index):
def callback(query):
assert query.all_fields == index.fields
index.query(callback=callback)
def test_query_callback_can_configure_query(self, index):
def callback(query):
query.clause("study")
results = index.query(callback=callback)
assert len(results) == 2
assert results[0]["ref"] == "b"
assert results[1]["ref"] == "a"
class TestIndexSerialization:
def test_serialization(self, index):
serialized_index = index.serialize()
assert serialized_index["version"] == __TARGET_JS_VERSION__
assert serialized_index["fields"] == index.fields
for ref, vector in serialized_index["fieldVectors"]:
assert ref in index.field_vectors
assert_vectors_equal(vector, index.field_vectors[ref])
def test_json_deserialization(self, index):
serialized_index = index.serialize()
json_serialized_index = json.dumps(serialized_index)
idx = Index.load(json_serialized_index)
assert idx == index
def test_load_warns_on_js_version_mismatch(self, index):
serialized_index = index.serialize()
serialized_index["version"] = "1.0.0"
with patch("lunr.index.logger") as mock_log:
Index.load(serialized_index)
mock_log.warning.assert_called_once()
|
