chewy 0.9.0 → 5.2.0

Sign up to get free protection for your applications and to get access to all the features.
Files changed (275) hide show
  1. checksums.yaml +5 -5
  2. data/.circleci/config.yml +214 -0
  3. data/.gitignore +1 -0
  4. data/.rubocop.yml +41 -19
  5. data/.rubocop_todo.yml +2 -2
  6. data/.yardopts +5 -0
  7. data/Appraisals +58 -28
  8. data/CHANGELOG.md +153 -12
  9. data/Gemfile +20 -12
  10. data/LEGACY_DSL.md +497 -0
  11. data/LICENSE.txt +1 -1
  12. data/README.md +338 -528
  13. data/chewy.gemspec +11 -12
  14. data/gemfiles/rails.5.2.activerecord.gemfile +17 -0
  15. data/gemfiles/rails.5.2.mongoid.6.4.gemfile +17 -0
  16. data/gemfiles/rails.6.0.activerecord.gemfile +17 -0
  17. data/gemfiles/rails.6.1.activerecord.gemfile +19 -0
  18. data/gemfiles/ruby3.gemfile +10 -0
  19. data/gemfiles/sequel.4.45.gemfile +11 -0
  20. data/lib/chewy.rb +79 -44
  21. data/lib/chewy/backports/duplicable.rb +1 -1
  22. data/lib/chewy/config.rb +43 -17
  23. data/lib/chewy/errors.rb +2 -2
  24. data/lib/chewy/fields/base.rb +56 -31
  25. data/lib/chewy/fields/root.rb +44 -11
  26. data/lib/chewy/index.rb +237 -149
  27. data/lib/chewy/index/actions.rb +100 -35
  28. data/lib/chewy/index/aliases.rb +2 -1
  29. data/lib/chewy/index/settings.rb +11 -5
  30. data/lib/chewy/index/specification.rb +60 -0
  31. data/lib/chewy/journal.rb +40 -92
  32. data/lib/chewy/minitest/helpers.rb +6 -6
  33. data/lib/chewy/minitest/search_index_receiver.rb +17 -17
  34. data/lib/chewy/query.rb +182 -122
  35. data/lib/chewy/query/compose.rb +13 -13
  36. data/lib/chewy/query/criteria.rb +13 -13
  37. data/lib/chewy/query/filters.rb +21 -4
  38. data/lib/chewy/query/loading.rb +1 -2
  39. data/lib/chewy/query/nodes/and.rb +2 -2
  40. data/lib/chewy/query/nodes/bool.rb +1 -1
  41. data/lib/chewy/query/nodes/equal.rb +2 -2
  42. data/lib/chewy/query/nodes/exists.rb +1 -1
  43. data/lib/chewy/query/nodes/field.rb +1 -1
  44. data/lib/chewy/query/nodes/has_relation.rb +2 -2
  45. data/lib/chewy/query/nodes/match_all.rb +1 -1
  46. data/lib/chewy/query/nodes/missing.rb +1 -1
  47. data/lib/chewy/query/nodes/not.rb +2 -2
  48. data/lib/chewy/query/nodes/or.rb +2 -2
  49. data/lib/chewy/query/nodes/prefix.rb +1 -1
  50. data/lib/chewy/query/nodes/query.rb +2 -2
  51. data/lib/chewy/query/nodes/range.rb +4 -4
  52. data/lib/chewy/query/nodes/regexp.rb +4 -4
  53. data/lib/chewy/query/nodes/script.rb +3 -3
  54. data/lib/chewy/query/pagination.rb +10 -1
  55. data/lib/chewy/railtie.rb +4 -3
  56. data/lib/chewy/rake_helper.rb +265 -48
  57. data/lib/chewy/rspec/update_index.rb +33 -27
  58. data/lib/chewy/search.rb +79 -26
  59. data/lib/chewy/search/loader.rb +83 -0
  60. data/lib/chewy/{query → search}/pagination/kaminari.rb +13 -5
  61. data/lib/chewy/search/pagination/will_paginate.rb +43 -0
  62. data/lib/chewy/search/parameters.rb +168 -0
  63. data/lib/chewy/search/parameters/aggs.rb +16 -0
  64. data/lib/chewy/search/parameters/allow_partial_search_results.rb +27 -0
  65. data/lib/chewy/search/parameters/concerns/bool_storage.rb +24 -0
  66. data/lib/chewy/search/parameters/concerns/hash_storage.rb +23 -0
  67. data/lib/chewy/search/parameters/concerns/integer_storage.rb +14 -0
  68. data/lib/chewy/search/parameters/concerns/query_storage.rb +238 -0
  69. data/lib/chewy/search/parameters/concerns/string_array_storage.rb +23 -0
  70. data/lib/chewy/search/parameters/concerns/string_storage.rb +14 -0
  71. data/lib/chewy/search/parameters/docvalue_fields.rb +12 -0
  72. data/lib/chewy/search/parameters/explain.rb +16 -0
  73. data/lib/chewy/search/parameters/filter.rb +47 -0
  74. data/lib/chewy/search/parameters/highlight.rb +16 -0
  75. data/lib/chewy/search/parameters/indices.rb +123 -0
  76. data/lib/chewy/search/parameters/indices_boost.rb +52 -0
  77. data/lib/chewy/search/parameters/limit.rb +17 -0
  78. data/lib/chewy/search/parameters/load.rb +32 -0
  79. data/lib/chewy/search/parameters/min_score.rb +16 -0
  80. data/lib/chewy/search/parameters/none.rb +27 -0
  81. data/lib/chewy/search/parameters/offset.rb +17 -0
  82. data/lib/chewy/search/parameters/order.rb +64 -0
  83. data/lib/chewy/search/parameters/post_filter.rb +19 -0
  84. data/lib/chewy/search/parameters/preference.rb +16 -0
  85. data/lib/chewy/search/parameters/profile.rb +16 -0
  86. data/lib/chewy/search/parameters/query.rb +19 -0
  87. data/lib/chewy/search/parameters/request_cache.rb +27 -0
  88. data/lib/chewy/search/parameters/rescore.rb +29 -0
  89. data/lib/chewy/search/parameters/script_fields.rb +16 -0
  90. data/lib/chewy/search/parameters/search_after.rb +20 -0
  91. data/lib/chewy/search/parameters/search_type.rb +16 -0
  92. data/lib/chewy/search/parameters/source.rb +73 -0
  93. data/lib/chewy/search/parameters/storage.rb +95 -0
  94. data/lib/chewy/search/parameters/stored_fields.rb +63 -0
  95. data/lib/chewy/search/parameters/suggest.rb +16 -0
  96. data/lib/chewy/search/parameters/terminate_after.rb +16 -0
  97. data/lib/chewy/search/parameters/timeout.rb +16 -0
  98. data/lib/chewy/search/parameters/track_scores.rb +16 -0
  99. data/lib/chewy/search/parameters/types.rb +20 -0
  100. data/lib/chewy/search/parameters/version.rb +16 -0
  101. data/lib/chewy/search/query_proxy.rb +257 -0
  102. data/lib/chewy/search/request.rb +1046 -0
  103. data/lib/chewy/search/response.rb +119 -0
  104. data/lib/chewy/search/scoping.rb +50 -0
  105. data/lib/chewy/search/scrolling.rb +134 -0
  106. data/lib/chewy/stash.rb +79 -0
  107. data/lib/chewy/strategy.rb +10 -3
  108. data/lib/chewy/strategy/active_job.rb +2 -1
  109. data/lib/chewy/strategy/atomic.rb +2 -4
  110. data/lib/chewy/strategy/bypass.rb +1 -1
  111. data/lib/chewy/strategy/resque.rb +1 -0
  112. data/lib/chewy/strategy/shoryuken.rb +40 -0
  113. data/lib/chewy/strategy/sidekiq.rb +13 -3
  114. data/lib/chewy/type.rb +29 -7
  115. data/lib/chewy/type/actions.rb +26 -2
  116. data/lib/chewy/type/adapter/active_record.rb +44 -29
  117. data/lib/chewy/type/adapter/base.rb +27 -7
  118. data/lib/chewy/type/adapter/mongoid.rb +19 -10
  119. data/lib/chewy/type/adapter/object.rb +187 -26
  120. data/lib/chewy/type/adapter/orm.rb +59 -32
  121. data/lib/chewy/type/adapter/sequel.rb +33 -19
  122. data/lib/chewy/type/crutch.rb +1 -1
  123. data/lib/chewy/type/import.rb +146 -191
  124. data/lib/chewy/type/import/bulk_builder.rb +122 -0
  125. data/lib/chewy/type/import/bulk_request.rb +78 -0
  126. data/lib/chewy/type/import/journal_builder.rb +45 -0
  127. data/lib/chewy/type/import/routine.rb +138 -0
  128. data/lib/chewy/type/mapping.rb +51 -35
  129. data/lib/chewy/type/observe.rb +17 -13
  130. data/lib/chewy/type/syncer.rb +222 -0
  131. data/lib/chewy/type/witchcraft.rb +32 -16
  132. data/lib/chewy/type/wrapper.rb +30 -4
  133. data/lib/chewy/version.rb +1 -1
  134. data/lib/sequel/plugins/chewy_observe.rb +4 -19
  135. data/lib/tasks/chewy.rake +84 -26
  136. data/spec/chewy/config_spec.rb +98 -1
  137. data/spec/chewy/fields/base_spec.rb +170 -135
  138. data/spec/chewy/fields/root_spec.rb +124 -20
  139. data/spec/chewy/fields/time_fields_spec.rb +2 -3
  140. data/spec/chewy/index/actions_spec.rb +214 -52
  141. data/spec/chewy/index/aliases_spec.rb +2 -2
  142. data/spec/chewy/index/settings_spec.rb +67 -38
  143. data/spec/chewy/index/specification_spec.rb +169 -0
  144. data/spec/chewy/index_spec.rb +108 -64
  145. data/spec/chewy/journal_spec.rb +150 -55
  146. data/spec/chewy/minitest/helpers_spec.rb +4 -4
  147. data/spec/chewy/minitest/search_index_receiver_spec.rb +1 -1
  148. data/spec/chewy/query/criteria_spec.rb +179 -179
  149. data/spec/chewy/query/filters_spec.rb +16 -16
  150. data/spec/chewy/query/loading_spec.rb +22 -20
  151. data/spec/chewy/query/nodes/and_spec.rb +2 -2
  152. data/spec/chewy/query/nodes/bool_spec.rb +4 -4
  153. data/spec/chewy/query/nodes/equal_spec.rb +19 -19
  154. data/spec/chewy/query/nodes/exists_spec.rb +6 -6
  155. data/spec/chewy/query/nodes/has_child_spec.rb +19 -19
  156. data/spec/chewy/query/nodes/has_parent_spec.rb +19 -19
  157. data/spec/chewy/query/nodes/missing_spec.rb +5 -5
  158. data/spec/chewy/query/nodes/not_spec.rb +4 -2
  159. data/spec/chewy/query/nodes/or_spec.rb +2 -2
  160. data/spec/chewy/query/nodes/prefix_spec.rb +5 -5
  161. data/spec/chewy/query/nodes/query_spec.rb +2 -2
  162. data/spec/chewy/query/nodes/range_spec.rb +18 -18
  163. data/spec/chewy/query/nodes/raw_spec.rb +1 -1
  164. data/spec/chewy/query/nodes/regexp_spec.rb +14 -14
  165. data/spec/chewy/query/nodes/script_spec.rb +4 -4
  166. data/spec/chewy/query/pagination/kaminari_spec.rb +3 -55
  167. data/spec/chewy/query/pagination/will_paginate_spec.rb +5 -0
  168. data/spec/chewy/query/pagination_spec.rb +25 -21
  169. data/spec/chewy/query_spec.rb +503 -561
  170. data/spec/chewy/rake_helper_spec.rb +381 -0
  171. data/spec/chewy/repository_spec.rb +4 -4
  172. data/spec/chewy/rspec/update_index_spec.rb +89 -56
  173. data/spec/chewy/runtime_spec.rb +2 -2
  174. data/spec/chewy/search/loader_spec.rb +117 -0
  175. data/spec/chewy/search/pagination/kaminari_examples.rb +71 -0
  176. data/spec/chewy/search/pagination/kaminari_spec.rb +21 -0
  177. data/spec/chewy/search/pagination/will_paginate_examples.rb +63 -0
  178. data/spec/chewy/search/pagination/will_paginate_spec.rb +23 -0
  179. data/spec/chewy/search/parameters/aggs_spec.rb +5 -0
  180. data/spec/chewy/search/parameters/bool_storage_examples.rb +53 -0
  181. data/spec/chewy/search/parameters/docvalue_fields_spec.rb +5 -0
  182. data/spec/chewy/search/parameters/explain_spec.rb +5 -0
  183. data/spec/chewy/search/parameters/filter_spec.rb +5 -0
  184. data/spec/chewy/search/parameters/hash_storage_examples.rb +59 -0
  185. data/spec/chewy/search/parameters/highlight_spec.rb +5 -0
  186. data/spec/chewy/search/parameters/indices_spec.rb +191 -0
  187. data/spec/chewy/search/parameters/integer_storage_examples.rb +32 -0
  188. data/spec/chewy/search/parameters/limit_spec.rb +5 -0
  189. data/spec/chewy/search/parameters/load_spec.rb +60 -0
  190. data/spec/chewy/search/parameters/min_score_spec.rb +32 -0
  191. data/spec/chewy/search/parameters/none_spec.rb +5 -0
  192. data/spec/chewy/search/parameters/offset_spec.rb +5 -0
  193. data/spec/chewy/search/parameters/order_spec.rb +65 -0
  194. data/spec/chewy/search/parameters/post_filter_spec.rb +5 -0
  195. data/spec/chewy/search/parameters/preference_spec.rb +5 -0
  196. data/spec/chewy/search/parameters/profile_spec.rb +5 -0
  197. data/spec/chewy/search/parameters/query_spec.rb +5 -0
  198. data/spec/chewy/search/parameters/query_storage_examples.rb +388 -0
  199. data/spec/chewy/search/parameters/request_cache_spec.rb +67 -0
  200. data/spec/chewy/search/parameters/rescore_spec.rb +62 -0
  201. data/spec/chewy/search/parameters/script_fields_spec.rb +5 -0
  202. data/spec/chewy/search/parameters/search_after_spec.rb +32 -0
  203. data/spec/chewy/search/parameters/search_type_spec.rb +5 -0
  204. data/spec/chewy/search/parameters/source_spec.rb +156 -0
  205. data/spec/chewy/search/parameters/storage_spec.rb +60 -0
  206. data/spec/chewy/search/parameters/stored_fields_spec.rb +126 -0
  207. data/spec/chewy/search/parameters/string_array_storage_examples.rb +63 -0
  208. data/spec/chewy/search/parameters/string_storage_examples.rb +32 -0
  209. data/spec/chewy/search/parameters/suggest_spec.rb +5 -0
  210. data/spec/chewy/search/parameters/terminate_after_spec.rb +5 -0
  211. data/spec/chewy/search/parameters/timeout_spec.rb +5 -0
  212. data/spec/chewy/search/parameters/track_scores_spec.rb +5 -0
  213. data/spec/chewy/search/parameters/types_spec.rb +5 -0
  214. data/spec/chewy/search/parameters/version_spec.rb +5 -0
  215. data/spec/chewy/search/parameters_spec.rb +147 -0
  216. data/spec/chewy/search/query_proxy_spec.rb +68 -0
  217. data/spec/chewy/search/request_spec.rb +685 -0
  218. data/spec/chewy/search/response_spec.rb +198 -0
  219. data/spec/chewy/search/scrolling_spec.rb +169 -0
  220. data/spec/chewy/search_spec.rb +33 -16
  221. data/spec/chewy/stash_spec.rb +95 -0
  222. data/spec/chewy/strategy/active_job_spec.rb +21 -2
  223. data/spec/chewy/strategy/resque_spec.rb +6 -0
  224. data/spec/chewy/strategy/shoryuken_spec.rb +70 -0
  225. data/spec/chewy/strategy/sidekiq_spec.rb +13 -1
  226. data/spec/chewy/strategy_spec.rb +6 -6
  227. data/spec/chewy/type/actions_spec.rb +29 -10
  228. data/spec/chewy/type/adapter/active_record_spec.rb +203 -91
  229. data/spec/chewy/type/adapter/mongoid_spec.rb +112 -54
  230. data/spec/chewy/type/adapter/object_spec.rb +101 -28
  231. data/spec/chewy/type/adapter/sequel_spec.rb +149 -82
  232. data/spec/chewy/type/import/bulk_builder_spec.rb +279 -0
  233. data/spec/chewy/type/import/bulk_request_spec.rb +102 -0
  234. data/spec/chewy/type/import/journal_builder_spec.rb +95 -0
  235. data/spec/chewy/type/import/routine_spec.rb +110 -0
  236. data/spec/chewy/type/import_spec.rb +356 -271
  237. data/spec/chewy/type/mapping_spec.rb +96 -29
  238. data/spec/chewy/type/observe_spec.rb +9 -5
  239. data/spec/chewy/type/syncer_spec.rb +123 -0
  240. data/spec/chewy/type/witchcraft_spec.rb +61 -29
  241. data/spec/chewy/type/wrapper_spec.rb +63 -23
  242. data/spec/chewy/type_spec.rb +28 -7
  243. data/spec/chewy_spec.rb +75 -7
  244. data/spec/spec_helper.rb +17 -3
  245. data/spec/support/active_record.rb +5 -1
  246. data/spec/support/class_helpers.rb +0 -14
  247. data/spec/support/mongoid.rb +15 -3
  248. data/spec/support/sequel.rb +6 -1
  249. metadata +219 -58
  250. data/.travis.yml +0 -36
  251. data/gemfiles/rails.3.2.activerecord.gemfile +0 -16
  252. data/gemfiles/rails.3.2.activerecord.kaminari.gemfile +0 -15
  253. data/gemfiles/rails.3.2.activerecord.will_paginate.gemfile +0 -15
  254. data/gemfiles/rails.4.2.activerecord.gemfile +0 -17
  255. data/gemfiles/rails.4.2.activerecord.kaminari.gemfile +0 -16
  256. data/gemfiles/rails.4.2.activerecord.will_paginate.gemfile +0 -16
  257. data/gemfiles/rails.4.2.mongoid.4.0.gemfile +0 -16
  258. data/gemfiles/rails.4.2.mongoid.4.0.kaminari.gemfile +0 -15
  259. data/gemfiles/rails.4.2.mongoid.4.0.will_paginate.gemfile +0 -15
  260. data/gemfiles/rails.4.2.mongoid.5.1.gemfile +0 -16
  261. data/gemfiles/rails.4.2.mongoid.5.1.kaminari.gemfile +0 -15
  262. data/gemfiles/rails.4.2.mongoid.5.1.will_paginate.gemfile +0 -15
  263. data/gemfiles/rails.5.0.activerecord.gemfile +0 -17
  264. data/gemfiles/rails.5.0.activerecord.kaminari.gemfile +0 -16
  265. data/gemfiles/rails.5.0.activerecord.will_paginate.gemfile +0 -16
  266. data/gemfiles/sequel.4.38.gemfile +0 -14
  267. data/lib/chewy/journal/apply.rb +0 -31
  268. data/lib/chewy/journal/clean.rb +0 -24
  269. data/lib/chewy/journal/entry.rb +0 -83
  270. data/lib/chewy/journal/query.rb +0 -87
  271. data/lib/chewy/query/pagination/will_paginate.rb +0 -27
  272. data/lib/chewy/query/scoping.rb +0 -20
  273. data/spec/chewy/journal/apply_spec.rb +0 -120
  274. data/spec/chewy/journal/entry_spec.rb +0 -237
  275. data/spec/chewy/query/pagination/will_paginage_spec.rb +0 -59
