chewy 0.9.0 → 0.10.0

Sign up to get free protection for your applications and to get access to all the features.
Files changed (265) hide show
  1. checksums.yaml +4 -4
  2. data/.gitignore +1 -0
  3. data/.rubocop.yml +24 -2
  4. data/.rubocop_todo.yml +2 -2
  5. data/.travis.yml +38 -21
  6. data/.yardopts +5 -0
  7. data/Appraisals +55 -27
  8. data/CHANGELOG.md +57 -12
  9. data/Gemfile +14 -10
  10. data/LEGACY_DSL.md +497 -0
  11. data/README.md +249 -515
  12. data/chewy.gemspec +5 -4
  13. data/gemfiles/rails.4.0.activerecord.gemfile +14 -0
  14. data/gemfiles/rails.4.1.activerecord.gemfile +14 -0
  15. data/gemfiles/rails.4.2.activerecord.gemfile +8 -10
  16. data/gemfiles/rails.4.2.mongoid.5.1.gemfile +9 -10
  17. data/gemfiles/rails.5.0.activerecord.gemfile +8 -10
  18. data/gemfiles/rails.5.0.mongoid.6.0.gemfile +15 -0
  19. data/gemfiles/rails.5.1.activerecord.gemfile +15 -0
  20. data/gemfiles/rails.5.1.mongoid.6.1.gemfile +15 -0
  21. data/gemfiles/sequel.4.45.gemfile +11 -0
  22. data/lib/chewy.rb +77 -43
  23. data/lib/chewy/config.rb +44 -7
  24. data/lib/chewy/errors.rb +2 -2
  25. data/lib/chewy/fields/base.rb +39 -32
  26. data/lib/chewy/fields/root.rb +33 -7
  27. data/lib/chewy/index.rb +237 -149
  28. data/lib/chewy/index/actions.rb +85 -28
  29. data/lib/chewy/index/aliases.rb +2 -1
  30. data/lib/chewy/index/settings.rb +9 -5
  31. data/lib/chewy/index/specification.rb +58 -0
  32. data/lib/chewy/journal.rb +40 -92
  33. data/lib/chewy/query.rb +43 -27
  34. data/lib/chewy/query/compose.rb +13 -13
  35. data/lib/chewy/query/criteria.rb +13 -13
  36. data/lib/chewy/query/filters.rb +1 -1
  37. data/lib/chewy/query/loading.rb +1 -1
  38. data/lib/chewy/query/nodes/and.rb +2 -2
  39. data/lib/chewy/query/nodes/bool.rb +1 -1
  40. data/lib/chewy/query/nodes/equal.rb +2 -2
  41. data/lib/chewy/query/nodes/exists.rb +1 -1
  42. data/lib/chewy/query/nodes/has_relation.rb +2 -2
  43. data/lib/chewy/query/nodes/match_all.rb +1 -1
  44. data/lib/chewy/query/nodes/missing.rb +1 -1
  45. data/lib/chewy/query/nodes/not.rb +2 -2
  46. data/lib/chewy/query/nodes/or.rb +2 -2
  47. data/lib/chewy/query/nodes/prefix.rb +1 -1
  48. data/lib/chewy/query/nodes/query.rb +2 -2
  49. data/lib/chewy/query/nodes/range.rb +4 -4
  50. data/lib/chewy/query/nodes/regexp.rb +4 -4
  51. data/lib/chewy/query/nodes/script.rb +3 -3
  52. data/lib/chewy/query/pagination.rb +10 -1
  53. data/lib/chewy/railtie.rb +1 -0
  54. data/lib/chewy/rake_helper.rb +265 -48
  55. data/lib/chewy/rspec/update_index.rb +30 -22
  56. data/lib/chewy/search.rb +78 -21
  57. data/lib/chewy/search/loader.rb +83 -0
  58. data/lib/chewy/{query → search}/pagination/kaminari.rb +13 -5
  59. data/lib/chewy/search/pagination/will_paginate.rb +41 -0
  60. data/lib/chewy/search/parameters.rb +150 -0
  61. data/lib/chewy/search/parameters/aggs.rb +16 -0
  62. data/lib/chewy/search/parameters/concerns/bool_storage.rb +24 -0
  63. data/lib/chewy/search/parameters/concerns/hash_storage.rb +23 -0
  64. data/lib/chewy/search/parameters/concerns/integer_storage.rb +14 -0
  65. data/lib/chewy/search/parameters/concerns/query_storage.rb +237 -0
  66. data/lib/chewy/search/parameters/concerns/string_array_storage.rb +23 -0
  67. data/lib/chewy/search/parameters/concerns/string_storage.rb +14 -0
  68. data/lib/chewy/search/parameters/docvalue_fields.rb +12 -0
  69. data/lib/chewy/search/parameters/explain.rb +16 -0
  70. data/lib/chewy/search/parameters/filter.rb +47 -0
  71. data/lib/chewy/search/parameters/highlight.rb +16 -0
  72. data/lib/chewy/search/parameters/indices_boost.rb +52 -0
  73. data/lib/chewy/search/parameters/limit.rb +17 -0
  74. data/lib/chewy/search/parameters/load.rb +32 -0
  75. data/lib/chewy/search/parameters/min_score.rb +16 -0
  76. data/lib/chewy/search/parameters/none.rb +27 -0
  77. data/lib/chewy/search/parameters/offset.rb +17 -0
  78. data/lib/chewy/search/parameters/order.rb +64 -0
  79. data/lib/chewy/search/parameters/post_filter.rb +19 -0
  80. data/lib/chewy/search/parameters/preference.rb +16 -0
  81. data/lib/chewy/search/parameters/profile.rb +16 -0
  82. data/lib/chewy/search/parameters/query.rb +19 -0
  83. data/lib/chewy/search/parameters/request_cache.rb +27 -0
  84. data/lib/chewy/search/parameters/rescore.rb +29 -0
  85. data/lib/chewy/search/parameters/script_fields.rb +16 -0
  86. data/lib/chewy/search/parameters/search_after.rb +20 -0
  87. data/lib/chewy/search/parameters/search_type.rb +16 -0
  88. data/lib/chewy/search/parameters/source.rb +73 -0
  89. data/lib/chewy/search/parameters/storage.rb +95 -0
  90. data/lib/chewy/search/parameters/stored_fields.rb +63 -0
  91. data/lib/chewy/search/parameters/suggest.rb +16 -0
  92. data/lib/chewy/search/parameters/terminate_after.rb +16 -0
  93. data/lib/chewy/search/parameters/timeout.rb +16 -0
  94. data/lib/chewy/search/parameters/track_scores.rb +16 -0
  95. data/lib/chewy/search/parameters/types.rb +20 -0
  96. data/lib/chewy/search/parameters/version.rb +16 -0
  97. data/lib/chewy/search/query_proxy.rb +257 -0
  98. data/lib/chewy/search/request.rb +1021 -0
  99. data/lib/chewy/search/response.rb +119 -0
  100. data/lib/chewy/search/scoping.rb +50 -0
  101. data/lib/chewy/search/scrolling.rb +136 -0
  102. data/lib/chewy/stash.rb +70 -0
  103. data/lib/chewy/strategy.rb +10 -3
  104. data/lib/chewy/strategy/active_job.rb +1 -0
  105. data/lib/chewy/strategy/atomic.rb +1 -3
  106. data/lib/chewy/strategy/bypass.rb +1 -1
  107. data/lib/chewy/strategy/resque.rb +1 -0
  108. data/lib/chewy/strategy/shoryuken.rb +40 -0
  109. data/lib/chewy/strategy/sidekiq.rb +13 -3
  110. data/lib/chewy/type.rb +29 -7
  111. data/lib/chewy/type/actions.rb +26 -2
  112. data/lib/chewy/type/adapter/active_record.rb +44 -29
  113. data/lib/chewy/type/adapter/base.rb +27 -7
  114. data/lib/chewy/type/adapter/mongoid.rb +18 -7
  115. data/lib/chewy/type/adapter/object.rb +187 -26
  116. data/lib/chewy/type/adapter/orm.rb +59 -32
  117. data/lib/chewy/type/adapter/sequel.rb +32 -16
  118. data/lib/chewy/type/import.rb +145 -191
  119. data/lib/chewy/type/import/bulk_builder.rb +122 -0
  120. data/lib/chewy/type/import/bulk_request.rb +76 -0
  121. data/lib/chewy/type/import/journal_builder.rb +45 -0
  122. data/lib/chewy/type/import/routine.rb +138 -0
  123. data/lib/chewy/type/mapping.rb +11 -1
  124. data/lib/chewy/type/observe.rb +1 -1
  125. data/lib/chewy/type/syncer.rb +220 -0
  126. data/lib/chewy/type/witchcraft.rb +27 -13
  127. data/lib/chewy/type/wrapper.rb +28 -2
  128. data/lib/chewy/version.rb +1 -1
  129. data/lib/tasks/chewy.rake +84 -26
  130. data/spec/chewy/config_spec.rb +82 -1
  131. data/spec/chewy/fields/base_spec.rb +147 -112
  132. data/spec/chewy/fields/root_spec.rb +75 -18
  133. data/spec/chewy/fields/time_fields_spec.rb +2 -3
  134. data/spec/chewy/index/actions_spec.rb +180 -50
  135. data/spec/chewy/index/aliases_spec.rb +2 -2
  136. data/spec/chewy/index/settings_spec.rb +67 -38
  137. data/spec/chewy/index/specification_spec.rb +160 -0
  138. data/spec/chewy/index_spec.rb +57 -66
  139. data/spec/chewy/journal_spec.rb +149 -54
  140. data/spec/chewy/minitest/helpers_spec.rb +4 -4
  141. data/spec/chewy/minitest/search_index_receiver_spec.rb +1 -1
  142. data/spec/chewy/query/criteria_spec.rb +179 -179
  143. data/spec/chewy/query/filters_spec.rb +15 -15
  144. data/spec/chewy/query/loading_spec.rb +22 -20
  145. data/spec/chewy/query/nodes/and_spec.rb +2 -2
  146. data/spec/chewy/query/nodes/bool_spec.rb +4 -4
  147. data/spec/chewy/query/nodes/equal_spec.rb +19 -19
  148. data/spec/chewy/query/nodes/exists_spec.rb +6 -6
  149. data/spec/chewy/query/nodes/has_child_spec.rb +19 -19
  150. data/spec/chewy/query/nodes/has_parent_spec.rb +19 -19
  151. data/spec/chewy/query/nodes/missing_spec.rb +5 -5
  152. data/spec/chewy/query/nodes/not_spec.rb +3 -2
  153. data/spec/chewy/query/nodes/or_spec.rb +2 -2
  154. data/spec/chewy/query/nodes/prefix_spec.rb +5 -5
  155. data/spec/chewy/query/nodes/query_spec.rb +2 -2
  156. data/spec/chewy/query/nodes/range_spec.rb +18 -18
  157. data/spec/chewy/query/nodes/raw_spec.rb +1 -1
  158. data/spec/chewy/query/nodes/regexp_spec.rb +14 -14
  159. data/spec/chewy/query/nodes/script_spec.rb +4 -4
  160. data/spec/chewy/query/pagination/kaminari_spec.rb +3 -55
  161. data/spec/chewy/query/pagination/will_paginate_spec.rb +5 -0
  162. data/spec/chewy/query/pagination_spec.rb +25 -21
  163. data/spec/chewy/query_spec.rb +501 -560
  164. data/spec/chewy/rake_helper_spec.rb +368 -0
  165. data/spec/chewy/repository_spec.rb +4 -4
  166. data/spec/chewy/rspec/update_index_spec.rb +89 -56
  167. data/spec/chewy/runtime_spec.rb +2 -2
  168. data/spec/chewy/search/loader_spec.rb +117 -0
  169. data/spec/chewy/search/pagination/kaminari_examples.rb +71 -0
  170. data/spec/chewy/search/pagination/kaminari_spec.rb +17 -0
  171. data/spec/chewy/search/pagination/will_paginate_examples.rb +63 -0
  172. data/spec/chewy/search/pagination/will_paginate_spec.rb +17 -0
  173. data/spec/chewy/search/parameters/aggs_spec.rb +5 -0
  174. data/spec/chewy/search/parameters/bool_storage_examples.rb +53 -0
  175. data/spec/chewy/search/parameters/docvalue_fields_spec.rb +5 -0
  176. data/spec/chewy/search/parameters/explain_spec.rb +5 -0
  177. data/spec/chewy/search/parameters/filter_spec.rb +5 -0
  178. data/spec/chewy/search/parameters/hash_storage_examples.rb +59 -0
  179. data/spec/chewy/search/parameters/highlight_spec.rb +5 -0
  180. data/spec/chewy/search/parameters/indices_boost_spec.rb +83 -0
  181. data/spec/chewy/search/parameters/integer_storage_examples.rb +32 -0
  182. data/spec/chewy/search/parameters/limit_spec.rb +5 -0
  183. data/spec/chewy/search/parameters/load_spec.rb +60 -0
  184. data/spec/chewy/search/parameters/min_score_spec.rb +32 -0
  185. data/spec/chewy/search/parameters/none_spec.rb +5 -0
  186. data/spec/chewy/search/parameters/offset_spec.rb +5 -0
  187. data/spec/chewy/search/parameters/order_spec.rb +65 -0
  188. data/spec/chewy/search/parameters/post_filter_spec.rb +5 -0
  189. data/spec/chewy/search/parameters/preference_spec.rb +5 -0
  190. data/spec/chewy/search/parameters/profile_spec.rb +5 -0
  191. data/spec/chewy/search/parameters/query_spec.rb +5 -0
  192. data/spec/chewy/search/parameters/query_storage_examples.rb +388 -0
  193. data/spec/chewy/search/parameters/request_cache_spec.rb +67 -0
  194. data/spec/chewy/search/parameters/rescore_spec.rb +62 -0
  195. data/spec/chewy/search/parameters/script_fields_spec.rb +5 -0
  196. data/spec/chewy/search/parameters/search_after_spec.rb +32 -0
  197. data/spec/chewy/search/parameters/search_type_spec.rb +5 -0
  198. data/spec/chewy/search/parameters/source_spec.rb +156 -0
  199. data/spec/chewy/search/parameters/storage_spec.rb +60 -0
  200. data/spec/chewy/search/parameters/stored_fields_spec.rb +126 -0
  201. data/spec/chewy/search/parameters/string_array_storage_examples.rb +63 -0
  202. data/spec/chewy/search/parameters/string_storage_examples.rb +32 -0
  203. data/spec/chewy/search/parameters/suggest_spec.rb +5 -0
  204. data/spec/chewy/search/parameters/terminate_after_spec.rb +5 -0
  205. data/spec/chewy/search/parameters/timeout_spec.rb +5 -0
  206. data/spec/chewy/search/parameters/track_scores_spec.rb +5 -0
  207. data/spec/chewy/search/parameters/types_spec.rb +5 -0
  208. data/spec/chewy/search/parameters/version_spec.rb +5 -0
  209. data/spec/chewy/search/parameters_spec.rb +130 -0
  210. data/spec/chewy/search/query_proxy_spec.rb +68 -0
  211. data/spec/chewy/search/request_spec.rb +669 -0
  212. data/spec/chewy/search/response_spec.rb +192 -0
  213. data/spec/chewy/search/scrolling_spec.rb +169 -0
  214. data/spec/chewy/search_spec.rb +13 -6
  215. data/spec/chewy/stash_spec.rb +95 -0
  216. data/spec/chewy/strategy/active_job_spec.rb +6 -0
  217. data/spec/chewy/strategy/resque_spec.rb +6 -0
  218. data/spec/chewy/strategy/shoryuken_spec.rb +64 -0
  219. data/spec/chewy/strategy/sidekiq_spec.rb +8 -0
  220. data/spec/chewy/strategy_spec.rb +6 -6
  221. data/spec/chewy/type/actions_spec.rb +29 -10
  222. data/spec/chewy/type/adapter/active_record_spec.rb +203 -91
  223. data/spec/chewy/type/adapter/mongoid_spec.rb +112 -54
  224. data/spec/chewy/type/adapter/object_spec.rb +101 -28
  225. data/spec/chewy/type/adapter/sequel_spec.rb +149 -82
  226. data/spec/chewy/type/import/bulk_builder_spec.rb +279 -0
  227. data/spec/chewy/type/import/bulk_request_spec.rb +102 -0
  228. data/spec/chewy/type/import/journal_builder_spec.rb +95 -0
  229. data/spec/chewy/type/import/routine_spec.rb +110 -0
  230. data/spec/chewy/type/import_spec.rb +350 -271
  231. data/spec/chewy/type/mapping_spec.rb +54 -18
  232. data/spec/chewy/type/observe_spec.rb +5 -1
  233. data/spec/chewy/type/syncer_spec.rb +123 -0
  234. data/spec/chewy/type/witchcraft_spec.rb +45 -29
  235. data/spec/chewy/type/wrapper_spec.rb +63 -23
  236. data/spec/chewy/type_spec.rb +28 -7
  237. data/spec/chewy_spec.rb +75 -7
  238. data/spec/spec_helper.rb +5 -2
  239. data/spec/support/active_record.rb +5 -1
  240. data/spec/support/class_helpers.rb +0 -14
  241. data/spec/support/mongoid.rb +15 -3
  242. data/spec/support/sequel.rb +6 -1
  243. metadata +198 -37
  244. data/gemfiles/rails.3.2.activerecord.gemfile +0 -16
  245. data/gemfiles/rails.3.2.activerecord.kaminari.gemfile +0 -15
  246. data/gemfiles/rails.3.2.activerecord.will_paginate.gemfile +0 -15
  247. data/gemfiles/rails.4.2.activerecord.kaminari.gemfile +0 -16
  248. data/gemfiles/rails.4.2.activerecord.will_paginate.gemfile +0 -16
  249. data/gemfiles/rails.4.2.mongoid.4.0.gemfile +0 -16
  250. data/gemfiles/rails.4.2.mongoid.4.0.kaminari.gemfile +0 -15
  251. data/gemfiles/rails.4.2.mongoid.4.0.will_paginate.gemfile +0 -15
  252. data/gemfiles/rails.4.2.mongoid.5.1.kaminari.gemfile +0 -15
  253. data/gemfiles/rails.4.2.mongoid.5.1.will_paginate.gemfile +0 -15
  254. data/gemfiles/rails.5.0.activerecord.kaminari.gemfile +0 -16
  255. data/gemfiles/rails.5.0.activerecord.will_paginate.gemfile +0 -16
  256. data/gemfiles/sequel.4.38.gemfile +0 -14
  257. data/lib/chewy/journal/apply.rb +0 -31
  258. data/lib/chewy/journal/clean.rb +0 -24
  259. data/lib/chewy/journal/entry.rb +0 -83
  260. data/lib/chewy/journal/query.rb +0 -87
  261. data/lib/chewy/query/pagination/will_paginate.rb +0 -27
  262. data/lib/chewy/query/scoping.rb +0 -20
  263. data/spec/chewy/journal/apply_spec.rb +0 -120
  264. data/spec/chewy/journal/entry_spec.rb +0 -237
  265. data/spec/chewy/query/pagination/will_paginage_spec.rb +0 -59
