forked from karmi/tire-contrib
-
Notifications
You must be signed in to change notification settings - Fork 0
/
custom_filters_score_test.rb
96 lines (82 loc) · 3.23 KB
/
custom_filters_score_test.rb
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
require 'test_helper'
require 'tire'
require 'tire/queries/custom_filters_score'
module Tire
module Search
class CustomFiltersScoreTest < Test::Unit::TestCase
context "CustomFiltersScoreQuery" do
should "not raise an error when no block is given" do
assert_nothing_raised { Query.new.custom_filters_score }
end
should "provides a default filter if no filter is given" do
query = Query.new.custom_filters_score do
query { term :foo, 'bar' }
end
query[:custom_filters_score].tap do |f|
assert_equal( { :term => { :foo => { :term => 'bar' } } }, f[:query].to_hash )
assert_equal( { :match_all => {} }, f[:filters].first[:filter])
assert_equal( 1.0, f[:filters].first[:boost])
end
end
should "properly encode filter with boost" do
query = Query.new.custom_filters_score do
query { term :foo, 'bar' }
filter do
filter :terms, :tags => ['ruby']
boost 2.0
end
end
query[:custom_filters_score].tap do |f|
assert_equal( { :term => { :foo => { :term => 'bar' } } }, f[:query].to_hash )
assert_equal( { :tags => ['ruby'] }, f[:filters].first[:filter][:terms])
assert_equal( 2.0, f[:filters].first[:boost])
end
end
should "properly encode filter with script" do
query = Query.new.custom_filters_score do
query { term :foo, 'bar' }
filter do
filter :terms, :tags => ['ruby']
script '_score * 2.0'
end
end
query[:custom_filters_score].tap do |f|
assert_equal( { :term => { :foo => { :term => 'bar' } } }, f[:query].to_hash )
assert_equal( { :tags => ['ruby'] }, f[:filters].first[:filter][:terms])
assert_equal( '_score * 2.0', f[:filters].first[:script])
end
end
should "properly encode multiple filters" do
query = Query.new.custom_filters_score do
query { term :foo, 'bar' }
filter do
filter :terms, :tags => ['ruby']
boost 2.0
end
filter do
filter :terms, :tags => ['python']
script '_score * 2.0'
end
end
query[:custom_filters_score].tap do |f|
assert_equal( { :term => { :foo => { :term => 'bar' } } }, f[:query].to_hash )
assert_equal( { :tags => ['ruby'] }, f[:filters].first[:filter][:terms])
assert_equal( 2.0, f[:filters].first[:boost])
assert_equal( { :tags => ['python'] }, f[:filters].last[:filter][:terms])
assert_equal( '_score * 2.0', f[:filters].last[:script])
end
end
should "allow setting the score_mode" do
query = Query.new.custom_filters_score do
query { term :foo, 'bar' }
score_mode 'total'
end
query[:custom_filters_score].tap do |f|
assert_equal( { :term => { :foo => { :term => 'bar' } } }, f[:query].to_hash )
assert_equal( 'total', f[:score_mode])
end
end
end
end
end
end