@@ -4,6 +4,8 @@ describe Chewy::Type::Adapter::Sequel, :sequel do
4
4
  before do
5
5
  stub_model(:city)
6
6
  stub_model(:country)
7
+ City.many_to_one :country
8
+ Country.one_to_many :cities
7
9
  end
8
10
 
9
11
  describe '#name' do
@@ -75,54 +77,54 @@ describe Chewy::Type::Adapter::Sequel, :sequel do
75
77
  let!(:deleted) { Array.new(4) { City.create!.tap(&:destroy) } }
76
78
  subject { described_class.new(City) }
77
79
 
78
- specify { expect(import).to eq([{ index: cities }]) }
80
+ specify { expect(import).to eq([{index: cities}]) }
79
81
  specify { expect(import(nil)).to eq([]) }
80
82
 
81
- specify { expect(import(City.order(:id))).to eq([{ index: cities }]) }
83
+ specify { expect(import(City.order(:id))).to eq([{index: cities}]) }
82
84
  specify do
83
85
  expect(import(City.order(:id), batch_size: 2))
84
- .to eq([{ index: cities.first(2) }, { index: cities.last(1) }])
86
+ .to eq([{index: cities.first(2)}, {index: cities.last(1)}])
85
87
  end
86
88
 
87
- specify { expect(import(cities)).to eq([{ index: cities }]) }
89
+ specify { expect(import(cities)).to eq([{index: cities}]) }
88
90
  specify do