@@ -0,0 +1,63 @@
1
+ require 'spec_helper'
2
+
3
+ shared_examples :string_array_storage do |param_name|
4
+ subject { described_class.new(%i[foo bar]) }
5
+
6
+ describe '#initialize' do
7
+ specify { expect(described_class.new.value).to eq([]) }
8
+ specify { expect(described_class.new(nil).value).to eq([]) }
9
+ specify { expect(described_class.new(:foo).value).to eq(%w[foo]) }
10
+ specify { expect(described_class.new([:foo, nil]).value).to eq(%w[foo]) }
11
+ specify { expect(subject.value).to eq(%w[foo bar]) }
12
+ end
13
+
14
+ describe '#replace!' do
15
+ specify do
16
+ expect { subject.replace!(nil) }
17
+ .to change { subject.value }
18
+ .from(%w[foo bar]).to([])
19
+ end
20
+
21
+ specify do
22
+ expect { subject.replace!(%i[foo baz]) }
23
+ .to change { subject.value }
24
+ .from(%w[foo bar]).to(%w[foo baz])
25
+ end
26
+ end
27
+
28
+ describe '#update!' do
29
+ specify do
30
+ expect { subject.update!(nil) }
31
+ .not_to change { subject.value }
32
+ end
33
+
34
+ specify do
35
+ expect { subject.update!(%i[foo baz]) }
36
+ .to change { subject.value }
37
+ .from(%w[foo bar]).to(%w[foo bar baz])
38
+ end
39
+ end
40
+
41
+ describe '#merge!' do
42
+ specify do
43
+ expect { subject.merge!(described_class.new) }
44
+ .not_to change { subject.value }
45
+ end
46
+
47
+ specify do
48
+ expect { subject.merge!(described_class.new(%i[foo baz])) }
49
+ .to change { subject.value }
50
+ .from(%w[foo bar]).to(%w[foo bar baz])
51
+ end
52
+ end
53
+
54
+ describe '#render' do
55
+ specify { expect(described_class.new.render).to be_nil }
56
+
57
+ if param_name
58
+ specify { expect(subject.render).to eq(param_name => %w[foo bar]) }
59
+ else
60
+ specify { expect(subject.render).to be_nil }
61
+ end
62
+ end
63
+ end
@@ -0,0 +1,32 @@
1
+ require 'spec_helper'
2
+
3
+ shared_examples :string_storage do |param_name|
4
+ subject { described_class.new(:foo) }
5
+
6
+ describe '#initialize' do
7
+ specify { expect(subject.value).to eq('foo') }
8
+ specify { expect(described_class.new(42).value).to eq('42') }
9
+ specify { expect(described_class.new('').value).to be_nil }
10
+ end
11
+
12
+ describe '#replace!' do
13
+ specify { expect { subject.replace!('bar') }.to change { subject.value }.from('foo').to('bar') }
14
+ specify { expect { subject.replace!('') }.to change { subject.value }.from('foo').to(nil) }
15
+ end
16
+
17
+ describe '#update!' do
18
+ specify { expect { subject.update!('bar') }.to change { subject.value }.from('foo').to('bar') }
19
+ specify { expect { subject.update!('') }.not_to change { subject.value }.from('foo') }
20
+ specify { expect { subject.update!(nil) }.not_to change { subject.value }.from('foo') }
21
+ end
22
+
23
+ describe '#merge!' do
24
+ specify { expect { subject.merge!(described_class.new('bar')) }.to change { subject.value }.from('foo').to('bar') }
25
+ specify { expect { subject.merge!(described_class.new) }.not_to change { subject.value }.from('foo') }
26
+ end
27
+
28
+ describe '#render' do
29
+ specify { expect(described_class.new.render).to be_nil }
30
+ specify { expect(subject.render).to eq(param_name => 'foo') }
31
+ end
32
+ end
@@ -0,0 +1,5 @@
1
+ require 'chewy/search/parameters/hash_storage_examples'
2
+
3
+ describe Chewy::Search::Parameters::Suggest do
4
+ it_behaves_like :hash_storage, :suggest
5
+ end
@@ -0,0 +1,5 @@
1
+ require 'chewy/search/parameters/integer_storage_examples'
2
+
3
+ describe Chewy::Search::Parameters::TerminateAfter do
4
+ it_behaves_like :integer_storage, :terminate_after
5
+ end
@@ -0,0 +1,5 @@
1
+ require 'chewy/search/parameters/string_storage_examples'
2
+
3
+ describe Chewy::Search::Parameters::Timeout do
4
+ it_behaves_like :string_storage, :timeout
5
+ end
@@ -0,0 +1,5 @@
1
+ require 'chewy/search/parameters/bool_storage_examples'
2
+
3
+ describe Chewy::Search::Parameters::TrackScores do
4
+ it_behaves_like :bool_storage, :track_scores
5
+ end
@@ -0,0 +1,5 @@
1
+ require 'chewy/search/parameters/string_array_storage_examples'
2
+
3
+ describe Chewy::Search::Parameters::Types do
4
+ it_behaves_like :string_array_storage, nil
5
+ end
@@ -0,0 +1,5 @@
1
+ require 'chewy/search/parameters/bool_storage_examples'
2
+
3
+ describe Chewy::Search::Parameters::Version do
4
+ it_behaves_like :bool_storage, :version
5
+ end
@@ -0,0 +1,130 @@
1
+ require 'spec_helper'
2
+
3
+ describe Chewy::Search::Parameters do
4
+ subject { described_class.new }
5
+
6
+ describe '.storages' do
7
+ specify { expect(described_class.storages[:limit]).to eq(Chewy::Search::Parameters::Limit) }
8
+ end
9
+
10
+ describe '#initialize' do
11
+ let(:limit) { described_class.storages[:limit].new(3) }
12
+ subject { described_class.new(limit: limit, order: :foo) }
13
+
14
+ specify { expect(subject.storages[:limit]).to equal(limit) }
15
+ specify { expect(subject.storages[:limit].value).to eq(3) }
16
+ specify { expect(subject.storages[:order].value).to eq('foo' => nil) }
17
+
18
+ specify { expect { described_class.new(offset: limit) }.to raise_error(TypeError) }
19
+ end
20
+
21
+ describe '#storages' do
22
+ specify { expect(subject.storages).to eq({}) }
23
+ specify { expect(subject.storages[:limit]).to be_a(Chewy::Search::Parameters::Limit) }
24
+ end
25
+
26
+ describe '#==' do
27
+ specify { expect(described_class.new(limit: 10)).to eq(described_class.new(limit: 10)) }
28
+ specify { expect(described_class.new(limit: 10)).to eq(described_class.new(limit: 10, offset: nil)) }
29
+ specify { expect(described_class.new(limit: 10, offset: 20)).to eq(described_class.new(limit: 10, offset: 20)) }
30
+ specify { expect(described_class.new(limit: 10)).not_to eq(described_class.new(limit: 20)) }
31
+ specify { expect(described_class.new(limit: 10)).not_to eq(described_class.new(limit: 10, offset: 20)) }
32
+ end
33
+
34
+ describe '#modify!' do
35
+ it 'updates the storage value' do
36
+ expect { subject.modify!(:limit) { replace!(42) } }
37
+ .to change { subject[:limit].value }
38
+ .from(nil).to(42)
39
+ end
40
+
41
+ it 'replaces the storage' do
42
+ expect { subject.modify!(:limit) { replace!(42) } }
43
+ .to change { subject[:limit].object_id }
44
+ end
45
+
46
+ it 'doesn\'t change the old object' do
47
+ subject.modify!(:limit) { replace!(42) }
48
+ old_limit = subject[:limit]
49
+ subject.modify!(:limit) { replace!(43) }
50
+ expect(old_limit.value).to eq(42)
51
+ end
52
+ end
53
+
54
+ describe '#only!' do
55
+ subject { described_class.new(limit: 10, offset: 20, order: :foo) }
56
+
57
+ specify { expect { subject.only!([:limit]) }.to change { subject.clone }.to(described_class.new(limit: 10)) }
58
+ specify { expect { subject.only!(%i[offset order]) }.to change { subject.clone }.to(described_class.new(offset: 20, order: :foo)) }
59
+ specify { expect { subject.only!(%i[limit something]) }.to raise_error NameError }
60
+ specify { expect { subject.only!([]) }.to raise_error ArgumentError }
61
+ end
62
+
63
+ describe '#except!' do
64
+ subject { described_class.new(limit: 10, offset: 20, order: :foo) }
65
+
66
+ specify { expect { subject.except!([:limit]) }.to change { subject.clone }.to(described_class.new(offset: 20, order: :foo)) }
67
+ specify { expect { subject.except!(%i[offset order]) }.to change { subject.clone }.to(described_class.new(limit: 10)) }
68
+ specify { expect { subject.except!(%i[limit something]) }.to raise_error NameError }
69
+ specify { expect { subject.except!([]) }.to raise_error ArgumentError }
70
+ end
71
+
72
+ describe '#merge!' do
73
+ let(:first) { described_class.new(offset: 10, order: 'foo') }
74
+ let(:second) { described_class.new(limit: 20, offset: 20, order: 'bar') }
75
+
76
+ specify do
77
+ expect { first.merge!(second) }.to change { first.clone }
78
+ .to(described_class.new(limit: 20, offset: 20, order: %w[foo bar]))
79
+ end
80
+ specify { expect { first.merge!(second) }.not_to change { second.clone } }
81
+
82
+ specify do
83
+ expect { second.merge!(first) }.to change { second.clone }
84
+ .to(described_class.new(limit: 20, offset: 10, order: %w[bar foo]))
85
+ end
86
+ specify { expect { second.merge!(first) }.not_to change { first.clone } }
87
+
88
+ context 'spawns new storages for the merge' do
89
+ let(:names) { %i[limit offset order] }
90
+ def storage_object_ids(params)
91
+ params.storages.values_at(*names).map(&:object_id)
92
+ end
93
+
94
+ specify { expect(storage_object_ids(first) | storage_object_ids(subject)).to have(6).items }
95
+ specify { expect(storage_object_ids(second) | storage_object_ids(subject)).to have(6).items }
96
+ end
97
+ end
98
+
99
+ describe '#render' do
100
+ subject { described_class.new(offset: 10, order: 'foo') }
101
+
102
+ specify { expect(subject.render).to eq(body: {from: 10, sort: ['foo']}) }
103
+ specify { expect(described_class.new.render).to eq({}) }
104
+
105
+ context do
106
+ subject { described_class.new(query: {foo: 'bar'}, filter: {moo: 'baz'}) }
107
+ specify { expect(subject.render).to eq(body: {query: {bool: {must: {foo: 'bar'}, filter: {moo: 'baz'}}}}) }
108
+ end
109
+
110
+ context do
111
+ subject { described_class.new(query: {should: {foo: 'bar'}}, filter: {moo: 'baz'}) }
112
+ specify { expect(subject.render).to eq(body: {query: {bool: {should: {foo: 'bar'}, filter: {moo: 'baz'}}}}) }
113
+ end
114
+
115
+ context do
116
+ subject { described_class.new(query: {must_not: {foo: 'bar'}}, filter: {moo: 'baz'}) }
117
+ specify { expect(subject.render).to eq(body: {query: {bool: {must_not: {foo: 'bar'}, filter: {moo: 'baz'}}}}) }
118
+ end
119
+
120
+ context do
121
+ subject { described_class.new(filter: {moo: 'baz'}) }
122
+ specify { expect(subject.render).to eq(body: {query: {bool: {filter: {moo: 'baz'}}}}) }
123
+ end
124
+
125
+ context do
126
+ subject { described_class.new(filter: {moo: 'baz'}, none: true) }
127
+ specify { expect(subject.render).to eq(body: {query: {bool: {filter: {bool: {must_not: {match_all: {}}}}}}}) }
128
+ end
129
+ end
130
+ end
@@ -0,0 +1,68 @@
1
+ require 'spec_helper'
2
+
3
+ describe Chewy::Search::QueryProxy do
4
+ before { stub_index(:products) }
5
+ let(:request) { Chewy::Search::Request.new(ProductsIndex).query(match: {foo: 'bar'}) }
6
+ let(:scope) { Chewy::Search::Request.new(ProductsIndex).query.must_not(match: {foo: 'bar'}) }
7
+ subject { described_class.new(:query, request) }
8
+
9
+ describe '#must' do
10
+ specify { expect { subject.must }.to raise_error ArgumentError }
11
+ specify { expect(subject.must(multi_match: {foo: 'bar'}).render[:body]).to eq(query: {bool: {must: [{match: {foo: 'bar'}}, {multi_match: {foo: 'bar'}}]}}) }
12
+ specify { expect(subject.must { multi_match foo: 'bar' }.render[:body]).to eq(query: {bool: {must: [{match: {foo: 'bar'}}, {multi_match: {foo: 'bar'}}]}}) }
13
+ end
14
+
15
+ describe '#should' do
16
+ specify { expect { subject.should }.to raise_error ArgumentError }
17
+ specify { expect(subject.should(multi_match: {foo: 'bar'}).render[:body]).to eq(query: {bool: {must: {match: {foo: 'bar'}}, should: {multi_match: {foo: 'bar'}}}}) }
18
+ specify { expect(subject.should { multi_match foo: 'bar' }.render[:body]).to eq(query: {bool: {must: {match: {foo: 'bar'}}, should: {multi_match: {foo: 'bar'}}}}) }
19
+ end
20
+
21
+ describe '#must_not' do
22
+ specify { expect { subject.must_not }.to raise_error ArgumentError }
23
+ specify { expect(subject.must_not(multi_match: {foo: 'bar'}).render[:body]).to eq(query: {bool: {must: {match: {foo: 'bar'}}, must_not: {multi_match: {foo: 'bar'}}}}) }
24
+ specify { expect(subject.must_not { multi_match foo: 'bar' }.render[:body]).to eq(query: {bool: {must: {match: {foo: 'bar'}}, must_not: {multi_match: {foo: 'bar'}}}}) }
25
+ end
26
+
27
+ describe '#and' do
28
+ specify { expect { subject.and }.to raise_error ArgumentError }
29
+ specify { expect(subject.and(multi_match: {foo: 'bar'}).render[:body]).to eq(query: {bool: {must: [{match: {foo: 'bar'}}, {multi_match: {foo: 'bar'}}]}}) }
30
+ specify { expect(subject.and { multi_match foo: 'bar' }.render[:body]).to eq(query: {bool: {must: [{match: {foo: 'bar'}}, {multi_match: {foo: 'bar'}}]}}) }
31
+ specify { expect(subject.and(scope).render[:body]).to eq(query: {bool: {must: [{match: {foo: 'bar'}}, {bool: {must_not: {match: {foo: 'bar'}}}}]}}) }
32
+ end
33
+
34
+ describe '#or' do
35
+ specify { expect { subject.or }.to raise_error ArgumentError }
36
+ specify { expect(subject.or(multi_match: {foo: 'bar'}).render[:body]).to eq(query: {bool: {should: [{match: {foo: 'bar'}}, {multi_match: {foo: 'bar'}}]}}) }
37
+ specify { expect(subject.or { multi_match foo: 'bar' }.render[:body]).to eq(query: {bool: {should: [{match: {foo: 'bar'}}, {multi_match: {foo: 'bar'}}]}}) }
38
+ specify { expect(subject.or(scope).render[:body]).to eq(query: {bool: {should: [{match: {foo: 'bar'}}, {bool: {must_not: {match: {foo: 'bar'}}}}]}}) }
39
+ end
40
+
41
+ describe '#not' do
42
+ specify { expect { subject.not }.to raise_error ArgumentError }
43
+ specify { expect(subject.not(multi_match: {foo: 'bar'}).render[:body]).to eq(query: {bool: {must: {match: {foo: 'bar'}}, must_not: {multi_match: {foo: 'bar'}}}}) }
44
+ specify { expect(subject.not { multi_match foo: 'bar' }.render[:body]).to eq(query: {bool: {must: {match: {foo: 'bar'}}, must_not: {multi_match: {foo: 'bar'}}}}) }
45
+ specify { expect(subject.not(scope).render[:body]).to eq(query: {bool: {must: {match: {foo: 'bar'}}, must_not: {bool: {must_not: {match: {foo: 'bar'}}}}}}) }
46
+ end
47
+
48
+ describe '#minimum_should_match' do
49
+ specify { expect(subject.minimum_should_match('100%').render[:body]).to eq(query: {match: {foo: 'bar'}}) }
50
+
51
+ context do
52
+ let(:request) do
53
+ Chewy::Search::Request.new(ProductsIndex)
54
+ .query.should(match: {foo: 'bar'})
55
+ end
56
+ specify { expect(subject.minimum_should_match('100%').render[:body]).to eq(query: {bool: {should: {match: {foo: 'bar'}}, minimum_should_match: '100%'}}) }
57
+ end
58
+
59
+ context do
60
+ let(:request) do
61
+ Chewy::Search::Request.new(ProductsIndex)
62
+ .query.should(match: {foo: 'bar'})
63
+ .query.minimum_should_match(2)
64
+ end
65
+ specify { expect(subject.minimum_should_match(nil).render[:body]).to eq(query: {bool: {should: {match: {foo: 'bar'}}}}) }
66
+ end
67
+ end
68
+ end
@@ -0,0 +1,669 @@
1
+ require 'spec_helper'
2
+
3
+ describe Chewy::Search::Request do
4
+ before { Chewy.massacre }
5
+
6
+ before do
7
+ stub_index(:products) do
8
+ define_type :product do
9
+ field :id, type: :integer
10
+ field :name
11
+ field :age, type: :integer
12
+ end
13
+ define_type :city do
14
+ field :id, type: :integer
15
+ end
16
+ define_type :country do
17
+ field :id, type: :integer
18
+ end
19
+ end
20
+
21
+ stub_index(:cities) do
22
+ define_type :city
23
+ end
24
+ end
25
+
26
+ subject { described_class.new(ProductsIndex) }
27
+
28
+ describe '#==' do
29
+ specify { expect(described_class.new(ProductsIndex)).to eq(described_class.new(ProductsIndex)) }
30
+ specify { expect(described_class.new(ProductsIndex)).not_to eq(described_class.new(CitiesIndex)) }
31
+ specify { expect(described_class.new(ProductsIndex)).not_to eq(described_class.new(ProductsIndex, CitiesIndex)) }
32
+ specify { expect(described_class.new(CitiesIndex, ProductsIndex)).to eq(described_class.new(ProductsIndex, CitiesIndex)) }
33
+ specify { expect(described_class.new(ProductsIndex::Product)).to eq(described_class.new(ProductsIndex::Product)) }
34
+ specify { expect(described_class.new(ProductsIndex::Product)).not_to eq(described_class.new(ProductsIndex::City)) }
35
+ specify { expect(described_class.new(ProductsIndex::Product)).not_to eq(described_class.new(ProductsIndex::Product, ProductsIndex::City)) }
36
+ specify { expect(described_class.new(ProductsIndex::City, ProductsIndex::Product)).to eq(described_class.new(ProductsIndex::Product, ProductsIndex::City)) }
37
+ specify { expect(described_class.new(ProductsIndex::City, CitiesIndex::City)).to eq(described_class.new(CitiesIndex::City, ProductsIndex::City)) }
38
+
39
+ specify { expect(described_class.new(ProductsIndex).limit(10)).to eq(described_class.new(ProductsIndex).limit(10)) }
40
+ specify { expect(described_class.new(ProductsIndex).limit(10)).not_to eq(described_class.new(ProductsIndex).limit(20)) }
41
+
42
+ specify { expect(ProductsIndex.limit(10)).to eq(ProductsIndex.limit(10)) }
43
+ specify { expect(ProductsIndex.limit(10)).not_to eq(CitiesIndex.limit(10)) }
44
+ end
45
+
46
+ describe '#render' do
47
+ specify do
48
+ expect(subject.render)
49
+ .to match(
50
+ index: %w[products],
51
+ type: array_including(%w[product city country]),
52
+ body: {}
53
+ )
54
+ end
55
+ end
56
+
57
+ describe '#inspect' do
58
+ specify do
59
+ expect(described_class.new(ProductsIndex).inspect)
60
+ .to eq('<Chewy::Search::Request {:index=>["products"], :type=>["product", "city", "country"], :body=>{}}>')
61
+ end
62
+ specify do
63
+ expect(ProductsIndex.limit(10).inspect)
64
+ .to eq('<ProductsIndex::Query {:index=>["products"], :type=>["product", "city", "country"], :body=>{:size=>10}}>')
65
+ end
66
+ end
67
+
68
+ %i[query post_filter].each do |name|
69
+ describe "##{name}" do
70
+ specify { expect(subject.send(name, match: {foo: 'bar'}).render[:body]).to include(name => {match: {foo: 'bar'}}) }
71
+ specify { expect(subject.send(name, nil)).to be_a described_class }
72
+ specify { expect(subject.send(name) { match foo: 'bar' }.render[:body]).to include(name => {match: {foo: 'bar'}}) }
73
+ specify do
74
+ expect(subject.send(name, match: {foo: 'bar'}).send(name) { multi_match foo: 'bar' }.render[:body])
75
+ .to include(name => {bool: {must: [{match: {foo: 'bar'}}, {multi_match: {foo: 'bar'}}]}})
76
+ end
77
+ specify { expect { subject.send(name, match: {foo: 'bar'}) }.not_to change { subject.render } }
78
+ specify do
79
+ expect(subject.send(name).should(match: {foo: 'bar'}).send(name).must_not { multi_match foo: 'bar' }.render[:body])
80
+ .to include(name => {bool: {should: {match: {foo: 'bar'}}, must_not: {multi_match: {foo: 'bar'}}}})
81
+ end
82
+
83
+ context do
84
+ let(:other_scope) { subject.send(name).should { multi_match foo: 'bar' }.send(name) { match foo: 'bar' } }
85
+
86
+ specify do
87
+ expect(subject.send(name).not(other_scope).render[:body])
88
+ .to include(name => {bool: {must_not: {bool: {must: {match: {foo: 'bar'}}, should: {multi_match: {foo: 'bar'}}}}}})
89
+ end
90
+ end
91
+ end
92
+ end
93
+
94
+ describe '#filter' do
95
+ specify { expect(subject.filter(match: {foo: 'bar'}).render[:body]).to include(query: {bool: {filter: {match: {foo: 'bar'}}}}) }
96
+ specify { expect(subject.filter(nil)).to be_a described_class }
97
+ specify { expect(subject.filter { match foo: 'bar' }.render[:body]).to include(query: {bool: {filter: {match: {foo: 'bar'}}}}) }
98
+ specify do
99
+ expect(subject.filter(match: {foo: 'bar'}).filter { multi_match foo: 'bar' }.render[:body])
100
+ .to include(query: {bool: {filter: [{match: {foo: 'bar'}}, {multi_match: {foo: 'bar'}}]}})
101
+ end
102
+ specify { expect { subject.filter(match: {foo: 'bar'}) }.not_to change { subject.render } }
103
+ specify do
104
+ expect(subject.filter.should(match: {foo: 'bar'}).filter.must_not { multi_match foo: 'bar' }.render[:body])
105
+ .to include(query: {bool: {filter: {bool: {should: {match: {foo: 'bar'}}, must_not: {multi_match: {foo: 'bar'}}}}}})
106
+ end
107
+
108
+ context do
109
+ let(:other_scope) { subject.filter.should { multi_match foo: 'bar' }.filter { match foo: 'bar' } }
110
+
111
+ specify do
112
+ expect(subject.filter.not(other_scope).render[:body])
113
+ .to include(query: {bool: {filter: {bool: {must_not: {bool: {must: {match: {foo: 'bar'}}, should: {multi_match: {foo: 'bar'}}}}}}}})
114
+ end
115
+ end
116
+ end
117
+
118
+ {limit: :size, offset: :from, terminate_after: :terminate_after}.each do |name, param_name|
119
+ describe "##{name}" do
120
+ specify { expect(subject.send(name, 10).render[:body]).to include(param_name => 10) }
121
+ specify { expect(subject.send(name, 10).send(name, 20).render[:body]).to include(param_name => 20) }
122
+ specify { expect(subject.send(name, 10).send(name, nil).render[:body]).to be_blank }
123
+ specify { expect { subject.send(name, 10) }.not_to change { subject.render } }
124
+ end
125
+ end
126
+
127
+ describe '#order' do
128
+ specify { expect(subject.order(:foo).render[:body]).to include(sort: ['foo']) }
129
+ specify { expect(subject.order(foo: 42).order(nil).render[:body]).to include(sort: ['foo' => 42]) }
130
+ specify { expect(subject.order(foo: 42).order(foo: 43).render[:body]).to include(sort: ['foo' => 43]) }
131
+ specify { expect(subject.order(:foo).order(:bar, :baz).render[:body]).to include(sort: %w[foo bar baz]) }
132
+ specify { expect(subject.order(nil).render[:body]).to be_blank }
133
+ specify { expect { subject.order(:foo) }.not_to change { subject.render } }
134
+ end
135
+
136
+ describe '#reorder' do
137
+ specify { expect(subject.reorder(:foo).render[:body]).to include(sort: ['foo']) }
138
+ specify { expect(subject.reorder(:foo).reorder(:bar, :baz).render[:body]).to include(sort: %w[bar baz]) }
139
+ specify { expect(subject.reorder(foo: 42).reorder(foo: 43).render[:body]).to include(sort: ['foo' => 43]) }
140
+ specify { expect(subject.reorder(foo: 42).reorder(nil).render[:body]).to be_blank }
141
+ specify { expect(subject.reorder(nil).render[:body]).to be_blank }
142
+ specify { expect { subject.reorder(:foo) }.not_to change { subject.render } }
143
+ end
144
+
145
+ %i[track_scores explain version profile].each do |name|
146
+ describe "##{name}" do
147
+ specify { expect(subject.send(name).render[:body]).to include(name => true) }
148
+ specify { expect(subject.send(name).send(name, false).render[:body]).to be_blank }
149
+ specify { expect { subject.send(name) }.not_to change { subject.render } }
150
+ end
151
+ end
152
+
153
+ describe '#request_cache' do
154
+ specify { expect(subject.request_cache(true).render[:body]).to include(request_cache: true) }
155
+ specify { expect(subject.request_cache(true).request_cache(false).render[:body]).to include(request_cache: false) }
156
+ specify { expect(subject.request_cache(true).request_cache(nil).render[:body]).to be_blank }
157
+ specify { expect { subject.request_cache(true) }.not_to change { subject.render } }
158
+ end
159
+
160
+ %i[search_type preference timeout].each do |name|
161
+ describe "##{name}" do
162
+ specify { expect(subject.send(name, :foo).render[:body]).to include(name => 'foo') }
163
+ specify { expect(subject.send(name, :foo).send(name, :bar).render[:body]).to include(name => 'bar') }
164
+ specify { expect(subject.send(name, :foo).send(name, nil).render[:body]).to be_blank }
165
+ specify { expect { subject.send(name, :foo) }.not_to change { subject.render } }
166
+ end
167
+ end
168
+
169
+ describe '#source' do
170
+ specify { expect(subject.source(:foo).render[:body]).to include(_source: ['foo']) }
171
+ specify { expect(subject.source(:foo, :bar).source(nil).render[:body]).to include(_source: %w[foo bar]) }
172
+ specify { expect(subject.source(%i[foo bar]).source(nil).render[:body]).to include(_source: %w[foo bar]) }
173
+ specify { expect(subject.source(excludes: :foo).render[:body]).to include(_source: {excludes: %w[foo]}) }
174
+ specify { expect(subject.source(excludes: :foo).source(excludes: %i[foo bar]).render[:body]).to include(_source: {excludes: %w[foo bar]}) }
175
+ specify { expect(subject.source(excludes: :foo).source(excludes: %i[foo bar]).render[:body]).to include(_source: {excludes: %w[foo bar]}) }
176
+ specify { expect(subject.source(excludes: :foo).source(:bar).render[:body]).to include(_source: {includes: %w[bar], excludes: %w[foo]}) }
177
+ specify { expect(subject.source(excludes: :foo).source(false).render[:body]).to include(_source: false) }
178
+ specify { expect(subject.source(excludes: :foo).source(false).source(excludes: :bar).render[:body]).to include(_source: {excludes: %w[foo bar]}) }
179
+ specify { expect(subject.source(excludes: :foo).source(false).source(true).render[:body]).to include(_source: {excludes: %w[foo]}) }
180
+ specify { expect(subject.source(nil).render[:body]).to be_blank }
181
+ specify { expect { subject.source(:foo) }.not_to change { subject.render } }
182
+ end
183
+
184
+ describe '#stored_fields' do
185
+ specify { expect(subject.stored_fields(:foo).render[:body]).to include(stored_fields: ['foo']) }
186
+ specify { expect(subject.stored_fields(%i[foo bar]).stored_fields(nil).render[:body]).to include(stored_fields: %w[foo bar]) }
187
+ specify { expect(subject.stored_fields(:foo).stored_fields(:foo, :bar).render[:body]).to include(stored_fields: %w[foo bar]) }
188
+ specify { expect(subject.stored_fields(:foo).stored_fields(false).render[:body]).to include(stored_fields: '_none_') }
189
+ specify { expect(subject.stored_fields(:foo).stored_fields(false).stored_fields(:bar).render[:body]).to include(stored_fields: %w[foo bar]) }
190
+ specify { expect(subject.stored_fields(:foo).stored_fields(false).stored_fields(true).render[:body]).to include(stored_fields: %w[foo]) }
191
+ specify { expect(subject.stored_fields(nil).render[:body]).to be_blank }
192
+ specify { expect { subject.stored_fields(:foo) }.not_to change { subject.render } }
193
+ end
194
+
195
+ %i[script_fields highlight].each do |name|
196
+ describe "##{name}" do
197
+ specify { expect(subject.send(name, foo: {bar: 42}).render[:body]).to include(name => {'foo' => {bar: 42}}) }
198
+ specify { expect(subject.send(name, foo: {bar: 42}).send(name, moo: {baz: 43}).render[:body]).to include(name => {'foo' => {bar: 42}, 'moo' => {baz: 43}}) }
199
+ specify { expect(subject.send(name, foo: {bar: 42}).send(name, nil).render[:body]).to include(name => {'foo' => {bar: 42}}) }
200
+ specify { expect { subject.send(name, foo: {bar: 42}) }.not_to change { subject.render } }
201
+ end
202
+ end
203
+
204
+ %i[suggest aggs].each do |name|
205
+ describe "##{name}" do
206
+ specify { expect(subject.send(name, foo: {bar: 42}).render[:body]).to include(name => {'foo' => {bar: 42}}) }
207
+ specify { expect(subject.send(name, foo: {bar: 42}).send(name, moo: {baz: 43}).render[:body]).to include(name => {'foo' => {bar: 42}, 'moo' => {baz: 43}}) }
208
+ specify { expect(subject.send(name, foo: {bar: 42}).send(name, nil).render[:body]).to include(name => {'foo' => {bar: 42}}) }
209
+ specify { expect { subject.send(name, foo: {bar: 42}) }.not_to change { subject.render } }
210
+ end
211
+ end
212
+
213
+ describe '#docvalue_fields' do
214
+ specify { expect(subject.docvalue_fields(:foo).render[:body]).to include(docvalue_fields: ['foo']) }
215
+ specify { expect(subject.docvalue_fields(%i[foo bar]).docvalue_fields(nil).render[:body]).to include(docvalue_fields: %w[foo bar]) }
216
+ specify { expect(subject.docvalue_fields(:foo).docvalue_fields(:foo, :bar).render[:body]).to include(docvalue_fields: %w[foo bar]) }
217
+ specify { expect(subject.docvalue_fields(nil).render[:body]).to be_blank }
218
+ specify { expect { subject.docvalue_fields(:foo) }.not_to change { subject.render } }
219
+ end
220
+
221
+ describe '#types' do
222
+ specify { expect(subject.types(:product).render[:type]).to contain_exactly('product') }
223
+ specify { expect(subject.types(%i[product city]).types(nil).render[:type]).to match_array(%w[product city]) }
224
+ specify { expect(subject.types(:product).types(:product, :city, :something).render[:type]).to match_array(%w[product city]) }
225
+ specify { expect(subject.types(nil).render[:body]).to be_blank }
226
+ specify { expect { subject.types(:product) }.not_to change { subject.render } }
227
+ end
228
+
229
+ describe '#indices_boost' do
230
+ specify { expect(subject.indices_boost(foo: 1.2).render[:body]).to include(indices_boost: [{'foo' => 1.2}]) }
231
+ specify { expect(subject.indices_boost(foo: 1.2).indices_boost(moo: 1.3).render[:body]).to include(indices_boost: [{'foo' => 1.2}, {'moo' => 1.3}]) }
232
+ specify { expect(subject.indices_boost(foo: 1.2).indices_boost(nil).render[:body]).to include(indices_boost: [{'foo' => 1.2}]) }
233
+ specify { expect { subject.indices_boost(foo: 1.2) }.not_to change { subject.render } }
234
+ end
235
+
236
+ describe '#rescore' do
237
+ specify { expect(subject.rescore(foo: 42).render[:body]).to include(rescore: [{foo: 42}]) }
238
+ specify { expect(subject.rescore(foo: 42).rescore(moo: 43).render[:body]).to include(rescore: [{foo: 42}, {moo: 43}]) }
239
+ specify { expect(subject.rescore(foo: 42).rescore(nil).render[:body]).to include(rescore: [{foo: 42}]) }
240
+ specify { expect { subject.rescore(foo: 42) }.not_to change { subject.render } }
241
+ end
242
+
243
+ describe '#min_score' do
244
+ specify { expect(subject.min_score(1.2).render[:body]).to include(min_score: 1.2) }
245
+ specify { expect(subject.min_score(1.2).min_score(0.5).render[:body]).to include(min_score: 0.5) }
246
+ specify { expect(subject.min_score(1.2).min_score(nil).render[:body]).to be_blank }
247
+ specify { expect { subject.min_score(1.2) }.not_to change { subject.render } }
248
+ end
249
+
250
+ describe '#search_after' do
251
+ specify { expect(subject.search_after(:foo, :bar).render[:body]).to include(search_after: %i[foo bar]) }
252
+ specify { expect(subject.search_after(%i[foo bar]).search_after(:baz).render[:body]).to include(search_after: [:baz]) }
253
+ specify { expect(subject.search_after(:foo).search_after(nil).render[:body]).to be_blank }
254
+ specify { expect { subject.search_after(:foo) }.not_to change { subject.render } }
255
+ end
256
+
257
+ context 'loading', :orm do
258
+ before do
259
+ stub_model(:city)
260
+ stub_model(:country)
261
+
262
+ stub_index(:places) do
263
+ define_type City do
264
+ field :rating, type: 'integer'
265
+ end
266
+
267
+ define_type Country do
268
+ field :rating, type: 'integer'
269
+ end
270
+ end
271
+ end
272
+
273
+ before { PlacesIndex.import!(cities: cities, countries: countries) }
274
+
275
+ let(:cities) { Array.new(2) { |i| City.create!(rating: i) } }
276
+ let(:countries) { Array.new(2) { |i| Country.create!(rating: i + 2) } }
277
+
278
+ subject { described_class.new(PlacesIndex).order(:rating) }
279
+
280
+ describe '#objects' do
281
+ specify { expect(subject.objects).to eq([*cities, *countries]) }
282
+ specify { expect(subject.objects.class).to eq(Array) }
283
+ end
284
+
285
+ describe '#load' do
286
+ specify { expect(subject.load(only: 'city')).to eq([*cities, *countries]) }
287
+ specify { expect(subject.load(only: 'city').map(&:class).uniq).to eq([PlacesIndex::City, PlacesIndex::Country]) }
288
+ specify { expect(subject.load(only: 'city').objects).to eq([*cities, nil, nil]) }
289
+ end
290
+ end
291
+
292
+ describe '#merge' do
293
+ let(:first) { described_class.new(ProductsIndex).limit(10).offset(10) }
294
+ let(:second) { described_class.new(ProductsIndex).offset(20).order(:foo) }
295
+
296
+ specify { expect(first.merge(second)).to eq(described_class.new(ProductsIndex).limit(10).offset(20).order(:foo)) }
297
+ specify { expect { first.merge(second) }.not_to change { first.render } }
298
+ specify { expect { first.merge(second) }.not_to change { second.render } }
299
+
300
+ specify { expect(second.merge(first)).to eq(described_class.new(ProductsIndex).limit(10).offset(10).order(:foo)) }
301
+ specify { expect { second.merge(first) }.not_to change { first.render } }
302
+ specify { expect { second.merge(first) }.not_to change { second.render } }
303
+ end
304
+
305
+ context do
306
+ let(:first_scope) { subject.query(foo: 'bar').filter.should(moo: 'baz').post_filter.must_not(boo: 'baf').limit(10) }
307
+ let(:second_scope) { subject.filter(foo: 'bar').post_filter.should(moo: 'baz').query.must_not(boo: 'baf').limit(20) }
308
+
309
+ describe '#and' do
310
+ specify do
311
+ expect(first_scope.and(second_scope).render[:body]).to eq(
312
+ query: {bool: {
313
+ must: [{foo: 'bar'}, {bool: {must_not: {boo: 'baf'}}}],
314
+ filter: [{bool: {should: {moo: 'baz'}}}, {foo: 'bar'}]
315
+ }},
316
+ post_filter: {bool: {must: [{bool: {must_not: {boo: 'baf'}}}, {bool: {should: {moo: 'baz'}}}]}},
317
+ size: 10
318
+ )
319
+ end
320
+ specify { expect { first_scope.and(second_scope) }.not_to change { first_scope.render } }
321
+ specify { expect { first_scope.and(second_scope) }.not_to change { second_scope.render } }
322
+ end
323
+
324
+ describe '#or' do
325
+ specify do
326
+ expect(first_scope.or(second_scope).render[:body]).to eq(
327
+ query: {bool: {
328
+ should: [{foo: 'bar'}, {bool: {must_not: {boo: 'baf'}}}],
329
+ filter: {bool: {should: [{bool: {should: {moo: 'baz'}}}, {foo: 'bar'}]}}
330
+ }},
331
+ post_filter: {bool: {should: [{bool: {must_not: {boo: 'baf'}}}, {bool: {should: {moo: 'baz'}}}]}},
332
+ size: 10
333
+ )
334
+ end
335
+ specify { expect { first_scope.or(second_scope) }.not_to change { first_scope.render } }
336
+ specify { expect { first_scope.or(second_scope) }.not_to change { second_scope.render } }
337
+ end
338
+
339
+ describe '#not' do
340
+ specify do
341
+ expect(first_scope.not(second_scope).render[:body]).to eq(
342
+ query: {bool: {
343
+ must: {foo: 'bar'}, must_not: {bool: {must_not: {boo: 'baf'}}},
344
+ filter: {bool: {should: {moo: 'baz'}, must_not: {foo: 'bar'}}}
345
+ }},
346
+ post_filter: {bool: {must_not: [{boo: 'baf'}, {bool: {should: {moo: 'baz'}}}]}},
347
+ size: 10
348
+ )
349
+ end
350
+ specify { expect { first_scope.not(second_scope) }.not_to change { first_scope.render } }
351
+ specify { expect { first_scope.not(second_scope) }.not_to change { second_scope.render } }
352
+ end
353
+ end
354
+
355
+ describe '#only' do
356
+ subject { described_class.new(ProductsIndex).limit(10).offset(10) }
357
+
358
+ specify { expect(subject.only(:limit)).to eq(described_class.new(ProductsIndex).limit(10)) }
359
+ specify { expect { subject.only(:limit) }.not_to change { subject.render } }
360
+ end
361
+
362
+ describe '#except' do
363
+ subject { described_class.new(ProductsIndex).limit(10).offset(10) }
364
+
365
+ specify { expect(subject.except(:limit)).to eq(described_class.new(ProductsIndex).offset(10)) }
366
+ specify { expect { subject.except(:limit) }.not_to change { subject.render } }
367
+ end
368
+
369
+ context 'index does not exist' do
370
+ specify { expect(subject).to eq([]) }
371
+ specify { expect(subject.count).to eq(0) }
372
+ end
373
+
374
+ context 'integration' do
375
+ let(:products) { Array.new(3) { |i| {id: i.next.to_i, name: "Name#{i.next}", age: 10 * i.next}.stringify_keys! } }
376
+ let(:cities) { Array.new(3) { |i| {id: (i.next + 3).to_i}.stringify_keys! } }
377
+ let(:countries) { Array.new(3) { |i| {id: (i.next + 6).to_i}.stringify_keys! } }
378
+ before do
379
+ ProductsIndex::Product.import!(products.map { |h| double(h) })
380
+ ProductsIndex::City.import!(cities.map { |h| double(h) })
381
+ ProductsIndex::Country.import!(countries.map { |h| double(h) })
382
+ CitiesIndex::City.import!(cities.map { |h| double(h) })
383
+ end
384
+
385
+ specify { expect(subject[0]._data).to be_a Hash }
386
+
387
+ context 'another index' do
388
+ subject { described_class.new(CitiesIndex) }
389
+
390
+ specify { expect(subject.count).to eq(3) }
391
+ specify { expect(subject.size).to eq(3) }
392
+ end
393
+
394
+ context 'limited types' do
395
+ subject { described_class.new(ProductsIndex::City, ProductsIndex::Country) }
396
+
397
+ specify { expect(subject.count).to eq(6) }
398
+ specify { expect(subject.size).to eq(6) }
399
+ end
400
+
401
+ context 'mixed types' do
402
+ subject { described_class.new(CitiesIndex, ProductsIndex::Product) }
403
+
404
+ specify { expect(subject.count).to eq(9) }
405
+ specify { expect(subject.size).to eq(9) }
406
+ end
407
+
408
+ context 'instrumentation' do
409
+ specify do
410
+ outer_payload = nil
411
+ ActiveSupport::Notifications.subscribe('search_query.chewy') do |_name, _start, _finish, _id, payload|
412
+ outer_payload = payload
413
+ end
414
+ subject.query(match: {name: 'name3'}).to_a
415
+ expect(outer_payload).to eq(
416
+ index: ProductsIndex,
417
+ indexes: [ProductsIndex],
418
+ request: {index: ['products'], type: %w[product city country], body: {query: {match: {name: 'name3'}}}},
419
+ type: [ProductsIndex::Product, ProductsIndex::City, ProductsIndex::Country],
420
+ types: [ProductsIndex::Product, ProductsIndex::City, ProductsIndex::Country]
421
+ )
422
+ end
423
+ end
424
+
425
+ describe '#none' do
426
+ specify { expect(subject.none).to eq([]) }
427
+ end
428
+
429
+ describe '#highlight' do
430
+ specify { expect(subject.query(match: {name: 'name3'}).highlight(fields: {name: {}}).first.name).to eq('Name3') }
431
+ specify { expect(subject.query(match: {name: 'name3'}).highlight(fields: {name: {}}).first.name_highlight).to eq('<em>Name3</em>') }
432
+ specify { expect(subject.query(match: {name: 'name3'}).highlight(fields: {name: {}}).first._data['_source']['name']).to eq('Name3') }
433
+ end
434
+
435
+ describe '#suggest' do
436
+ specify do
437
+ expect(subject.suggest(
438
+ foo: {
439
+ text: 'name',
440
+ term: {field: 'name'}
441
+ }
442
+ ).suggest).to eq(
443
+ 'foo' => [
444
+ {'text' => 'name', 'offset' => 0, 'length' => 4, 'options' => [
445
+ {'text' => 'name1', 'score' => 0.75, 'freq' => 1},
446
+ {'text' => 'name2', 'score' => 0.75, 'freq' => 1},
447
+ {'text' => 'name3', 'score' => 0.75, 'freq' => 1}
448
+ ]}
449
+ ]
450
+ )
451
+ end
452
+ end
453
+
454
+ describe '#aggs' do
455
+ specify do
456
+ expect(subject.aggs(avg_age: {avg: {field: :age}}).aggs)
457
+ .to eq('avg_age' => {'value' => 20.0})
458
+ end
459
+ end
460
+
461
+ describe '#size' do
462
+ specify { expect(subject.size).to eq(9) }
463
+ specify { expect(subject.limit(6).size).to eq(6) }
464
+ specify { expect(subject.offset(6).size).to eq(3) }
465
+ end
466
+
467
+ describe '#total' do
468
+ specify { expect(subject.limit(6).total).to eq(9) }
469
+ specify { expect(subject.limit(6).total_count).to eq(9) }
470
+ specify { expect(subject.offset(6).total_entries).to eq(9) }
471
+ end
472
+
473
+ describe '#count' do
474
+ specify { expect(subject.count).to eq(9) }
475
+ specify { expect(subject.limit(6).count).to eq(9) }
476
+ specify { expect(subject.offset(6).count).to eq(9) }
477
+ specify { expect(subject.types(:product, :something).count).to eq(3) }
478
+ specify { expect(subject.types(:product, :country).count).to eq(6) }
479
+ specify { expect(subject.filter(term: {age: 10}).count).to eq(1) }
480
+ specify { expect(subject.query(term: {age: 10}).count).to eq(1) }
481
+ specify { expect(subject.order(nil).count).to eq(9) }
482
+ specify { expect(subject.none.count).to eq(0) }
483
+
484
+ context do
485
+ before { expect(Chewy.client).to receive(:count).and_call_original }
486
+ specify { subject.count }
487
+ end
488
+
489
+ context do
490
+ subject { described_class.new(ProductsIndex).limit(6) }
491
+ before do
492
+ expect(Chewy.client).not_to receive(:count)
493
+ subject.total
494
+ end
495
+ specify { expect(subject.count).to eq(9) }
496
+ end
497
+ end
498
+
499
+ describe '#exists?' do
500
+ before { expect(Chewy.client).to receive(:search).once.and_call_original }
501
+
502
+ specify { expect(subject.exists?).to be(true) }
503
+ specify { expect(subject.filter(match: {name: 'foo'}).exist?).to be(false) }
504
+
505
+ context do
506
+ before { subject.total }
507
+ specify { expect(subject.exists?).to eq(true) }
508
+ end
509
+ end
510
+
511
+ describe '#first' do
512
+ subject { described_class.new(ProductsIndex).order(id: {order: 'desc'}) }
513
+
514
+ context do
515
+ before { expect(Chewy.client).to receive(:search).once.and_call_original }
516
+
517
+ specify { expect(subject.first).to be_a(ProductsIndex::Country).and have_attributes(id: 9) }
518
+ specify { expect(subject.first(3).map(&:id)).to eq([9, 8, 7]) }
519
+ specify { expect(subject.first(10).map(&:id)).to have(9).items }
520
+ specify { expect(subject.limit(5).first(10).map(&:id)).to have(9).items }
521
+ specify { expect(subject.terminate_after(5).first(10).map(&:id)).to have(5).items }
522
+ end
523
+
524
+ context do
525
+ before do
526
+ subject.response
527
+ expect(Chewy.client).not_to receive(:search)
528
+ end
529
+
530
+ specify { expect(subject.first).to be_a(ProductsIndex::Country).and have_attributes(id: 9) }
531
+ specify { expect(subject.first(3).map(&:id)).to eq([9, 8, 7]) }
532
+ specify { expect(subject.first(10).map(&:id)).to have(9).items }
533
+
534
+ context do
535
+ subject { described_class.new(ProductsIndex).terminate_after(5) }
536
+ specify { expect(subject.first(10).map(&:id)).to have(5).items }
537
+ end
538
+ end
539
+
540
+ context do
541
+ subject { described_class.new(ProductsIndex).limit(5) }
542
+ before do
543
+ subject.response
544
+ expect(Chewy.client).to receive(:search).once.and_call_original
545
+ end
546
+ specify { expect(subject.first(10).map(&:id)).to have(9).items }
547
+ end
548
+ end
549
+
550
+ describe '#find' do
551
+ specify { expect(subject.find('1')).to be_a(ProductsIndex::Product).and have_attributes(id: 1) }
552
+ specify { expect(subject.find { |w| w.id == 2 }).to be_a(ProductsIndex::Product).and have_attributes(id: 2) }
553
+ specify { expect(subject.limit(2).find('1', '3', '7').map(&:id)).to contain_exactly(1, 3, 7) }
554
+ specify { expect(subject.find(1, 3, 7).map(&:id)).to contain_exactly(1, 3, 7) }
555
+ specify { expect { subject.find('1', '3', '42') }.to raise_error Chewy::DocumentNotFound, 'Could not find documents for ids: 42' }
556
+ specify { expect { subject.find(1, 3, 42) }.to raise_error Chewy::DocumentNotFound, 'Could not find documents for ids: 42' }
557
+ specify { expect { subject.query(match: {name: 'name3'}).find('1', '3') }.to raise_error Chewy::DocumentNotFound, 'Could not find documents for ids: 1' }
558
+ specify { expect { subject.query(match: {name: 'name2'}).find('1', '3') }.to raise_error Chewy::DocumentNotFound, 'Could not find documents for ids: 1 and 3' }
559
+ specify { expect { subject.filter(match: {name: 'name2'}).find('1', '3') }.to raise_error Chewy::DocumentNotFound, 'Could not find documents for ids: 1 and 3' }
560
+ specify { expect { subject.post_filter(match: {name: 'name2'}).find('1', '3') }.to raise_error Chewy::DocumentNotFound, 'Could not find documents for ids: 1 and 3' }
561
+
562
+ context 'make sure it returns everything' do
563
+ let(:countries) { Array.new(6) { |i| {id: (i.next + 6).to_i}.stringify_keys! } }
564
+ before { expect(Chewy.client).not_to receive(:scroll) }
565
+
566
+ specify { expect(subject.find((1..12).to_a)).to have(12).items }
567
+ end
568
+
569
+ context 'make sure it returns everything in batches if needed' do
570
+ before { stub_const("#{described_class}::DEFAULT_BATCH_SIZE", 5) }
571
+ before { expect(Chewy.client).to receive(:scroll).once.and_call_original }
572
+
573
+ specify { expect(subject.find((1..9).to_a)).to have(9).items }
574
+ specify { expect(subject.find((1..9).to_a)).to all be_a(Chewy::Type) }
575
+ end
576
+ end
577
+
578
+ describe '#pluck' do
579
+ specify { expect(subject.limit(5).pluck(:_id)).to eq(%w[1 2 3 4 5]) }
580
+ specify { expect(subject.limit(5).pluck(:_id, :age)).to eq([['1', 10], ['2', 20], ['3', 30], ['4', nil], ['5', nil]]) }
581
+ specify { expect(subject.limit(5).source(:name).pluck(:id, :age)).to eq([[1, 10], [2, 20], [3, 30], [4, nil], [5, nil]]) }
582
+ specify do
583
+ expect(subject.limit(5).pluck(:_index, :_type, :name)).to eq([
584
+ %w[products product Name1],
585
+ %w[products product Name2],
586
+ %w[products product Name3],
587
+ ['products', 'city', nil],
588
+ ['products', 'city', nil]
589
+ ])
590
+ end
591
+
592
+ context 'make sure it returns everything in batches if needed' do
593
+ before { stub_const("#{described_class}::DEFAULT_PLUCK_BATCH_SIZE", 5) }
594
+ before { expect(Chewy.client).to receive(:scroll).once.and_call_original }
595
+
596
+ specify { expect(subject.pluck(:_id)).to eq((1..9).to_a.map(&:to_s)) }
597
+ end
598
+ end
599
+
600
+ describe '#delete_all' do
601
+ specify do
602
+ expect do
603
+ subject.none.delete_all
604
+ Chewy.client.indices.refresh(index: 'products')
605
+ end.not_to change { described_class.new(ProductsIndex).total }.from(9)
606
+ end
607
+ specify do
608
+ expect do
609
+ subject.query(match: {name: 'name3'}).delete_all
610
+ Chewy.client.indices.refresh(index: 'products')
611
+ end.to change { described_class.new(ProductsIndex).total }.from(9).to(8)
612
+ end
613
+ specify do
614
+ expect do
615
+ subject.filter(range: {age: {gte: 10, lte: 20}}).delete_all
616
+ Chewy.client.indices.refresh(index: 'products')
617
+ end.to change { described_class.new(ProductsIndex).total_count }.from(9).to(7)
618
+ end
619
+ specify do
620
+ expect do
621
+ subject.types(:product).delete_all
622
+ Chewy.client.indices.refresh(index: 'products')
623
+ end.to change { described_class.new(ProductsIndex::Product).total_entries }.from(3).to(0)
624
+ end
625
+ specify do
626
+ expect do
627
+ subject.delete_all
628
+ Chewy.client.indices.refresh(index: 'products')
629
+ end.to change { described_class.new(ProductsIndex).total }.from(9).to(0)
630
+ end
631
+ specify do
632
+ expect do
633
+ described_class.new(ProductsIndex::City).delete_all
634
+ Chewy.client.indices.refresh(index: 'products')
635
+ end.to change { described_class.new(ProductsIndex).total }.from(9).to(6)
636
+ end
637
+
638
+ specify do
639
+ outer_payload = nil
640
+ ActiveSupport::Notifications.subscribe('delete_query.chewy') do |_name, _start, _finish, _id, payload|
641
+ outer_payload = payload
642
+ end
643
+ subject.query(match: {name: 'name3'}).delete_all
644
+ expect(outer_payload).to eq(
645
+ index: ProductsIndex,
646
+ indexes: [ProductsIndex],
647
+ request: {index: ['products'], type: %w[product city country], body: {query: {match: {name: 'name3'}}}, refresh: true},
648
+ type: [ProductsIndex::Product, ProductsIndex::City, ProductsIndex::Country],
649
+ types: [ProductsIndex::Product, ProductsIndex::City, ProductsIndex::Country]
650
+ )
651
+ end
652
+
653
+ specify do
654
+ outer_payload = nil
655
+ ActiveSupport::Notifications.subscribe('delete_query.chewy') do |_name, _start, _finish, _id, payload|
656
+ outer_payload = payload
657
+ end
658
+ subject.query(match: {name: 'name3'}).delete_all(refresh: false)
659
+ expect(outer_payload).to eq(
660
+ index: ProductsIndex,
661
+ indexes: [ProductsIndex],
662
+ request: {index: ['products'], type: %w[product city country], body: {query: {match: {name: 'name3'}}}, refresh: false},
663
+ type: [ProductsIndex::Product, ProductsIndex::City, ProductsIndex::Country],
664
+ types: [ProductsIndex::Product, ProductsIndex::City, ProductsIndex::Country]
665
+ )
666
+ end
667
+ end
668
+ end
669
+ end