suggest_test.rb
3.03 KB
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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
require_relative "test_helper"
class SuggestTest < Minitest::Test
def setup
super
Product.reindex
end
def test_basic
store_names ["Great White Shark", "Hammerhead Shark", "Tiger Shark"]
assert_suggest "How Big is a Tigre Shar", "how big is a tiger shark", fields: [:name]
end
def test_perfect
store_names ["Tiger Shark", "Great White Shark"]
assert_suggest "Tiger Shark", nil, fields: [:name] # no correction
end
def test_phrase
store_names ["Big Tiger Shark", "Tiger Sharp Teeth", "Tiger Sharp Mind"]
assert_suggest "How to catch a big tiger shar", "how to catch a big tiger shark", fields: [:name]
end
def test_empty
assert_suggest "hi", nil
end
def test_without_option
store_names ["hi"] # needed to prevent ElasticsearchException - seed 668
assert_raises(RuntimeError) { Product.search("hi").suggestions }
end
def test_multiple_fields
store [
{name: "Shark", color: "Sharp"},
{name: "Shark", color: "Sharp"}
]
assert_suggest_all "shar", ["shark", "sharp"]
end
def test_multiple_fields_highest_score_first
store [
{name: "Tiger Shark", color: "Sharp"}
]
assert_suggest "tiger shar", "tiger shark"
end
def test_multiple_fields_same_value
store [
{name: "Shark", color: "Shark"}
]
assert_suggest_all "shar", ["shark"]
end
def test_fields_option
store [
{name: "Shark", color: "Sharp"}
]
assert_suggest_all "shar", ["shark"], fields: [:name]
end
def test_fields_option_multiple
store [
{name: "Shark"}
]
assert_suggest "shar", "shark", fields: [:name, :unknown]
end
def test_fields_partial_match
store_names ["Great White Shark", "Hammerhead Shark", "Tiger Shark"]
assert_suggest "How Big is a Tigre Shar", "how big is a tiger shark", fields: [{name: :word_start}]
end
def test_fields_partial_match_boost
store_names ["Great White Shark", "Hammerhead Shark", "Tiger Shark"]
assert_suggest "How Big is a Tigre Shar", "how big is a tiger shark", fields: [{"name^2" => :word_start}]
end
def test_multiple_models
skip # flaky test
store_names ["Great White Shark", "Hammerhead Shark", "Tiger Shark"]
assert_equal "how big is a tiger shark", Searchkick.search("How Big is a Tigre Shar", suggest: [:name], fields: [:name]).suggestions.first
end
def test_multiple_models_no_fields
store_names ["Great White Shark", "Hammerhead Shark", "Tiger Shark"]
assert_raises(ArgumentError) { Searchkick.search("How Big is a Tigre Shar", suggest: true) }
end
def test_star
assert_equal [], Product.search("*", suggest: true).suggestions
end
protected
def assert_suggest(term, expected, options = {})
result = Product.search(term, suggest: true, **options).suggestions.first
if expected.nil?
assert_nil result
else
assert_equal expected, result
end
end
# any order
def assert_suggest_all(term, expected, options = {})
assert_equal expected.sort, Product.search(term, suggest: true, **options).suggestions.sort
end
end