89
91
  expect(import(cities, batch_size: 2))
90
- .to eq([{ index: cities.first(2) }, { index: cities.last(1) }])
92
+ .to eq([{index: cities.first(2)}, {index: cities.last(1)}])
91
93
  end
92
94
  specify do
93
95
  expect(import(cities, deleted))
94
- .to eq([{ index: cities }, { delete: deleted }])
96
+ .to eq([{index: cities}, {delete: deleted}])
95
97
  end
96
98
  specify do
97
99
  expect(import(cities, deleted, batch_size: 2)).to eq([
98
- { index: cities.first(2) },
99
- { index: cities.last(1) },
100
- { delete: deleted.first(2) },
101
- { delete: deleted.last(2) }
100
+ {index: cities.first(2)},
101
+ {index: cities.last(1)},
102
+ {delete: deleted.first(2)},
103
+ {delete: deleted.last(2)}
102
104
  ])
103
105
  end
104
106
 
105
- specify { expect(import(cities.map(&:id))).to eq([{ index: cities }]) }
106
- specify { expect(import(deleted.map(&:id))).to eq([{ delete: deleted.map(&:id) }]) }
107
+ specify { expect(import(cities.map(&:id))).to eq([{index: cities}]) }
108
+ specify { expect(import(deleted.map(&:id))).to eq([{delete: deleted.map(&:id)}]) }
107
109
  specify do
108
110
  expect(import(cities.map(&:id), batch_size: 2))
109
- .to eq([{ index: cities.first(2) }, { index: cities.last(1) }])
111
+ .to eq([{index: cities.first(2)}, {index: cities.last(1)}])
110
112
  end
111
113
  specify do
112
114
  expect(import(cities.map(&:id), deleted.map(&:id)))
113
- .to eq([{ index: cities }, { delete: deleted.map(&:id) }])
115
+ .to eq([{index: cities}, {delete: deleted.map(&:id)}])
114
116
  end
115
117
  specify do
116
118
  expect(import(cities.map(&:id), deleted.map(&:id), batch_size: 2)).to eq([
117
- { index: cities.first(2) },
118
- { index: cities.last(1) },
119
- { delete: deleted.first(2).map(&:id) },
120
- { delete: deleted.last(2).map(&:id) }
119
+ {index: cities.first(2)},
120
+ {index: cities.last(1)},
121
+ {delete: deleted.first(2).map(&:id)},
122
+ {delete: deleted.last(2).map(&:id)}
121
123
  ])
122
124
  end
123
125
 
124
- specify { expect(import(cities.first, nil)).to eq([{ index: [cities.first] }]) }
125
- specify { expect(import(cities.first.id, nil)).to eq([{ index: [cities.first] }]) }
126
+ specify { expect(import(cities.first, nil)).to eq([{index: [cities.first]}]) }
127
+ specify { expect(import(cities.first.id, nil)).to eq([{index: [cities.first]}]) }
126
128
  end
127
129
 
128
130
  context 'additional delete conditions' do
@@ -141,19 +143,19 @@ describe Chewy::Type::Adapter::Sequel, :sequel do
141
143
 
142
144
  specify do
143
145
  expect(import(City.where(nil))).to eq([
144
- { index: [cities[0]], delete: [cities[1]] }
146
+ {index: [cities[0]], delete: [cities[1]]}
145
147
  ])
146
148
  end
147
149
  specify do
148
150
  expect(import(cities)).to eq([
149
- { index: [cities[0]], delete: [cities[1]] },
150
- { delete: cities.last(2) }
151
+ {index: [cities[0]], delete: [cities[1]]},
152
+ {delete: cities.last(2)}
151
153
  ])
152
154
  end
153
155
  specify do
154
156
  expect(import(cities.map(&:id))).to eq([
155
- { index: [cities[0]], delete: [cities[1]] },
156
- { delete: cities.last(2).map(&:id) }
157
+ {index: [cities[0]], delete: [cities[1]]},
158
+ {delete: cities.last(2).map(&:id)}
157
159
  ])
158
160
  end
159
161
  end
@@ -164,47 +166,47 @@ describe Chewy::Type::Adapter::Sequel, :sequel do
164
166
  let!(:deleted) { Array.new(3) { |i| City.create! { |c| c.rating = i + 10 }.tap(&:destroy) } }
165
167
  subject { described_class.new(City) }
166
168
 
167
- specify { expect(import).to eq([{ index: cities }]) }
169
+ specify { expect(import).to eq([{index: cities}]) }
168
170
 
169
- specify { expect(import(City.order(:rating))).to eq([{ index: cities }]) }
171
+ specify { expect(import(City.order(:rating))).to eq([{index: cities}]) }
170
172
  specify do
171
173
  expect(import(City.order(:rating), batch_size: 2))
172
- .to eq([{ index: cities.first(2) }, { index: cities.last(1) }])
174
+ .to eq([{index: cities.first(2)}, {index: cities.last(1)}])
173
175
  end
174
176
 
175
- specify { expect(import(cities)).to eq([{ index: cities }]) }
177
+ specify { expect(import(cities)).to eq([{index: cities}]) }
176
178
  specify do
177
179
  expect(import(cities, batch_size: 2))
178
- .to eq([{ index: cities.first(2) }, { index: cities.last(1) }])
180
+ .to eq([{index: cities.first(2)}, {index: cities.last(1)}])
179
181
  end
180
182
  specify do
181
183
  expect(import(cities, deleted))
182
- .to eq([{ index: cities }, { delete: deleted }])
184
+ .to eq([{index: cities}, {delete: deleted}])
183
185
  end
184
186
  specify do
185
187
  expect(import(cities, deleted, batch_size: 2)).to eq([
186
- { index: cities.first(2) },
187
- { index: cities.last(1) },
188
- { delete: deleted.first(2) },
189
- { delete: deleted.last(1) }
188
+ {index: cities.first(2)},
189
+ {index: cities.last(1)},
190
+ {delete: deleted.first(2)},
191
+ {delete: deleted.last(1)}
190
192
  ])
191
193
  end
192
194
 
193
- specify { expect(import(cities.map(&:rating))).to eq([{ index: cities }]) }
195
+ specify { expect(import(cities.map(&:rating))).to eq([{index: cities}]) }
194
196
  specify do
195
197
  expect(import(cities.map(&:rating), batch_size: 2))
196
- .to eq([{ index: cities.first(2) }, { index: cities.last(1) }])
198
+ .to eq([{index: cities.first(2)}, {index: cities.last(1)}])
197
199
  end
198
200
  specify do
199
201
  expect(import(cities.map(&:rating), deleted.map(&:rating)))
200
- .to eq([{ index: cities }, { delete: deleted.map(&:rating) }])
202
+ .to eq([{index: cities}, {delete: deleted.map(&:rating)}])
201
203
  end
202
204
  specify do
203
205
  expect(import(cities.map(&:rating), deleted.map(&:rating), batch_size: 2)).to eq([
204
- { index: cities.first(2) },
205
- { index: cities.last(1) },
206
- { delete: deleted.first(2).map(&:rating) },
207
- { delete: deleted.last(1).map(&:rating) }
206
+ {index: cities.first(2)},
207
+ {index: cities.last(1)},
208
+ {delete: deleted.first(2).map(&:rating)},
209
+ {delete: deleted.last(1).map(&:rating)}
208
210
  ])
209
211
  end
210
212
  end
@@ -214,67 +216,67 @@ describe Chewy::Type::Adapter::Sequel, :sequel do
214
216
  let!(:deleted) { Array.new(3) { City.create!.tap(&:destroy) } }
215
217
  subject { described_class.new(City.where(rating: 0)) }
216
218
 
217
- specify { expect(import).to eq([{ index: cities.first(3) }]) }
219
+ specify { expect(import).to eq([{index: cities.first(3)}]) }
218
220
 
219
221
  specify do
220
222
  expect(import(City.where('rating < 2')))
221
- .to eq([{ index: cities.first(3) }])
223
+ .to eq([{index: cities.first(3)}])
222
224
  end
223
225
  specify do
224
226
  expect(import(City.where('rating < 2'), batch_size: 2))
225
- .to eq([{ index: cities.first(2) }, { index: [cities[2]] }])
227
+ .to eq([{index: cities.first(2)}, {index: [cities[2]]}])
226
228
  end
227
229
  specify do
228
230
  expect(import(City.where('rating < 1')))
229
- .to eq([{ index: cities.first(3) }])
231
+ .to eq([{index: cities.first(3)}])
230
232
  end
231
233
  specify { expect(import(City.where('rating > 1'))).to eq([]) }
232
234
 
233
235
  specify do
234
236
  expect(import(cities.first(2)))
235
- .to eq([{ index: cities.first(2) }])
237
+ .to eq([{index: cities.first(2)}])
236
238
  end
237
239
  specify do
238
240
  expect(import(cities))
239
- .to eq([{ index: cities.first(3) }, { delete: cities.last(1) }])
241
+ .to eq([{index: cities.first(3)}, {delete: cities.last(1)}])
240
242
  end
241
243
  specify do
242
244
  expect(import(cities, batch_size: 2))
243
- .to eq([{ index: cities.first(2) }, { index: [cities[2]] }, { delete: cities.last(1) }])
245
+ .to eq([{index: cities.first(2)}, {index: [cities[2]]}, {delete: cities.last(1)}])
244
246
  end
245
247
  specify do
246
248
  expect(import(cities, deleted))
247
- .to eq([{ index: cities.first(3) }, { delete: cities.last(1) + deleted }])
249
+ .to eq([{index: cities.first(3)}, {delete: cities.last(1) + deleted}])
248
250
  end
249
251
  specify do
250
252
  expect(import(cities, deleted, batch_size: 3)).to eq([
251
- { index: cities.first(3) },
252
- { delete: cities.last(1) + deleted.first(2) },
253
- { delete: deleted.last(1) }
253
+ {index: cities.first(3)},
254
+ {delete: cities.last(1) + deleted.first(2)},
255
+ {delete: deleted.last(1)}
254
256
  ])
255
257
  end
256
258
 
257
259
  specify do
258
260
  expect(import(cities.first(2).map(&:id)))
259
- .to eq([{ index: cities.first(2) }])
261
+ .to eq([{index: cities.first(2)}])
260
262
  end
261
263
  specify do
262
264
  expect(import(cities.map(&:id)))
263
- .to eq([{ index: cities.first(3) }, { delete: [cities.last.id] }])
265
+ .to eq([{index: cities.first(3)}, {delete: [cities.last.id]}])
264
266
  end
265
267
  specify do
266
268
  expect(import(cities.map(&:id), batch_size: 2))
267
- .to eq([{ index: cities.first(2) }, { index: [cities[2]] }, { delete: [cities.last.id] }])
269
+ .to eq([{index: cities.first(2)}, {index: [cities[2]]}, {delete: [cities.last.id]}])
268
270
  end
269
271
  specify do
270
272
  expect(import(cities.map(&:id), deleted.map(&:id)))
271
- .to eq([{ index: cities.first(3) }, { delete: [cities.last.id] + deleted.map(&:id) }])
273
+ .to eq([{index: cities.first(3)}, {delete: [cities.last.id] + deleted.map(&:id)}])
272
274
  end
273
275
  specify do
274
276
  expect(import(cities.map(&:id), deleted.map(&:id), batch_size: 3)).to eq([
275
- { index: cities.first(3) },
276
- { delete: [cities.last.id] + deleted.first(2).map(&:id) },
277
- { delete: deleted.last(1).map(&:id) }
277
+ {index: cities.first(3)},
278
+ {delete: [cities.last.id] + deleted.first(2).map(&:id)},
279
+ {delete: deleted.last(1).map(&:id)}
278
280
  ])
279
281
  end
280
282
  end
@@ -336,35 +338,98 @@ describe Chewy::Type::Adapter::Sequel, :sequel do
336
338
  end
337
339
  end
338
340
 
341
+ describe '#import_fields' do
342
+ subject { described_class.new(Country) }
343
+ let!(:countries) { Array.new(3) { |i| Country.create!(rating: i) { |c| c.id = i + 1 } } }
344
+ let!(:cities) { Array.new(6) { |i| City.create!(rating: i + 3, country_id: (i + 4) / 2) { |c| c.id = i + 3 } } }
345
+
346
+ specify { expect(subject.import_fields).to match([contain_exactly(1, 2, 3)]) }
347
+ specify { expect(subject.import_fields(fields: [:rating])).to match([contain_exactly([1, 0], [2, 1], [3, 2])]) }
348
+
349
+ context 'scopes' do
350
+ context do
351
+ subject { described_class.new(Country.eager_graph(:cities)) }
352
+
353
+ specify { expect(subject.import_fields).to match([contain_exactly(1, 2, 3)]) }
354
+ specify { expect(subject.import_fields(fields: [:rating])).to match([contain_exactly([1, 0], [2, 1], [3, 2])]) }
355
+ end
356
+
357
+ context do
358
+ subject { described_class.new(Country.eager(:cities)) }
359
+
360
+ specify { expect(subject.import_fields).to match([contain_exactly(1, 2, 3)]) }
361
+ specify { expect(subject.import_fields(fields: [:rating])).to match([contain_exactly([1, 0], [2, 1], [3, 2])]) }
362
+ end
363
+
364
+ context do
365
+ subject { described_class.new(Country.join(:cities, country_id: :id)) }
366
+
367
+ specify { expect(subject.import_fields).to match([contain_exactly(2, 3)]) }
368
+ specify { expect(subject.import_fields(fields: [:rating])).to match([contain_exactly([2, 1], [3, 2])]) }
369
+ end
370
+
371
+ context 'ignores default scope if another scope is passed' do
372
+ subject { described_class.new(Country.join(:cities, country_id: :id)) }
373
+
374
+ specify { expect(subject.import_fields(Country.where('rating < 2'))).to match([contain_exactly(1, 2)]) }
375
+ specify { expect(subject.import_fields(Country.where('rating < 2'), fields: [:rating])).to match([contain_exactly([1, 0], [2, 1])]) }
376
+ end
377
+ end
378
+
379
+ context 'objects/ids' do
380
+ specify { expect(subject.import_fields(1, 2)).to match([contain_exactly(1, 2)]) }
381
+ specify { expect(subject.import_fields(1, 2, fields: [:rating])).to match([contain_exactly([1, 0], [2, 1])]) }
382
+
383
+ specify { expect(subject.import_fields(countries.first(2))).to match([contain_exactly(1, 2)]) }
384
+ specify { expect(subject.import_fields(countries.first(2), fields: [:rating])).to match([contain_exactly([1, 0], [2, 1])]) }
385
+ end
386
+
387
+ context 'batch_size' do
388
+ specify { expect(subject.import_fields(batch_size: 2)).to match([contain_exactly(1, 2), [3]]) }
389
+ specify { expect(subject.import_fields(batch_size: 2, fields: [:rating])).to match([contain_exactly([1, 0], [2, 1]), [[3, 2]]]) }
390
+
391
+ specify { expect(subject.import_fields(Country.where('rating < 2'), batch_size: 2)).to match([contain_exactly(1, 2)]) }
392
+ specify { expect(subject.import_fields(Country.where('rating < 2'), batch_size: 2, fields: [:rating])).to match([contain_exactly([1, 0], [2, 1])]) }
393
+
394
+ specify { expect(subject.import_fields(1, 2, batch_size: 1)).to match([[1], [2]]) }
395
+ specify { expect(subject.import_fields(1, 2, batch_size: 1, fields: [:rating])).to match([[[1, 0]], [[2, 1]]]) }
396
+
397
+ specify { expect(subject.import_fields(countries.first(2), batch_size: 1)).to match([[1], [2]]) }
398
+ specify { expect(subject.import_fields(countries.first(2), batch_size: 1, fields: [:rating])).to match([[[1, 0]], [[2, 1]]]) }
399
+ end
400
+ end
401
+
339
402
  describe '#load' do
340
403
  context do
341
404
  let!(:cities) { Array.new(3) { |i| City.create!(rating: i / 2) } }
342
405
  let!(:deleted) { Array.new(2) { City.create!.tap(&:destroy) } }
406
+ let(:city_ids) { cities.map(&:id) }
407
+ let(:deleted_ids) { deleted.map(&:id) }
343
408
 
344
409
  let(:type) { double(type_name: 'user') }
345
410
 
346
411
  subject { described_class.new(City) }
347
412
 
348
- specify { expect(subject.load(cities.map { |c| double(id: c.id) }, _type: type)).to eq(cities) }
349
- specify { expect(subject.load(cities.map { |c| double(id: c.id) }.reverse, _type: type)).to eq(cities.reverse) }
350
- specify { expect(subject.load(deleted.map { |c| double(id: c.id) }, _type: type)).to eq([nil, nil]) }
351
- specify { expect(subject.load((cities + deleted).map { |c| double(id: c.id) }, _type: type)).to eq([*cities, nil, nil]) }
413
+ specify { expect(subject.load(city_ids, _type: type)).to eq(cities) }
414
+ specify { expect(subject.load(city_ids.reverse, _type: type)).to eq(cities.reverse) }
415
+ specify { expect(subject.load(deleted_ids, _type: type)).to eq([nil, nil]) }
416
+ specify { expect(subject.load(city_ids + deleted_ids, _type: type)).to eq([*cities, nil, nil]) }
352
417
  specify do
353
- expect(subject.load(cities.map { |c| double(id: c.id) }, _type: type, scope: -> { where(rating: 0) }))
418
+ expect(subject.load(city_ids, _type: type, scope: -> { where(rating: 0) }))
354
419
  .to eq(cities.first(2) + [nil])
355
420
  end
356
421
  specify do
357
- expect(subject.load(cities.map { |c| double(id: c.id) },
358
- _type: type, scope: -> { where(rating: 0) }, user: { scope: -> { where(rating: 1) } }))
422
+ expect(subject.load(city_ids,
423
+ _type: type, scope: -> { where(rating: 0) }, user: {scope: -> { where(rating: 1) }}))
359
424
  .to eq([nil, nil] + cities.last(1))
360
425
  end
361
- xspecify do
362
- expect(subject.load(cities.map { |c| double(id: c.id) }, _type: type, scope: City.where(rating: 1)))
426
+ xspecify 'sequel does not support scopes merge' do
427
+ expect(subject.load(city_ids, _type: type, scope: City.where(rating: 1)))
363
428
  .to eq([nil, nil] + cities.last(1))
364
429
  end
365
430
  specify do
366
- expect(subject.load(cities.map { |c| double(id: c.id) },
367
- _type: type, scope: City.where(rating: 1), user: { scope: -> { where(rating: 0) } }))
431
+ expect(subject.load(city_ids,
432
+ _type: type, scope: City.where(rating: 1), user: {scope: -> { where(rating: 0) }}))
368
433
  .to eq(cities.first(2) + [nil])
369
434
  end
370
435
  end
@@ -373,31 +438,33 @@ describe Chewy::Type::Adapter::Sequel, :sequel do
373
438
  before { stub_model(:city).set_dataset :rating_cities }
374
439
  let!(:cities) { Array.new(3) { |i| City.create!(country_id: i / 2) { |c| c.rating = i + 7 } } }
375
440
  let!(:deleted) { Array.new(2) { |i| City.create! { |c| c.rating = i + 10 }.tap(&:destroy) } }
441
+ let(:city_ids) { cities.map(&:rating) }
442
+ let(:deleted_ids) { deleted.map(&:rating) }
376
443
 
377
444
  let(:type) { double(type_name: 'user') }
378
445
 
379
446
  subject { described_class.new(City) }
380
447
 
381
- specify { expect(subject.load(cities.map { |c| double(rating: c.rating) }, _type: type)).to eq(cities) }
382
- specify { expect(subject.load(cities.map { |c| double(rating: c.rating) }.reverse, _type: type)).to eq(cities.reverse) }
383
- specify { expect(subject.load(deleted.map { |c| double(rating: c.rating) }, _type: type)).to eq([nil, nil]) }
384
- specify { expect(subject.load((cities + deleted).map { |c| double(rating: c.rating) }, _type: type)).to eq([*cities, nil, nil]) }
448
+ specify { expect(subject.load(city_ids, _type: type)).to eq(cities) }
449
+ specify { expect(subject.load(city_ids.reverse, _type: type)).to eq(cities.reverse) }
450
+ specify { expect(subject.load(deleted_ids, _type: type)).to eq([nil, nil]) }
451
+ specify { expect(subject.load(city_ids + deleted_ids, _type: type)).to eq([*cities, nil, nil]) }
385
452
  specify do
386
- expect(subject.load(cities.map { |c| double(rating: c.rating) }, _type: type, scope: -> { where(country_id: 0) }))
453
+ expect(subject.load(city_ids, _type: type, scope: -> { where(country_id: 0) }))
387
454
  .to eq(cities.first(2) + [nil])
388
455
  end
389
456
  specify do
390
- expect(subject.load(cities.map { |c| double(rating: c.rating) },
391
- _type: type, scope: -> { where(country_id: 0) }, user: { scope: -> { where(country_id: 1) } }))
457
+ expect(subject.load(city_ids,
458
+ _type: type, scope: -> { where(country_id: 0) }, user: {scope: -> { where(country_id: 1) }}))
392
459
  .to eq([nil, nil] + cities.last(1))
393
460
  end
394
- xspecify do
395
- expect(subject.load(cities.map { |c| double(rating: c.rating) }, _type: type, scope: City.where(country_id: 1)))
461
+ xspecify 'sequel does not support scopes merge' do
462
+ expect(subject.load(city_ids, _type: type, scope: City.where(country_id: 1)))
396
463
  .to eq([nil, nil] + cities.last(1))
397
464
  end
398
465
  specify do
399
- expect(subject.load(cities.map { |c| double(rating: c.rating) },
400
- _type: type, scope: City.where(country_id: 1), user: { scope: -> { where(country_id: 0) } }))
466
+ expect(subject.load(city_ids,
467
+ _type: type, scope: City.where(country_id: 1), user: {scope: -> { where(country_id: 0) }}))
401
468
  .to eq(cities.first(2) + [nil])
402
469
  end
403
470
  end
@@ -0,0 +1,279 @@
1
+ require 'spec_helper'
2
+
3
+ describe Chewy::Type::Import::BulkBuilder do
4
+ before { Chewy.massacre }
5
+
6
+ subject { described_class.new(type, index: index, delete: delete, fields: fields) }
7
+ let(:type) { PlacesIndex::City }
8
+ let(:index) { [] }
9
+ let(:delete) { [] }
10
+ let(:fields) { [] }
11
+
12
+ describe '#bulk_body' do
13
+ context 'simple bulk', :orm do
14
+ before do
15
+ stub_model(:city)
16
+ stub_index(:places) do
17
+ define_type City do
18
+ field :name, :rating
19
+ end
20
+ end
21
+ end
22
+ let(:cities) { Array.new(3) { |i| City.create!(id: i + 1, name: "City#{i + 17}", rating: 42) } }
23
+
24
+ specify { expect(subject.bulk_body).to eq([]) }
25
+
26
+ context do
27
+ let(:index) { cities }
28
+ specify do
29
+ expect(subject.bulk_body).to eq([
30
+ {index: {_id: 1, data: {'name' => 'City17', 'rating' => 42}}},
31
+ {index: {_id: 2, data: {'name' => 'City18', 'rating' => 42}}},
32
+ {index: {_id: 3, data: {'name' => 'City19', 'rating' => 42}}}
33
+ ])
34
+ end
35
+ end
36
+
37
+ context do
38
+ let(:delete) { cities }
39
+ specify do
40
+ expect(subject.bulk_body).to eq([
41
+ {delete: {_id: 1}}, {delete: {_id: 2}}, {delete: {_id: 3}}
42
+ ])
43
+ end
44
+ end
45
+
46
+ context do
47
+ let(:index) { cities.first(2) }
48
+ let(:delete) { [cities.last] }
49
+ specify do
50
+ expect(subject.bulk_body).to eq([
51
+ {index: {_id: 1, data: {'name' => 'City17', 'rating' => 42}}},
52
+ {index: {_id: 2, data: {'name' => 'City18', 'rating' => 42}}},
53
+ {delete: {_id: 3}}
54
+ ])
55
+ end
56
+
57
+ context ':fields' do
58
+ let(:fields) { %w[name] }
59
+ specify do
60
+ expect(subject.bulk_body).to eq([
61
+ {update: {_id: 1, data: {doc: {'name' => 'City17'}}}},
62
+ {update: {_id: 2, data: {doc: {'name' => 'City18'}}}},
63
+ {delete: {_id: 3}}
64
+ ])
65
+ end
66
+ end
67
+ end
68
+ end
69
+
70
+ context 'parent-child relationship', :orm do
71
+ before do
72
+ stub_model(:country)
73
+ stub_model(:city)
74
+ adapter == :sequel ? City.many_to_one(:country) : City.belongs_to(:country)
75
+ end
76
+
77
+ before do
78
+ stub_index(:places) do
79
+ define_type Country do
80
+ field :name
81
+ end
82
+
83
+ define_type City do
84
+ root parent: 'country', parent_id: -> { country_id } do
85
+ field :name
86
+ field :rating
87
+ end
88
+ end
89
+ end
90
+ end
91
+
92
+ before { PlacesIndex::Country.import(country) }
93
+ let(:country) { Country.create!(id: 1, name: 'country') }
94
+ let(:another_country) { Country.create!(id: 2, name: 'another country') }
95
+ let(:city) { City.create!(id: 4, country_id: country.id, name: 'city', rating: 42) }
96
+
97
+ context 'indexing' do
98
+ let(:index) { [city] }
99
+
100
+ specify do
101
+ expect(subject.bulk_body).to eq([
102
+ {index: {_id: city.id, parent: country.id, data: {'name' => 'city', 'rating' => 42}}}
103
+ ])
104
+ end
105
+
106
+ context do
107
+ let(:fields) { %w[name] }
108
+
109
+ specify do
110
+ expect(subject.bulk_body).to eq([
111
+ {update: {_id: city.id, parent: country.id, data: {doc: {'name' => 'city'}}}}
112
+ ])
113
+ end
114
+ end
115
+ end
116
+
117
+ context 'updating parent' do
118
+ before do
119
+ PlacesIndex::City.import(city)
120
+ city.update(country_id: another_country.id)
121
+ end
122
+ let(:index) { [city] }
123
+
124
+ specify do
125
+ expect(subject.bulk_body).to eq([
126
+ {delete: {_id: city.id, parent: country.id.to_s}},
127
+ {index: {_id: city.id, parent: another_country.id, data: {'name' => 'city', 'rating' => 42}}}
128
+ ])
129
+ end
130
+
131
+ context do
132
+ let(:fields) { %w[name] }
133
+
134
+ specify do
135
+ expect(subject.bulk_body).to eq([
136
+ {delete: {_id: city.id, parent: country.id.to_s}},
137
+ {index: {_id: city.id, parent: another_country.id, data: {'name' => 'city', 'rating' => 42}}}
138
+ ])
139
+ end
140
+ end
141
+ end
142
+
143
+ context 'destroying' do
144
+ before { PlacesIndex::City.import(city) }
145
+ let(:delete) { [city] }
146
+
147
+ specify do
148
+ expect(subject.bulk_body).to eq([
149
+ {delete: {_id: city.id, parent: country.id.to_s}}
150
+ ])
151
+ end
152
+ end
153
+ end
154
+
155
+ context 'custom id', :orm do
156
+ before do
157
+ stub_model(:city)
158
+ end
159
+
160
+ before do
161
+ stub_index(:places) do
162
+ define_type City do
163
+ root id: -> { name } do
164
+ field :rating
165
+ end
166
+ end
167
+ end
168
+ end
169
+
170
+ let(:london) { City.create(id: 1, name: 'London', rating: 4) }
171
+
172
+ specify do
173
+ expect { PlacesIndex::City.import(london) }
174
+ .to update_index(PlacesIndex::City).and_reindex(london.name)
175
+ end
176
+
177
+ context 'indexing' do
178
+ let(:index) { [london] }
179
+
180
+ specify do
181
+ expect(subject.bulk_body).to eq([
182
+ {index: {_id: london.name, data: {'rating' => 4}}}
183
+ ])
184
+ end
185
+ end
186
+
187
+ context 'destroying' do
188
+ let(:delete) { [london] }
189
+
190
+ specify do
191
+ expect(subject.bulk_body).to eq([
192
+ {delete: {_id: london.name}}
193
+ ])
194
+ end
195
+ end
196
+ end
197
+
198
+ context 'crutches' do
199
+ before do
200
+ stub_index(:places) do
201
+ define_type :city do
202
+ crutch :names do |collection|
203
+ collection.map { |item| [item.id, "Name#{item.id}"] }.to_h
204
+ end
205
+
206
+ field :name, value: ->(o, c) { c.names[o.id] }
207
+ end
208
+ end
209
+ end
210
+
211
+ let(:index) { [double(id: 42)] }
212
+
213
+ specify do
214
+ expect(subject.bulk_body).to eq([
215
+ {index: {_id: 42, data: {'name' => 'Name42'}}}
216
+ ])
217
+ end
218
+
219
+ context 'witchcraft' do
220
+ before { PlacesIndex::City.witchcraft! }
221
+ specify do
222
+ expect(subject.bulk_body).to eq([
223
+ {index: {_id: 42, data: {'name' => 'Name42'}}}
224
+ ])
225
+ end
226
+ end
227
+ end
228
+
229
+ context 'empty ids' do
230
+ before do
231
+ stub_index(:places) do
232
+ define_type :city do
233
+ field :name
234
+ end
235
+ end
236
+ end
237
+
238
+ let(:index) { [{id: 1, name: 'Name0'}, double(id: '', name: 'Name1'), double(name: 'Name2')] }
239
+ let(:delete) { [double(id: '', name: 'Name3'), {name: 'Name4'}, '', 2] }
240
+
241
+ specify do
242
+ expect(subject.bulk_body).to eq([
243
+ {index: {_id: 1, data: {'name' => 'Name0'}}},
244
+ {index: {data: {'name' => 'Name1'}}},
245
+ {index: {data: {'name' => 'Name2'}}},
246
+ {delete: {_id: {'name' => 'Name4'}}},
247
+ {delete: {_id: 2}}
248
+ ])
249
+ end
250
+
251
+ context do
252
+ let(:fields) { %w[name] }
253
+
254
+ specify do
255
+ expect(subject.bulk_body).to eq([
256
+ {update: {_id: 1, data: {doc: {'name' => 'Name0'}}}},
257
+ {delete: {_id: {'name' => 'Name4'}}},
258
+ {delete: {_id: 2}}
259
+ ])
260
+ end
261
+ end
262
+ end
263
+ end
264
+
265
+ describe '#index_objects_by_id' do
266
+ before do
267
+ stub_index(:places) do
268
+ define_type :city do
269
+ field :name
270
+ end
271
+ end
272
+ end
273
+
274
+ let(:index) { [double(id: 1), double(id: 2), double(id: ''), double] }
275
+ let(:delete) { [double(id: 3)] }
276
+
277
+ specify { expect(subject.index_objects_by_id).to eq('1' => index.first, '2' => index.second) }
278
+ end
279
+ end